datastore: Update AllocateIDs to take keys.

impl/memory/datastore.go

Index: impl/memory/datastore.go
diff --git a/impl/memory/datastore.go b/impl/memory/datastore.go
index 46c32e9864a55063d5b04555b2b4ac1e47c379c2..533f2f62b9c232a4060f83779370bf65b3b4d538 100644
--- a/impl/memory/datastore.go
+++ b/impl/memory/datastore.go
@@ -82,8 +82,8 @@ type dsImpl struct {
 
 var _ ds.RawInterface = (*dsImpl)(nil)
 
-func (d *dsImpl) AllocateIDs(incomplete *ds.Key, n int) (int64, error) {
-	return d.data.allocateIDs(incomplete, n)
+func (d *dsImpl) AllocateIDs(keys []*ds.Key, cb ds.PutMultiCB) error {
+	return d.data.allocateIDs(keys, cb)
 }
 
 func (d *dsImpl) PutMulti(keys []*ds.Key, vals []ds.PropertyMap, cb ds.PutMultiCB) error {
@@ -188,8 +188,8 @@ type txnDsImpl struct {
 
 var _ ds.RawInterface = (*txnDsImpl)(nil)
 
-func (d *txnDsImpl) AllocateIDs(incomplete *ds.Key, n int) (int64, error) {
-	return d.data.parent.allocateIDs(incomplete, n)
+func (d *txnDsImpl) AllocateIDs(keys []*ds.Key, cb ds.PutMultiCB) error {
+	return d.data.parent.allocateIDs(keys, cb)
 }
 
 func (d *txnDsImpl) PutMulti(keys []*ds.Key, vals []ds.PropertyMap, cb ds.PutMultiCB) error {