Add missing AllocateIDs API.

filter/featureBreaker/rds.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 featureBreaker
 
 import (
         "golang.org/x/net/context"
 
         ds "github.com/luci/gae/service/datastore"
 )
 
 type dsState struct {
         *state
 
         rds ds.RawInterface
 }
 
+func (r *dsState) AllocateIDs(incomplete *ds.Key, n int) (int64, error) {
+        start := int64(0)
+        err := r.run(func() (err error) {
+                start, err = r.rds.AllocateIDs(incomplete, n)
+                return
+        })
+        return start, err
+}
+
 func (r *dsState) DecodeCursor(s string) (ds.Cursor, error) {
         curs := ds.Cursor(nil)
         err := r.run(func() (err error) {
                 curs, err = r.rds.DecodeCursor(s)
                 return
         })
         return curs, err
 }
 
 func (r *dsState) Run(q *ds.FinalizedQuery, cb ds.RawRunCB) error {
@@ skipping 33 matching lines @@
         return r.rds.Testable()
 }
 
 // FilterRDS installs a counter datastore filter in the context.
 func FilterRDS(c context.Context, defaultError error) (context.Context, FeatureBreaker) {
         state := newState(defaultError)
         return ds.AddRawFilters(c, func(ic context.Context, RawDatastore ds.RawInterface) ds.RawInterface {
                 return &dsState{state, RawDatastore}
         }), state
 }