datastore: Update AllocateIDs to take keys.

filter/txnBuf/ds_txn.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 txnBuf
 
 import (
         ds "github.com/luci/gae/service/datastore"
         "github.com/luci/luci-go/common/errors"
         "golang.org/x/net/context"
@@ skipping 14 matching lines @@
         state    *txnBufState
         haveLock bool
 }
 
 var _ ds.RawInterface = (*dsTxnBuf)(nil)
 
 func (d *dsTxnBuf) DecodeCursor(s string) (ds.Cursor, error) {
         return d.state.parentDS.DecodeCursor(s)
 }
 
-func (d *dsTxnBuf) AllocateIDs(incomplete *ds.Key, n int) (start int64, err error) {
-        return d.state.parentDS.AllocateIDs(incomplete, n)
+func (d *dsTxnBuf) AllocateIDs(keys []*ds.Key, cb ds.NewKeyCB) error {
+        return d.state.parentDS.AllocateIDs(keys, cb)
 }
 
 func (d *dsTxnBuf) GetMulti(keys []*ds.Key, metas ds.MultiMetaGetter, cb ds.GetMultiCB) error {
         return d.state.getMulti(keys, metas, cb, d.haveLock)
 }
 
-func (d *dsTxnBuf) PutMulti(keys []*ds.Key, vals []ds.PropertyMap, cb ds.PutMultiCB) error {
+func (d *dsTxnBuf) PutMulti(keys []*ds.Key, vals []ds.PropertyMap, cb ds.NewKeyCB) error {
         return d.state.putMulti(keys, vals, cb, d.haveLock)
 }
 
 func (d *dsTxnBuf) DeleteMulti(keys []*ds.Key, cb ds.DeleteMultiCB) error {
         return d.state.deleteMulti(keys, cb, d.haveLock)
 }
 
 func (d *dsTxnBuf) Count(fq *ds.FinalizedQuery) (count int64, err error) {
         // Unfortunately there's no fast-path here. We literally have to run the
         // query and count. Fortunately we can optimize to count keys if it's not
@@ skipping 58 matching lines @@
         if !d.haveLock {
                 d.state.Lock()
                 defer d.state.Unlock()
         }
         return withTxnBuf(d.ic, cb, opts)
 }
 
 func (d *dsTxnBuf) Testable() ds.Testable {
         return d.state.parentDS.Testable()
 }