datastore: return a mix of auto-inserted and already existing keys. f…

…ixes #17.
googleapis · Jul 11, 2014 · 578c64d · 578c64d
1 parent 6f2146f
commit 578c64d
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/lib/datastore/index.js b/lib/datastore/index.js
@@ -195,9 +195,10 @@ Transaction.prototype.saveAll = function(keys, objs, callback) {
     if (err || !resp) {
       return callback(err);
     }
-    var keys = [];
-    resp.mutationResult.insertAutoIdKeys && resp.mutationResult.insertAutoIdKeys.forEach(function(item) {
-      keys.push(entity.keyFromKeyProto(item));
+    resp.mutationResult.insertAutoIdKeys = resp.mutationResult.insertAutoIdKeys || [];
+    var i = 0;
+    resp.mutationResult.insertAutoIdKeys.forEach(function(item) {
+      keys[insertIndexes[i++]] = entity.keyFromKeyProto(item);
     });
     callback(null, keys);
   });