@Override public void init() throws DBException { host = getProperties().getProperty("documentdb.host", null); masterKey = getProperties().getProperty("documentdb.masterKey", null); if (host == null) { System.err.println("ERROR: 'documentdb.host' must be set!"); System.exit(1); } if (masterKey == null) { System.err.println("ERROR: 'documentdb.masterKey' must be set!"); System.exit(1); } databaseId = getProperties().getProperty("documentdb.databaseId", "ycsb"); String collectionId = getProperties().getProperty("documentdb.collectionId", "usertable"); documentClient = new DocumentClient(host, masterKey, ConnectionPolicy.GetDefault(), ConsistencyLevel.Session); try { // Initialize test database and collection. collection = getCollection(collectionId); } catch (DocumentClientException e) { throw new DBException("Initialze collection failed", e); } feedOptions = new FeedOptions(); feedOptions.setEmitVerboseTracesInQuery(false); }
private void configConnectionPolicy(DocumentDBProperties properties, ConnectionPolicy connectionPolicy) { // This is a temp fix as DocumentDbFactory does not support loading ConnectionPolicy bean from context final ConnectionPolicy policy = connectionPolicy == null ? ConnectionPolicy.GetDefault() : connectionPolicy; properties.setConnectionPolicy(policy); } }
private void configConnectionPolicy(DocumentDBProperties properties, ConnectionPolicy connectionPolicy) { // This is a temp fix as DocumentDbFactory does not support loading ConnectionPolicy bean from context final ConnectionPolicy policy = connectionPolicy == null ? ConnectionPolicy.GetDefault() : connectionPolicy; properties.setConnectionPolicy(policy); } }
public DocumentDbFactory(String host, String key, boolean isBiEnabled) { Assert.hasText(host, "host must not be empty!"); Assert.hasText(key, "key must not be empty!"); final ConnectionPolicy policy = ConnectionPolicy.GetDefault(); String userAgent = ";" + USER_AGENT_SUFFIX; if (isBiEnabled && GetHashMac.getHashMac() != null) { userAgent += ";" + GetHashMac.getHashMac(); } policy.setUserAgentSuffix(userAgent); documentClient = new DocumentClient(host, key, policy, ConsistencyLevel.Session); }
public DocumentDbFactory(String host, String key, boolean isBiEnabled) { Assert.hasText(host, "host must not be empty!"); Assert.hasText(key, "key must not be empty!"); final ConnectionPolicy policy = ConnectionPolicy.GetDefault(); String userAgent = ";" + USER_AGENT_SUFFIX; if (isBiEnabled && GetHashMac.getHashMac() != null) { userAgent += ";" + GetHashMac.getHashMac(); } policy.setUserAgentSuffix(userAgent); documentClient = new DocumentClient(host, key, policy, ConsistencyLevel.Session); }
public static DocumentDBConfigBuilder builder(String uri, String key, String database) { return defaultBuilder() .uri(uri) .key(key) .database(database) .connectionPolicy(ConnectionPolicy.GetDefault()) .consistencyLevel(ConsistencyLevel.Session); }
private DocumentClient createDocumentClient() { LOG.debug("createDocumentClient"); final ConnectionPolicy policy = connectionPolicy == null ? ConnectionPolicy.GetDefault() : connectionPolicy; String userAgent = (policy.getUserAgentSuffix() == null ? "" : ";" + policy.getUserAgentSuffix()) + ";" + USER_AGENT_SUFFIX; if (properties.isAllowTelemetry() && GetHashMac.getHashMac() != null) { userAgent += ";" + GetHashMac.getHashMac(); } policy.setUserAgentSuffix(userAgent); return new DocumentClient(properties.getUri(), properties.getKey(), policy, properties.getConsistencyLevel() == null ? ConsistencyLevel.Session : properties.getConsistencyLevel()); }
private DocumentClient createDocumentClient() { LOG.debug("createDocumentClient"); final ConnectionPolicy policy = connectionPolicy == null ? ConnectionPolicy.GetDefault() : connectionPolicy; String userAgent = (policy.getUserAgentSuffix() == null ? "" : ";" + policy.getUserAgentSuffix()) + ";" + USER_AGENT_SUFFIX; if (properties.isAllowTelemetry() && GetHashMac.getHashMac() != null) { userAgent += ";" + GetHashMac.getHashMac(); } policy.setUserAgentSuffix(userAgent); return new DocumentClient(properties.getUri(), properties.getKey(), policy, properties.getConsistencyLevel() == null ? ConsistencyLevel.Session : properties.getConsistencyLevel()); }
/** * Create document client. * * @param properties the properties * @return the document client */ public DocumentClient createDocumentClient(final BaseCosmosDbProperties properties) { val policy = ConnectionPolicy.GetDefault(); var userAgent = (policy.getUserAgentSuffix() == null ? StringUtils.EMPTY : ';' + policy.getUserAgentSuffix()) + ';' + USER_AGENT_SUFFIX; if (properties.isAllowTelemetry() && GetHashMac.getHashMac() != null) { userAgent += ';' + GetHashMac.getHashMac(); } policy.setUserAgentSuffix(userAgent); return new DocumentClient(properties.getUri(), properties.getKey(), policy, ConsistencyLevel.valueOf(properties.getConsistencyLevel())); }