Index: service/datastore/checkfilter.go |
diff --git a/service/datastore/checkfilter.go b/service/datastore/checkfilter.go |
index ff4fbb7d68898d76adcd2d4cc430f33efd07b8c7..6018a60a3dd0d9956ec31c13120b25fde16d3a1c 100644 |
--- a/service/datastore/checkfilter.go |
+++ b/service/datastore/checkfilter.go |
@@ -19,6 +19,16 @@ type checkFilter struct { |
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") |