@webiny/db-dynamodb 0.0.0-unstable.9bd236cf5e → 0.0.0-unstable.9f53ea597d

package/DynamoDbDriver.d.ts

@@ -1,17 +1,13 @@
-import type { DynamoDBDocument } from "@webiny/aws-sdk/client-dynamodb";
+import type { DynamoDBDocument } from "@webiny/aws-sdk/client-dynamodb/index.js";
 import type { DbDriver, GetValueResult, GetValuesResult, IListValuesParams, ListValuesResult, RemoveValueResult, RemoveValuesResult, StorageKey, StoreValueResult, StoreValuesResult } from "@webiny/db";
-import type { Entity } from "dynamodb-toolbox";
-import type { Table } from "./utils/createTable";
-import type { GenericRecord } from "@webiny/api/types";
+import type { GenericRecord } from "@webiny/api/types.js";
 interface ConstructorArgs {
     documentClient: DynamoDBDocument;
 }
 declare class DynamoDbDriver implements DbDriver<DynamoDBDocument> {
     readonly documentClient: DynamoDBDocument;
-    private _table;
-    private _entity;
-    table(): Table;
-    entity(): Entity;
+    readonly table: import("./utils").ITable<string, "PK", "SK">;
+    readonly entity: import("./store/types").IStoreEntity;
     constructor({ documentClient }: ConstructorArgs);
     getClient(): DynamoDBDocument;
     storeValue<V>(key: string, input: V): Promise<StoreValueResult<V>>;

package/DynamoDbDriver.js

@@ -1,264 +1,228 @@
-"use strict";
-
-Object.defineProperty(exports, "__esModule", {
-  value: true
-});
-exports.default = void 0;
-var _createTable = require("./utils/createTable");
-var _entity = require("./store/entity");
-var _utils = require("./utils");
-var _keys = require("./store/keys");
+import { createTable } from "./utils/createTable.js";
+import { createEntity } from "./store/entity.js";
+import { createPartitionKey, createSortKey, createType } from "./store/keys.js";
 class DynamoDbDriver {
-  _table = undefined;
-  _entity = undefined;
-  table() {
-    if (this._table) {
-      return this._table;
-    }
-    this._table = (0, _createTable.createTable)({
-      documentClient: this.documentClient
-    });
-    return this._table;
-  }
-  entity() {
-    if (this._entity) {
-      return this._entity;
+    constructor({ documentClient }){
+        this.documentClient = documentClient;
+        this.table = createTable({
+            name: process.env.DB_TABLE,
+            documentClient
+        });
+        this.entity = createEntity({
+            table: this.table
+        });
     }
-    this._entity = (0, _entity.createEntity)({
-      table: this.table()
-    });
-    return this._entity;
-  }
-  constructor({
-    documentClient
-  }) {
-    this.documentClient = documentClient;
-  }
-  getClient() {
-    return this.documentClient;
-  }
-  async storeValue(key, input) {
-    let value;
-    try {
-      value = JSON.stringify(input);
-    } catch (ex) {
-      return {
-        key,
-        error: ex
-      };
+    getClient() {
+        return this.documentClient;
     }
-    try {
-      await (0, _utils.put)({
-        entity: this.entity(),
-        item: {
-          PK: (0, _keys.createPartitionKey)(),
-          SK: (0, _keys.createSortKey)({
-            key
-          }),
-          TYPE: (0, _keys.createType)(),
-          key,
-          value
+    async storeValue(key, input) {
+        let value;
+        try {
+            value = JSON.stringify(input);
+        } catch (ex) {
+            return {
+                key,
+                error: ex
+            };
+        }
+        try {
+            await this.entity.put({
+                PK: createPartitionKey(),
+                SK: createSortKey({
+                    key
+                }),
+                TYPE: createType(),
+                data: {
+                    key,
+                    value
+                }
+            });
+            return {
+                key,
+                data: input
+            };
+        } catch (ex) {
+            return {
+                key,
+                error: ex
+            };
         }
-      });
-      return {
-        key,
-        data: input
-      };
-    } catch (ex) {
-      return {
-        key,
-        error: ex
-      };
     }
-  }
-  async storeValues(values) {
-    const keys = Object.keys(values);
-    try {
-      const batch = keys.map(key => {
-        const input = values[key];
-        let value;
+    async storeValues(values) {
+        const keys = Object.keys(values);
+        const batchWrite = this.entity.createEntityWriter();
         try {
-          value = JSON.stringify(input);
+            for (const key of keys){
+                const input = values[key];
+                let value;
+                try {
+                    value = JSON.stringify(input);
+                } catch (ex) {
+                    throw ex;
+                }
+                batchWrite.put({
+                    PK: createPartitionKey(),
+                    SK: createSortKey({
+                        key
+                    }),
+                    TYPE: createType(),
+                    data: {
+                        key,
+                        value
+                    }
+                });
+            }
+            await batchWrite.execute();
+            return {
+                keys,
+                data: values
+            };
         } catch (ex) {
-          throw ex;
+            return {
+                keys,
+                error: ex
+            };
         }
-        const item = {
-          PK: (0, _keys.createPartitionKey)(),
-          SK: (0, _keys.createSortKey)({
-            key
-          }),
-          TYPE: (0, _keys.createType)(),
-          key,
-          value
-        };
-        return this.entity().putBatch(item);
-      });
-      await (0, _utils.batchWriteAll)({
-        table: this.table(),
-        items: batch
-      });
-      return {
-        keys,
-        data: values
-      };
-    } catch (ex) {
-      return {
-        keys,
-        error: ex
-      };
     }
-  }
-  async getValue(key) {
-    try {
-      const result = await (0, _utils.get)({
-        entity: this.entity(),
-        keys: {
-          PK: (0, _keys.createPartitionKey)(),
-          SK: (0, _keys.createSortKey)({
-            key
-          })
+    async getValue(key) {
+        try {
+            const result = await this.entity.get({
+                PK: createPartitionKey(),
+                SK: createSortKey({
+                    key
+                })
+            });
+            return {
+                key,
+                data: result?.data?.value ? JSON.parse(result.data.value) : null
+            };
+        } catch (ex) {
+            return {
+                key,
+                error: ex
+            };
         }
-      });
-      return {
-        key,
-        data: result ? JSON.parse(result.value) : null
-      };
-    } catch (ex) {
-      return {
-        key,
-        error: ex
-      };
     }
-  }
-  async getValues(input) {
-    const keys = [...input];
-    const batch = keys.map(key => {
-      return this.entity().getBatch({
-        PK: (0, _keys.createPartitionKey)(),
-        SK: (0, _keys.createSortKey)({
-          key
-        })
-      });
-    });
-    try {
-      const results = await (0, _utils.batchReadAll)({
-        table: this.table(),
-        items: batch
-      });
-      const data = keys.reduce((collection, key) => {
-        const result = results.find(item => {
-          return item.PK === (0, _keys.createPartitionKey)() && item.SK === (0, _keys.createSortKey)({
-            key
-          });
+    async getValues(input) {
+        const keys = [
+            ...input
+        ];
+        const batchRead = this.entity.createEntityReader({
+            read: keys.map((key)=>({
+                    PK: createPartitionKey(),
+                    SK: createSortKey({
+                        key
+                    })
+                }))
         });
-        if (!result?.value) {
-          // @ts-expect-error
-          collection[key] = null;
-          return collection;
-        }
         try {
-          // @ts-expect-error
-          collection[key] = JSON.parse(result.value);
-        } catch {
-          // @ts-expect-error
-          collection[key] = null;
+            const results = await batchRead.execute();
+            const data = keys.reduce((collection, initialKey)=>{
+                const key = initialKey;
+                const result = results.find((item)=>item.PK === createPartitionKey() && item.SK === createSortKey({
+                        key
+                    }));
+                if (!result?.data?.value) {
+                    collection[key] = null;
+                    return collection;
+                }
+                try {
+                    collection[key] = JSON.parse(result.data.value);
+                } catch  {
+                    collection[key] = null;
+                }
+                return collection;
+            }, {});
+            return {
+                keys,
+                data
+            };
+        } catch (ex) {
+            return {
+                keys,
+                error: ex
+            };
         }
-        return collection;
-      }, {});
-      return {
-        keys,
-        data
-      };
-    } catch (ex) {
-      return {
-        keys,
-        error: ex
-      };
     }
-  }
-  async listValues(params) {
-    try {
-      const partitionKey = (0, _keys.createPartitionKey)();
-      const options = {
-        ...params
-      };
-      const results = await (0, _utils.queryAll)({
-        entity: this.entity(),
-        partitionKey,
-        options
-      });
-      const data = results.reduce((collection, item) => {
+    async listValues(params) {
         try {
-          // @ts-expect-error
-          collection[item.key] = JSON.parse(item.value);
+            const partitionKey = createPartitionKey();
+            const options = {
+                ...params
+            };
+            const results = await this.entity.queryAll({
+                partitionKey,
+                options
+            });
+            const data = results.reduce((collection, item)=>{
+                const key = item.data.key;
+                try {
+                    collection[key] = JSON.parse(item.data.value);
+                } catch  {
+                    collection[key] = null;
+                }
+                return collection;
+            }, {});
+            return {
+                keys: Object.keys(data),
+                data
+            };
         } catch (ex) {
-          // @ts-expect-error
-          collection[item.key] = null;
+            return {
+                error: ex
+            };
         }
-        return collection;
-      }, {});
-      return {
-        keys: Object.keys(data),
-        data
-      };
-    } catch (ex) {
-      return {
-        error: ex
-      };
     }
-  }
-  async removeValue(key) {
-    const result = await this.getValue(key);
-    if (result.error) {
-      return {
-        key,
-        error: result.error
-      };
-    }
-    try {
-      await this.entity().delete({
-        PK: (0, _keys.createPartitionKey)(),
-        SK: (0, _keys.createSortKey)({
-          key
-        })
-      });
-      return {
-        key,
-        data: result.data
-      };
-    } catch (ex) {
-      return {
-        key,
-        error: ex
-      };
+    async removeValue(key) {
+        const result = await this.getValue(key);
+        if (result.error) return {
+            key,
+            error: result.error
+        };
+        try {
+            await this.entity.delete({
+                PK: createPartitionKey(),
+                SK: createSortKey({
+                    key
+                })
+            });
+            return {
+                key,
+                data: result.data
+            };
+        } catch (ex) {
+            return {
+                key,
+                error: ex
+            };
+        }
     }
-  }
-  async removeValues(input) {
-    const keys = [...input];
-    const batch = keys.map(key => {
-      return this.entity().deleteBatch({
-        PK: (0, _keys.createPartitionKey)(),
-        SK: (0, _keys.createSortKey)({
-          key
-        })
-      });
-    });
-    try {
-      await (0, _utils.batchWriteAll)({
-        table: this.table(),
-        items: batch
-      });
-      return {
-        keys
-      };
-    } catch (ex) {
-      return {
-        keys,
-        error: ex
-      };
+    async removeValues(input) {
+        const keys = [
+            ...input
+        ];
+        const batchDelete = this.entity.createEntityWriter({
+            delete: keys.map((key)=>({
+                    PK: createPartitionKey(),
+                    SK: createSortKey({
+                        key
+                    })
+                }))
+        });
+        try {
+            await batchDelete.execute();
+            return {
+                keys
+            };
+        } catch (ex) {
+            return {
+                keys,
+                error: ex
+            };
+        }
     }
-  }
 }
-var _default = exports.default = DynamoDbDriver;
+const src_DynamoDbDriver = DynamoDbDriver;
+export default src_DynamoDbDriver;
 
 //# sourceMappingURL=DynamoDbDriver.js.map

package/DynamoDbDriver.js.map

@@ -1 +1 @@
-{"version":3,"names":["_createTable","require","_entity","_utils","_keys","DynamoDbDriver","_table","undefined","table","createTable","documentClient","entity","createEntity","constructor","getClient","storeValue","key","input","value","JSON","stringify","ex","error","put","item","PK","createPartitionKey","SK","createSortKey","TYPE","createType","data","storeValues","values","keys","Object","batch","map","putBatch","batchWriteAll","items","getValue","result","get","parse","getValues","getBatch","results","batchReadAll","reduce","collection","find","listValues","params","partitionKey","options","queryAll","removeValue","delete","removeValues","deleteBatch","_default","exports","default"],"sources":["DynamoDbDriver.ts"],"sourcesContent":["import type { DynamoDBDocument } from \"@webiny/aws-sdk/client-dynamodb\";\nimport type {\n    DbDriver,\n    GetValueResult,\n    GetValuesResult,\n    IListValuesParams,\n    ListValuesResult,\n    RemoveValueResult,\n    RemoveValuesResult,\n    StorageKey,\n    StoreValueResult,\n    StoreValuesResult\n} from \"@webiny/db\";\nimport type { Entity } from \"dynamodb-toolbox\";\nimport type { Table } from \"~/utils/createTable\";\nimport { createTable } from \"~/utils/createTable\";\nimport type { GenericRecord } from \"@webiny/api/types\";\nimport { createEntity } from \"~/store/entity\";\nimport { batchReadAll, batchWriteAll, get, put, queryAll } from \"~/utils\";\nimport { createPartitionKey, createSortKey, createType } from \"~/store/keys\";\nimport type { IStoreItem } from \"~/store/types\";\n\ninterface ConstructorArgs {\n    documentClient: DynamoDBDocument;\n}\n\nclass DynamoDbDriver implements DbDriver<DynamoDBDocument> {\n    public readonly documentClient: DynamoDBDocument;\n\n    private _table: Table | undefined = undefined;\n    private _entity: Entity | undefined = undefined;\n\n    public table(): Table {\n        if (this._table) {\n            return this._table;\n        }\n        this._table = createTable({\n            documentClient: this.documentClient\n        });\n        return this._table;\n    }\n\n    public entity(): Entity {\n        if (this._entity) {\n            return this._entity;\n        }\n        this._entity = createEntity({\n            table: this.table()\n        });\n        return this._entity;\n    }\n\n    constructor({ documentClient }: ConstructorArgs) {\n        this.documentClient = documentClient;\n    }\n\n    public getClient() {\n        return this.documentClient;\n    }\n\n    public async storeValue<V>(key: string, input: V): Promise<StoreValueResult<V>> {\n        let value: string | undefined;\n        try {\n            value = JSON.stringify(input);\n        } catch (ex) {\n            return {\n                key,\n                error: ex\n            };\n        }\n\n        try {\n            await put<IStoreItem>({\n                entity: this.entity(),\n                item: {\n                    PK: createPartitionKey(),\n                    SK: createSortKey({ key }),\n                    TYPE: createType(),\n                    key,\n                    value\n                }\n            });\n\n            return {\n                key,\n                data: input\n            };\n        } catch (ex) {\n            return {\n                key,\n                error: ex\n            };\n        }\n    }\n    public async storeValues<V extends GenericRecord<StorageKey>>(\n        values: V\n    ): Promise<StoreValuesResult<V>> {\n        const keys = Object.keys(values);\n        try {\n            const batch = keys.map(key => {\n                const input = values[key];\n                let value: string | undefined;\n                try {\n                    value = JSON.stringify(input);\n                } catch (ex) {\n                    throw ex;\n                }\n                const item: IStoreItem = {\n                    PK: createPartitionKey(),\n                    SK: createSortKey({ key }),\n                    TYPE: createType(),\n                    key,\n                    value\n                };\n                return this.entity().putBatch(item);\n            });\n\n            await batchWriteAll({\n                table: this.table(),\n                items: batch\n            });\n            return {\n                keys,\n                data: values\n            };\n        } catch (ex) {\n            return {\n                keys,\n                error: ex\n            };\n        }\n    }\n    public async getValue<V>(key: StorageKey): Promise<GetValueResult<V>> {\n        try {\n            const result = await get<IStoreItem>({\n                entity: this.entity(),\n                keys: {\n                    PK: createPartitionKey(),\n                    SK: createSortKey({ key })\n                }\n            });\n            return {\n                key,\n                data: result ? JSON.parse(result.value) : null\n            };\n        } catch (ex) {\n            return {\n                key,\n                error: ex\n            };\n        }\n    }\n    public async getValues<V extends GenericRecord<StorageKey>>(\n        input: (keyof V)[]\n    ): Promise<GetValuesResult<V>> {\n        const keys = [...input] as string[];\n        const batch = keys.map(key => {\n            return this.entity().getBatch({\n                PK: createPartitionKey(),\n                SK: createSortKey({ key })\n            });\n        });\n\n        try {\n            const results = await batchReadAll<IStoreItem>({\n                table: this.table(),\n                items: batch\n            });\n            const data = keys.reduce((collection, key) => {\n                const result = results.find(item => {\n                    return item.PK === createPartitionKey() && item.SK === createSortKey({ key });\n                });\n                if (!result?.value) {\n                    // @ts-expect-error\n                    collection[key] = null;\n                    return collection;\n                }\n                try {\n                    // @ts-expect-error\n                    collection[key] = JSON.parse(result.value);\n                } catch {\n                    // @ts-expect-error\n                    collection[key] = null;\n                }\n\n                return collection;\n            }, {} as V);\n            return {\n                keys,\n                data\n            };\n        } catch (ex) {\n            return {\n                keys,\n                error: ex\n            };\n        }\n    }\n    public async listValues<V extends GenericRecord<StorageKey>>(\n        params?: IListValuesParams\n    ): Promise<ListValuesResult<V>> {\n        try {\n            const partitionKey = createPartitionKey();\n            const options = {\n                ...params\n            };\n            const results = await queryAll<IStoreItem>({\n                entity: this.entity(),\n                partitionKey,\n                options\n            });\n\n            const data = results.reduce((collection, item) => {\n                try {\n                    // @ts-expect-error\n                    collection[item.key] = JSON.parse(item.value);\n                } catch (ex) {\n                    // @ts-expect-error\n                    collection[item.key] = null;\n                }\n\n                return collection;\n            }, {} as V);\n\n            return {\n                keys: Object.keys(data),\n                data\n            };\n        } catch (ex) {\n            return {\n                error: ex\n            };\n        }\n    }\n\n    public async removeValue<V>(key: StorageKey): Promise<RemoveValueResult<V>> {\n        const result = await this.getValue<V>(key);\n        if (result.error) {\n            return {\n                key,\n                error: result.error\n            };\n        }\n        try {\n            await this.entity().delete({\n                PK: createPartitionKey(),\n                SK: createSortKey({ key })\n            });\n            return {\n                key,\n                data: result.data\n            };\n        } catch (ex) {\n            return {\n                key,\n                error: ex\n            };\n        }\n    }\n\n    public async removeValues<V extends GenericRecord<StorageKey>>(\n        input: (keyof V)[]\n    ): Promise<RemoveValuesResult<V>> {\n        const keys = [...input] as string[];\n        const batch = keys.map(key => {\n            return this.entity().deleteBatch({\n                PK: createPartitionKey(),\n                SK: createSortKey({ key })\n            });\n        });\n\n        try {\n            await batchWriteAll({\n                table: this.table(),\n                items: batch\n            });\n            return {\n                keys\n            };\n        } catch (ex) {\n            return {\n                keys,\n                error: ex\n            };\n        }\n    }\n}\n\nexport default DynamoDbDriver;\n"],"mappings":";;;;;;AAeA,IAAAA,YAAA,GAAAC,OAAA;AAEA,IAAAC,OAAA,GAAAD,OAAA;AACA,IAAAE,MAAA,GAAAF,OAAA;AACA,IAAAG,KAAA,GAAAH,OAAA;AAOA,MAAMI,cAAc,CAAuC;EAG/CC,MAAM,GAAsBC,SAAS;EACrCL,OAAO,GAAuBK,SAAS;EAExCC,KAAKA,CAAA,EAAU;IAClB,IAAI,IAAI,CAACF,MAAM,EAAE;MACb,OAAO,IAAI,CAACA,MAAM;IACtB;IACA,IAAI,CAACA,MAAM,GAAG,IAAAG,wBAAW,EAAC;MACtBC,cAAc,EAAE,IAAI,CAACA;IACzB,CAAC,CAAC;IACF,OAAO,IAAI,CAACJ,MAAM;EACtB;EAEOK,MAAMA,CAAA,EAAW;IACpB,IAAI,IAAI,CAACT,OAAO,EAAE;MACd,OAAO,IAAI,CAACA,OAAO;IACvB;IACA,IAAI,CAACA,OAAO,GAAG,IAAAU,oBAAY,EAAC;MACxBJ,KAAK,EAAE,IAAI,CAACA,KAAK,CAAC;IACtB,CAAC,CAAC;IACF,OAAO,IAAI,CAACN,OAAO;EACvB;EAEAW,WAAWA,CAAC;IAAEH;EAAgC,CAAC,EAAE;IAC7C,IAAI,CAACA,cAAc,GAAGA,cAAc;EACxC;EAEOI,SAASA,CAAA,EAAG;IACf,OAAO,IAAI,CAACJ,cAAc;EAC9B;EAEA,MAAaK,UAAUA,CAAIC,GAAW,EAAEC,KAAQ,EAAgC;IAC5E,IAAIC,KAAyB;IAC7B,IAAI;MACAA,KAAK,GAAGC,IAAI,CAACC,SAAS,CAACH,KAAK,CAAC;IACjC,CAAC,CAAC,OAAOI,EAAE,EAAE;MACT,OAAO;QACHL,GAAG;QACHM,KAAK,EAAED;MACX,CAAC;IACL;IAEA,IAAI;MACA,MAAM,IAAAE,UAAG,EAAa;QAClBZ,MAAM,EAAE,IAAI,CAACA,MAAM,CAAC,CAAC;QACrBa,IAAI,EAAE;UACFC,EAAE,EAAE,IAAAC,wBAAkB,EAAC,CAAC;UACxBC,EAAE,EAAE,IAAAC,mBAAa,EAAC;YAAEZ;UAAI,CAAC,CAAC;UAC1Ba,IAAI,EAAE,IAAAC,gBAAU,EAAC,CAAC;UAClBd,GAAG;UACHE;QACJ;MACJ,CAAC,CAAC;MAEF,OAAO;QACHF,GAAG;QACHe,IAAI,EAAEd;MACV,CAAC;IACL,CAAC,CAAC,OAAOI,EAAE,EAAE;MACT,OAAO;QACHL,GAAG;QACHM,KAAK,EAAED;MACX,CAAC;IACL;EACJ;EACA,MAAaW,WAAWA,CACpBC,MAAS,EACoB;IAC7B,MAAMC,IAAI,GAAGC,MAAM,CAACD,IAAI,CAACD,MAAM,CAAC;IAChC,IAAI;MACA,MAAMG,KAAK,GAAGF,IAAI,CAACG,GAAG,CAACrB,GAAG,IAAI;QAC1B,MAAMC,KAAK,GAAGgB,MAAM,CAACjB,GAAG,CAAC;QACzB,IAAIE,KAAyB;QAC7B,IAAI;UACAA,KAAK,GAAGC,IAAI,CAACC,SAAS,CAACH,KAAK,CAAC;QACjC,CAAC,CAAC,OAAOI,EAAE,EAAE;UACT,MAAMA,EAAE;QACZ;QACA,MAAMG,IAAgB,GAAG;UACrBC,EAAE,EAAE,IAAAC,wBAAkB,EAAC,CAAC;UACxBC,EAAE,EAAE,IAAAC,mBAAa,EAAC;YAAEZ;UAAI,CAAC,CAAC;UAC1Ba,IAAI,EAAE,IAAAC,gBAAU,EAAC,CAAC;UAClBd,GAAG;UACHE;QACJ,CAAC;QACD,OAAO,IAAI,CAACP,MAAM,CAAC,CAAC,CAAC2B,QAAQ,CAACd,IAAI,CAAC;MACvC,CAAC,CAAC;MAEF,MAAM,IAAAe,oBAAa,EAAC;QAChB/B,KAAK,EAAE,IAAI,CAACA,KAAK,CAAC,CAAC;QACnBgC,KAAK,EAAEJ;MACX,CAAC,CAAC;MACF,OAAO;QACHF,IAAI;QACJH,IAAI,EAAEE;MACV,CAAC;IACL,CAAC,CAAC,OAAOZ,EAAE,EAAE;MACT,OAAO;QACHa,IAAI;QACJZ,KAAK,EAAED;MACX,CAAC;IACL;EACJ;EACA,MAAaoB,QAAQA,CAAIzB,GAAe,EAA8B;IAClE,IAAI;MACA,MAAM0B,MAAM,GAAG,MAAM,IAAAC,UAAG,EAAa;QACjChC,MAAM,EAAE,IAAI,CAACA,MAAM,CAAC,CAAC;QACrBuB,IAAI,EAAE;UACFT,EAAE,EAAE,IAAAC,wBAAkB,EAAC,CAAC;UACxBC,EAAE,EAAE,IAAAC,mBAAa,EAAC;YAAEZ;UAAI,CAAC;QAC7B;MACJ,CAAC,CAAC;MACF,OAAO;QACHA,GAAG;QACHe,IAAI,EAAEW,MAAM,GAAGvB,IAAI,CAACyB,KAAK,CAACF,MAAM,CAACxB,KAAK,CAAC,GAAG;MAC9C,CAAC;IACL,CAAC,CAAC,OAAOG,EAAE,EAAE;MACT,OAAO;QACHL,GAAG;QACHM,KAAK,EAAED;MACX,CAAC;IACL;EACJ;EACA,MAAawB,SAASA,CAClB5B,KAAkB,EACS;IAC3B,MAAMiB,IAAI,GAAG,CAAC,GAAGjB,KAAK,CAAa;IACnC,MAAMmB,KAAK,GAAGF,IAAI,CAACG,GAAG,CAACrB,GAAG,IAAI;MAC1B,OAAO,IAAI,CAACL,MAAM,CAAC,CAAC,CAACmC,QAAQ,CAAC;QAC1BrB,EAAE,EAAE,IAAAC,wBAAkB,EAAC,CAAC;QACxBC,EAAE,EAAE,IAAAC,mBAAa,EAAC;UAAEZ;QAAI,CAAC;MAC7B,CAAC,CAAC;IACN,CAAC,CAAC;IAEF,IAAI;MACA,MAAM+B,OAAO,GAAG,MAAM,IAAAC,mBAAY,EAAa;QAC3CxC,KAAK,EAAE,IAAI,CAACA,KAAK,CAAC,CAAC;QACnBgC,KAAK,EAAEJ;MACX,CAAC,CAAC;MACF,MAAML,IAAI,GAAGG,IAAI,CAACe,MAAM,CAAC,CAACC,UAAU,EAAElC,GAAG,KAAK;QAC1C,MAAM0B,MAAM,GAAGK,OAAO,CAACI,IAAI,CAAC3B,IAAI,IAAI;UAChC,OAAOA,IAAI,CAACC,EAAE,KAAK,IAAAC,wBAAkB,EAAC,CAAC,IAAIF,IAAI,CAACG,EAAE,KAAK,IAAAC,mBAAa,EAAC;YAAEZ;UAAI,CAAC,CAAC;QACjF,CAAC,CAAC;QACF,IAAI,CAAC0B,MAAM,EAAExB,KAAK,EAAE;UAChB;UACAgC,UAAU,CAAClC,GAAG,CAAC,GAAG,IAAI;UACtB,OAAOkC,UAAU;QACrB;QACA,IAAI;UACA;UACAA,UAAU,CAAClC,GAAG,CAAC,GAAGG,IAAI,CAACyB,KAAK,CAACF,MAAM,CAACxB,KAAK,CAAC;QAC9C,CAAC,CAAC,MAAM;UACJ;UACAgC,UAAU,CAAClC,GAAG,CAAC,GAAG,IAAI;QAC1B;QAEA,OAAOkC,UAAU;MACrB,CAAC,EAAE,CAAC,CAAM,CAAC;MACX,OAAO;QACHhB,IAAI;QACJH;MACJ,CAAC;IACL,CAAC,CAAC,OAAOV,EAAE,EAAE;MACT,OAAO;QACHa,IAAI;QACJZ,KAAK,EAAED;MACX,CAAC;IACL;EACJ;EACA,MAAa+B,UAAUA,CACnBC,MAA0B,EACE;IAC5B,IAAI;MACA,MAAMC,YAAY,GAAG,IAAA5B,wBAAkB,EAAC,CAAC;MACzC,MAAM6B,OAAO,GAAG;QACZ,GAAGF;MACP,CAAC;MACD,MAAMN,OAAO,GAAG,MAAM,IAAAS,eAAQ,EAAa;QACvC7C,MAAM,EAAE,IAAI,CAACA,MAAM,CAAC,CAAC;QACrB2C,YAAY;QACZC;MACJ,CAAC,CAAC;MAEF,MAAMxB,IAAI,GAAGgB,OAAO,CAACE,MAAM,CAAC,CAACC,UAAU,EAAE1B,IAAI,KAAK;QAC9C,IAAI;UACA;UACA0B,UAAU,CAAC1B,IAAI,CAACR,GAAG,CAAC,GAAGG,IAAI,CAACyB,KAAK,CAACpB,IAAI,CAACN,KAAK,CAAC;QACjD,CAAC,CAAC,OAAOG,EAAE,EAAE;UACT;UACA6B,UAAU,CAAC1B,IAAI,CAACR,GAAG,CAAC,GAAG,IAAI;QAC/B;QAEA,OAAOkC,UAAU;MACrB,CAAC,EAAE,CAAC,CAAM,CAAC;MAEX,OAAO;QACHhB,IAAI,EAAEC,MAAM,CAACD,IAAI,CAACH,IAAI,CAAC;QACvBA;MACJ,CAAC;IACL,CAAC,CAAC,OAAOV,EAAE,EAAE;MACT,OAAO;QACHC,KAAK,EAAED;MACX,CAAC;IACL;EACJ;EAEA,MAAaoC,WAAWA,CAAIzC,GAAe,EAAiC;IACxE,MAAM0B,MAAM,GAAG,MAAM,IAAI,CAACD,QAAQ,CAAIzB,GAAG,CAAC;IAC1C,IAAI0B,MAAM,CAACpB,KAAK,EAAE;MACd,OAAO;QACHN,GAAG;QACHM,KAAK,EAAEoB,MAAM,CAACpB;MAClB,CAAC;IACL;IACA,IAAI;MACA,MAAM,IAAI,CAACX,MAAM,CAAC,CAAC,CAAC+C,MAAM,CAAC;QACvBjC,EAAE,EAAE,IAAAC,wBAAkB,EAAC,CAAC;QACxBC,EAAE,EAAE,IAAAC,mBAAa,EAAC;UAAEZ;QAAI,CAAC;MAC7B,CAAC,CAAC;MACF,OAAO;QACHA,GAAG;QACHe,IAAI,EAAEW,MAAM,CAACX;MACjB,CAAC;IACL,CAAC,CAAC,OAAOV,EAAE,EAAE;MACT,OAAO;QACHL,GAAG;QACHM,KAAK,EAAED;MACX,CAAC;IACL;EACJ;EAEA,MAAasC,YAAYA,CACrB1C,KAAkB,EACY;IAC9B,MAAMiB,IAAI,GAAG,CAAC,GAAGjB,KAAK,CAAa;IACnC,MAAMmB,KAAK,GAAGF,IAAI,CAACG,GAAG,CAACrB,GAAG,IAAI;MAC1B,OAAO,IAAI,CAACL,MAAM,CAAC,CAAC,CAACiD,WAAW,CAAC;QAC7BnC,EAAE,EAAE,IAAAC,wBAAkB,EAAC,CAAC;QACxBC,EAAE,EAAE,IAAAC,mBAAa,EAAC;UAAEZ;QAAI,CAAC;MAC7B,CAAC,CAAC;IACN,CAAC,CAAC;IAEF,IAAI;MACA,MAAM,IAAAuB,oBAAa,EAAC;QAChB/B,KAAK,EAAE,IAAI,CAACA,KAAK,CAAC,CAAC;QACnBgC,KAAK,EAAEJ;MACX,CAAC,CAAC;MACF,OAAO;QACHF;MACJ,CAAC;IACL,CAAC,CAAC,OAAOb,EAAE,EAAE;MACT,OAAO;QACHa,IAAI;QACJZ,KAAK,EAAED;MACX,CAAC;IACL;EACJ;AACJ;AAAC,IAAAwC,QAAA,GAAAC,OAAA,CAAAC,OAAA,GAEc1D,cAAc","ignoreList":[]}
+{"version":3,"file":"DynamoDbDriver.js","sources":["../src/DynamoDbDriver.ts"],"sourcesContent":["import type { DynamoDBDocument } from \"@webiny/aws-sdk/client-dynamodb/index.js\";\nimport type {\n    DbDriver,\n    GetValueResult,\n    GetValuesResult,\n    IListValuesParams,\n    ListValuesResult,\n    RemoveValueResult,\n    RemoveValuesResult,\n    StorageKey,\n    StoreValueResult,\n    StoreValuesResult\n} from \"@webiny/db\";\nimport { createTable } from \"~/utils/createTable.js\";\nimport type { GenericRecord } from \"@webiny/api/types.js\";\nimport { createEntity } from \"~/store/entity.js\";\nimport { createPartitionKey, createSortKey, createType } from \"~/store/keys.js\";\n\ninterface ConstructorArgs {\n    documentClient: DynamoDBDocument;\n}\n\nclass DynamoDbDriver implements DbDriver<DynamoDBDocument> {\n    public readonly documentClient;\n\n    public readonly table;\n    public readonly entity;\n\n    constructor({ documentClient }: ConstructorArgs) {\n        this.documentClient = documentClient;\n        this.table = createTable({\n            name: process.env.DB_TABLE as string,\n            documentClient\n        });\n        this.entity = createEntity({\n            table: this.table\n        });\n    }\n\n    public getClient() {\n        return this.documentClient;\n    }\n\n    public async storeValue<V>(key: string, input: V): Promise<StoreValueResult<V>> {\n        let value: string | undefined;\n        try {\n            value = JSON.stringify(input);\n        } catch (ex) {\n            return {\n                key,\n                error: ex\n            };\n        }\n\n        try {\n            await this.entity.put({\n                PK: createPartitionKey(),\n                SK: createSortKey({ key }),\n                TYPE: createType(),\n                data: {\n                    key,\n                    value\n                }\n            });\n\n            return {\n                key,\n                data: input\n            };\n        } catch (ex) {\n            return {\n                key,\n                error: ex\n            };\n        }\n    }\n    public async storeValues<V extends GenericRecord<StorageKey>>(\n        values: V\n    ): Promise<StoreValuesResult<V>> {\n        const keys = Object.keys(values);\n\n        const batchWrite = this.entity.createEntityWriter();\n        try {\n            for (const key of keys) {\n                const input = values[key];\n                let value: string | undefined;\n                try {\n                    value = JSON.stringify(input);\n                } catch (ex) {\n                    throw ex;\n                }\n                batchWrite.put({\n                    PK: createPartitionKey(),\n                    SK: createSortKey({ key }),\n                    TYPE: createType(),\n                    data: {\n                        key,\n                        value\n                    }\n                });\n            }\n\n            await batchWrite.execute();\n            return {\n                keys,\n                data: values\n            };\n        } catch (ex) {\n            return {\n                keys,\n                error: ex\n            };\n        }\n    }\n    public async getValue<V>(key: StorageKey): Promise<GetValueResult<V>> {\n        try {\n            const result = await this.entity.get({\n                PK: createPartitionKey(),\n                SK: createSortKey({ key })\n            });\n            return {\n                key,\n                data: result?.data?.value ? JSON.parse(result.data.value) : null\n            };\n        } catch (ex) {\n            return {\n                key,\n                error: ex\n            };\n        }\n    }\n    public async getValues<V extends GenericRecord<StorageKey>>(\n        input: (keyof V)[]\n    ): Promise<GetValuesResult<V>> {\n        const keys = [...input] as string[];\n\n        const batchRead = this.entity.createEntityReader({\n            read: keys.map(key => {\n                return {\n                    PK: createPartitionKey(),\n                    SK: createSortKey({ key })\n                };\n            })\n        });\n\n        try {\n            const results = await batchRead.execute();\n            const data = keys.reduce((collection, initialKey) => {\n                const key = initialKey as keyof V;\n                const result = results.find(item => {\n                    return item.PK === createPartitionKey() && item.SK === createSortKey({ key });\n                });\n                if (!result?.data?.value) {\n                    collection[key] = null;\n                    return collection;\n                }\n                try {\n                    collection[key] = JSON.parse(result.data.value);\n                } catch {\n                    collection[key] = null;\n                }\n\n                return collection;\n            }, {} as GenericRecord);\n            return {\n                keys,\n                data\n            };\n        } catch (ex) {\n            return {\n                keys,\n                error: ex\n            };\n        }\n    }\n    public async listValues<V extends GenericRecord<StorageKey>>(\n        params?: IListValuesParams\n    ): Promise<ListValuesResult<V>> {\n        try {\n            const partitionKey = createPartitionKey();\n            const options = {\n                ...params\n            };\n            const results = await this.entity.queryAll({\n                partitionKey,\n                options\n            });\n\n            const data = results.reduce((collection, item) => {\n                const key = item.data.key as keyof V;\n                try {\n                    collection[key] = JSON.parse(item.data.value);\n                } catch {\n                    collection[key] = null;\n                }\n\n                return collection;\n            }, {} as GenericRecord);\n\n            return {\n                keys: Object.keys(data),\n                data\n            };\n        } catch (ex) {\n            return {\n                error: ex\n            };\n        }\n    }\n\n    public async removeValue<V>(key: StorageKey): Promise<RemoveValueResult<V>> {\n        const result = await this.getValue<V>(key);\n        if (result.error) {\n            return {\n                key,\n                error: result.error\n            };\n        }\n        try {\n            await this.entity.delete({\n                PK: createPartitionKey(),\n                SK: createSortKey({ key })\n            });\n            return {\n                key,\n                data: result.data\n            };\n        } catch (ex) {\n            return {\n                key,\n                error: ex\n            };\n        }\n    }\n\n    public async removeValues<V extends GenericRecord<StorageKey>>(\n        input: (keyof V)[]\n    ): Promise<RemoveValuesResult<V>> {\n        const keys = [...input] as string[];\n\n        const batchDelete = this.entity.createEntityWriter({\n            delete: keys.map(key => {\n                return {\n                    PK: createPartitionKey(),\n                    SK: createSortKey({ key })\n                };\n            })\n        });\n\n        try {\n            await batchDelete.execute();\n            return {\n                keys\n            };\n        } catch (ex) {\n            return {\n                keys,\n                error: ex\n            };\n        }\n    }\n}\n\nexport default DynamoDbDriver;\n"],"names":["DynamoDbDriver","documentClient","createTable","process","createEntity","key","input","value","JSON","ex","createPartitionKey","createSortKey","createType","values","keys","Object","batchWrite","result","batchRead","results","data","collection","initialKey","item","params","partitionKey","options","batchDelete"],"mappings":";;;AAsBA,MAAMA;IAMF,YAAY,EAAEC,cAAc,EAAmB,CAAE;QAC7C,IAAI,CAAC,cAAc,GAAGA;QACtB,IAAI,CAAC,KAAK,GAAGC,YAAY;YACrB,MAAMC,QAAQ,GAAG,CAAC,QAAQ;YAC1BF;QACJ;QACA,IAAI,CAAC,MAAM,GAAGG,aAAa;YACvB,OAAO,IAAI,CAAC,KAAK;QACrB;IACJ;IAEO,YAAY;QACf,OAAO,IAAI,CAAC,cAAc;IAC9B;IAEA,MAAa,WAAcC,GAAW,EAAEC,KAAQ,EAAgC;QAC5E,IAAIC;QACJ,IAAI;YACAA,QAAQC,KAAK,SAAS,CAACF;QAC3B,EAAE,OAAOG,IAAI;YACT,OAAO;gBACHJ;gBACA,OAAOI;YACX;QACJ;QAEA,IAAI;YACA,MAAM,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC;gBAClB,IAAIC;gBACJ,IAAIC,cAAc;oBAAEN;gBAAI;gBACxB,MAAMO;gBACN,MAAM;oBACFP;oBACAE;gBACJ;YACJ;YAEA,OAAO;gBACHF;gBACA,MAAMC;YACV;QACJ,EAAE,OAAOG,IAAI;YACT,OAAO;gBACHJ;gBACA,OAAOI;YACX;QACJ;IACJ;IACA,MAAa,YACTI,MAAS,EACoB;QAC7B,MAAMC,OAAOC,OAAO,IAAI,CAACF;QAEzB,MAAMG,aAAa,IAAI,CAAC,MAAM,CAAC,kBAAkB;QACjD,IAAI;YACA,KAAK,MAAMX,OAAOS,KAAM;gBACpB,MAAMR,QAAQO,MAAM,CAACR,IAAI;gBACzB,IAAIE;gBACJ,IAAI;oBACAA,QAAQC,KAAK,SAAS,CAACF;gBAC3B,EAAE,OAAOG,IAAI;oBACT,MAAMA;gBACV;gBACAO,WAAW,GAAG,CAAC;oBACX,IAAIN;oBACJ,IAAIC,cAAc;wBAAEN;oBAAI;oBACxB,MAAMO;oBACN,MAAM;wBACFP;wBACAE;oBACJ;gBACJ;YACJ;YAEA,MAAMS,WAAW,OAAO;YACxB,OAAO;gBACHF;gBACA,MAAMD;YACV;QACJ,EAAE,OAAOJ,IAAI;YACT,OAAO;gBACHK;gBACA,OAAOL;YACX;QACJ;IACJ;IACA,MAAa,SAAYJ,GAAe,EAA8B;QAClE,IAAI;YACA,MAAMY,SAAS,MAAM,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC;gBACjC,IAAIP;gBACJ,IAAIC,cAAc;oBAAEN;gBAAI;YAC5B;YACA,OAAO;gBACHA;gBACA,MAAMY,QAAQ,MAAM,QAAQT,KAAK,KAAK,CAACS,OAAO,IAAI,CAAC,KAAK,IAAI;YAChE;QACJ,EAAE,OAAOR,IAAI;YACT,OAAO;gBACHJ;gBACA,OAAOI;YACX;QACJ;IACJ;IACA,MAAa,UACTH,KAAkB,EACS;QAC3B,MAAMQ,OAAO;eAAIR;SAAM;QAEvB,MAAMY,YAAY,IAAI,CAAC,MAAM,CAAC,kBAAkB,CAAC;YAC7C,MAAMJ,KAAK,GAAG,CAACT,CAAAA,MACJ;oBACH,IAAIK;oBACJ,IAAIC,cAAc;wBAAEN;oBAAI;gBAC5B;QAER;QAEA,IAAI;YACA,MAAMc,UAAU,MAAMD,UAAU,OAAO;YACvC,MAAME,OAAON,KAAK,MAAM,CAAC,CAACO,YAAYC;gBAClC,MAAMjB,MAAMiB;gBACZ,MAAML,SAASE,QAAQ,IAAI,CAACI,CAAAA,OACjBA,KAAK,EAAE,KAAKb,wBAAwBa,KAAK,EAAE,KAAKZ,cAAc;wBAAEN;oBAAI;gBAE/E,IAAI,CAACY,QAAQ,MAAM,OAAO;oBACtBI,UAAU,CAAChB,IAAI,GAAG;oBAClB,OAAOgB;gBACX;gBACA,IAAI;oBACAA,UAAU,CAAChB,IAAI,GAAGG,KAAK,KAAK,CAACS,OAAO,IAAI,CAAC,KAAK;gBAClD,EAAE,OAAM;oBACJI,UAAU,CAAChB,IAAI,GAAG;gBACtB;gBAEA,OAAOgB;YACX,GAAG,CAAC;YACJ,OAAO;gBACHP;gBACAM;YACJ;QACJ,EAAE,OAAOX,IAAI;YACT,OAAO;gBACHK;gBACA,OAAOL;YACX;QACJ;IACJ;IACA,MAAa,WACTe,MAA0B,EACE;QAC5B,IAAI;YACA,MAAMC,eAAef;YACrB,MAAMgB,UAAU;gBACZ,GAAGF,MAAM;YACb;YACA,MAAML,UAAU,MAAM,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC;gBACvCM;gBACAC;YACJ;YAEA,MAAMN,OAAOD,QAAQ,MAAM,CAAC,CAACE,YAAYE;gBACrC,MAAMlB,MAAMkB,KAAK,IAAI,CAAC,GAAG;gBACzB,IAAI;oBACAF,UAAU,CAAChB,IAAI,GAAGG,KAAK,KAAK,CAACe,KAAK,IAAI,CAAC,KAAK;gBAChD,EAAE,OAAM;oBACJF,UAAU,CAAChB,IAAI,GAAG;gBACtB;gBAEA,OAAOgB;YACX,GAAG,CAAC;YAEJ,OAAO;gBACH,MAAMN,OAAO,IAAI,CAACK;gBAClBA;YACJ;QACJ,EAAE,OAAOX,IAAI;YACT,OAAO;gBACH,OAAOA;YACX;QACJ;IACJ;IAEA,MAAa,YAAeJ,GAAe,EAAiC;QACxE,MAAMY,SAAS,MAAM,IAAI,CAAC,QAAQ,CAAIZ;QACtC,IAAIY,OAAO,KAAK,EACZ,OAAO;YACHZ;YACA,OAAOY,OAAO,KAAK;QACvB;QAEJ,IAAI;YACA,MAAM,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC;gBACrB,IAAIP;gBACJ,IAAIC,cAAc;oBAAEN;gBAAI;YAC5B;YACA,OAAO;gBACHA;gBACA,MAAMY,OAAO,IAAI;YACrB;QACJ,EAAE,OAAOR,IAAI;YACT,OAAO;gBACHJ;gBACA,OAAOI;YACX;QACJ;IACJ;IAEA,MAAa,aACTH,KAAkB,EACY;QAC9B,MAAMQ,OAAO;eAAIR;SAAM;QAEvB,MAAMqB,cAAc,IAAI,CAAC,MAAM,CAAC,kBAAkB,CAAC;YAC/C,QAAQb,KAAK,GAAG,CAACT,CAAAA,MACN;oBACH,IAAIK;oBACJ,IAAIC,cAAc;wBAAEN;oBAAI;gBAC5B;QAER;QAEA,IAAI;YACA,MAAMsB,YAAY,OAAO;YACzB,OAAO;gBACHb;YACJ;QACJ,EAAE,OAAOL,IAAI;YACT,OAAO;gBACHK;gBACA,OAAOL;YACX;QACJ;IACJ;AACJ;AAEA,2BAAeT"}