@model-ts/dynamodb 4.0.0 → 4.2.0

package/dist/esm/in-memory/treap.js

@@ -0,0 +1,183 @@
+export class DeterministicTreap {
+    constructor() {
+        this.root = null;
+        this._size = 0;
+    }
+    get size() {
+        return this._size;
+    }
+    insert(key, value, priority) {
+        const [nextRoot, inserted] = this.insertNode(this.root, key, value, priority);
+        this.root = nextRoot;
+        if (inserted)
+            this._size += 1;
+    }
+    remove(key) {
+        const [nextRoot, removed] = this.removeNode(this.root, key);
+        this.root = nextRoot;
+        if (removed)
+            this._size -= 1;
+        return removed;
+    }
+    has(key) {
+        let current = this.root;
+        while (current) {
+            if (key === current.key)
+                return true;
+            current = key < current.key ? current.left : current.right;
+        }
+        return false;
+    }
+    *iterate(direction, bounds = {}) {
+        if (direction === "asc") {
+            yield* this.iterateAsc(bounds);
+            return;
+        }
+        yield* this.iterateDesc(bounds);
+    }
+    clear() {
+        this.root = null;
+        this._size = 0;
+    }
+    rotateRight(node) {
+        const left = node.left;
+        if (!left)
+            return node;
+        node.left = left.right;
+        left.right = node;
+        return left;
+    }
+    rotateLeft(node) {
+        const right = node.right;
+        if (!right)
+            return node;
+        node.right = right.left;
+        right.left = node;
+        return right;
+    }
+    insertNode(node, key, value, priority) {
+        if (!node) {
+            return [
+                {
+                    key,
+                    priority,
+                    value,
+                    left: null,
+                    right: null,
+                },
+                true,
+            ];
+        }
+        if (key === node.key) {
+            node.value = value;
+            return [node, false];
+        }
+        if (key < node.key) {
+            const [next, inserted] = this.insertNode(node.left, key, value, priority);
+            node.left = next;
+            if (node.left && node.left.priority < node.priority) {
+                node = this.rotateRight(node);
+            }
+            return [node, inserted];
+        }
+        const [next, inserted] = this.insertNode(node.right, key, value, priority);
+        node.right = next;
+        if (node.right && node.right.priority < node.priority) {
+            node = this.rotateLeft(node);
+        }
+        return [node, inserted];
+    }
+    removeNode(node, key) {
+        if (!node)
+            return [null, false];
+        if (key < node.key) {
+            const [next, removed] = this.removeNode(node.left, key);
+            node.left = next;
+            return [node, removed];
+        }
+        if (key > node.key) {
+            const [next, removed] = this.removeNode(node.right, key);
+            node.right = next;
+            return [node, removed];
+        }
+        if (!node.left)
+            return [node.right, true];
+        if (!node.right)
+            return [node.left, true];
+        if (node.left.priority < node.right.priority) {
+            const rotated = this.rotateRight(node);
+            const [next, removed] = this.removeNode(rotated.right, key);
+            rotated.right = next;
+            return [rotated, removed];
+        }
+        const rotated = this.rotateLeft(node);
+        const [next, removed] = this.removeNode(rotated.left, key);
+        rotated.left = next;
+        return [rotated, removed];
+    }
+    *iterateAsc(bounds) {
+        const stack = [];
+        const pushLeft = (node) => {
+            while (node) {
+                if (this.isBelowLowerBound(node.key, bounds.lower)) {
+                    node = node.right;
+                }
+                else {
+                    stack.push(node);
+                    node = node.left;
+                }
+            }
+        };
+        pushLeft(this.root);
+        while (stack.length) {
+            const node = stack.pop();
+            if (this.isAboveUpperBound(node.key, bounds.upper)) {
+                return;
+            }
+            if (!this.isBelowLowerBound(node.key, bounds.lower)) {
+                yield { key: node.key, value: node.value };
+            }
+            pushLeft(node.right);
+        }
+    }
+    *iterateDesc(bounds) {
+        const stack = [];
+        const pushRight = (node) => {
+            while (node) {
+                if (this.isAboveUpperBound(node.key, bounds.upper)) {
+                    node = node.left;
+                }
+                else {
+                    stack.push(node);
+                    node = node.right;
+                }
+            }
+        };
+        pushRight(this.root);
+        while (stack.length) {
+            const node = stack.pop();
+            if (this.isBelowLowerBound(node.key, bounds.lower)) {
+                return;
+            }
+            if (!this.isAboveUpperBound(node.key, bounds.upper)) {
+                yield { key: node.key, value: node.value };
+            }
+            pushRight(node.left);
+        }
+    }
+    isBelowLowerBound(key, lower) {
+        if (!lower)
+            return false;
+        if (lower.inclusive)
+            return key < lower.key;
+        return key <= lower.key;
+    }
+    isAboveUpperBound(key, upper) {
+        if (!upper)
+            return false;
+        if (upper.inclusive)
+            return key > upper.key;
+        return key >= upper.key;
+    }
+}
+//# sourceMappingURL=treap.js.map

