Add missing AllocateIDs API.

impl/prod/raw_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 prod
 
 import (
         ds "github.com/luci/gae/service/datastore"
         "github.com/luci/gae/service/info"
         "github.com/luci/luci-go/common/errors"
@@ skipping 28 matching lines @@
         me, ok := err.(errors.MultiError)
         if ok {
                 for i, err := range me {
                         cb(i, err)
                 }
                 return nil
         }
         return err
 }
 
+func (d rdsImpl) AllocateIDs(incomplete *ds.Key, n int) (start int64, err error) {
+        par, err := dsF2R(d, incomplete.Parent())
+        if err != nil {
+                return
+        }
+
+        start, _, err = datastore.AllocateIDs(d, incomplete.Last().Kind, par, n)

[dnj (Google), 2015/09/22 06:36:06]

(nit/thought): incomplete.Last().Kind() should probably be a top-level Key
function, like incomplete.Kind()

[iannucci, 2015/09/22 16:41:05]

Yeah... at the time I thought it was a good idea... I think I'll file a CL to do
this and rename Last() to LastTok().

+        return
+}
+
 func (d rdsImpl) DeleteMulti(ks []*ds.Key, cb ds.DeleteMultiCB) error {
         keys, err := dsMF2R(d, ks)
         if err == nil {
                 err = datastore.DeleteMulti(d, keys)
         }
         return idxCallbacker(err, len(ks), func(_ int, err error) {
                 cb(err)
         })
 }
 
@@ skipping 139 matching lines @@
 }
 
 func (d rdsImpl) RunInTransaction(f func(c context.Context) error, opts *ds.TransactionOptions) error {
         ropts := (*datastore.TransactionOptions)(opts)
         return datastore.RunInTransaction(d, f, ropts)
 }
 
 func (d rdsImpl) Testable() ds.Testable {
         return nil
 }