datastore: Update AllocateIDs to take keys.

filter/dscache/ds.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 dscache
 
 import (
         "time"
 
         ds "github.com/luci/gae/service/datastore"
@@ skipping 10 matching lines @@
 }
 
 var _ ds.RawInterface = (*dsCache)(nil)
 
 func (d *dsCache) DeleteMulti(keys []*ds.Key, cb ds.DeleteMultiCB) error {
         return d.mutation(keys, func() error {
                 return d.RawInterface.DeleteMulti(keys, cb)
         })
 }
 
-func (d *dsCache) PutMulti(keys []*ds.Key, vals []ds.PropertyMap, cb ds.PutMultiCB) error {
+func (d *dsCache) PutMulti(keys []*ds.Key, vals []ds.PropertyMap, cb ds.NewKeyCB) error {
         return d.mutation(keys, func() error {
                 return d.RawInterface.PutMulti(keys, vals, cb)
         })
 }
 
 func (d *dsCache) GetMulti(keys []*ds.Key, metas ds.MultiMetaGetter, cb ds.GetMultiCB) error {
         lockItems, nonce := d.mkRandLockItems(keys, metas)
         if len(lockItems) == 0 {
                 return d.RawInterface.GetMulti(keys, metas, cb)
         }
@@ skipping 93 matching lines @@
                 if err == nil {
                         err = txnState.apply(d.supportContext)
                 }
                 return err
         }, opts)
         if err == nil {
                 txnState.release(d.supportContext)
         }
         return err
 }