package/dist/esm/in-memory/treap.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"treap.js","sourceRoot":"","sources":["../../../src/in-memory/treap.ts"],"names":[],"mappings":"AAaA,MAAM,OAAO,kBAAkB;IAA/B;QACU,SAAI,GAAwB,IAAI,CAAA;QAChC,UAAK,GAAG,CAAC,CAAA;IA+NnB,CAAC;IA7NC,IAAI,IAAI;QACN,OAAO,IAAI,CAAC,KAAK,CAAA;IACnB,CAAC;IAED,MAAM,CAAC,GAAW,EAAE,KAAQ,EAAE,QAAgB;QAC5C,MAAM,CAAC,QAAQ,EAAE,QAAQ,CAAC,GAAG,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,EAAE,GAAG,EAAE,KAAK,EAAE,QAAQ,CAAC,CAAA;QAC7E,IAAI,CAAC,IAAI,GAAG,QAAQ,CAAA;QACpB,IAAI,QAAQ;YAAE,IAAI,CAAC,KAAK,IAAI,CAAC,CAAA;IAC/B,CAAC;IAED,MAAM,CAAC,GAAW;QAChB,MAAM,CAAC,QAAQ,EAAE,OAAO,CAAC,GAAG,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,EAAE,GAAG,CAAC,CAAA;QAC3D,IAAI,CAAC,IAAI,GAAG,QAAQ,CAAA;QACpB,IAAI,OAAO;YAAE,IAAI,CAAC,KAAK,IAAI,CAAC,CAAA;QAC5B,OAAO,OAAO,CAAA;IAChB,CAAC;IAED,GAAG,CAAC,GAAW;QACb,IAAI,OAAO,GAAG,IAAI,CAAC,IAAI,CAAA;QACvB,OAAO,OAAO,EAAE;YACd,IAAI,GAAG,KAAK,OAAO,CAAC,GAAG;gBAAE,OAAO,IAAI,CAAA;YACpC,OAAO,GAAG,GAAG,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAA;SAC3D;QACD,OAAO,KAAK,CAAA;IACd,CAAC;IAED,CAAC,OAAO,CACN,SAAyB,EACzB,SAAsB,EAAE;QAExB,IAAI,SAAS,KAAK,KAAK,EAAE;YACvB,KAAK,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,CAAA;YAC9B,OAAM;SACP;QAED,KAAK,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,CAAA;IACjC,CAAC;IAED,KAAK;QACH,IAAI,CAAC,IAAI,GAAG,IAAI,CAAA;QAChB,IAAI,CAAC,KAAK,GAAG,CAAC,CAAA;IAChB,CAAC;IAEO,WAAW,CAAC,IAAkB;QACpC,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAA;QACtB,IAAI,CAAC,IAAI;YAAE,OAAO,IAAI,CAAA;QAEtB,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC,KAAK,CAAA;QACtB,IAAI,CAAC,KAAK,GAAG,IAAI,CAAA;QACjB,OAAO,IAAI,CAAA;IACb,CAAC;IAEO,UAAU,CAAC,IAAkB;QACnC,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAA;QACxB,IAAI,CAAC,KAAK;YAAE,OAAO,IAAI,CAAA;QAEvB,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC,IAAI,CAAA;QACvB,KAAK,CAAC,IAAI,GAAG,IAAI,CAAA;QACjB,OAAO,KAAK,CAAA;IACd,CAAC;IAEO,UAAU,CAChB,IAAyB,EACzB,GAAW,EACX,KAAQ,EACR,QAAgB;QAEhB,IAAI,CAAC,IAAI,EAAE;YACT,OAAO;gBACL;oBACE,GAAG;oBACH,QAAQ;oBACR,KAAK;oBACL,IAAI,EAAE,IAAI;oBACV,KAAK,EAAE,IAAI;iBACZ;gBACD,IAAI;aACL,CAAA;SACF;QAED,IAAI,GAAG,KAAK,IAAI,CAAC,GAAG,EAAE;YACpB,IAAI,CAAC,KAAK,GAAG,KAAK,CAAA;YAClB,OAAO,CAAC,IAAI,EAAE,KAAK,CAAC,CAAA;SACrB;QAED,IAAI,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE;YAClB,MAAM,CAAC,IAAI,EAAE,QAAQ,CAAC,GAAG,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,EAAE,GAAG,EAAE,KAAK,EAAE,QAAQ,CAAC,CAAA;YACzE,IAAI,CAAC,IAAI,GAAG,IAAI,CAAA;YAEhB,IAAI,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,QAAQ,EAAE;gBACnD,IAAI,GAAG,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,CAAA;aAC9B;YAED,OAAO,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAA;SACxB;QAED,MAAM,CAAC,IAAI,EAAE,QAAQ,CAAC,GAAG,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,KAAK,EAAE,GAAG,EAAE,KAAK,EAAE,QAAQ,CAAC,CAAA;QAC1E,IAAI,CAAC,KAAK,GAAG,IAAI,CAAA;QAEjB,IAAI,IAAI,CAAC,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC,QAAQ,GAAG,IAAI,CAAC,QAAQ,EAAE;YACrD,IAAI,GAAG,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,CAAA;SAC7B;QAED,OAAO,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAA;IACzB,CAAC;IAEO,UAAU,CAChB,IAAyB,EACzB,GAAW;QAEX,IAAI,CAAC,IAAI;YAAE,OAAO,CAAC,IAAI,EAAE,KAAK,CAAC,CAAA;QAE/B,IAAI,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE;YAClB,MAAM,CAAC,IAAI,EAAE,OAAO,CAAC,GAAG,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,EAAE,GAAG,CAAC,CAAA;YACvD,IAAI,CAAC,IAAI,GAAG,IAAI,CAAA;YAChB,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC,CAAA;SACvB;QAED,IAAI,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE;YAClB,MAAM,CAAC,IAAI,EAAE,OAAO,CAAC,GAAG,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,KAAK,EAAE,GAAG,CAAC,CAAA;YACxD,IAAI,CAAC,KAAK,GAAG,IAAI,CAAA;YACjB,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC,CAAA;SACvB;QAED,IAAI,CAAC,IAAI,CAAC,IAAI;YAAE,OAAO,CAAC,IAAI,CAAC,KAAK,EAAE,IAAI,CAAC,CAAA;QACzC,IAAI,CAAC,IAAI,CAAC,KAAK;YAAE,OAAO,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,CAAA;QAEzC,IAAI,IAAI,CAAC,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE;YAC5C,MAAM,OAAO,GAAG,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,CAAA;YACtC,MAAM,CAAC,IAAI,EAAE,OAAO,CAAC,GAAG,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC,CAAA;YAC3D,OAAO,CAAC,KAAK,GAAG,IAAI,CAAA;YACpB,OAAO,CAAC,OAAO,EAAE,OAAO,CAAC,CAAA;SAC1B;QAED,MAAM,OAAO,GAAG,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,CAAA;QACrC,MAAM,CAAC,IAAI,EAAE,OAAO,CAAC,GAAG,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,IAAI,EAAE,GAAG,CAAC,CAAA;QAC1D,OAAO,CAAC,IAAI,GAAG,IAAI,CAAA;QACnB,OAAO,CAAC,OAAO,EAAE,OAAO,CAAC,CAAA;IAC3B,CAAC;IAEO,CAAC,UAAU,CAAC,MAAmB;QACrC,MAAM,KAAK,GAAmB,EAAE,CAAA;QAEhC,MAAM,QAAQ,GAAG,CAAC,IAAyB,EAAE,EAAE;YAC7C,OAAO,IAAI,EAAE;gBACX,IAAI,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,GAAG,EAAE,MAAM,CAAC,KAAK,CAAC,EAAE;oBAClD,IAAI,GAAG,IAAI,CAAC,KAAK,CAAA;iBAClB;qBAAM;oBACL,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;oBAChB,IAAI,GAAG,IAAI,CAAC,IAAI,CAAA;iBACjB;aACF;QACH,CAAC,CAAA;QAED,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QAEnB,OAAO,KAAK,CAAC,MAAM,EAAE;YACnB,MAAM,IAAI,GAAG,KAAK,CAAC,GAAG,EAAG,CAAA;YAEzB,IAAI,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,GAAG,EAAE,MAAM,CAAC,KAAK,CAAC,EAAE;gBAClD,OAAM;aACP;YAED,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,GAAG,EAAE,MAAM,CAAC,KAAK,CAAC,EAAE;gBACnD,MAAM,EAAE,GAAG,EAAE,IAAI,CAAC,GAAG,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,CAAA;aAC3C;YAED,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;SACrB;IACH,CAAC;IAEO,CAAC,WAAW,CAAC,MAAmB;QACtC,MAAM,KAAK,GAAmB,EAAE,CAAA;QAEhC,MAAM,SAAS,GAAG,CAAC,IAAyB,EAAE,EAAE;YAC9C,OAAO,IAAI,EAAE;gBACX,IAAI,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,GAAG,EAAE,MAAM,CAAC,KAAK,CAAC,EAAE;oBAClD,IAAI,GAAG,IAAI,CAAC,IAAI,CAAA;iBACjB;qBAAM;oBACL,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;oBAChB,IAAI,GAAG,IAAI,CAAC,KAAK,CAAA;iBAClB;aACF;QACH,CAAC,CAAA;QAED,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QAEpB,OAAO,KAAK,CAAC,MAAM,EAAE;YACnB,MAAM,IAAI,GAAG,KAAK,CAAC,GAAG,EAAG,CAAA;YAEzB,IAAI,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,GAAG,EAAE,MAAM,CAAC,KAAK,CAAC,EAAE;gBAClD,OAAM;aACP;YAED,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,GAAG,EAAE,MAAM,CAAC,KAAK,CAAC,EAAE;gBACnD,MAAM,EAAE,GAAG,EAAE,IAAI,CAAC,GAAG,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,CAAA;aAC3C;YAED,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;SACrB;IACH,CAAC;IAEO,iBAAiB,CACvB,GAAW,EACX,KAA2C;QAE3C,IAAI,CAAC,KAAK;YAAE,OAAO,KAAK,CAAA;QAExB,IAAI,KAAK,CAAC,SAAS;YAAE,OAAO,GAAG,GAAG,KAAK,CAAC,GAAG,CAAA;QAC3C,OAAO,GAAG,IAAI,KAAK,CAAC,GAAG,CAAA;IACzB,CAAC;IAEO,iBAAiB,CACvB,GAAW,EACX,KAA2C;QAE3C,IAAI,CAAC,KAAK;YAAE,OAAO,KAAK,CAAA;QAExB,IAAI,KAAK,CAAC,SAAS;YAAE,OAAO,GAAG,GAAG,KAAK,CAAC,GAAG,CAAA;QAC3C,OAAO,GAAG,IAAI,KAAK,CAAC,GAAG,CAAA;IACzB,CAAC;CACF"}

