java/src/main/com/google/appengine/api/datastore/WrappedQueryResultV4.java

   1 package com.google.appengine.api.datastore;
   2
   3 import com.google.appengine.api.datastore.BaseQueryResultsSource.WrappedQueryResult;
   4 import com.google.apphosting.datastore.DatastoreV4;
   5 import com.google.apphosting.datastore.DatastoreV4.QueryResultBatch;
   6 import com.google.apphosting.datastore.DatastoreV4.QueryResultBatch.MoreResultsType;
   7 import com.google.common.collect.ImmutableList;
   8 import com.google.common.collect.Lists;
   9
  10 import java.util.Collection;
  11 import java.util.Collections;
  12 import java.util.List;
  13
  14 /**
  15  * Wrapper for V4 protos with common functions.
  16  */
  17 class WrappedQueryResultV4 implements WrappedQueryResult {
  18   private final QueryResultBatch batch;
  19
  20   WrappedQueryResultV4(QueryResultBatch batch) {
  21     this.batch = batch;
  22   }
  23
  24   @Override
  25   public Cursor getEndCursor() {
  26     if (batch.hasEndCursor()) {
  27       return new Cursor(batch.getEndCursor());
  28     }
  29     return null;
  30   }
  31
  32   @Override
  33   public List<Entity> getEntities(Collection<Projection> projections) {
  34     List<Entity> entityList = Lists.newArrayListWithCapacity(batch.getEntityResultCount());
  35     if (projections.isEmpty()) {
  36       for (DatastoreV4.EntityResult entityResult : batch.getEntityResultList()) {
  37         entityList.add(DataTypeTranslator.toEntity(entityResult.getEntity()));
  38       }
  39     } else {
  40       for (DatastoreV4.EntityResult entityResult : batch.getEntityResultList()) {
  41         entityList.add(DataTypeTranslator.toEntity(entityResult.getEntity(), projections));
  42       }
  43     }
  44     return entityList;
  45   }
  46
  47   @Override
  48   public List<Cursor> getResultCursors() {
  49     return Collections.<Cursor>nCopies(batch.getEntityResultCount(), null);
  50   }
  51
  52   @Override
  53   public Cursor getSkippedResultsCursor() {
  54     return null;
  55   }
  56
  57   @Override
  58   public boolean hasMoreResults() {
  59     return batch.getMoreResults() == MoreResultsType.NOT_FINISHED;
  60   }
  61
  62   @Override
  63   public int numSkippedResults() {
  64     return batch.getSkippedResults();
  65   }
  66
  67   @Override
  68   public List<Index> getIndexInfo(Collection<Index> monitoredIndexBuffer) {
  69     return ImmutableList.of();
  70   }
  71
  72   QueryResultBatch getBatch() {
  73     return batch;
  74   }
  75 }