datastore: Update AllocateIDs to take keys.

service/datastore/checkfilter.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 datastore
 
 import (
         "fmt"
 
         "github.com/luci/gae/service/info"
         "github.com/luci/luci-go/common/errors"
         "golang.org/x/net/context"
 )
 
 type checkFilter struct {
         RawInterface
 
         aid string
         ns  string
 }
 
-func (tcf *checkFilter) AllocateIDs(incomplete *Key, n int) (start int64, err error) {
-        if n <= 0 {
-                return 0, fmt.Errorf("datastore: invalid `n` parameter in AllocateIDs: %d", n)
-        }
-        if !incomplete.PartialValid(tcf.aid, tcf.ns) {
-                return 0, ErrInvalidKey
-        }
-        return tcf.RawInterface.AllocateIDs(incomplete, n)
-}
-
 func (tcf *checkFilter) RunInTransaction(f func(c context.Context) error, opts *TransactionOptions) error {
         if f == nil {
                 return fmt.Errorf("datastore: RunInTransaction function is nil")
         }
         return tcf.RawInterface.RunInTransaction(f, opts)
 }
 
 func (tcf *checkFilter) Run(fq *FinalizedQuery, cb RawRunCB) error {
         if fq == nil {
                 return fmt.Errorf("datastore: Run query is nil")
         }
         if cb == nil {
                 return fmt.Errorf("datastore: Run callback is nil")
         }
         return tcf.RawInterface.Run(fq, cb)
 }
 
 func (tcf *checkFilter) GetMulti(keys []*Key, meta MultiMetaGetter, cb GetMultiCB) error {
         if len(keys) == 0 {
                 return nil
         }
         if cb == nil {
                 return fmt.Errorf("datastore: GetMulti callback is nil")
         }
         lme := errors.NewLazyMultiError(len(keys))
         for i, k := range keys {
-                if k.Incomplete() || !k.Valid(true, tcf.aid, tcf.ns) {
+                if k.IsIncomplete() || !k.Valid(true, tcf.aid, tcf.ns) {
                         lme.Assign(i, ErrInvalidKey)
                 }
         }
         if me := lme.Get(); me != nil {
                 for _, err := range me.(errors.MultiError) {
                         cb(nil, err)
                 }
                 return nil
         }
         return tcf.RawInterface.GetMulti(keys, meta, cb)
 }
 
-func (tcf *checkFilter) PutMulti(keys []*Key, vals []PropertyMap, cb PutMultiCB) error {
+func (tcf *checkFilter) PutMulti(keys []*Key, vals []PropertyMap, cb NewKeyCB) error {
         if len(keys) != len(vals) {
                 return fmt.Errorf("datastore: PutMulti with mismatched keys/vals lengths (%d/%d)", len(keys), len(vals))
         }
         if len(keys) == 0 {
                 return nil
         }
         if cb == nil {
                 return fmt.Errorf("datastore: PutMulti callback is nil")
         }
         lme := errors.NewLazyMultiError(len(keys))
@@ skipping 19 matching lines @@
 
 func (tcf *checkFilter) DeleteMulti(keys []*Key, cb DeleteMultiCB) error {
         if len(keys) == 0 {
                 return nil
         }
         if cb == nil {
                 return fmt.Errorf("datastore: DeleteMulti callback is nil")
         }
         lme := errors.NewLazyMultiError(len(keys))
         for i, k := range keys {
-                if k.Incomplete() || !k.Valid(false, tcf.aid, tcf.ns) {
+                if k.IsIncomplete() || !k.Valid(false, tcf.aid, tcf.ns) {
                         lme.Assign(i, ErrInvalidKey)
                 }
         }
         if me := lme.Get(); me != nil {
                 for _, err := range me.(errors.MultiError) {
                         cb(err)
                 }
                 return nil
         }
         return tcf.RawInterface.DeleteMulti(keys, cb)
 }
 
 func applyCheckFilter(c context.Context, i RawInterface) RawInterface {
         inf := info.Get(c)
         ns, _ := inf.GetNamespace()
         return &checkFilter{i, inf.FullyQualifiedAppID(), ns}
 }