package/dist/esm/in-memory/utils.d.ts

@@ -0,0 +1,10 @@
+export declare type InMemoryItem = {
+    [key: string]: any;
+};
+export declare const ITEM_KEY_SEPARATOR = "\0";
+export declare const encodeCompositeKey: (...parts: string[]) => string;
+export declare const encodeItemKey: (pk: string, sk: string) => string;
+export declare const encodeIndexEntryKey: (rangeKey: string, itemKey: string) => string;
+export declare const cloneItem: <T>(item: T) => T;
+export declare const stablePriority: (indexName: string, hashKey: string, rangeKey: string, itemKey: string) => number;
+export declare const sortItemsByPKSK: (items: InMemoryItem[]) => InMemoryItem[];

package/dist/esm/in-memory/utils.js

@@ -0,0 +1,28 @@
+import crypto from "crypto";
+export const ITEM_KEY_SEPARATOR = "\u0000";
+export const encodeCompositeKey = (...parts) => parts
+    .map((part) => `${part.length}:${part}`)
+    .join(ITEM_KEY_SEPARATOR);
+export const encodeItemKey = (pk, sk) => encodeCompositeKey(pk, sk);
+export const encodeIndexEntryKey = (rangeKey, itemKey) => `${rangeKey}${ITEM_KEY_SEPARATOR}${itemKey}`;
+export const cloneItem = (item) => JSON.parse(JSON.stringify(item));
+export const stablePriority = (indexName, hashKey, rangeKey, itemKey) => {
+    const hash = crypto
+        .createHash("sha256")
+        .update(`${indexName}::${hashKey}::${rangeKey}::${itemKey}`)
+        .digest();
+    return hash.readUInt32BE(0);
+};
+export const sortItemsByPKSK = (items) => [...items].sort((a, b) => {
+    var _a, _b, _c, _d;
+    const pkA = String((_a = a.PK) !== null && _a !== void 0 ? _a : "");
+    const pkB = String((_b = b.PK) !== null && _b !== void 0 ? _b : "");
+    if (pkA !== pkB)
+        return pkA < pkB ? -1 : 1;
+    const skA = String((_c = a.SK) !== null && _c !== void 0 ? _c : "");
+    const skB = String((_d = b.SK) !== null && _d !== void 0 ? _d : "");
+    if (skA !== skB)
+        return skA < skB ? -1 : 1;
+    return 0;
+});
+//# sourceMappingURL=utils.js.map

package/dist/esm/in-memory/utils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"utils.js","sourceRoot":"","sources":["../../../src/in-memory/utils.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,QAAQ,CAAA;AAI3B,MAAM,CAAC,MAAM,kBAAkB,GAAG,QAAQ,CAAA;AAE1C,MAAM,CAAC,MAAM,kBAAkB,GAAG,CAAC,GAAG,KAAe,EAAU,EAAE,CAC/D,KAAK;KACF,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,GAAG,IAAI,CAAC,MAAM,IAAI,IAAI,EAAE,CAAC;KACvC,IAAI,CAAC,kBAAkB,CAAC,CAAA;AAE7B,MAAM,CAAC,MAAM,aAAa,GAAG,CAAC,EAAU,EAAE,EAAU,EAAU,EAAE,CAC9D,kBAAkB,CAAC,EAAE,EAAE,EAAE,CAAC,CAAA;AAE5B,MAAM,CAAC,MAAM,mBAAmB,GAAG,CAAC,QAAgB,EAAE,OAAe,EAAU,EAAE,CAC/E,GAAG,QAAQ,GAAG,kBAAkB,GAAG,OAAO,EAAE,CAAA;AAE9C,MAAM,CAAC,MAAM,SAAS,GAAG,CAAI,IAAO,EAAK,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAA;AAE5E,MAAM,CAAC,MAAM,cAAc,GAAG,CAC5B,SAAiB,EACjB,OAAe,EACf,QAAgB,EAChB,OAAe,EACP,EAAE;IACV,MAAM,IAAI,GAAG,MAAM;SAChB,UAAU,CAAC,QAAQ,CAAC;SACpB,MAAM,CAAC,GAAG,SAAS,KAAK,OAAO,KAAK,QAAQ,KAAK,OAAO,EAAE,CAAC;SAC3D,MAAM,EAAE,CAAA;IAEX,OAAO,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;AAC7B,CAAC,CAAA;AAED,MAAM,CAAC,MAAM,eAAe,GAAG,CAAC,KAAqB,EAAkB,EAAE,CACvE,CAAC,GAAG,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE;;IACvB,MAAM,GAAG,GAAG,MAAM,CAAC,MAAA,CAAC,CAAC,EAAE,mCAAI,EAAE,CAAC,CAAA;IAC9B,MAAM,GAAG,GAAG,MAAM,CAAC,MAAA,CAAC,CAAC,EAAE,mCAAI,EAAE,CAAC,CAAA;IAC9B,IAAI,GAAG,KAAK,GAAG;QAAE,OAAO,GAAG,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAA;IAE1C,MAAM,GAAG,GAAG,MAAM,CAAC,MAAA,CAAC,CAAC,EAAE,mCAAI,EAAE,CAAC,CAAA;IAC9B,MAAM,GAAG,GAAG,MAAM,CAAC,MAAA,CAAC,CAAC,EAAE,mCAAI,EAAE,CAAC,CAAA;IAC9B,IAAI,GAAG,KAAK,GAAG;QAAE,OAAO,GAAG,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAA;IAE1C,OAAO,CAAC,CAAA;AACV,CAAC,CAAC,CAAA"}

package/dist/esm/sandbox.d.ts

@@ -16,5 +16,7 @@ export interface Sandbox {
     diff: (before: {
         [key: string]: any;
     }) => Promise<string>;
+    startTracking: () => void;
+    rollback: () => Promise<void>;
 }
 export declare const createSandbox: (client: Client) => Promise<Sandbox>;

package/dist/esm/sandbox.js

@@ -4,6 +4,7 @@ import { chunksOf } from "fp-ts/lib/Array";
 import DynamoDB from "aws-sdk/clients/dynamodb";
 import { formatSnapshotDiff } from "./diff";
 import { GSI_NAMES } from "./gsi";
