Add initial Query generation, correctness checking and index generation.

go/src/infra/gae/libs/wrapper/memory/datastore.go

 // Copyright 2015 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 package memory
 
 import (
         "errors"
         "fmt"
         "infra/gae/libs/wrapper"
@@ skipping 98 matching lines @@
 
         data *txnDataStoreData
         ns   string
 }
 
 var (
         _ = wrapper.Datastore((*txnDsImpl)(nil))
         _ = wrapper.Testable((*txnDsImpl)(nil))
 )
 
+func (d *dsImpl) NewQuery(kind string) wrapper.DSQuery {
+        return &queryImpl{DSQuery: wrapper.DummyQY(), ns: d.ns, kind: kind}
+}
+
+func (d *dsImpl) Run(q wrapper.DSQuery) wrapper.DSIterator {
+        rq := q.(*queryImpl)
+        rq = rq.normalize().checkCorrectness(d.ns, false)

[M-A Ruel, 2015/05/31 23:03:15]

return &queryIterImp{q.(*queryImpl).normalize().checkCorrectness(d.ns, false)}

or you fine this too intense?

[iannucci, 2015/05/31 23:31:33]

I think I had that originally and it seemed too line-noisey :)

+        return &queryIterImpl{rq}
+}
+
+func (d *dsImpl) GetAll(q wrapper.DSQuery, dst interface{}) ([]*datastore.Key, error) {
+        // TODO(riannucci): assert that dst is a slice of structs
+        return nil, nil
+}
+
+func (d *dsImpl) Count(q wrapper.DSQuery) (ret int, err error) {
+        itr := d.Run(q.KeysOnly())
+        for _, err = itr.Next(nil); err != nil; _, err = itr.Next(nil) {
+                ret++
+        }
+        if err == datastore.Done {
+                err = nil
+        }
+        return
+}
+
 func (d *txnDsImpl) BreakFeatures(err error, features ...string) {
         d.data.BreakFeatures(err, features...)
 }
 func (d *txnDsImpl) UnbreakFeatures(features ...string) {
         d.data.UnbreakFeatures(features...)
 }
 
 func (d *txnDsImpl) Kind(src interface{}) string {
         return kind(d.ns, d.KindNameResolver(), src)
 }
@@ skipping 42 matching lines @@
 
 ////////////////////////////// private functions ///////////////////////////////
 
 func newDSError(code pb.Error_ErrorCode, message ...string) *appengine_internal.APIError {
         return &appengine_internal.APIError{
                 Detail:  strings.Join(message, ""),
                 Service: "datastore_v3",
                 Code:    int32(code),
         }
 }