@Override public void run(Compute compute, Void arg) { for (Zone zone : compute.listZones().iterateAll()) { System.out.println(zone); } } }
PageIterator(Page<T> currentPage) { this.currentPageIterator = currentPage.getValues().iterator(); this.currentPage = currentPage; }
@Override protected T computeNext() { while (!currentPageIterator.hasNext()) { currentPage = currentPage.getNextPage(); if (currentPage == null) { return endOfData(); } currentPageIterator = currentPage.getValues().iterator(); } return currentPageIterator.next(); } }
@Override public TableResult getNextPage() { if (pageNoSchema.hasNextPage()) { return new TableResult(schema, totalRows, pageNoSchema.getNextPage()); } return null; }
@Test public void testListTableData() { EasyMock.expect(bigqueryRpcMock.listTableData(PROJECT, DATASET, TABLE, EMPTY_RPC_OPTIONS)) .andReturn(TABLE_DATA_PB); EasyMock.replay(bigqueryRpcMock); bigquery = options.getService(); Page<FieldValueList> page = bigquery.listTableData(DATASET, TABLE); assertEquals(CURSOR, page.getNextPageToken()); assertArrayEquals(TABLE_DATA.toArray(), Iterables.toArray(page.getValues(), List.class)); }
@Test public void testListPaging() { RESOURCE_MANAGER.create(PARTIAL_PROJECT); RESOURCE_MANAGER.create(COMPLETE_PROJECT); Page<Project> page = RESOURCE_MANAGER.list(ProjectListOption.pageSize(1)); assertNotNull(page.getNextPageToken()); Iterator<Project> iterator = page.getValues().iterator(); compareReadWriteFields(COMPLETE_PROJECT, iterator.next()); assertFalse(iterator.hasNext()); page = page.getNextPage(); iterator = page.getValues().iterator(); compareReadWriteFields(PARTIAL_PROJECT, iterator.next()); assertFalse(iterator.hasNext()); assertNull(page.getNextPageToken()); }
@Override public String getNextPageToken() { return pageNoSchema.getNextPageToken(); }
@Test public void testListEmptyZones() { ImmutableList<com.google.api.services.compute.model.Zone> zones = ImmutableList.of(); Tuple<String, Iterable<com.google.api.services.compute.model.Zone>> result = Tuple.<String, Iterable<com.google.api.services.compute.model.Zone>>of(null, zones); EasyMock.expect(computeRpcMock.listZones(EMPTY_RPC_OPTIONS)).andReturn(result); EasyMock.replay(computeRpcMock); compute = options.getService(); Page<Zone> page = compute.listZones(); assertNull(page.getNextPageToken()); assertArrayEquals(zones.toArray(), Iterables.toArray(page.getValues(), Zone.class)); }
@Test public void testListSinksEmpty() { EasyMock.replay(rpcFactoryMock); logging = options.getService(); ListSinksRequest request = ListSinksRequest.newBuilder().setParent(PROJECT_PB).build(); List<Sink> sinkList = ImmutableList.of(); ListSinksResponse response = ListSinksResponse.newBuilder() .setNextPageToken("") .addAllSinks(Lists.transform(sinkList, SINK_TO_PB_FUNCTION)) .build(); ApiFuture<ListSinksResponse> futureResponse = ApiFutures.immediateFuture(response); EasyMock.expect(loggingRpcMock.list(request)).andReturn(futureResponse); EasyMock.replay(loggingRpcMock); Page<Sink> page = logging.listSinks(); assertNull(page.getNextPageToken()); assertNull(page.getNextPage()); assertArrayEquals(sinkList.toArray(), Iterables.toArray(page.getValues(), Sink.class)); }
@Override public void run(Compute compute, Void arg) { for (Image image : compute.listImages().iterateAll()) { System.out.println(image); } } }
@Test public void testAggregatedListEmptyDisks() { compute = options.getService(); ImmutableList<com.google.api.services.compute.model.Disk> diskList = ImmutableList.of(); Tuple<String, Iterable<com.google.api.services.compute.model.Disk>> result = Tuple.<String, Iterable<com.google.api.services.compute.model.Disk>>of(null, diskList); EasyMock.expect(computeRpcMock.listDisks(EMPTY_RPC_OPTIONS)).andReturn(result); EasyMock.replay(computeRpcMock); Page<Disk> page = compute.listDisks(); assertNull(page.getNextPageToken()); assertArrayEquals(diskList.toArray(), Iterables.toArray(page.getValues(), Disk.class)); }
@Override public void run(ResourceManager resourceManager, String... args) { System.out.println("Projects you can view:"); for (ProjectInfo project : resourceManager.list().getValues()) { System.out.println(projectDetails(project)); } }
/** 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 void run(BigQuery bigquery, Void arg) { for (Job job : bigquery.listJobs().iterateAll()) { System.out.println(job); } } }
@Test public void testListRegions() { String cursor = "cursor"; compute = options.getService(); ImmutableList<Region> regionList = ImmutableList.of(REGION, REGION); Tuple<String, Iterable<com.google.api.services.compute.model.Region>> result = Tuple.of(cursor, Iterables.transform(regionList, Region.TO_PB_FUNCTION)); EasyMock.expect(computeRpcMock.listRegions(EMPTY_RPC_OPTIONS)).andReturn(result); EasyMock.replay(computeRpcMock); Page<Region> page = compute.listRegions(); assertEquals(cursor, page.getNextPageToken()); assertArrayEquals(regionList.toArray(), Iterables.toArray(page.getValues(), Region.class)); }
@Override public Iterable<FieldValueList> getValues() { return addSchema(pageNoSchema.getValues()); }
@Override public void run(Logging logging, Void arg) { for (Metric metric : logging.listMetrics().iterateAll()) { System.out.println(metric); } } }
@Test public void testListBucketsEmpty() { EasyMock.expect(storageRpcMock.list(EMPTY_RPC_OPTIONS)) .andReturn( Tuple.<String, Iterable<com.google.api.services.storage.model.Bucket>>of(null, null)); EasyMock.replay(storageRpcMock); initializeService(); Page<Bucket> page = storage.list(); assertNull(page.getNextPageToken()); assertArrayEquals( ImmutableList.of().toArray(), Iterables.toArray(page.getValues(), Bucket.class)); }
@Test public void testListRecordSets() { EasyMock.expect(dnsRpcMock.listRecordSets(ZONE_INFO.getName(), EMPTY_RPC_OPTIONS)) .andReturn(LIST_OF_PB_DNS_RECORDS); EasyMock.replay(dnsRpcMock); dns = options.getService(); // creates DnsImpl Page<RecordSet> dnsPage = dns.listRecordSets(ZONE_INFO.getName()); assertEquals(2, Lists.newArrayList(dnsPage.getValues()).size()); assertTrue(Lists.newArrayList(dnsPage.getValues()).contains(DNS_RECORD1)); assertTrue(Lists.newArrayList(dnsPage.getValues()).contains(DNS_RECORD2)); }
@Override public void run(Compute compute, Void arg) { for (Snapshot snapshot : compute.listSnapshots().iterateAll()) { System.out.println(snapshot); } } }