@webiny/db-dynamodb 0.0.0-unstable.78f581c1d2 → 0.0.0-unstable.7be00a75a9

package/DynamoDbDriver.d.ts

@@ -1,35 +1,21 @@
-/**
- * Remove this when no apps are using our internal db drivers anymore
- */
-import { DocumentClient } from "aws-sdk/clients/dynamodb";
-import BatchProcess from "./BatchProcess";
-import { DbDriver, Args, Result, ArgsBatch } from "@webiny/db";
-declare type ConstructorArgs = {
-    documentClient?: DocumentClient;
-};
-interface ReadLogsParams {
-    table: string;
+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 { GenericRecord } from "@webiny/api/types.js";
+interface ConstructorArgs {
+    documentClient: DynamoDBDocument;
 }
-interface CreateLogParams {
-    id: string;
-    operation: string;
-    /**
-     * TODO: determine the data type.
-     */
-    data: any;
-    table: string;
-}
-declare class DynamoDbDriver implements DbDriver {
-    batchProcesses: Record<string, BatchProcess>;
-    documentClient: DocumentClient;
-    constructor({ documentClient }?: ConstructorArgs);
-    getClient(): DocumentClient;
-    create({ table, data, meta, __batch: batch }: Args): Promise<Result>;
-    update({ query, data, table, meta, __batch: batch }: Args): Promise<Result>;
-    delete({ query, table, meta, __batch: batch }: Args): Promise<Result>;
-    read<T>({ table, query, sort, limit, keys, meta, __batch: batch }: Args): Promise<Result<T[]>>;
-    createLog({ id, operation, data, table }: CreateLogParams): Promise<Result>;
-    readLogs<T>({ table }: ReadLogsParams): Promise<Result<T[]>>;
-    getBatchProcess(__batch: ArgsBatch): BatchProcess;
+declare class DynamoDbDriver implements DbDriver<DynamoDBDocument> {
+    readonly documentClient: DynamoDBDocument;
+    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>>;
+    storeValues<V extends GenericRecord<StorageKey>>(values: V): Promise<StoreValuesResult<V>>;
+    getValue<V>(key: StorageKey): Promise<GetValueResult<V>>;
+    getValues<V extends GenericRecord<StorageKey>>(input: (keyof V)[]): Promise<GetValuesResult<V>>;
+    listValues<V extends GenericRecord<StorageKey>>(params?: IListValuesParams): Promise<ListValuesResult<V>>;
+    removeValue<V>(key: StorageKey): Promise<RemoveValueResult<V>>;
+    removeValues<V extends GenericRecord<StorageKey>>(input: (keyof V)[]): Promise<RemoveValuesResult<V>>;
 }
 export default DynamoDbDriver;

package/DynamoDbDriver.js

@@ -1,274 +1,228 @@
-"use strict";
-
-var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault").default;
-
-Object.defineProperty(exports, "__esModule", {
-  value: true
-});
-exports.default = void 0;
-
-var _objectSpread2 = _interopRequireDefault(require("@babel/runtime/helpers/objectSpread2"));
-
-var _defineProperty2 = _interopRequireDefault(require("@babel/runtime/helpers/defineProperty"));
-
-var _dynamodb = require("aws-sdk/clients/dynamodb");
-
-var _BatchProcess = _interopRequireDefault(require("./BatchProcess"));
-
-var _QueryGenerator = _interopRequireDefault(require("./QueryGenerator"));
-
-/**
- * Remove this when no apps are using our internal db drivers anymore
- */
-// @ts-nocheck
-const LOG_KEYS = [{
-  primary: true,
-  unique: true,
-  name: "primary",
-  fields: [{
-    name: "PK"
-  }, {
-    name: "SK"
-  }]
-}];
-
+import { createTable } from "./utils/createTable.js";
+import { createEntity } from "./store/entity.js";
+import { createPartitionKey, createSortKey, createType } from "./store/keys.js";
 class DynamoDbDriver {
-  constructor({
-    documentClient
-  } = {}) {
-    (0, _defineProperty2.default)(this, "batchProcesses", void 0);
-    (0, _defineProperty2.default)(this, "documentClient", void 0);
-    this.batchProcesses = {};
-    this.documentClient = documentClient || new _dynamodb.DocumentClient();
-  }
-
-  getClient() {
-    return this.documentClient;
-  }
-
-  async create({
-    table,
-    data,
-    meta,
-    __batch: batch
-  }) {
-    if (!batch) {
-      const result = await this.documentClient.put({
-        TableName: table,
-        Item: data,
-        ReturnConsumedCapacity: meta ? "TOTAL" : "NONE"
-      }).promise();
-      return [true, {
-        response: result.$response
-      }];
+    constructor({ documentClient }){
+        this.documentClient = documentClient;
+        this.table = createTable({
+            name: process.env.DB_TABLE,
+            documentClient
+        });
+        this.entity = createEntity({
+            table: this.table
+        });
     }
-
-    const batchProcess = this.getBatchProcess(batch);
-    batchProcess.addBatchWrite({
-      table,
-      data
-    });
-
-    if (batchProcess.allOperationsAdded()) {
-      batchProcess.startExecution();
-    } else {
-      await batchProcess.waitStartExecution();
+    getClient() {
+        return this.documentClient;
     }
-
-    await batchProcess.waitExecution();
-    return [true, {
-      response: batchProcess.response
-    }];
-  }
-
-  async update({
-    query,
-    data,
-    table,
-    meta,
-    __batch: batch
-  }) {
-    if (!batch) {
-      const update = {
-        UpdateExpression: "SET ",
-        ExpressionAttributeNames: {},
-        ExpressionAttributeValues: {}
-      };
-      const updateExpression = [];
-
-      for (const key in data) {
-        updateExpression.push(`#${key} = :${key}`);
-        update.ExpressionAttributeNames[`#${key}`] = key;
-        update.ExpressionAttributeValues[`:${key}`] = data[key];
-      }
-
-      update.UpdateExpression += updateExpression.join(", ");
-      const result = await this.documentClient.update((0, _objectSpread2.default)({
-        TableName: table,
-        Key: query,
-        ReturnConsumedCapacity: meta ? "TOTAL" : "NONE"
-      }, update)).promise();
-      return [true, {
-        response: result.$response
-      }];
+    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
+            };
+        }
     }
-
-    const batchProcess = this.getBatchProcess(batch);
-    batchProcess.addBatchWrite({
-      table,
-      data
-    });
-
-    if (batchProcess.allOperationsAdded()) {
-      batchProcess.startExecution();
-    } else {
-      await batchProcess.waitStartExecution();
+    async storeValues(values) {
+        const keys = Object.keys(values);
+        const batchWrite = this.entity.createEntityWriter();
+        try {
+            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) {
+            return {
+                keys,
+                error: ex
+            };
+        }
     }
-
-    await batchProcess.waitExecution();
-    return [true, {
-      response: batchProcess.response
-    }];
-  }
-
-  async delete({
-    query,
-    table,
-    meta,
-    __batch: batch
-  }) {
-    if (!batch) {
-      const result = await this.documentClient.delete({
-        TableName: table,
-        Key: query,
-        ReturnConsumedCapacity: meta ? "TOTAL" : "NONE"
-      }).promise();
-      return [true, {
-        response: result.$response
-      }];
+    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
+            };
+        }
     }
-
-    const batchProcess = this.getBatchProcess(batch);
-    batchProcess.addBatchDelete({
-      table,
-      query
-    });
-
-    if (batchProcess.allOperationsAdded()) {
-      batchProcess.startExecution();
-    } else {
-      await batchProcess.waitStartExecution();
+    async getValues(input) {
+        const keys = [
+            ...input
+        ];
+        const batchRead = this.entity.createEntityReader({
+            read: keys.map((key)=>({
+                    PK: createPartitionKey(),
+                    SK: createSortKey({
+                        key
+                    })
+                }))
+        });
+        try {
+            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
+            };
+        }
     }
-
-    await batchProcess.waitExecution();
-    return [true, {
-      response: batchProcess.response
-    }];
-  }
-
-  async read({
-    table,
-    query,
-    sort,
-    limit,
-    keys,
-    meta,
-    __batch: batch
-  }) {
-    if (!batch) {
-      const queryGenerator = new _QueryGenerator.default();
-      const queryParams = queryGenerator.generate({
-        query,
-        keys,
-        sort,
-        limit,
-        tableName: table
-      });
-      const response = await this.documentClient.query((0, _objectSpread2.default)((0, _objectSpread2.default)({}, queryParams), {}, {
-        ReturnConsumedCapacity: meta ? "TOTAL" : "NONE"
-      })).promise();
-
-      if (Array.isArray(response.Items)) {
-        return [response.Items, {
-          response: response.$response
-        }];
-      }
-
-      return [[], {
-        response: response.$response
-      }];
-    } // DynamoDb doesn't support batch queries, so we can immediately assume the GetRequest operation.
-
-
-    const batchProcess = this.getBatchProcess(batch);
-    const getResult = batchProcess.addBatchGet({
-      table,
-      query
-    });
-
-    if (batchProcess.allOperationsAdded()) {
-      batchProcess.startExecution();
-    } else {
-      await batchProcess.waitStartExecution();
+    async listValues(params) {
+        try {
+            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) {
+            return {
+                error: ex
+            };
+        }
     }
-
-    await batchProcess.waitExecution();
-    const result = getResult();
-
-    if (result) {
-      return [[result], {
-        response: batchProcess.response
-      }];
+    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
+            };
+        }
     }
-
-    return [[], {
-      response: batchProcess.response
-    }];
-  }
-
-  async createLog({
-    id,
-    operation,
-    data,
-    table
-  }) {
-    await this.create({
-      table: table,
-      keys: LOG_KEYS,
-      data: (0, _objectSpread2.default)({
-        PK: "log",
-        SK: id,
-        id,
-        operation
-      }, data)
-    });
-    return [true, {}];
-  }
-
-  async readLogs({
-    table
-  }) {
-    return this.read({
-      table,
-      keys: LOG_KEYS,
-      query: {
-        PK: "log",
-        SK: {
-          $gte: " "
+    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
+            };
         }
-      }
-    });
-  }
-
-  getBatchProcess(__batch) {
-    if (!this.batchProcesses[__batch.instance.id]) {
-      this.batchProcesses[__batch.instance.id] = new _BatchProcess.default(__batch.instance, this.documentClient);
     }
-
-    return this.batchProcesses[__batch.instance.id];
-  }
-
 }
