---

ajaaym · chingor13 · commit e7d088c6296e · 2018-11-27T10:54:19.000-08:00
yaml --- r: 11877 b: refs/heads/autosynth-bigquerydatatransfer c: 2501613 h: refs/heads/master i: 11875: d08c228
diff --git a/[refs] b/[refs]
@@ -124,7 +124,7 @@ refs/tags/v0.68.0: 9cc799fcf68c82ab431d425fefa58ef615ce8e5b
 refs/tags/v0.69.0: 78f67a29e8b9c46ba01de566a2eae0fd1c03edea
 refs/heads/autosynth-asset: af3772eb76acbfc401f7c3f9310f09f2b0e17602
 refs/heads/autosynth-automl: 8c05f2af9d8bcf10536f8255093c2484bfe9243c
-refs/heads/autosynth-bigquerydatatransfer: 38ec041b6ddc84f7ea475df1618a1fe1a697acfe
+refs/heads/autosynth-bigquerydatatransfer: 25016131f77f3038df3ca754fc784412bd1c162f
 refs/heads/autosynth-bigquerystorage: 167e37d9259c6460538187b96e6fb74542af84a1
 refs/heads/autosynth-bigtable: 79be26443b70481fdb9a7c5159b798ae6ed0edea
 refs/heads/autosynth-bigtable-admin: 3cd9cc8e61a8a8e02cbdfc4a1c1adb95e4103555
diff --git a/branches/autosynth-bigquerydatatransfer/google-cloud-clients/google-cloud-datastore/src/main/java/com/google/cloud/datastore/QueryResults.java b/branches/autosynth-bigquerydatatransfer/google-cloud-clients/google-cloud-datastore/src/main/java/com/google/cloud/datastore/QueryResults.java
@@ -55,4 +55,19 @@ public interface QueryResults<V> extends Iterator<V> {
    * }</pre>
    */
   Cursor getCursorAfter();
+
+  /**
+   * Returns the number of results skipped, typically because of an offset.
+   *
+   * <p>A simple use case to count entities:
+   *
+   * <pre>{@code
+   * Query<Key> query = Query.newKeyQueryBuilder().setOffset(Integer.MAX_VALUE).build();
+   * QueryResults<Key> result = datasore.datastore.run(query);
+   * if (!result.hasNext()) {
+   *  int numberOfEntities = result.getSkippedResults();
+   * }
+   * }</pre>
+   */
+  int getSkippedResults();
 }
diff --git a/branches/autosynth-bigquerydatatransfer/google-cloud-clients/google-cloud-datastore/src/main/java/com/google/cloud/datastore/QueryResultsImpl.java b/branches/autosynth-bigquerydatatransfer/google-cloud-clients/google-cloud-datastore/src/main/java/com/google/cloud/datastore/QueryResultsImpl.java
@@ -114,4 +114,9 @@ public Class<?> getResultClass() {
   public Cursor getCursorAfter() {
     return new Cursor(cursor);
   }
+
+  @Override
+  public int getSkippedResults() {
+    return runQueryResponsePb.getBatch().getSkippedResults();
+  }
 }
diff --git a/branches/autosynth-bigquerydatatransfer/google-cloud-clients/google-cloud-datastore/src/test/java/com/google/cloud/datastore/DatastoreTest.java b/branches/autosynth-bigquerydatatransfer/google-cloud-clients/google-cloud-datastore/src/test/java/com/google/cloud/datastore/DatastoreTest.java
@@ -633,6 +633,13 @@ public void testQueryPaginationWithLimit() throws DatastoreException {
     EasyMock.verify(rpcFactoryMock, rpcMock);
   }
 
+  @Test
+  public void testRunKeyQueryWithOffset() {
+    Query<Key> query = Query.newKeyQueryBuilder().setOffset(Integer.MAX_VALUE).build();
+    int numberOfEntities = datastore.run(query).getSkippedResults();
+    assertEquals(2, numberOfEntities);
+  }
+
   private List<RunQueryResponse> buildResponsesForQueryPaginationWithLimit() {
     Entity entity4 = Entity.newBuilder(KEY4).set("value", StringValue.of("value")).build();
     Entity entity5 = Entity.newBuilder(KEY5).set("value", "value").build();
diff --git a/branches/autosynth-bigquerydatatransfer/google-cloud-clients/google-cloud-datastore/src/test/java/com/google/cloud/datastore/it/ITDatastoreTest.java b/branches/autosynth-bigquerydatatransfer/google-cloud-clients/google-cloud-datastore/src/test/java/com/google/cloud/datastore/it/ITDatastoreTest.java
@@ -851,4 +851,11 @@ public Integer run(DatastoreReaderWriter transaction) {
       assertEquals(3, ((DatastoreException) expected.getCause()).getCode());
     }
   }
+
+  @Test
+  public void testSkippedResults() {
+    Query<Key> query = Query.newKeyQueryBuilder().setOffset(Integer.MAX_VALUE).build();
+    int numberOfEntities = DATASTORE.run(query).getSkippedResults();
+    assertEquals(2, numberOfEntities);
+  }
 }

Original file line number	Diff line number	Diff line change
`@@ -114,4 +114,9 @@ public Class<?> getResultClass() {`
`114`	`114`	`public Cursor getCursorAfter() {`
`115`	`115`	`return new Cursor(cursor);`
`116`	`116`	`}`
	`117`	`+`
	`118`	`+ @Override`
	`119`	`+ public int getSkippedResults() {`
	`120`	`+ return runQueryResponsePb.getBatch().getSkippedResults();`
	`121`	`+ }`
`117`	`122`	`}`
Original file line number	Diff line number	Diff line change
`@@ -851,4 +851,11 @@ public Integer run(DatastoreReaderWriter transaction) {`
`851`	`851`	`assertEquals(3, ((DatastoreException) expected.getCause()).getCode());`
`852`	`852`	`}`
`853`	`853`	`}`
	`854`	`+`
	`855`	`+ @Test`
	`856`	`+ public void testSkippedResults() {`
	`857`	`+ Query<Key> query = Query.newKeyQueryBuilder().setOffset(Integer.MAX_VALUE).build();`
	`858`	`+ int numberOfEntities = DATASTORE.run(query).getSkippedResults();`
	`859`	`+ assertEquals(2, numberOfEntities);`
	`860`	`+ }`
`854`	`861`	`}`