impl/memory: Implement Queries

impl/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"
 
@@ skipping 51 matching lines @@
 func (d *dsImpl) DeleteMulti(keys []ds.Key, cb ds.DeleteMultiCB) error {
         d.data.delMulti(keys, cb)
         return nil
 }
 
 func (d *dsImpl) NewQuery(kind string) ds.Query {
         return &queryImpl{ns: d.ns, kind: kind}
 }
 
 func (d *dsImpl) Run(q ds.Query, cb ds.RawRunCB) error {
-        rq := q.(*queryImpl)
-        done, err := rq.valid(d.ns, true)
-        if done || err != nil {
-                return err // will be nil if done
+        _, err := q.(*queryImpl).prep(d.ns, true)
+        if err != nil {
+                if err == errQueryDone {
+                        return nil
+                }
+                return err
         }
         return nil
 }
 
 func (d *dsImpl) AddIndexes(idxs ...*ds.IndexDefinition) {
         for _, i := range idxs {
                 if !i.Compound() {
                         panic(fmt.Errorf("Attempted to add non-compound index: %s", i))
                 }
         }
@@ skipping 49 matching lines @@
         })
 }
 
 func (d *txnDsImpl) DeleteMulti(keys []ds.Key, cb ds.DeleteMultiCB) error {
         return d.data.run(func() error {
                 return d.data.delMulti(keys, cb)
         })
 }
 
 func (d *txnDsImpl) Run(q ds.Query, cb ds.RawRunCB) error {
-        rq := q.(*queryImpl)
-        done, err := rq.valid(d.ns, true)
-        if done || err != nil {
-                return err // will be nil if done
+        _, err := q.(*queryImpl).prep(d.ns, true)
+        if err != nil {
+                if err == errQueryDone {
+                        return nil
+                }
+                return err
         }
-        if rq.eventualConsistency {
-                rq = rq.checkMutateClone(nil, nil)
-                rq.eventualConsistency = false
+        if q.(*queryImpl).eventualConsistency {
+                // TODO(riannucci): use head instead of snap for indexes
         }
-        // TODO(riannucci): use head instead of snap for indexes
         panic("NOT IMPLEMENTED")
 }
 
 func (*txnDsImpl) RunInTransaction(func(c context.Context) error, *ds.TransactionOptions) error {
         return errors.New("datastore: nested transactions are not supported")
 }
 
 func (d *txnDsImpl) NewQuery(kind string) ds.Query {
         return &queryImpl{ns: d.ns, kind: kind}
 }
 
 func (*txnDsImpl) Testable() ds.Testable {
         return nil
 }