datastore: Update AllocateIDs to take keys.

filter/count/rds.go

Index: filter/count/rds.go
diff --git a/filter/count/rds.go b/filter/count/rds.go
index d96fe6681d2943a5c41fb5ac5739ab5e6ae05802..418878fdf3927eb2705f03d0ed34bd10ed66e2ce 100644
--- a/filter/count/rds.go
+++ b/filter/count/rds.go
@@ -30,9 +30,8 @@ type dsCounter struct {
 
 var _ ds.RawInterface = (*dsCounter)(nil)
 
-func (r *dsCounter) AllocateIDs(incomplete *ds.Key, n int) (int64, error) {
-	start, err := r.ds.AllocateIDs(incomplete, n)
-	return start, r.c.AllocateIDs.up(err)
+func (r *dsCounter) AllocateIDs(keys []*ds.Key, cb ds.NewKeyCB) error {
+	return r.c.AllocateIDs.up(r.ds.AllocateIDs(keys, cb))
 }
 
 func (r *dsCounter) DecodeCursor(s string) (ds.Cursor, error) {
@@ -61,7 +60,7 @@ func (r *dsCounter) GetMulti(keys []*ds.Key, meta ds.MultiMetaGetter, cb ds.GetM
 	return r.c.GetMulti.upFilterStop(r.ds.GetMulti(keys, meta, cb))
 }
 
-func (r *dsCounter) PutMulti(keys []*ds.Key, vals []ds.PropertyMap, cb ds.PutMultiCB) error {
+func (r *dsCounter) PutMulti(keys []*ds.Key, vals []ds.PropertyMap, cb ds.NewKeyCB) error {
 	return r.c.PutMulti.upFilterStop(r.ds.PutMulti(keys, vals, cb))
 }