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; }
/** * 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); } }
@Before public void setUp() throws Exception { testHelper = env.getTestHelper(); dbAdminClient = testHelper.getClient().getDatabaseAdminClient(); }
DatabaseAdminClient dbAdminClient = spanner.getDatabaseAdminClient();
@Bean @ConditionalOnMissingBean public SpannerDatabaseAdminTemplate spannerDatabaseAdminTemplate( Spanner spanner, DatabaseId databaseId) { return new SpannerDatabaseAdminTemplate(spanner.getDatabaseAdminClient(), spanner.getDatabaseClient(databaseId), databaseId); } }
/** * 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); } }
@Before public void setUp() throws Exception { PipelineOptionsFactory.register(SpannerTestPipelineOptions.class); options = TestPipeline.testingPipelineOptions().as(SpannerTestPipelineOptions.class); project = options.as(GcpOptions.class).getProject(); spanner = SpannerOptions.newBuilder().setProjectId(project).build().getService(); databaseName = generateDatabaseName(); databaseAdminClient = spanner.getDatabaseAdminClient(); // Delete database if exists. databaseAdminClient.dropDatabase(options.getInstanceId(), databaseName); Operation<Database, CreateDatabaseMetadata> op = databaseAdminClient.createDatabase( options.getInstanceId(), databaseName, Collections.singleton( "CREATE TABLE " + options.getTable() + " (" + " Key INT64," + " Value STRING(MAX) NOT NULL," + ") PRIMARY KEY (Key)")); op.waitFor(); }
@Before public void setUp() throws Exception { PipelineOptionsFactory.register(SpannerTestPipelineOptions.class); options = TestPipeline.testingPipelineOptions().as(SpannerTestPipelineOptions.class); project = options.as(GcpOptions.class).getProject(); spanner = SpannerOptions.newBuilder().setProjectId(project).build().getService(); databaseName = generateDatabaseName(); databaseAdminClient = spanner.getDatabaseAdminClient(); // Delete database if exists. databaseAdminClient.dropDatabase(options.getInstanceId(), databaseName); Operation<Database, CreateDatabaseMetadata> op = databaseAdminClient.createDatabase( options.getInstanceId(), databaseName, Collections.singleton( "CREATE TABLE " + options.getTable() + " (" + " Key INT64," + " Value STRING(MAX)," + ") PRIMARY KEY (Key)")); op.waitFor(); makeTestData(); }
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); } }