datastore: Update AllocateIDs to take keys.

service/datastore/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 datastore
 
 import (
         "fmt"
         "io"
         "io/ioutil"
@@ skipping 30 matching lines @@
 }
 
 func (d *datastoreImpl) MakeKey(elems ...interface{}) *Key {
         return MakeKey(d.aid, d.ns, elems...)
 }
 
 func (d *datastoreImpl) NewKey(kind, stringID string, intID int64, parent *Key) *Key {
         return NewKey(d.aid, d.ns, kind, stringID, intID, parent)
 }
 
+func (d *datastoreImpl) NewIncompleteKeys(count int, kind string, parent *Key) (keys []*Key) {
+        if count > 0 {
+                keys = make([]*Key, count)
+                for i := range keys {
+                        keys[i] = d.NewKey(kind, "", 0, parent)
+                }
+        }
+        return
+}
+
 func (d *datastoreImpl) NewKeyToks(toks []KeyTok) *Key {
         return NewKeyToks(d.aid, d.ns, toks)
 }
 
 // PopulateKey loads key into obj.
 //
 // obj is any object that Interface.Get is able to accept.
 //
 // This method will panic if obj is an invalid datastore model. If the key could
 // not be applied to the object, nothing will happen.
@@ skipping 53 matching lines @@
         if cbTyp.NumOut() > 1 {
                 badSig()
         } else if cbTyp.NumOut() == 1 && cbTyp.Out(0) != typeOfError {
                 badSig()
         }
         hasErr = cbTyp.NumOut() == 1
 
         return
 }
 
+func (d *datastoreImpl) AllocateIDs(keys ...*Key) ([]*Key, error) {
+        if len(keys) == 0 {
+                return nil, nil
+        }
+
+        // Duplicate the keys slice, then pass to our raw interface.
+        allocKeys := make([]*Key, len(keys))
+        copy(allocKeys, keys)
+
+        if err := d.RawInterface.AllocateIDs(allocKeys); err != nil {
+                return nil, err
+        }
+        return allocKeys, nil
+}
+
 func (d *datastoreImpl) Run(q *Query, cbIface interface{}) error {
         isKey, hasErr, hasCursorCB, mat := runParseCallback(cbIface)
 
         if isKey {
                 q = q.KeysOnly(true)
         }
         fq, err := q.Finalize()
         if err != nil {
                 return err
         }
@@ skipping 327 matching lines @@
                 currentDir = filepath.Dir(currentDir)
         }
 }
 
 func filterStop(err error) error {
         if err == Stop {
                 err = nil
         }
         return err
 }