spanner.getDatabaseClient(DatabaseId.of(options.getProjectId(), instanceId, databaseId)); spanner.close();
DatabaseAdminClient getDatabaseAdminClient() { // [START get_dbadmin_client] SpannerOptions options = SpannerOptions.newBuilder().build(); Spanner spanner = options.getService(); DatabaseAdminClient dbAdminClient = spanner.getDatabaseAdminClient(); // [END get_dbadmin_client] return dbAdminClient; }
public BatchClient getBatchClient(Database db) { return getClient().getBatchClient(db.getId()); }
spanner = getSpanner(properties, host, project); if (project == null) { project = spanner.getOptions().getProjectId(); dbClient = spanner.getDatabaseClient(DatabaseId.of(project, instance, database)); } catch (Exception e) { LOGGER.log(Level.SEVERE, "init()", e); .append("\nHost: ").append(spanner.getOptions().getHost()) .append("\nProject: ").append(project) .append("\nInstance: ").append(instance)
String clientProject = spanner.getOptions().getProjectId(); if (!db.getInstanceId().getProject().equals(clientProject)) { System.err.println( DatabaseClient dbClient = spanner.getDatabaseClient(db); DatabaseAdminClient dbAdminClient = spanner.getDatabaseAdminClient(); spanner.close();
public DatabaseClient getDatabaseClient(Database db) { return getClient().getDatabaseClient(db.getId()); }
public SpannerAccessor connectToSpanner() { SpannerOptions.Builder builder = SpannerOptions.newBuilder(); if (getProjectId() != null) { builder.setProjectId(getProjectId().get()); } if (getServiceFactory() != null) { builder.setServiceFactory(this.getServiceFactory()); } if (getHost() != null) { builder.setHost(getHost().get()); } String userAgentString = USER_AGENT_PREFIX + "/" + ReleaseInfo.getReleaseInfo().getVersion(); builder.setHeaderProvider(FixedHeaderProvider.create("user-agent", userAgentString)); SpannerOptions options = builder.build(); Spanner spanner = options.getService(); DatabaseClient databaseClient = spanner.getDatabaseClient( DatabaseId.of(options.getProjectId(), getInstanceId().get(), getDatabaseId().get())); BatchClient batchClient = spanner.getBatchClient( DatabaseId.of(options.getProjectId(), getInstanceId().get(), getDatabaseId().get())); DatabaseAdminClient databaseAdminClient = spanner.getDatabaseAdminClient(); return new SpannerAccessor(spanner, databaseClient, databaseAdminClient, batchClient); } }
@Bean @ConditionalOnMissingBean public SpannerDatabaseAdminTemplate spannerDatabaseAdminTemplate( Spanner spanner, DatabaseId databaseId) { return new SpannerDatabaseAdminTemplate(spanner.getDatabaseAdminClient(), spanner.getDatabaseClient(databaseId), databaseId); } }
@Override public void run() { spanner.close(); } });
public FakeServiceFactory() { synchronized (lock) { index = count++; mockSpanners.add(mock(Spanner.class, withSettings().serializable())); mockDatabaseClients.add(mock(DatabaseClient.class, withSettings().serializable())); mockBatchClients.add(mock(BatchClient.class, withSettings().serializable())); } when(mockSpanner().getDatabaseClient(Matchers.any(DatabaseId.class))) .thenReturn(mockDatabaseClient()); when(mockSpanner().getBatchClient(Matchers.any(DatabaseId.class))) .thenReturn(mockBatchClient()); }
BatchClient batchClient = spanner.getBatchClient( DatabaseId.of(options.getProjectId(), instanceId, databaseId)); executor.shutdown(); executor.awaitTermination(1, TimeUnit.HOURS); spanner.close();
InstanceAdminClient getInstanceAdminClient() { // [START get_instance_admin_client] SpannerOptions options = SpannerOptions.newBuilder().build(); Spanner spanner = options.getService(); InstanceAdminClient instanceAdminClient = spanner.getInstanceAdminClient(); // [END get_instance_admin_client] return instanceAdminClient; }
DatabaseClient getDatabaseClient() { // [START get_db_client] SpannerOptions options = SpannerOptions.newBuilder().build(); Spanner spanner = options.getService(); final String project = "test-project"; final String instance = "test-instance"; final String database = "example-db"; DatabaseId db = DatabaseId.of(project, instance, database); DatabaseClient dbClient = spanner.getDatabaseClient(db); // [END get_db_client] return dbClient; }
/** Deletes all the databases created via {@code createTestDatabase}. Shuts down the client. */ public void cleanUp() { // Drop all the databases we created explicitly. int numDropped = 0; for (Database db : dbs) { try { logger.log(Level.FINE, "Dropping test database {0}", db.getId()); db.drop(); ++numDropped; } catch (SpannerException e) { logger.log(Level.SEVERE, "Failed to drop test database " + db.getId(), e); } } logger.log(Level.INFO, "Dropped {0} test database(s)", numDropped); client.close(); }
@Before public void setUp() { instanceClient = env.getTestHelper().getClient().getInstanceAdminClient(); }
try { DatabaseClient dbClient = spanner.getDatabaseClient(DatabaseId.of( options.getProjectId(), instanceId, databaseId)); spanner.close();
@Test public void invalidDatabase() { RemoteSpannerHelper helper = env.getTestHelper(); DatabaseClient invalidClient = helper.getClient().getDatabaseClient(DatabaseId.of(helper.getInstanceId(), "invalid")); expectedException.expect(isSpannerException(ErrorCode.NOT_FOUND)); invalidClient .singleUse(TimestampBound.strong()) .readRow(TABLE_NAME, Key.of("k99"), ALL_COLUMNS); }
/** * Creates a test database defined by {@code statements} in the test instance. A {@code CREATE * DATABASE ...} statement should not be included; an appropriate name will be chosen and the * statement generated accordingly. */ public Database createTestDatabase(Iterable<String> statements) throws SpannerException { String dbId = getUniqueDatabaseId(); try { OperationFuture<Database, CreateDatabaseMetadata> op = client .getDatabaseAdminClient() .createDatabase(instanceId.getInstance(), dbId, statements); Database db = op.get(); logger.log(Level.FINE, "Created test database {0}", db.getId()); dbs.add(db); return db; } catch (Exception e) { throw SpannerExceptionFactory.newSpannerException(e); } }
@Override public void close() { spanner.close(); } }
BatchClient getBatchClient() { // [START get_batch_client] SpannerOptions options = SpannerOptions.newBuilder().build(); Spanner spanner = options.getService(); final String project = "test-project"; final String instance = "test-instance"; final String database = "example-db"; DatabaseId db = DatabaseId.of(project, instance, database); BatchClient batchClient = spanner.getBatchClient(db); // [END get_batch_client] return batchClient; } }