+import { createInMemoryDocumentClient } from "./in-memory";
 const ddb = new DynamoDB({
     accessKeyId: "xxx",
     secretAccessKey: "xxx",
@@ -91,9 +92,177 @@ export const getTableContents = (tableName) => __awaiter(void 0, void 0, void 0,
     items.forEach((item) => (acc[`${item.PK}__${item.SK}`] = item));
     return acc;
 });
+const WRITE_METHODS = new Set([
+    "put",
+    "update",
+    "delete",
+    "batchWrite",
+    "transactWrite",
+]);
+function createTrackedDocClient(original, tableName) {
+    let isTracking = false;
+    const trackedKeys = new Map();
+    const captureKey = (pk, sk) => __awaiter(this, void 0, void 0, function* () {
+        const compositeKey = `${pk}__${sk}`;
+        if (trackedKeys.has(compositeKey))
+            return;
+        const { Item } = yield original
+            .get({ TableName: tableName, Key: { PK: pk, SK: sk } })
+            .promise();
+        trackedKeys.set(compositeKey, { pk, sk, original: Item !== null && Item !== void 0 ? Item : null });
+    });
+    const captureKeysForOperation = (method, params) => __awaiter(this, void 0, void 0, function* () {
+        var _a;
+        switch (method) {
+            case "put":
+                if (params.TableName === tableName && params.Item) {
+                    yield captureKey(params.Item.PK, params.Item.SK);
+                }
+                break;
+            case "update":
+            case "delete":
+                if (params.TableName === tableName && params.Key) {
+                    yield captureKey(params.Key.PK, params.Key.SK);
+                }
+                break;
+            case "batchWrite": {
+                const tableItems = ((_a = params.RequestItems) === null || _a === void 0 ? void 0 : _a[tableName]) || [];
+                yield Promise.all(tableItems.map((item) => {
+                    if (item.PutRequest) {
+                        return captureKey(item.PutRequest.Item.PK, item.PutRequest.Item.SK);
+                    }
+                    if (item.DeleteRequest) {
+                        return captureKey(item.DeleteRequest.Key.PK, item.DeleteRequest.Key.SK);
+                    }
+                }));
+                break;
+            }
+            case "transactWrite": {
+                const transactItems = params.TransactItems || [];
+                yield Promise.all(transactItems
+                    .map((item) => {
+                    var _a, _b, _c;
+                    if (((_a = item.Put) === null || _a === void 0 ? void 0 : _a.TableName) === tableName) {
+                        return captureKey(item.Put.Item.PK, item.Put.Item.SK);
+                    }
+                    if (((_b = item.Update) === null || _b === void 0 ? void 0 : _b.TableName) === tableName) {
+                        return captureKey(item.Update.Key.PK, item.Update.Key.SK);
+                    }
+                    if (((_c = item.Delete) === null || _c === void 0 ? void 0 : _c.TableName) === tableName) {
+                        return captureKey(item.Delete.Key.PK, item.Delete.Key.SK);
+                    }
+                })
+                    .filter(Boolean));
+                break;
+            }
+        }
+    });
+    const proxy = new Proxy(original, {
+        get(target, prop) {
+            const value = target[prop];
+            if (value === undefined)
+                return undefined;
+            if (typeof value === "function") {
+                if (isTracking && WRITE_METHODS.has(prop)) {
+                    return (params) => {
+                        const request = value.call(target, params);
+                        const origPromise = request.promise.bind(request);
+                        request.promise = () => __awaiter(this, void 0, void 0, function* () {
+                            yield captureKeysForOperation(prop, params);
+                            return origPromise();
+                        });
+                        return request;
+                    };
+                }
+                return value.bind(target);
+            }
+            return value;
+        },
+    });
+    return {
+        proxy: proxy,
+        startTracking: () => {
+            isTracking = true;
+            trackedKeys.clear();
+        },
+        rollback: () => __awaiter(this, void 0, void 0, function* () {
+            isTracking = false;
+            const entries = Array.from(trackedKeys.values());
+            const toDelete = entries.filter((e) => e.original === null);
+            const toRestore = entries.filter((e) => e.original !== null);
+            const deleteChunks = chunksOf(25)(toDelete);
+            const restoreChunks = chunksOf(25)(toRestore);
+            yield Promise.all([
+                ...deleteChunks.map((chunk) => original
+                    .batchWrite({
+                    RequestItems: {
+                        [tableName]: chunk.map(({ pk, sk }) => ({
+                            DeleteRequest: { Key: { PK: pk, SK: sk } },
+                        })),
+                    },
+                })
+                    .promise()),
+                ...restoreChunks.map((chunk) => original
+                    .batchWrite({
+                    RequestItems: {
+                        [tableName]: chunk.map(({ original: item }) => ({
+                            PutRequest: { Item: item },
+                        })),
+                    },
+                })
+                    .promise()),
+            ]);
+            trackedKeys.clear();
+        }),
+    };
+}
 export const createSandbox = (client) => __awaiter(void 0, void 0, void 0, function* () {
+    if (process.env.EXPERIMENTAL_DYNAMODB_IN_MEMORY === "1") {
+        const tableName = crypto.randomBytes(20).toString("hex");
+        const inMemoryClient = createInMemoryDocumentClient();
+        const tracked = createTrackedDocClient(inMemoryClient, tableName);
+        client.setDocumentClient(tracked.proxy);
+        client.setTableName(tableName);
+        return {
+            destroy: () => __awaiter(void 0, void 0, void 0, function* () {
+                inMemoryClient.__inMemoryResetTable(tableName);
+            }),
+            snapshot: () => __awaiter(void 0, void 0, void 0, function* () { return inMemoryClient.__inMemorySnapshot(tableName); }),
+            seed: (...args) => __awaiter(void 0, void 0, void 0, function* () {
+                const chunks = chunksOf(25)(args);
+                yield Promise.all(chunks.map((chunk) => __awaiter(void 0, void 0, void 0, function* () {
+                    const items = chunk.map((i) => {
+                        var _a;
+                        return typeof ((_a = i === null || i === void 0 ? void 0 : i._model) === null || _a === void 0 ? void 0 : _a.__dynamoDBEncode) === "function"
+                            ? i._model.__dynamoDBEncode(i)
+                            : typeof i.encode === "function"
+                                ? i.encode()
+                                : i;
+                    });
+                    return client.documentClient
+                        .batchWrite({
+                        RequestItems: {
+                            [tableName]: items.map((i) => ({ PutRequest: { Item: i } })),
+                        },
+                    })
+                        .promise();
+                })));
+            }),
+            get: (pk, sk) => client.documentClient
+                .get({ TableName: tableName, Key: { PK: pk, SK: sk } })
+                .promise()
+                .then(({ Item }) => Item !== null && Item !== void 0 ? Item : null),
+            diff: (before) => __awaiter(void 0, void 0, void 0, function* () {
+                const snapshot = inMemoryClient.__inMemorySnapshot(tableName);
+                return formatSnapshotDiff(before, snapshot);
+            }),
+            startTracking: tracked.startTracking,
+            rollback: tracked.rollback,
+        };
+    }
     const tableName = yield createTable();
-    client.setDocumentClient(docClient);
+    const tracked = createTrackedDocClient(docClient, tableName);
+    client.setDocumentClient(tracked.proxy);
     client.setTableName(tableName);
     return {
         destroy: () => destroyTable(tableName),
@@ -126,6 +295,8 @@ export const createSandbox = (client) => __awaiter(void 0, void 0, void 0, funct
             const snapshot = yield getTableContents(tableName);
             return formatSnapshotDiff(before, snapshot);
         }),
+        startTracking: tracked.startTracking,
+        rollback: tracked.rollback,
     };
 });
 //# sourceMappingURL=sandbox.js.map

package/dist/esm/sandbox.js.map