+const src_DynamoDbDriver = DynamoDbDriver;
+export default src_DynamoDbDriver;
 
-var _default = DynamoDbDriver;
-exports.default = _default;
+//# sourceMappingURL=DynamoDbDriver.js.map

package/DynamoDbDriver.js.map

@@ -1 +1 @@
-{"version":3,"names":["LOG_KEYS","primary","unique","name","fields","DynamoDbDriver","constructor","documentClient","batchProcesses","DocumentClient","getClient","create","table","data","meta","__batch","batch","result","put","TableName","Item","ReturnConsumedCapacity","promise","response","$response","batchProcess","getBatchProcess","addBatchWrite","allOperationsAdded","startExecution","waitStartExecution","waitExecution","update","query","UpdateExpression","ExpressionAttributeNames","ExpressionAttributeValues","updateExpression","key","push","join","Key","delete","addBatchDelete","read","sort","limit","keys","queryGenerator","QueryGenerator","queryParams","generate","tableName","Array","isArray","Items","getResult","addBatchGet","createLog","id","operation","PK","SK","readLogs","$gte","instance","BatchProcess"],"sources":["DynamoDbDriver.ts"],"sourcesContent":["/**\n * Remove this when no apps are using our internal db drivers anymore\n */\n// @ts-nocheck\nimport { DocumentClient } from \"aws-sdk/clients/dynamodb\";\nimport BatchProcess from \"./BatchProcess\";\nimport QueryGenerator from \"./QueryGenerator\";\nimport { DbDriver, Args, Result, ArgsBatch } from \"@webiny/db\";\nimport { QueryKeys } from \"~/types\";\n\ntype ConstructorArgs = {\n    documentClient?: DocumentClient;\n};\n\nconst LOG_KEYS: QueryKeys = [\n    {\n        primary: true,\n        unique: true,\n        name: \"primary\",\n        fields: [{ name: \"PK\" }, { name: \"SK\" }]\n    }\n];\n\ninterface Update {\n    UpdateExpression: string;\n    ExpressionAttributeNames: Record<string, any>;\n    ExpressionAttributeValues: Record<string, any>;\n}\n\ninterface ReadLogsParams {\n    table: string;\n}\n\ninterface CreateLogParams {\n    id: string;\n    operation: string;\n    /**\n     * TODO: determine the data type.\n     */\n    data: any;\n    table: string;\n}\n\nclass DynamoDbDriver implements DbDriver {\n    batchProcesses: Record<string, BatchProcess>;\n    documentClient: DocumentClient;\n    constructor({ documentClient }: ConstructorArgs = {}) {\n        this.batchProcesses = {};\n        this.documentClient = documentClient || new DocumentClient();\n    }\n\n    getClient() {\n        return this.documentClient;\n    }\n\n    async create({ table, data, meta, __batch: batch }: Args): Promise<Result> {\n        if (!batch) {\n            const result = await this.documentClient\n                .put({\n                    TableName: table,\n                    Item: data,\n                    ReturnConsumedCapacity: meta ? \"TOTAL\" : \"NONE\"\n                })\n                .promise();\n            return [true, { response: result.$response }];\n        }\n\n        const batchProcess = this.getBatchProcess(batch);\n        batchProcess.addBatchWrite({ table, data });\n\n        if (batchProcess.allOperationsAdded()) {\n            batchProcess.startExecution();\n        } else {\n            await batchProcess.waitStartExecution();\n        }\n\n        await batchProcess.waitExecution();\n\n        return [true, { response: batchProcess.response }];\n    }\n\n    async update({ query, data, table, meta, __batch: batch }: Args): Promise<Result> {\n        if (!batch) {\n            const update: Update = {\n                UpdateExpression: \"SET \",\n                ExpressionAttributeNames: {},\n                ExpressionAttributeValues: {}\n            };\n\n            const updateExpression = [];\n            for (const key in data) {\n                updateExpression.push(`#${key} = :${key}`);\n                update.ExpressionAttributeNames[`#${key}`] = key;\n                update.ExpressionAttributeValues[`:${key}`] = data[key];\n            }\n\n            update.UpdateExpression += updateExpression.join(\", \");\n\n            const result = await this.documentClient\n                .update({\n                    TableName: table,\n                    Key: query,\n                    ReturnConsumedCapacity: meta ? \"TOTAL\" : \"NONE\",\n                    ...update\n                })\n                .promise();\n\n            return [true, { response: result.$response }];\n        }\n\n        const batchProcess = this.getBatchProcess(batch);\n\n        batchProcess.addBatchWrite({\n            table,\n            data\n        });\n\n        if (batchProcess.allOperationsAdded()) {\n            batchProcess.startExecution();\n        } else {\n            await batchProcess.waitStartExecution();\n        }\n\n        await batchProcess.waitExecution();\n\n        return [true, { response: batchProcess.response }];\n    }\n\n    async delete({ query, table, meta, __batch: batch }: Args): Promise<Result> {\n        if (!batch) {\n            const result = await this.documentClient\n                .delete({\n                    TableName: table,\n                    Key: query,\n                    ReturnConsumedCapacity: meta ? \"TOTAL\" : \"NONE\"\n                })\n                .promise();\n\n            return [true, { response: result.$response }];\n        }\n\n        const batchProcess = this.getBatchProcess(batch);\n        batchProcess.addBatchDelete({\n            table,\n            query\n        });\n\n        if (batchProcess.allOperationsAdded()) {\n            batchProcess.startExecution();\n        } else {\n            await batchProcess.waitStartExecution();\n        }\n\n        await batchProcess.waitExecution();\n\n        return [true, { response: batchProcess.response }];\n    }\n\n    async read<T>({\n        table,\n        query,\n        sort,\n        limit,\n        keys,\n        meta,\n        __batch: batch\n    }: Args): Promise<Result<T[]>> {\n        if (!batch) {\n            const queryGenerator = new QueryGenerator();\n            const queryParams = queryGenerator.generate({\n                query,\n                keys,\n                sort,\n                limit,\n                tableName: table\n            });\n\n            const response = await this.documentClient\n                .query({ ...queryParams, ReturnConsumedCapacity: meta ? \"TOTAL\" : \"NONE\" })\n                .promise();\n\n            if (Array.isArray(response.Items)) {\n                return [response.Items as T[], { response: response.$response }];\n            }\n            return [[], { response: response.$response }];\n        }\n\n        // DynamoDb doesn't support batch queries, so we can immediately assume the GetRequest operation.\n        const batchProcess = this.getBatchProcess(batch);\n        const getResult = batchProcess.addBatchGet({\n            table,\n            query\n        });\n\n        if (batchProcess.allOperationsAdded()) {\n            batchProcess.startExecution();\n        } else {\n            await batchProcess.waitStartExecution();\n        }\n\n        await batchProcess.waitExecution();\n\n        const result = getResult() as T;\n        if (result) {\n            return [[result], { response: batchProcess.response }];\n        }\n\n        return [[], { response: batchProcess.response }];\n    }\n\n    async createLog({ id, operation, data, table }: CreateLogParams): Promise<Result> {\n        await this.create({\n            table: table,\n            keys: LOG_KEYS,\n            data: {\n                PK: \"log\",\n                SK: id,\n                id,\n                operation,\n                ...data\n            }\n        });\n\n        return [true, {}];\n    }\n\n    async readLogs<T>({ table }: ReadLogsParams) {\n        return this.read<T>({\n            table,\n            keys: LOG_KEYS,\n            query: {\n                PK: \"log\",\n                SK: { $gte: \" \" }\n            }\n        });\n    }\n\n    getBatchProcess(__batch: ArgsBatch): BatchProcess {\n        if (!this.batchProcesses[__batch.instance.id]) {\n            this.batchProcesses[__batch.instance.id] = new BatchProcess(\n                __batch.instance,\n                this.documentClient\n            );\n        }\n\n        return this.batchProcesses[__batch.instance.id];\n    }\n}\n\nexport default DynamoDbDriver;\n"],"mappings":";;;;;;;;;;;;;AAIA;;AACA;;AACA;;AANA;AACA;AACA;AACA;AAWA,MAAMA,QAAmB,GAAG,CACxB;EACIC,OAAO,EAAE,IADb;EAEIC,MAAM,EAAE,IAFZ;EAGIC,IAAI,EAAE,SAHV;EAIIC,MAAM,EAAE,CAAC;IAAED,IAAI,EAAE;EAAR,CAAD,EAAiB;IAAEA,IAAI,EAAE;EAAR,CAAjB;AAJZ,CADwB,CAA5B;;AA6BA,MAAME,cAAN,CAAyC;EAGrCC,WAAW,CAAC;IAAEC;EAAF,IAAsC,EAAvC,EAA2C;IAAA;IAAA;IAClD,KAAKC,cAAL,GAAsB,EAAtB;IACA,KAAKD,cAAL,GAAsBA,cAAc,IAAI,IAAIE,wBAAJ,EAAxC;EACH;;EAEDC,SAAS,GAAG;IACR,OAAO,KAAKH,cAAZ;EACH;;EAEW,MAANI,MAAM,CAAC;IAAEC,KAAF;IAASC,IAAT;IAAeC,IAAf;IAAqBC,OAAO,EAAEC;EAA9B,CAAD,EAA+D;IACvE,IAAI,CAACA,KAAL,EAAY;MACR,MAAMC,MAAM,GAAG,MAAM,KAAKV,cAAL,CAChBW,GADgB,CACZ;QACDC,SAAS,EAAEP,KADV;QAEDQ,IAAI,EAAEP,IAFL;QAGDQ,sBAAsB,EAAEP,IAAI,GAAG,OAAH,GAAa;MAHxC,CADY,EAMhBQ,OANgB,EAArB;MAOA,OAAO,CAAC,IAAD,EAAO;QAAEC,QAAQ,EAAEN,MAAM,CAACO;MAAnB,CAAP,CAAP;IACH;;IAED,MAAMC,YAAY,GAAG,KAAKC,eAAL,CAAqBV,KAArB,CAArB;IACAS,YAAY,CAACE,aAAb,CAA2B;MAAEf,KAAF;MAASC;IAAT,CAA3B;;IAEA,IAAIY,YAAY,CAACG,kBAAb,EAAJ,EAAuC;MACnCH,YAAY,CAACI,cAAb;IACH,CAFD,MAEO;MACH,MAAMJ,YAAY,CAACK,kBAAb,EAAN;IACH;;IAED,MAAML,YAAY,CAACM,aAAb,EAAN;IAEA,OAAO,CAAC,IAAD,EAAO;MAAER,QAAQ,EAAEE,YAAY,CAACF;IAAzB,CAAP,CAAP;EACH;;EAEW,MAANS,MAAM,CAAC;IAAEC,KAAF;IAASpB,IAAT;IAAeD,KAAf;IAAsBE,IAAtB;IAA4BC,OAAO,EAAEC;EAArC,CAAD,EAAsE;IAC9E,IAAI,CAACA,KAAL,EAAY;MACR,MAAMgB,MAAc,GAAG;QACnBE,gBAAgB,EAAE,MADC;QAEnBC,wBAAwB,EAAE,EAFP;QAGnBC,yBAAyB,EAAE;MAHR,CAAvB;MAMA,MAAMC,gBAAgB,GAAG,EAAzB;;MACA,KAAK,MAAMC,GAAX,IAAkBzB,IAAlB,EAAwB;QACpBwB,gBAAgB,CAACE,IAAjB,CAAuB,IAAGD,GAAI,OAAMA,GAAI,EAAxC;QACAN,MAAM,CAACG,wBAAP,CAAiC,IAAGG,GAAI,EAAxC,IAA6CA,GAA7C;QACAN,MAAM,CAACI,yBAAP,CAAkC,IAAGE,GAAI,EAAzC,IAA8CzB,IAAI,CAACyB,GAAD,CAAlD;MACH;;MAEDN,MAAM,CAACE,gBAAP,IAA2BG,gBAAgB,CAACG,IAAjB,CAAsB,IAAtB,CAA3B;MAEA,MAAMvB,MAAM,GAAG,MAAM,KAAKV,cAAL,CAChByB,MADgB;QAEbb,SAAS,EAAEP,KAFE;QAGb6B,GAAG,EAAER,KAHQ;QAIbZ,sBAAsB,EAAEP,IAAI,GAAG,OAAH,GAAa;MAJ5B,GAKVkB,MALU,GAOhBV,OAPgB,EAArB;MASA,OAAO,CAAC,IAAD,EAAO;QAAEC,QAAQ,EAAEN,MAAM,CAACO;MAAnB,CAAP,CAAP;IACH;;IAED,MAAMC,YAAY,GAAG,KAAKC,eAAL,CAAqBV,KAArB,CAArB;IAEAS,YAAY,CAACE,aAAb,CAA2B;MACvBf,KADuB;MAEvBC;IAFuB,CAA3B;;IAKA,IAAIY,YAAY,CAACG,kBAAb,EAAJ,EAAuC;MACnCH,YAAY,CAACI,cAAb;IACH,CAFD,MAEO;MACH,MAAMJ,YAAY,CAACK,kBAAb,EAAN;IACH;;IAED,MAAML,YAAY,CAACM,aAAb,EAAN;IAEA,OAAO,CAAC,IAAD,EAAO;MAAER,QAAQ,EAAEE,YAAY,CAACF;IAAzB,CAAP,CAAP;EACH;;EAEW,MAANmB,MAAM,CAAC;IAAET,KAAF;IAASrB,KAAT;IAAgBE,IAAhB;IAAsBC,OAAO,EAAEC;EAA/B,CAAD,EAAgE;IACxE,IAAI,CAACA,KAAL,EAAY;MACR,MAAMC,MAAM,GAAG,MAAM,KAAKV,cAAL,CAChBmC,MADgB,CACT;QACJvB,SAAS,EAAEP,KADP;QAEJ6B,GAAG,EAAER,KAFD;QAGJZ,sBAAsB,EAAEP,IAAI,GAAG,OAAH,GAAa;MAHrC,CADS,EAMhBQ,OANgB,EAArB;MAQA,OAAO,CAAC,IAAD,EAAO;QAAEC,QAAQ,EAAEN,MAAM,CAACO;MAAnB,CAAP,CAAP;IACH;;IAED,MAAMC,YAAY,GAAG,KAAKC,eAAL,CAAqBV,KAArB,CAArB;IACAS,YAAY,CAACkB,cAAb,CAA4B;MACxB/B,KADwB;MAExBqB;IAFwB,CAA5B;;IAKA,IAAIR,YAAY,CAACG,kBAAb,EAAJ,EAAuC;MACnCH,YAAY,CAACI,cAAb;IACH,CAFD,MAEO;MACH,MAAMJ,YAAY,CAACK,kBAAb,EAAN;IACH;;IAED,MAAML,YAAY,CAACM,aAAb,EAAN;IAEA,OAAO,CAAC,IAAD,EAAO;MAAER,QAAQ,EAAEE,YAAY,CAACF;IAAzB,CAAP,CAAP;EACH;;EAES,MAAJqB,IAAI,CAAI;IACVhC,KADU;IAEVqB,KAFU;IAGVY,IAHU;IAIVC,KAJU;IAKVC,IALU;IAMVjC,IANU;IAOVC,OAAO,EAAEC;EAPC,CAAJ,EAQqB;IAC3B,IAAI,CAACA,KAAL,EAAY;MACR,MAAMgC,cAAc,GAAG,IAAIC,uBAAJ,EAAvB;MACA,MAAMC,WAAW,GAAGF,cAAc,CAACG,QAAf,CAAwB;QACxClB,KADwC;QAExCc,IAFwC;QAGxCF,IAHwC;QAIxCC,KAJwC;QAKxCM,SAAS,EAAExC;MAL6B,CAAxB,CAApB;MAQA,MAAMW,QAAQ,GAAG,MAAM,KAAKhB,cAAL,CAClB0B,KADkB,6DACPiB,WADO;QACM7B,sBAAsB,EAAEP,IAAI,GAAG,OAAH,GAAa;MAD/C,IAElBQ,OAFkB,EAAvB;;MAIA,IAAI+B,KAAK,CAACC,OAAN,CAAc/B,QAAQ,CAACgC,KAAvB,CAAJ,EAAmC;QAC/B,OAAO,CAAChC,QAAQ,CAACgC,KAAV,EAAwB;UAAEhC,QAAQ,EAAEA,QAAQ,CAACC;QAArB,CAAxB,CAAP;MACH;;MACD,OAAO,CAAC,EAAD,EAAK;QAAED,QAAQ,EAAEA,QAAQ,CAACC;MAArB,CAAL,CAAP;IACH,CAnB0B,CAqB3B;;;IACA,MAAMC,YAAY,GAAG,KAAKC,eAAL,CAAqBV,KAArB,CAArB;IACA,MAAMwC,SAAS,GAAG/B,YAAY,CAACgC,WAAb,CAAyB;MACvC7C,KADuC;MAEvCqB;IAFuC,CAAzB,CAAlB;;IAKA,IAAIR,YAAY,CAACG,kBAAb,EAAJ,EAAuC;MACnCH,YAAY,CAACI,cAAb;IACH,CAFD,MAEO;MACH,MAAMJ,YAAY,CAACK,kBAAb,EAAN;IACH;;IAED,MAAML,YAAY,CAACM,aAAb,EAAN;IAEA,MAAMd,MAAM,GAAGuC,SAAS,EAAxB;;IACA,IAAIvC,MAAJ,EAAY;MACR,OAAO,CAAC,CAACA,MAAD,CAAD,EAAW;QAAEM,QAAQ,EAAEE,YAAY,CAACF;MAAzB,CAAX,CAAP;IACH;;IAED,OAAO,CAAC,EAAD,EAAK;MAAEA,QAAQ,EAAEE,YAAY,CAACF;IAAzB,CAAL,CAAP;EACH;;EAEc,MAATmC,SAAS,CAAC;IAAEC,EAAF;IAAMC,SAAN;IAAiB/C,IAAjB;IAAuBD;EAAvB,CAAD,EAAmE;IAC9E,MAAM,KAAKD,MAAL,CAAY;MACdC,KAAK,EAAEA,KADO;MAEdmC,IAAI,EAAE/C,QAFQ;MAGda,IAAI;QACAgD,EAAE,EAAE,KADJ;QAEAC,EAAE,EAAEH,EAFJ;QAGAA,EAHA;QAIAC;MAJA,GAKG/C,IALH;IAHU,CAAZ,CAAN;IAYA,OAAO,CAAC,IAAD,EAAO,EAAP,CAAP;EACH;;EAEa,MAARkD,QAAQ,CAAI;IAAEnD;EAAF,CAAJ,EAA+B;IACzC,OAAO,KAAKgC,IAAL,CAAa;MAChBhC,KADgB;MAEhBmC,IAAI,EAAE/C,QAFU;MAGhBiC,KAAK,EAAE;QACH4B,EAAE,EAAE,KADD;QAEHC,EAAE,EAAE;UAAEE,IAAI,EAAE;QAAR;MAFD;IAHS,CAAb,CAAP;EAQH;;EAEDtC,eAAe,CAACX,OAAD,EAAmC;IAC9C,IAAI,CAAC,KAAKP,cAAL,CAAoBO,OAAO,CAACkD,QAAR,CAAiBN,EAArC,CAAL,EAA+C;MAC3C,KAAKnD,cAAL,CAAoBO,OAAO,CAACkD,QAAR,CAAiBN,EAArC,IAA2C,IAAIO,qBAAJ,CACvCnD,OAAO,CAACkD,QAD+B,EAEvC,KAAK1D,cAFkC,CAA3C;IAIH;;IAED,OAAO,KAAKC,cAAL,CAAoBO,OAAO,CAACkD,QAAR,CAAiBN,EAArC,CAAP;EACH;;AA3MoC;;eA8M1BtD,c"}
+{"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"}