datastore: Update AllocateIDs to take keys.

impl/memory/datastore.go

 // Copyright 2015 The LUCI Authors. All rights reserved.
 // Use of this source code is governed under the Apache License, Version 2.0
 // that can be found in the LICENSE file.
 
 package memory
 
 import (
         "errors"
         "fmt"
 
@@ skipping 64 matching lines @@
 // dsImpl exists solely to bind the current c to the datastore data.
 type dsImpl struct {
         data  *dataStoreData
         ns    string
         hasNS bool
         c     context.Context
 }
 
 var _ ds.RawInterface = (*dsImpl)(nil)
 
-func (d *dsImpl) AllocateIDs(incomplete *ds.Key, n int) (int64, error) {
-        return d.data.allocateIDs(incomplete, n)
+func (d *dsImpl) AllocateIDs(keys []*ds.Key, cb ds.NewKeyCB) error {
+        return d.data.allocateIDs(keys, cb)
 }
 
-func (d *dsImpl) PutMulti(keys []*ds.Key, vals []ds.PropertyMap, cb ds.PutMultiCB) error {
+func (d *dsImpl) PutMulti(keys []*ds.Key, vals []ds.PropertyMap, cb ds.NewKeyCB) error {
         d.data.putMulti(keys, vals, cb)
         return nil
 }
 
 func (d *dsImpl) GetMulti(keys []*ds.Key, _meta ds.MultiMetaGetter, cb ds.GetMultiCB) error {
         return d.data.getMulti(keys, cb)
 }
 
 func (d *dsImpl) DeleteMulti(keys []*ds.Key, cb ds.DeleteMultiCB) error {
         d.data.delMulti(keys, cb)
@@ skipping 81 matching lines @@
 ////////////////////////////////// txnDsImpl ///////////////////////////////////
 
 type txnDsImpl struct {
         data  *txnDataStoreData
         ns    string
         hasNS bool
 }
 
 var _ ds.RawInterface = (*txnDsImpl)(nil)
 
-func (d *txnDsImpl) AllocateIDs(incomplete *ds.Key, n int) (int64, error) {
-        return d.data.parent.allocateIDs(incomplete, n)
+func (d *txnDsImpl) AllocateIDs(keys []*ds.Key, cb ds.NewKeyCB) error {
+        return d.data.parent.allocateIDs(keys, cb)
 }
 
-func (d *txnDsImpl) PutMulti(keys []*ds.Key, vals []ds.PropertyMap, cb ds.PutMultiCB) error {
+func (d *txnDsImpl) PutMulti(keys []*ds.Key, vals []ds.PropertyMap, cb ds.NewKeyCB) error {
         return d.data.run(func() error {
                 d.data.putMulti(keys, vals, cb)
                 return nil
         })
 }
 
 func (d *txnDsImpl) GetMulti(keys []*ds.Key, _meta ds.MultiMetaGetter, cb ds.GetMultiCB) error {
         return d.data.run(func() error {
                 return d.data.getMulti(keys, cb)
         })
@@ skipping 46 matching lines @@
         if ns == "" && hasNS {
                 // The user has set an empty namespace. Datastore does not support this
                 // for queries.
                 //
                 // Bug on file is:
                 // https://code.google.com/p/googleappengine/issues/detail?id=12914
                 return errors.New("namespace may not be present and empty")
         }
         return nil
 }