@@ -1 +1 @@
-{"version":3,"file":"sandbox.js","sourceRoot":"","sources":["../../src/sandbox.ts"],"names":[],"mappings":";AAAA,OAAO,MAAM,MAAM,QAAQ,CAAA;AAC3B,OAAO,EAAE,QAAQ,EAAE,MAAM,iBAAiB,CAAA;AAC1C,OAAO,QAAQ,MAAM,0BAA0B,CAAA;AAC/C,OAAO,EAAE,kBAAkB,EAAE,MAAM,QAAQ,CAAA;AAE3C,OAAO,EAAE,SAAS,EAAE,MAAM,OAAO,CAAA;AAEjC,MAAM,GAAG,GAAG,IAAI,QAAQ,CAAC;IACvB,WAAW,EAAE,KAAK;IAClB,eAAe,EAAE,KAAK;IACtB,QAAQ,EAAE,OAAO,CAAC,GAAG,CAAC,cAAc;IACpC,MAAM,EAAE,OAAO;CAChB,CAAC,CAAA;AAEF,MAAM,SAAS,GAAG,IAAI,QAAQ,CAAC,cAAc,CAAC;IAC5C,WAAW,EAAE,KAAK;IAClB,eAAe,EAAE,KAAK;IACtB,QAAQ,EAAE,OAAO,CAAC,GAAG,CAAC,cAAc;IACpC,MAAM,EAAE,OAAO;CAChB,CAAC,CAAA;AAEF,MAAM,CAAC,MAAM,WAAW,GAAG,GAAS,EAAE;IACpC,MAAM,SAAS,GAAG,MAAM,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAA;IAExD,MAAM,GAAG;SACN,WAAW,CAAC;QACX,SAAS,EAAE,SAAS;QACpB,oBAAoB,EAAE;YACpB,EAAE,aAAa,EAAE,IAAI,EAAE,aAAa,EAAE,GAAG,EAAE;YAC3C,EAAE,aAAa,EAAE,IAAI,EAAE,aAAa,EAAE,GAAG,EAAE;YAC3C,GAAG,SAAS,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC;gBAC5B,EAAE,aAAa,EAAE,GAAG,GAAG,IAAI,EAAE,aAAa,EAAE,GAAG,EAAE;gBACjD,EAAE,aAAa,EAAE,GAAG,GAAG,IAAI,EAAE,aAAa,EAAE,GAAG,EAAE;aAClD,CAAC;SACH;QACD,SAAS,EAAE;YACT,EAAE,aAAa,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,EAAE;YACxC,EAAE,aAAa,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE;SAC1C;QACD,sBAAsB,EAAE;YACtB;gBACE,SAAS,EAAE,MAAM;gBACjB,SAAS,EAAE;oBACT,EAAE,aAAa,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,EAAE;oBACxC,EAAE,aAAa,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE;iBAC1C;gBACD,UAAU,EAAE;oBACV,cAAc,EAAE,KAAK;iBACtB;aACF;YACD,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;gBACzB,SAAS,EAAE,GAAG;gBACd,SAAS,EAAE;oBACT,EAAE,aAAa,EAAE,GAAG,GAAG,IAAI,EAAE,OAAO,EAAE,MAAM,EAAE;oBAC9C,EAAE,aAAa,EAAE,GAAG,GAAG,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE;iBAChD;gBACD,UAAU,EAAE;oBACV,cAAc,EAAE,KAAK;iBACtB;aACF,CAAC,CAAC;SACJ;QACD,WAAW,EAAE,iBAAiB;KAC/B,CAAC;SACD,OAAO,EAAE;SACT,KAAK,CAAC,CAAC,CAAM,EAAE,EAAE;QAChB,OAAO,CAAC,GAAG,CAAC,kCAAkC,EAAE,CAAC,CAAC,CAAA;QAClD,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;IACjB,CAAC,CAAC,CAAA;IAEJ,OAAO,SAAS,CAAA;AAClB,CAAC,CAAA,CAAA;AAED,MAAM,CAAC,MAAM,YAAY,GAAG,CAAO,SAAiB,EAAE,EAAE;IACtD,OAAO,GAAG;SACP,WAAW,CAAC,EAAE,SAAS,EAAE,SAAS,EAAE,CAAC;SACrC,OAAO,EAAE;SACT,IAAI,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC;SACd,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE;QACX,OAAO,CAAC,GAAG,CAAC,mCAAmC,EAAE,CAAC,CAAC,CAAA;QACnD,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;IACjB,CAAC,CAAC,CAAA;AACN,CAAC,CAAA,CAAA;AAED,MAAM,CAAC,MAAM,gBAAgB,GAAG,CAC9B,SAAiB,EACgB,EAAE;IACnC,MAAM,IAAI,GAAG,CAAO,iBAAuB,EAAkB,EAAE;QAC7D,MAAM,EAAE,KAAK,GAAG,EAAE,EAAE,gBAAgB,EAAE,GAAG,MAAM,SAAS;aACrD,IAAI,CAAC;YACJ,SAAS,EAAE,SAAS;YACpB,iBAAiB;SAClB,CAAC;aACD,OAAO,EAAE,CAAA;QAEZ,IAAI,gBAAgB;YAAE,OAAO,CAAC,GAAG,KAAK,EAAE,GAAG,CAAC,MAAM,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC,CAAA;QAC1E,OAAO,KAAK,CAAA;IACd,CAAC,CAAA,CAAA;IAED,MAAM,KAAK,GAAG,MAAM,IAAI,EAAE,CAAA;IAE1B,MAAM,GAAG,GAA2B,EAAE,CAAA;IAEtC,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,EAAE,KAAK,IAAI,CAAC,EAAE,EAAE,CAAC,GAAG,IAAI,CAAC,CAAC,CAAA;IAE/D,OAAO,GAAG,CAAA;AACZ,CAAC,CAAA,CAAA;AAUD,MAAM,CAAC,MAAM,aAAa,GAAG,CAAO,MAAc,EAAoB,EAAE;IACtE,MAAM,SAAS,GAAG,MAAM,WAAW,EAAE,CAAA;IAErC,MAAM,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAA;IACnC,MAAM,CAAC,YAAY,CAAC,SAAS,CAAC,CAAA;IAE9B,OAAO;QACL,OAAO,EAAE,GAAG,EAAE,CAAC,YAAY,CAAC,SAAS,CAAC;QACtC,QAAQ,EAAE,GAAG,EAAE,CAAC,gBAAgB,CAAC,SAAS,CAAC;QAC3C,IAAI,EAAE,CAAO,GAAG,IAAmC,EAAE,EAAE;YACrD,MAAM,MAAM,GAAG,QAAQ,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,CAAA;YAEjC,MAAM,OAAO,CAAC,GAAG,CACf,MAAM,CAAC,GAAG,CAAC,CAAO,KAAK,EAAE,EAAE;gBACzB,MAAM,KAAK,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;;oBAC5B,OAAA,OAAO,CAAA,MAAA,CAAC,aAAD,CAAC,uBAAD,CAAC,CAAE,MAAM,0CAAE,gBAAgB,CAAA,KAAK,UAAU;wBAC/C,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC,CAAC;wBAC9B,CAAC,CAAC,OAAO,CAAC,CAAC,MAAM,KAAK,UAAU;4BAChC,CAAC,CAAC,CAAC,CAAC,MAAM,EAAE;4BACZ,CAAC,CAAC,CAAC,CAAA;iBAAA,CACN,CAAA;gBAED,OAAO,MAAM,CAAC,cAAc;qBACzB,UAAU,CAAC;oBACV,YAAY,EAAE;wBACZ,CAAC,SAAS,CAAC,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,UAAU,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;qBAC7D;iBACF,CAAC;qBACD,OAAO,EAAE,CAAA;YACd,CAAC,CAAA,CAAC,CACH,CAAA;QACH,CAAC,CAAA;QACD,GAAG,EAAE,CAAC,EAAU,EAAE,EAAU,EAAE,EAAE,CAC9B,MAAM,CAAC,cAAc;aAClB,GAAG,CAAC,EAAE,SAAS,EAAE,SAAS,EAAE,GAAG,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC;aACtD,OAAO,EAAE;aACT,IAAI,CAAC,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,IAAI,aAAJ,IAAI,cAAJ,IAAI,GAAI,IAAI,CAAC;QACrC,IAAI,EAAE,CAAO,MAAM,EAAE,EAAE;YACrB,MAAM,QAAQ,GAAG,MAAM,gBAAgB,CAAC,SAAS,CAAC,CAAA;YAElD,OAAO,kBAAkB,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAA;QAC7C,CAAC,CAAA;KACF,CAAA;AACH,CAAC,CAAA,CAAA"}
+{"version":3,"file":"sandbox.js","sourceRoot":"","sources":["../../src/sandbox.ts"],"names":[],"mappings":";AAAA,OAAO,MAAM,MAAM,QAAQ,CAAA;AAC3B,OAAO,EAAE,QAAQ,EAAE,MAAM,iBAAiB,CAAA;AAC1C,OAAO,QAAQ,MAAM,0BAA0B,CAAA;AAC/C,OAAO,EAAE,kBAAkB,EAAE,MAAM,QAAQ,CAAA;AAE3C,OAAO,EAAE,SAAS,EAAE,MAAM,OAAO,CAAA;AACjC,OAAO,EAAE,4BAA4B,EAAE,MAAM,aAAa,CAAA;AAE1D,MAAM,GAAG,GAAG,IAAI,QAAQ,CAAC;IACvB,WAAW,EAAE,KAAK;IAClB,eAAe,EAAE,KAAK;IACtB,QAAQ,EAAE,OAAO,CAAC,GAAG,CAAC,cAAc;IACpC,MAAM,EAAE,OAAO;CAChB,CAAC,CAAA;AAEF,MAAM,SAAS,GAAG,IAAI,QAAQ,CAAC,cAAc,CAAC;IAC5C,WAAW,EAAE,KAAK;IAClB,eAAe,EAAE,KAAK;IACtB,QAAQ,EAAE,OAAO,CAAC,GAAG,CAAC,cAAc;IACpC,MAAM,EAAE,OAAO;CAChB,CAAC,CAAA;AAEF,MAAM,CAAC,MAAM,WAAW,GAAG,GAAS,EAAE;IACpC,MAAM,SAAS,GAAG,MAAM,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAA;IAExD,MAAM,GAAG;SACN,WAAW,CAAC;QACX,SAAS,EAAE,SAAS;QACpB,oBAAoB,EAAE;YACpB,EAAE,aAAa,EAAE,IAAI,EAAE,aAAa,EAAE,GAAG,EAAE;YAC3C,EAAE,aAAa,EAAE,IAAI,EAAE,aAAa,EAAE,GAAG,EAAE;YAC3C,GAAG,SAAS,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC;gBAC5B,EAAE,aAAa,EAAE,GAAG,GAAG,IAAI,EAAE,aAAa,EAAE,GAAG,EAAE;gBACjD,EAAE,aAAa,EAAE,GAAG,GAAG,IAAI,EAAE,aAAa,EAAE,GAAG,EAAE;aAClD,CAAC;SACH;QACD,SAAS,EAAE;YACT,EAAE,aAAa,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,EAAE;YACxC,EAAE,aAAa,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE;SAC1C;QACD,sBAAsB,EAAE;YACtB;gBACE,SAAS,EAAE,MAAM;gBACjB,SAAS,EAAE;oBACT,EAAE,aAAa,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,EAAE;oBACxC,EAAE,aAAa,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE;iBAC1C;gBACD,UAAU,EAAE;oBACV,cAAc,EAAE,KAAK;iBACtB;aACF;YACD,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;gBACzB,SAAS,EAAE,GAAG;gBACd,SAAS,EAAE;oBACT,EAAE,aAAa,EAAE,GAAG,GAAG,IAAI,EAAE,OAAO,EAAE,MAAM,EAAE;oBAC9C,EAAE,aAAa,EAAE,GAAG,GAAG,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE;iBAChD;gBACD,UAAU,EAAE;oBACV,cAAc,EAAE,KAAK;iBACtB;aACF,CAAC,CAAC;SACJ;QACD,WAAW,EAAE,iBAAiB;KAC/B,CAAC;SACD,OAAO,EAAE;SACT,KAAK,CAAC,CAAC,CAAM,EAAE,EAAE;QAChB,OAAO,CAAC,GAAG,CAAC,kCAAkC,EAAE,CAAC,CAAC,CAAA;QAClD,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;IACjB,CAAC,CAAC,CAAA;IAEJ,OAAO,SAAS,CAAA;AAClB,CAAC,CAAA,CAAA;AAED,MAAM,CAAC,MAAM,YAAY,GAAG,CAAO,SAAiB,EAAE,EAAE;IACtD,OAAO,GAAG;SACP,WAAW,CAAC,EAAE,SAAS,EAAE,SAAS,EAAE,CAAC;SACrC,OAAO,EAAE;SACT,IAAI,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC;SACd,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE;QACX,OAAO,CAAC,GAAG,CAAC,mCAAmC,EAAE,CAAC,CAAC,CAAA;QACnD,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;IACjB,CAAC,CAAC,CAAA;AACN,CAAC,CAAA,CAAA;AAED,MAAM,CAAC,MAAM,gBAAgB,GAAG,CAC9B,SAAiB,EACgB,EAAE;IACnC,MAAM,IAAI,GAAG,CAAO,iBAAuB,EAAkB,EAAE;QAC7D,MAAM,EAAE,KAAK,GAAG,EAAE,EAAE,gBAAgB,EAAE,GAAG,MAAM,SAAS;aACrD,IAAI,CAAC;YACJ,SAAS,EAAE,SAAS;YACpB,iBAAiB;SAClB,CAAC;aACD,OAAO,EAAE,CAAA;QAEZ,IAAI,gBAAgB;YAAE,OAAO,CAAC,GAAG,KAAK,EAAE,GAAG,CAAC,MAAM,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC,CAAA;QAC1E,OAAO,KAAK,CAAA;IACd,CAAC,CAAA,CAAA;IAED,MAAM,KAAK,GAAG,MAAM,IAAI,EAAE,CAAA;IAE1B,MAAM,GAAG,GAA2B,EAAE,CAAA;IAEtC,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,EAAE,KAAK,IAAI,CAAC,EAAE,EAAE,CAAC,GAAG,IAAI,CAAC,CAAC,CAAA;IAE/D,OAAO,GAAG,CAAA;AACZ,CAAC,CAAA,CAAA;AAYD,MAAM,aAAa,GAAG,IAAI,GAAG,CAAC;IAC5B,KAAK;IACL,QAAQ;IACR,QAAQ;IACR,YAAY;IACZ,eAAe;CAChB,CAAC,CAAA;AAEF,SAAS,sBAAsB,CAC7B,QAAiC,EACjC,SAAiB;IAEjB,IAAI,UAAU,GAAG,KAAK,CAAA;IACtB,MAAM,WAAW,GAAG,IAAI,GAAG,EAAwB,CAAA;IAEnD,MAAM,UAAU,GAAG,CAAO,EAAU,EAAE,EAAU,EAAE,EAAE;QAClD,MAAM,YAAY,GAAG,GAAG,EAAE,KAAK,EAAE,EAAE,CAAA;QACnC,IAAI,WAAW,CAAC,GAAG,CAAC,YAAY,CAAC;YAAE,OAAM;QAEzC,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,QAAQ;aAC5B,GAAG,CAAC,EAAE,SAAS,EAAE,SAAS,EAAE,GAAG,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC;aACtD,OAAO,EAAE,CAAA;QAEZ,WAAW,CAAC,GAAG,CAAC,YAAY,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,QAAQ,EAAE,IAAI,aAAJ,IAAI,cAAJ,IAAI,GAAI,IAAI,EAAE,CAAC,CAAA;IACnE,CAAC,CAAA,CAAA;IAED,MAAM,uBAAuB,GAAG,CAAO,MAAc,EAAE,MAAW,EAAE,EAAE;;QACpE,QAAQ,MAAM,EAAE;YACd,KAAK,KAAK;gBACR,IAAI,MAAM,CAAC,SAAS,KAAK,SAAS,IAAI,MAAM,CAAC,IAAI,EAAE;oBACjD,MAAM,UAAU,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,EAAE,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAA;iBACjD;gBACD,MAAK;YACP,KAAK,QAAQ,CAAC;YACd,KAAK,QAAQ;gBACX,IAAI,MAAM,CAAC,SAAS,KAAK,SAAS,IAAI,MAAM,CAAC,GAAG,EAAE;oBAChD,MAAM,UAAU,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,EAAE,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAA;iBAC/C;gBACD,MAAK;YACP,KAAK,YAAY,CAAC,CAAC;gBACjB,MAAM,UAAU,GAAG,CAAA,MAAA,MAAM,CAAC,YAAY,0CAAG,SAAS,CAAC,KAAI,EAAE,CAAA;gBACzD,MAAM,OAAO,CAAC,GAAG,CACf,UAAU,CAAC,GAAG,CAAC,CAAC,IAAS,EAAE,EAAE;oBAC3B,IAAI,IAAI,CAAC,UAAU,EAAE;wBACnB,OAAO,UAAU,CACf,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,EAAE,EACvB,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,EAAE,CACxB,CAAA;qBACF;oBACD,IAAI,IAAI,CAAC,aAAa,EAAE;wBACtB,OAAO,UAAU,CACf,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,EAAE,EACzB,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,EAAE,CAC1B,CAAA;qBACF;gBACH,CAAC,CAAC,CACH,CAAA;gBACD,MAAK;aACN;YACD,KAAK,eAAe,CAAC,CAAC;gBACpB,MAAM,aAAa,GAAG,MAAM,CAAC,aAAa,IAAI,EAAE,CAAA;gBAChD,MAAM,OAAO,CAAC,GAAG,CACf,aAAa;qBACV,GAAG,CAAC,CAAC,IAAS,EAAE,EAAE;;oBACjB,IAAI,CAAA,MAAA,IAAI,CAAC,GAAG,0CAAE,SAAS,MAAK,SAAS,EAAE;wBACrC,OAAO,UAAU,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAA;qBACtD;oBACD,IAAI,CAAA,MAAA,IAAI,CAAC,MAAM,0CAAE,SAAS,MAAK,SAAS,EAAE;wBACxC,OAAO,UAAU,CAAC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,EAAE,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAA;qBAC1D;oBACD,IAAI,CAAA,MAAA,IAAI,CAAC,MAAM,0CAAE,SAAS,MAAK,SAAS,EAAE;wBACxC,OAAO,UAAU,CAAC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,EAAE,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAA;qBAC1D;gBACH,CAAC,CAAC;qBACD,MAAM,CAAC,OAAO,CAAC,CACnB,CAAA;gBACD,MAAK;aACN;SACF;IACH,CAAC,CAAA,CAAA;IAED,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,QAAQ,EAAE;QAChC,GAAG,CAAC,MAAM,EAAE,IAAI;YACd,MAAM,KAAK,GAAI,MAAc,CAAC,IAAI,CAAC,CAAA;YACnC,IAAI,KAAK,KAAK,SAAS;gBAAE,OAAO,SAAS,CAAA;YAEzC,IAAI,OAAO,KAAK,KAAK,UAAU,EAAE;gBAC/B,IAAI,UAAU,IAAI,aAAa,CAAC,GAAG,CAAC,IAAc,CAAC,EAAE;oBACnD,OAAO,CAAC,MAAW,EAAE,EAAE;wBACrB,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;wBAC1C,MAAM,WAAW,GAAG,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;wBACjD,OAAO,CAAC,OAAO,GAAG,GAAS,EAAE;4BAC3B,MAAM,uBAAuB,CAAC,IAAc,EAAE,MAAM,CAAC,CAAA;4BACrD,OAAO,WAAW,EAAE,CAAA;wBACtB,CAAC,CAAA,CAAA;wBACD,OAAO,OAAO,CAAA;oBAChB,CAAC,CAAA;iBACF;gBACD,OAAO,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;aAC1B;YAED,OAAO,KAAK,CAAA;QACd,CAAC;KACF,CAAC,CAAA;IAEF,OAAO;QACL,KAAK,EAAE,KAAgC;QACvC,aAAa,EAAE,GAAG,EAAE;YAClB,UAAU,GAAG,IAAI,CAAA;YACjB,WAAW,CAAC,KAAK,EAAE,CAAA;QACrB,CAAC;QACD,QAAQ,EAAE,GAAS,EAAE;YACnB,UAAU,GAAG,KAAK,CAAA;YAElB,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,CAAA;YAChD,MAAM,QAAQ,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,KAAK,IAAI,CAAC,CAAA;YAC3D,MAAM,SAAS,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,KAAK,IAAI,CAAC,CAAA;YAE5D,MAAM,YAAY,GAAG,QAAQ,CAAC,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAA;YAC3C,MAAM,aAAa,GAAG,QAAQ,CAAC,EAAE,CAAC,CAAC,SAAS,CAAC,CAAA;YAE7C,MAAM,OAAO,CAAC,GAAG,CAAC;gBAChB,GAAG,YAAY,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAC5B,QAAQ;qBACL,UAAU,CAAC;oBACV,YAAY,EAAE;wBACZ,CAAC,SAAS,CAAC,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC;4BACtC,aAAa,EAAE,EAAE,GAAG,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE;yBAC3C,CAAC,CAAC;qBACJ;iBACF,CAAC;qBACD,OAAO,EAAE,CACb;gBACD,GAAG,aAAa,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAC7B,QAAQ;qBACL,UAAU,CAAC;oBACV,YAAY,EAAE;wBACZ,CAAC,SAAS,CAAC,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC;4BAC9C,UAAU,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE;yBAC3B,CAAC,CAAC;qBACJ;iBACF,CAAC;qBACD,OAAO,EAAE,CACb;aACF,CAAC,CAAA;YAEF,WAAW,CAAC,KAAK,EAAE,CAAA;QACrB,CAAC,CAAA;KACF,CAAA;AACH,CAAC;AAgBD,MAAM,CAAC,MAAM,aAAa,GAAG,CAAO,MAAc,EAAoB,EAAE;IACtE,IAAI,OAAO,CAAC,GAAG,CAAC,+BAA+B,KAAK,GAAG,EAAE;QACvD,MAAM,SAAS,GAAG,MAAM,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAA;QACxD,MAAM,cAAc,GAClB,4BAA4B,EAG3B,CAAA;QAEH,MAAM,OAAO,GAAG,sBAAsB,CAAC,cAAc,EAAE,SAAS,CAAC,CAAA;QAEjE,MAAM,CAAC,iBAAiB,CAAC,OAAO,CAAC,KAAK,CAAC,CAAA;QACvC,MAAM,CAAC,YAAY,CAAC,SAAS,CAAC,CAAA;QAE9B,OAAO;YACL,OAAO,EAAE,GAAS,EAAE;gBAClB,cAAc,CAAC,oBAAoB,CAAC,SAAS,CAAC,CAAA;YAChD,CAAC,CAAA;YACD,QAAQ,EAAE,GAAS,EAAE,kDAAC,OAAA,cAAc,CAAC,kBAAkB,CAAC,SAAS,CAAC,CAAA,GAAA;YAClE,IAAI,EAAE,CAAO,GAAG,IAAmC,EAAE,EAAE;gBACrD,MAAM,MAAM,GAAG,QAAQ,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,CAAA;gBAEjC,MAAM,OAAO,CAAC,GAAG,CACf,MAAM,CAAC,GAAG,CAAC,CAAO,KAAK,EAAE,EAAE;oBACzB,MAAM,KAAK,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;;wBAC5B,OAAA,OAAO,CAAA,MAAA,CAAC,aAAD,CAAC,uBAAD,CAAC,CAAE,MAAM,0CAAE,gBAAgB,CAAA,KAAK,UAAU;4BAC/C,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC,CAAC;4BAC9B,CAAC,CAAC,OAAO,CAAC,CAAC,MAAM,KAAK,UAAU;gCAChC,CAAC,CAAC,CAAC,CAAC,MAAM,EAAE;gCACZ,CAAC,CAAC,CAAC,CAAA;qBAAA,CACN,CAAA;oBAED,OAAO,MAAM,CAAC,cAAc;yBACzB,UAAU,CAAC;wBACV,YAAY,EAAE;4BACZ,CAAC,SAAS,CAAC,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,UAAU,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;yBAC7D;qBACF,CAAC;yBACD,OAAO,EAAE,CAAA;gBACd,CAAC,CAAA,CAAC,CACH,CAAA;YACH,CAAC,CAAA;YACD,GAAG,EAAE,CAAC,EAAU,EAAE,EAAU,EAAE,EAAE,CAC9B,MAAM,CAAC,cAAc;iBAClB,GAAG,CAAC,EAAE,SAAS,EAAE,SAAS,EAAE,GAAG,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC;iBACtD,OAAO,EAAE;iBACT,IAAI,CAAC,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,IAAI,aAAJ,IAAI,cAAJ,IAAI,GAAI,IAAI,CAAC;YACrC,IAAI,EAAE,CAAO,MAAM,EAAE,EAAE;gBACrB,MAAM,QAAQ,GAAG,cAAc,CAAC,kBAAkB,CAAC,SAAS,CAAC,CAAA;gBAC7D,OAAO,kBAAkB,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAA;YAC7C,CAAC,CAAA;YACD,aAAa,EAAE,OAAO,CAAC,aAAa;YACpC,QAAQ,EAAE,OAAO,CAAC,QAAQ;SAC3B,CAAA;KACF;IAED,MAAM,SAAS,GAAG,MAAM,WAAW,EAAE,CAAA;IAErC,MAAM,OAAO,GAAG,sBAAsB,CAAC,SAAS,EAAE,SAAS,CAAC,CAAA;IAC5D,MAAM,CAAC,iBAAiB,CAAC,OAAO,CAAC,KAAK,CAAC,CAAA;IACvC,MAAM,CAAC,YAAY,CAAC,SAAS,CAAC,CAAA;IAE9B,OAAO;QACL,OAAO,EAAE,GAAG,EAAE,CAAC,YAAY,CAAC,SAAS,CAAC;QACtC,QAAQ,EAAE,GAAG,EAAE,CAAC,gBAAgB,CAAC,SAAS,CAAC;QAC3C,IAAI,EAAE,CAAO,GAAG,IAAmC,EAAE,EAAE;YACrD,MAAM,MAAM,GAAG,QAAQ,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,CAAA;YAEjC,MAAM,OAAO,CAAC,GAAG,CACf,MAAM,CAAC,GAAG,CAAC,CAAO,KAAK,EAAE,EAAE;gBACzB,MAAM,KAAK,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;;oBAC5B,OAAA,OAAO,CAAA,MAAA,CAAC,aAAD,CAAC,uBAAD,CAAC,CAAE,MAAM,0CAAE,gBAAgB,CAAA,KAAK,UAAU;wBAC/C,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC,CAAC;wBAC9B,CAAC,CAAC,OAAO,CAAC,CAAC,MAAM,KAAK,UAAU;4BAChC,CAAC,CAAC,CAAC,CAAC,MAAM,EAAE;4BACZ,CAAC,CAAC,CAAC,CAAA;iBAAA,CACN,CAAA;gBAED,OAAO,MAAM,CAAC,cAAc;qBACzB,UAAU,CAAC;oBACV,YAAY,EAAE;wBACZ,CAAC,SAAS,CAAC,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,UAAU,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;qBAC7D;iBACF,CAAC;qBACD,OAAO,EAAE,CAAA;YACd,CAAC,CAAA,CAAC,CACH,CAAA;QACH,CAAC,CAAA;QACD,GAAG,EAAE,CAAC,EAAU,EAAE,EAAU,EAAE,EAAE,CAC9B,MAAM,CAAC,cAAc;aAClB,GAAG,CAAC,EAAE,SAAS,EAAE,SAAS,EAAE,GAAG,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC;aACtD,OAAO,EAAE;aACT,IAAI,CAAC,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,IAAI,aAAJ,IAAI,cAAJ,IAAI,GAAI,IAAI,CAAC;QACrC,IAAI,EAAE,CAAO,MAAM,EAAE,EAAE;YACrB,MAAM,QAAQ,GAAG,MAAM,gBAAgB,CAAC,SAAS,CAAC,CAAA;YAElD,OAAO,kBAAkB,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAA;QAC7C,CAAC,CAAA;QACD,aAAa,EAAE,OAAO,CAAC,aAAa;QACpC,QAAQ,EAAE,OAAO,CAAC,QAAQ;KAC3B,CAAA;AACH,CAAC,CAAA,CAAA"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@model-ts/dynamodb",
-  "version": "4.0.0",
+  "version": "4.2.0",
   "main": "dist/cjs/index.js",
   "module": "dist/esm/index.js",
   "author": "Fabian Finke <finkef@icloud.com>",
@@ -9,6 +9,7 @@
     "build": "tsc -b",
     "build:esm": "tsc -b tsconfig.esm.json",
     "test": "TZ=UTC LOCAL_ENDPOINT='http://localhost:8000' jest",
+    "test:memory": "EXPERIMENTAL_DYNAMODB_IN_MEMORY=1 TZ=UTC LOCAL_ENDPOINT='http://localhost:8000' jest",
     "dynamodb:start": "docker run -d --name dynamodb --rm -p 8000:8000 amazon/dynamodb-local",
     "dynamodb:stop": "docker stop dynamodb"
   },

