---

Ajay Kannan · Ajay Kannan · commit 2b869c10b560 · 2015-10-21T15:54:39.000-07:00
yaml --- r: 4233 b: refs/heads/gcs-nio c: 6ffe927 h: refs/heads/master i: 4231: 98e3c88
diff --git a/[refs] b/[refs]
@@ -11,7 +11,7 @@ refs/tags/v0.0.12: 2fd8066e891fb3dfea69b65f6bf6461db79342b9
 refs/heads/compute-alpha: 969cba2627f1d53d352cc4a5ffe0879dacf65e6c
 refs/heads/dns-alpha: 2f90e7e338349287ace33375896907af0f032ca1
 refs/heads/dns-alpha-batch: 17442b07867021b85d0452f5f3eda29a3413288f
-refs/heads/gcs-nio: b270c64f96c5cad1e07c717aab1120b10dd7025d
+refs/heads/gcs-nio: 6ffe9273596c10d38cef62db73a77b38ac8b8c3c
 refs/heads/logging-alpha: db5312bffa7fccac194f6a7feb8cc3066de16aff
 refs/tags/v0.1.0: a615317f7424ed58621b1f65d5c4d8cbbe8a6ed8
 refs/tags/v0.1.1: 7a7f6985fe465e9dd6a075af55493f42b4933be0
diff --git a/branches/gcs-nio/gcloud-java-datastore/src/main/java/com/google/gcloud/datastore/GqlQuery.java b/branches/gcs-nio/gcloud-java-datastore/src/main/java/com/google/gcloud/datastore/GqlQuery.java
@@ -379,12 +379,9 @@ protected void populatePb(com.google.datastore.v1beta3.RunQueryRequest.Builder r
   }
 
   @Override
-  protected StructuredQuery<V> nextQuery(com.google.datastore.v1beta3.RunQueryResponse responsePb) {
-    return new StructuredQuery.Builder<>(type())
-        .namespace(namespace())
-        .mergeFrom(responsePb.getQuery())
-        .prepareNext(responsePb.getBatch())
-        .build();
+  protected Query<V> nextQuery(com.google.datastore.v1beta3.RunQueryResponse responsePb) {
+    return StructuredQuery.<V>fromPb(type(), namespace(), responsePb.getQuery())
+        .nextQuery(responsePb);
   }
 
   @Override
diff --git a/branches/gcs-nio/gcloud-java-datastore/src/main/java/com/google/gcloud/datastore/StructuredQuery.java b/branches/gcs-nio/gcloud-java-datastore/src/main/java/com/google/gcloud/datastore/StructuredQuery.java
@@ -669,23 +669,9 @@ B mergeFrom(com.google.datastore.v1beta3.Query queryPb) {
       return self();
     }
 
-    B prepareNext(com.google.datastore.v1beta3.QueryResultBatch queryResultBatchPb) {
-      startCursor(new Cursor(queryResultBatchPb.getEndCursor()));
-      if (offset > 0 && queryResultBatchPb.getSkippedResults() < offset) {
-        offset(offset - queryResultBatchPb.getSkippedResults());
-      } else {
-        offset(0);
-        if (limit != null) {
-          limit(limit - queryResultBatchPb.getEntityResultsCount());
-        }
-      }
-      return self();
-    }
-
     public StructuredQuery<V> build() {
       return new StructuredQuery<>(this);
     }
-
   }
 
   static final class Builder<V> extends BaseBuilder<V, Builder<V>> {
@@ -858,8 +844,19 @@ protected void populatePb(com.google.datastore.v1beta3.RunQueryRequest.Builder r
   }
 
   @Override
-  protected StructuredQuery<V> nextQuery(com.google.datastore.v1beta3.RunQueryResponse responsePb) {
-    return new Builder<>(type()).mergeFrom(toPb()).prepareNext(responsePb.getBatch()).build();
+  protected Query<V> nextQuery(com.google.datastore.v1beta3.RunQueryResponse responsePb) {
+    Builder<V> builder = new Builder<>(type());
+    builder.mergeFrom(toPb());
+    builder.startCursor(new Cursor(responsePb.getBatch().getEndCursor()));
+    if (offset > 0 && responsePb.getBatch().getSkippedResults() < offset) {
+      builder.offset(offset - responsePb.getBatch().getSkippedResults());
+    } else {
+      builder.offset(0);
+      if (limit != null) {
+        builder.limit(limit - responsePb.getBatch().getEntityResultsCount());
+      }
+    }
+    return builder.build();
   }
 
   @Override
@@ -907,7 +904,9 @@ protected Object fromPb(ResultType<V> resultType, String namespace, byte[] bytes
     return fromPb(resultType, namespace, com.google.datastore.v1beta3.Query.parseFrom(bytesPb));
   }
 
-  private static StructuredQuery<?> fromPb(ResultType<?> resultType, String namespace,
+  @SuppressWarnings("unchecked")
+  static <V> StructuredQuery<V> fromPb(
+      ResultType<?> resultType, String namespace,
       com.google.datastore.v1beta3.Query queryPb) {
     BaseBuilder<?, ?> builder;
     if (resultType.equals(ResultType.ENTITY)) {
@@ -917,6 +916,6 @@ private static StructuredQuery<?> fromPb(ResultType<?> resultType, String namesp
     } else {
       builder = new ProjectionEntityQueryBuilder();
     }
-    return builder.namespace(namespace).mergeFrom(queryPb).build();
+    return (StructuredQuery<V>) builder.namespace(namespace).mergeFrom(queryPb).build();
   }
 }

Original file line number	Diff line number	Diff line change
`@@ -379,12 +379,9 @@ protected void populatePb(com.google.datastore.v1beta3.RunQueryRequest.Builder r`
`379`	`379`	`}`
`380`	`380`
`381`	`381`	`@Override`
`382`		`- protected StructuredQuery<V> nextQuery(com.google.datastore.v1beta3.RunQueryResponse responsePb) {`
`383`		`- return new StructuredQuery.Builder<>(type())`
`384`		`- .namespace(namespace())`
`385`		`- .mergeFrom(responsePb.getQuery())`
`386`		`- .prepareNext(responsePb.getBatch())`
`387`		`- .build();`
	`382`	`+ protected Query<V> nextQuery(com.google.datastore.v1beta3.RunQueryResponse responsePb) {`
	`383`	`+ return StructuredQuery.<V>fromPb(type(), namespace(), responsePb.getQuery())`
	`384`	`+ .nextQuery(responsePb);`
`388`	`385`	`}`
`389`	`386`
`390`	`387`	`@Override`