Add missing AllocateIDs API.

service/datastore/checkfilter.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"
 
         "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")
@@ skipping 76 matching lines @@
                 }
                 return nil
         }
         return tcf.RawInterface.DeleteMulti(keys, cb)
 }
 
 func applyCheckFilter(c context.Context, i RawInterface) RawInterface {
         inf := info.Get(c)
         return &checkFilter{i, inf.FullyQualifiedAppID(), inf.GetNamespace()}
 }