/** Example to list instance configs. */ public List<InstanceConfig> listInstanceConfigs() { // [START instance_admin_client_list_configs] List<InstanceConfig> configs = Lists.newArrayList( instanceAdminClient.listInstanceConfigs(Options.pageSize(1)).iterateAll()); // [END instance_admin_client_list_configs] return configs; }
/** Example to list instances. */ public List<Instance> listInstances() { // [START instance_admin_client_list_instances] List<Instance> instances = Lists.newArrayList(instanceAdminClient.listInstances(Options.pageSize(1)).iterateAll()); // [END instance_admin_client_list_instances] return instances; }
/** Example to get the list of Cloud Spanner database in the given instance. */ // [TARGET listDatabases(String, ListOption...)] // [VARIABLE my_instance_id] public List<Database> listDatabases(String instanceId) { // [START listDatabases] Page<Database> page = dbAdminClient.listDatabases(instanceId, Options.pageSize(1)); List<Database> dbs = new ArrayList<>(); while (page != null) { Database db = Iterables.getOnlyElement(page.getValues()); dbs.add(db); page = page.getNextPage(); } // [END listDatabases] return dbs; } }
@Override public Page<InstanceConfig> listInstanceConfigs(ListOption... options) { final Options listOptions = Options.fromListOptions(options); Preconditions.checkArgument( !listOptions.hasFilter(), "Filter option is not supported by listInstanceConfigs"); final int pageSize = listOptions.hasPageSize() ? listOptions.pageSize() : 0; PageFetcher<InstanceConfig, com.google.spanner.admin.instance.v1.InstanceConfig> pageFetcher = new PageFetcher<InstanceConfig, com.google.spanner.admin.instance.v1.InstanceConfig>() { @Override public Paginated<com.google.spanner.admin.instance.v1.InstanceConfig> getNextPage( String nextPageToken) { return rpc.listInstanceConfigs(pageSize, nextPageToken); } @Override public InstanceConfig fromProto( com.google.spanner.admin.instance.v1.InstanceConfig proto) { return InstanceConfig.fromProto(proto, InstanceAdminClientImpl.this); } }; if (listOptions.hasPageToken()) { pageFetcher.nextPageToken = listOptions.pageToken(); } return pageFetcher.getNextPage(); }
@Override // Since Options mandates checking hasXX() before XX() is called, the equals & hashCode look more // complicated than usual. public boolean equals(Object o) { if (this == o) { return true; } if (o == null || getClass() != o.getClass()) { return false; } Options that = (Options) o; return (!hasLimit() && !that.hasLimit() || hasLimit() && that.hasLimit() && Objects.equals(limit(), that.limit())) && (!hasPrefetchChunks() && !that.hasPrefetchChunks() || hasPrefetchChunks() && that.hasPrefetchChunks() && Objects.equals(prefetchChunks(), that.prefetchChunks())) && (!hasPageSize() && !that.hasPageSize() || hasPageSize() && that.hasPageSize() && Objects.equals(pageSize(), that.pageSize())) && Objects.equals(pageToken(), that.pageToken()) && Objects.equals(filter(), that.filter()); }
@Override public Page<Database> listDatabases(String instanceId, ListOption... options) { final String instanceName = getInstanceName(instanceId); final Options listOptions = Options.fromListOptions(options); Preconditions.checkArgument( !listOptions.hasFilter(), "Filter option is not support by" + "listDatabases"); final int pageSize = listOptions.hasPageSize() ? listOptions.pageSize() : 0; PageFetcher<Database, com.google.spanner.admin.database.v1.Database> pageFetcher = new PageFetcher<Database, com.google.spanner.admin.database.v1.Database>() { @Override public Paginated<com.google.spanner.admin.database.v1.Database> getNextPage( String nextPageToken) { return rpc.listDatabases(instanceName, pageSize, nextPageToken); } @Override public Database fromProto(com.google.spanner.admin.database.v1.Database proto) { return Database.fromProto(proto, DatabaseAdminClientImpl.this); } }; if (listOptions.hasPageToken()) { pageFetcher.nextPageToken = listOptions.pageToken(); } return pageFetcher.getNextPage(); }
@Test public void listDatabases() { String pageToken = "token"; when(rpc.listDatabases(INSTANCE_NAME, 1, null)) .thenReturn(new Paginated<>(ImmutableList.<Database>of(getDatabaseProto()), pageToken)); when(rpc.listDatabases(INSTANCE_NAME, 1, pageToken)) .thenReturn(new Paginated<>(ImmutableList.<Database>of(getAnotherDatabaseProto()), "")); List<com.google.cloud.spanner.Database> dbs = Lists.newArrayList(client.listDatabases(INSTANCE_ID, Options.pageSize(1)).iterateAll()); assertThat(dbs.get(0).getId().getName()).isEqualTo(DB_NAME); assertThat(dbs.get(1).getId().getName()).isEqualTo(DB_NAME2); assertThat(dbs.size()).isEqualTo(2); } }
@Override public Page<Instance> listInstances(ListOption... options) throws SpannerException { final Options listOptions = Options.fromListOptions(options); final int pageSize = listOptions.hasPageSize() ? listOptions.pageSize() : 0; final String filter = listOptions.filter(); PageFetcher<Instance, com.google.spanner.admin.instance.v1.Instance> pageFetcher = new PageFetcher<Instance, com.google.spanner.admin.instance.v1.Instance>() { @Override public Paginated<com.google.spanner.admin.instance.v1.Instance> getNextPage( String nextPageToken) { return rpc.listInstances(pageSize, nextPageToken, filter); } @Override public Instance fromProto(com.google.spanner.admin.instance.v1.Instance proto) { return Instance.fromProto(proto, InstanceAdminClientImpl.this, dbClient); } }; if (listOptions.hasPageToken()) { pageFetcher.nextPageToken = listOptions.pageToken(); } return pageFetcher.getNextPage(); }
@Test public void listInstances() { String nextToken = "token"; String filter = "env:dev"; when(rpc.listInstances(1, null, filter)) .thenReturn( new Paginated<com.google.spanner.admin.instance.v1.Instance>( ImmutableList.of(getInstanceProto()), nextToken)); when(rpc.listInstances(1, nextToken, filter)) .thenReturn( new Paginated<com.google.spanner.admin.instance.v1.Instance>( ImmutableList.of(getAnotherInstanceProto()), "")); List<Instance> instances = Lists.newArrayList( client.listInstances(Options.pageSize(1), Options.filter(filter)).iterateAll()); assertThat(instances.get(0).getId().getName()).isEqualTo(INSTANCE_NAME); assertThat(instances.get(1).getId().getName()).isEqualTo(INSTANCE_NAME2); assertThat(instances.size()).isEqualTo(2); } }
@Test public void listInstanceConfigs() { String nextToken = "token"; when(rpc.listInstanceConfigs(1, null)) .thenReturn( new Paginated<InstanceConfig>( ImmutableList.of(InstanceConfig.newBuilder().setName(CONFIG_NAME).build()), nextToken)); when(rpc.listInstanceConfigs(1, nextToken)) .thenReturn( new Paginated<InstanceConfig>( ImmutableList.of(InstanceConfig.newBuilder().setName(CONFIG_NAME2).build()), "")); List<com.google.cloud.spanner.InstanceConfig> configs = Lists.newArrayList(client.listInstanceConfigs(Options.pageSize(1)).iterateAll()); assertThat(configs.get(0).getId().getName()).isEqualTo(CONFIG_NAME); assertThat(configs.get(1).getId().getName()).isEqualTo(CONFIG_NAME2); assertThat(configs.size()).isEqualTo(2); }
@Test public void listOptTest() { int pageSize = 3; String pageToken = "ptok"; String filter = "env"; Options opts = Options.fromListOptions( Options.pageSize(pageSize), Options.pageToken(pageToken), Options.filter(filter)); assertThat(opts.toString()) .isEqualTo( "pageSize: " + Integer.toString(pageSize) + " pageToken: " + pageToken + " filter: " + filter + " "); assertThat(opts.hasPageSize()).isTrue(); assertThat(opts.hasPageToken()).isTrue(); assertThat(opts.hasFilter()).isTrue(); assertThat(opts.pageSize()).isEqualTo(pageSize); assertThat(opts.pageToken()).isEqualTo(pageToken); assertThat(opts.filter()).isEqualTo(filter); assertThat(opts.hashCode()).isEqualTo(108027089); }
@Test public void listPagination() throws Exception { List<String> dbIds = ImmutableList.of( testHelper.getUniqueDatabaseId(), testHelper.getUniqueDatabaseId(), testHelper.getUniqueDatabaseId()); String instanceId = testHelper.getInstanceId().getInstance(); for (String dbId : dbIds) { dbs.add(dbAdminClient.createDatabase(instanceId, dbId, ImmutableList.<String>of()).get()); } Page<Database> page = dbAdminClient.listDatabases(instanceId, Options.pageSize(1)); List<String> dbIdsGot = new ArrayList<>(); while (page != null) { Database db = Iterables.getOnlyElement(page.getValues()); dbIdsGot.add(db.getId().getDatabase()); page = page.getNextPage(); } assertThat(dbIdsGot).containsAllIn(dbIds); } }
@Test public void listEquality() { Options o1; Options o2; Options o3; o1 = Options.fromListOptions(); o2 = Options.fromListOptions(); assertThat(o1.equals(o2)).isTrue(); o2 = Options.fromListOptions(Options.pageSize(1)); assertThat(o1.equals(o2)).isFalse(); assertThat(o2.equals(o1)).isFalse(); o3 = Options.fromListOptions(Options.pageSize(1)); assertThat(o2.equals(o3)).isTrue(); o3 = Options.fromListOptions(Options.pageSize(2)); assertThat(o2.equals(o3)).isFalse(); o2 = Options.fromListOptions(Options.pageToken("t1")); assertThat(o1.equals(o2)).isFalse(); o3 = Options.fromListOptions(Options.pageToken("t1")); assertThat(o2.equals(o3)).isTrue(); o3 = Options.fromListOptions(Options.pageToken("t2")); assertThat(o2.equals(o3)).isFalse(); o2 = Options.fromListOptions(Options.filter("f1")); assertThat(o1.equals(o2)).isFalse(); o3 = Options.fromListOptions(Options.filter("f1")); assertThat(o2.equals(o3)).isTrue(); o3 = Options.fromListOptions(Options.filter("f2")); assertThat(o2.equals(o3)).isFalse(); }
@Override // Since Options mandates checking hasXX() before XX() is called, the equals & hashCode look more // complicated than usual. public boolean equals(Object o) { if (this == o) { return true; } if (o == null || getClass() != o.getClass()) { return false; } Options that = (Options) o; return (!hasLimit() && !that.hasLimit() || hasLimit() && that.hasLimit() && Objects.equals(limit(), that.limit())) && (!hasPrefetchChunks() && !that.hasPrefetchChunks() || hasPrefetchChunks() && that.hasPrefetchChunks() && Objects.equals(prefetchChunks(), that.prefetchChunks())) && (!hasPageSize() && !that.hasPageSize() || hasPageSize() && that.hasPageSize() && Objects.equals(pageSize(), that.pageSize())) && Objects.equals(pageToken(), that.pageToken()) && Objects.equals(filter(), that.filter()); }
@Override public Page<InstanceConfig> listInstanceConfigs(ListOption... options) { final Options listOptions = Options.fromListOptions(options); Preconditions.checkArgument( !listOptions.hasFilter(), "Filter option is not supported by listInstanceConfigs"); final int pageSize = listOptions.hasPageSize() ? listOptions.pageSize() : 0; PageFetcher<InstanceConfig, com.google.spanner.admin.instance.v1.InstanceConfig> pageFetcher = new PageFetcher<InstanceConfig, com.google.spanner.admin.instance.v1.InstanceConfig>() { @Override public Paginated<com.google.spanner.admin.instance.v1.InstanceConfig> getNextPage( String nextPageToken) { return rpc.listInstanceConfigs(pageSize, nextPageToken); } @Override public InstanceConfig fromProto( com.google.spanner.admin.instance.v1.InstanceConfig proto) { return InstanceConfig.fromProto(proto, InstanceAdminClientImpl.this); } }; if (listOptions.hasPageToken()) { pageFetcher.nextPageToken = listOptions.pageToken(); } return pageFetcher.getNextPage(); }
@Override public Page<Database> listDatabases(String instanceId, ListOption... options) { final String instanceName = getInstanceName(instanceId); final Options listOptions = Options.fromListOptions(options); Preconditions.checkArgument( !listOptions.hasFilter(), "Filter option is not support by" + "listDatabases"); final int pageSize = listOptions.hasPageSize() ? listOptions.pageSize() : 0; PageFetcher<Database, com.google.spanner.admin.database.v1.Database> pageFetcher = new PageFetcher<Database, com.google.spanner.admin.database.v1.Database>() { @Override public Paginated<com.google.spanner.admin.database.v1.Database> getNextPage( String nextPageToken) { return rpc.listDatabases(instanceName, pageSize, nextPageToken); } @Override public Database fromProto(com.google.spanner.admin.database.v1.Database proto) { return Database.fromProto(proto, DatabaseAdminClientImpl.this); } }; if (listOptions.hasPageToken()) { pageFetcher.nextPageToken = listOptions.pageToken(); } return pageFetcher.getNextPage(); }
@Override public Page<Instance> listInstances(ListOption... options) throws SpannerException { final Options listOptions = Options.fromListOptions(options); final int pageSize = listOptions.hasPageSize() ? listOptions.pageSize() : 0; final String filter = listOptions.filter(); PageFetcher<Instance, com.google.spanner.admin.instance.v1.Instance> pageFetcher = new PageFetcher<Instance, com.google.spanner.admin.instance.v1.Instance>() { @Override public Paginated<com.google.spanner.admin.instance.v1.Instance> getNextPage( String nextPageToken) { return rpc.listInstances(pageSize, nextPageToken, filter); } @Override public Instance fromProto(com.google.spanner.admin.instance.v1.Instance proto) { return Instance.fromProto(proto, InstanceAdminClientImpl.this, dbClient); } }; if (listOptions.hasPageToken()) { pageFetcher.nextPageToken = listOptions.pageToken(); } return pageFetcher.getNextPage(); }