Index: filter/count/rds.go |
diff --git a/filter/count/rds.go b/filter/count/rds.go |
index b289ce65a344eebd57ec0af590d3646a6852c6e2..c10bc8350803da02a571deb6f98bec4c813cfb1e 100644 |
--- a/filter/count/rds.go |
+++ b/filter/count/rds.go |
@@ -12,6 +12,7 @@ import ( |
// DSCounter is the counter object for the datastore service. |
type DSCounter struct { |
+ AllocateIDs Entry |
DecodeCursor Entry |
RunInTransaction Entry |
Run Entry |
@@ -28,6 +29,11 @@ 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) DecodeCursor(s string) (ds.Cursor, error) { |
cursor, err := r.ds.DecodeCursor(s) |
return cursor, r.c.DecodeCursor.up(err) |