package/src/__test__/client-env-guard.test.ts

@@ -0,0 +1,31 @@
+import { Client } from "../client"
+
+describe("client env guard", () => {
+  const originalNodeEnv = process.env.NODE_ENV
+  const originalInMemory = process.env.EXPERIMENTAL_DYNAMODB_IN_MEMORY
+
+  afterEach(() => {
+    if (typeof originalNodeEnv === "undefined") delete process.env.NODE_ENV
+    else process.env.NODE_ENV = originalNodeEnv
+
+    if (typeof originalInMemory === "undefined")
+      delete process.env.EXPERIMENTAL_DYNAMODB_IN_MEMORY
+    else process.env.EXPERIMENTAL_DYNAMODB_IN_MEMORY = originalInMemory
+  })
+
+  test("allows in-memory mode in test env", () => {
+    process.env.NODE_ENV = "test"
+    process.env.EXPERIMENTAL_DYNAMODB_IN_MEMORY = "1"
+
+    expect(() => new Client({ tableName: "table" })).not.toThrow()
+  })
+
+  test("throws when in-memory mode is enabled outside test env", () => {
+    process.env.NODE_ENV = "development"
+    process.env.EXPERIMENTAL_DYNAMODB_IN_MEMORY = "1"
+
+    expect(() => new Client({ tableName: "table" })).toThrow(
+      'EXPERIMENTAL_DYNAMODB_IN_MEMORY=1 is only allowed when NODE_ENV is "test"'
+    )
+  })
+})