Index: impl/prod/raw_datastore.go |
diff --git a/impl/prod/raw_datastore.go b/impl/prod/raw_datastore.go |
index 0a638eecfd53c1bea34e99f08d3bf8bfac54d71b..e1c8e84055a359282d19351213112008b3a6d4b1 100644 |
--- a/impl/prod/raw_datastore.go |
+++ b/impl/prod/raw_datastore.go |
@@ -39,10 +39,12 @@ type rdsImpl struct { |
aeCtx context.Context |
} |
-func idxCallbacker(err error, amt int, cb func(idx int, err error)) error { |
+func idxCallbacker(err error, amt int, cb func(idx int, err error) error) error { |
if err == nil { |
for i := 0; i < amt; i++ { |
- cb(i, nil) |
+ if err := cb(i, nil); err != nil { |
+ return err |
+ } |
} |
return nil |
} |
@@ -50,7 +52,9 @@ func idxCallbacker(err error, amt int, cb func(idx int, err error)) error { |
me, ok := err.(errors.MultiError) |
if ok { |
for i, err := range me { |
- cb(i, err) |
+ if err := cb(i, err); err != nil { |
+ return err |
+ } |
} |
return nil |
} |
@@ -72,8 +76,8 @@ func (d rdsImpl) DeleteMulti(ks []*ds.Key, cb ds.DeleteMultiCB) error { |
if err == nil { |
err = datastore.DeleteMulti(d.aeCtx, keys) |
} |
- return idxCallbacker(err, len(ks), func(_ int, err error) { |
- cb(err) |
+ return idxCallbacker(err, len(ks), func(_ int, err error) error { |
+ return cb(err) |
}) |
} |
@@ -86,12 +90,11 @@ func (d rdsImpl) GetMulti(keys []*ds.Key, _meta ds.MultiMetaGetter, cb ds.GetMul |
} |
err = datastore.GetMulti(d.aeCtx, rkeys, vals) |
} |
- return idxCallbacker(err, len(keys), func(idx int, err error) { |
+ return idxCallbacker(err, len(keys), func(idx int, err error) error { |
if pls := vals[idx]; pls != nil { |
- cb(pls.(*typeFilter).pm, err) |
- } else { |
- cb(nil, err) |
+ return cb(pls.(*typeFilter).pm, err) |
} |
+ return cb(nil, err) |
}) |
} |
@@ -104,12 +107,12 @@ func (d rdsImpl) PutMulti(keys []*ds.Key, vals []ds.PropertyMap, cb ds.PutMultiC |
} |
rkeys, err = datastore.PutMulti(d.aeCtx, rkeys, rvals) |
} |
- return idxCallbacker(err, len(keys), func(idx int, err error) { |
+ return idxCallbacker(err, len(keys), func(idx int, err error) error { |
k := (*ds.Key)(nil) |
if err == nil { |
k = dsR2F(rkeys[idx]) |
} |
- cb(k, err) |
+ return cb(k, err) |
}) |
} |