refactor utils.map function

little-bear-labs · Jan 31, 2019 · 55ffc5e · 55ffc5e
1 parent 5fa0e4d
commit 55ffc5e
Showing 1 changed file with 6 additions and 12 deletions.
diff --git a/packages/appsync-emulator-serverless/util.js b/packages/appsync-emulator-serverless/util.js
@@ -159,20 +159,14 @@ const create = (errors = [], now = new Date()) => ({
   },
   map: {
     copyAndRetainAllKeys(map, keys = []) {
-      return Object.entries(map).reduce((sum, [key, value]) => {
-        if (keys.indexOf(key) === -1) return sum;
-        return {
-          ...sum,
-          [key]: value,
-        };
-      }, {});
+      const newMap = new Map();
+      keys.forEach(key => newMap.set(key, map.get(key)));
+      return newMap;
     },
     copyAndRemoveAllKeys(map, keys = []) {
-      const result = { ...map };
-      for (const key of keys) {
-        delete result[key];
-      }
-      return result;
+      const newMap = new Map(map);
+      keys.forEach(key => newMap.delete(key));
+      return newMap;
     },
   },
   dynamodb: {