Add the ability to disable special automatic entities in impl/memory

impl/memory/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 memory
 
 import (
         "errors"
         "fmt"
 
@@ skipping 23 matching lines @@
 // dsImpl exists solely to bind the current c to the datastore data.
 type dsImpl struct {
         data *dataStoreData
         ns   string
         c    context.Context
 }
 
 var _ ds.RawInterface = (*dsImpl)(nil)
 
 func (d *dsImpl) AllocateIDs(incomplete *ds.Key, n int) (int64, error) {
-        start := d.data.allocateIDs(incomplete, n)
-        return start, nil
+        return d.data.allocateIDs(incomplete, n)
 }
 
 func (d *dsImpl) PutMulti(keys []*ds.Key, vals []ds.PropertyMap, cb ds.PutMultiCB) error {
         d.data.putMulti(keys, vals, cb)
         return nil
 }
 
 func (d *dsImpl) GetMulti(keys []*ds.Key, _meta ds.MultiMetaGetter, cb ds.GetMultiCB) error {
         return d.data.getMulti(keys, cb)
 }
@@ skipping 58 matching lines @@
 }
 
 func (d *dsImpl) Consistent(always bool) {
         d.data.setConsistent(always)
 }
 
 func (d *dsImpl) AutoIndex(enable bool) {
         d.data.setAutoIndex(enable)
 }
 
+func (d *dsImpl) DisableSpecialEntities(enabled bool) {
+        d.data.setDisableSpecialEntities(enabled)
+}
+
 func (d *dsImpl) Testable() ds.Testable {
         return d
 }
 
 ////////////////////////////////// txnDsImpl ///////////////////////////////////
 
 type txnDsImpl struct {
         data *txnDataStoreData
         ns   string
 }
 
 var _ ds.RawInterface = (*txnDsImpl)(nil)
 
 func (d *txnDsImpl) AllocateIDs(incomplete *ds.Key, n int) (int64, error) {
-        start := d.data.parent.allocateIDs(incomplete, n)
-        return start, nil
+        return d.data.parent.allocateIDs(incomplete, n)
 }
 
 func (d *txnDsImpl) PutMulti(keys []*ds.Key, vals []ds.PropertyMap, cb ds.PutMultiCB) error {
         return d.data.run(func() error {
                 d.data.putMulti(keys, vals, cb)
                 return nil
         })
 }
 
 func (d *txnDsImpl) GetMulti(keys []*ds.Key, _meta ds.MultiMetaGetter, cb ds.GetMultiCB) error {
@@ skipping 29 matching lines @@
         return countQuery(fq, d.ns, true, d.data.snap, d.data.snap)
 }
 
 func (*txnDsImpl) RunInTransaction(func(c context.Context) error, *ds.TransactionOptions) error {
         return errors.New("datastore: nested transactions are not supported")
 }
 
 func (*txnDsImpl) Testable() ds.Testable {
         return nil
 }