@webiny/db-dynamodb 0.0.0-unstable.d7f521b032 → 0.0.0-unstable.dbdf5d6258

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (234) hide show
  1. package/DynamoDbDriver.d.ts +18 -32
  2. package/DynamoDbDriver.js +207 -248
  3. package/DynamoDbDriver.js.map +1 -1
  4. package/README.md +6 -62
  5. package/index.d.ts +4 -1
  6. package/index.js +5 -14
  7. package/index.js.map +1 -1
  8. package/package.json +20 -26
  9. package/plugins/definitions/AttributePlugin.d.ts +3 -5
  10. package/plugins/definitions/AttributePlugin.js +6 -29
  11. package/plugins/definitions/AttributePlugin.js.map +1 -1
  12. package/plugins/definitions/DateTimeTransformPlugin.d.ts +2 -1
  13. package/plugins/definitions/DateTimeTransformPlugin.js +18 -38
  14. package/plugins/definitions/DateTimeTransformPlugin.js.map +1 -1
  15. package/plugins/definitions/FieldPathPlugin.js +11 -31
  16. package/plugins/definitions/FieldPathPlugin.js.map +1 -1
  17. package/plugins/definitions/FieldPlugin.d.ts +2 -2
  18. package/plugins/definitions/FieldPlugin.js +3 -32
  19. package/plugins/definitions/FieldPlugin.js.map +1 -1
  20. package/plugins/definitions/TimeTransformPlugin.d.ts +2 -1
  21. package/plugins/definitions/TimeTransformPlugin.js +22 -37
  22. package/plugins/definitions/TimeTransformPlugin.js.map +1 -1
  23. package/plugins/definitions/ValueFilterPlugin.d.ts +3 -1
  24. package/plugins/definitions/ValueFilterPlugin.js +13 -27
  25. package/plugins/definitions/ValueFilterPlugin.js.map +1 -1
  26. package/plugins/definitions/ValueTransformPlugin.d.ts +1 -1
  27. package/plugins/definitions/ValueTransformPlugin.js +11 -32
  28. package/plugins/definitions/ValueTransformPlugin.js.map +1 -1
  29. package/plugins/definitions/assignFields.js +5 -20
  30. package/plugins/definitions/assignFields.js.map +1 -1
  31. package/plugins/filters/andIn.d.ts +2 -2
  32. package/plugins/filters/andIn.js +7 -19
  33. package/plugins/filters/andIn.js.map +1 -1
  34. package/plugins/filters/between.d.ts +2 -2
  35. package/plugins/filters/between.js +7 -19
  36. package/plugins/filters/between.js.map +1 -1
  37. package/plugins/filters/contains.d.ts +2 -2
  38. package/plugins/filters/contains.js +45 -25
  39. package/plugins/filters/contains.js.map +1 -1
  40. package/plugins/filters/eq.d.ts +2 -2
  41. package/plugins/filters/eq.js +7 -15
  42. package/plugins/filters/eq.js.map +1 -1
  43. package/plugins/filters/fuzzy.d.ts +2 -2
  44. package/plugins/filters/fuzzy.js +19 -21
  45. package/plugins/filters/fuzzy.js.map +1 -1
  46. package/plugins/filters/gt.d.ts +2 -2
  47. package/plugins/filters/gt.js +5 -12
  48. package/plugins/filters/gt.js.map +1 -1
  49. package/plugins/filters/gte.d.ts +2 -2
  50. package/plugins/filters/gte.js +5 -12
  51. package/plugins/filters/gte.js.map +1 -1
  52. package/plugins/filters/in.d.ts +2 -2
  53. package/plugins/filters/in.js +7 -19
  54. package/plugins/filters/in.js.map +1 -1
  55. package/plugins/filters/index.d.ts +1 -1
  56. package/plugins/filters/index.js +14 -34
  57. package/plugins/filters/index.js.map +1 -1
  58. package/plugins/filters/lt.d.ts +2 -2
  59. package/plugins/filters/lt.js +5 -12
  60. package/plugins/filters/lt.js.map +1 -1
  61. package/plugins/filters/lte.d.ts +2 -2
  62. package/plugins/filters/lte.js +5 -12
  63. package/plugins/filters/lte.js.map +1 -1
  64. package/plugins/filters/startsWith.d.ts +2 -2
  65. package/plugins/filters/startsWith.js +16 -16
  66. package/plugins/filters/startsWith.js.map +1 -1
  67. package/plugins/index.d.ts +9 -1
  68. package/plugins/index.js +12 -14
  69. package/plugins/index.js.map +1 -1
  70. package/store/entity.d.ts +6 -0
  71. package/store/entity.js +11 -0
  72. package/store/entity.js.map +1 -0
  73. package/store/keys.d.ts +6 -0
  74. package/store/keys.js +16 -0
  75. package/store/keys.js.map +1 -0
  76. package/store/types.d.ts +6 -0
  77. package/store/types.js +3 -0
  78. package/store/types.js.map +1 -0
  79. package/toolbox.d.ts +8 -0
  80. package/toolbox.js +5 -0
  81. package/toolbox.js.map +1 -0
  82. package/types.d.ts +10 -23
  83. package/types.js +2 -4
  84. package/types.js.map +1 -1
  85. package/utils/attributes.d.ts +5 -5
  86. package/utils/attributes.js +9 -21
  87. package/utils/attributes.js.map +1 -1
  88. package/utils/batch/batchRead.d.ts +15 -0
  89. package/utils/{batchRead.js → batch/batchRead.js} +9 -34
  90. package/utils/batch/batchRead.js.map +1 -0
  91. package/utils/batch/batchWrite.d.ts +13 -0
  92. package/utils/batch/batchWrite.js +57 -0
  93. package/utils/batch/batchWrite.js.map +1 -0
  94. package/utils/batch/index.d.ts +3 -0
  95. package/utils/batch/index.js +5 -0
  96. package/utils/batch/index.js.map +1 -0
  97. package/utils/batch/types.d.ts +30 -0
  98. package/utils/batch/types.js +3 -0
  99. package/utils/batch/types.js.map +1 -0
  100. package/utils/cleanup.d.ts +3 -3
  101. package/utils/cleanup.js +9 -24
  102. package/utils/cleanup.js.map +1 -1
  103. package/utils/count.d.ts +2 -0
  104. package/utils/count.js +16 -0
  105. package/utils/count.js.map +1 -0
  106. package/utils/createEntity.d.ts +36 -0
  107. package/utils/createEntity.js +85 -0
  108. package/utils/createEntity.js.map +1 -0
  109. package/utils/createTable.d.ts +11 -0
  110. package/utils/createTable.js +39 -0
  111. package/utils/createTable.js.map +1 -0
  112. package/utils/cursor.js +3 -15
  113. package/utils/cursor.js.map +1 -1
  114. package/utils/delete.d.ts +10 -0
  115. package/utils/delete.js +11 -0
  116. package/utils/delete.js.map +1 -0
  117. package/utils/entity/Entity.d.ts +28 -0
  118. package/utils/entity/Entity.js +99 -0
  119. package/utils/entity/Entity.js.map +1 -0
  120. package/utils/entity/EntityReadBatch.d.ts +18 -0
  121. package/utils/entity/EntityReadBatch.js +39 -0
  122. package/utils/entity/EntityReadBatch.js.map +1 -0
  123. package/utils/entity/EntityReadBatchBuilder.d.ts +10 -0
  124. package/utils/entity/EntityReadBatchBuilder.js +21 -0
  125. package/utils/entity/EntityReadBatchBuilder.js.map +1 -0
  126. package/utils/entity/EntityWriteBatch.d.ts +23 -0
  127. package/utils/entity/EntityWriteBatch.js +51 -0
  128. package/utils/entity/EntityWriteBatch.js.map +1 -0
  129. package/utils/entity/EntityWriteBatchBuilder.d.ts +11 -0
  130. package/utils/entity/EntityWriteBatchBuilder.js +20 -0
  131. package/utils/entity/EntityWriteBatchBuilder.js.map +1 -0
  132. package/utils/entity/getEntity.d.ts +4 -0
  133. package/utils/entity/getEntity.js +12 -0
  134. package/utils/entity/getEntity.js.map +1 -0
  135. package/utils/entity/index.d.ts +7 -0
  136. package/utils/entity/index.js +9 -0
  137. package/utils/entity/index.js.map +1 -0
  138. package/utils/entity/types.d.ts +75 -0
  139. package/utils/entity/types.js +4 -0
  140. package/utils/entity/types.js.map +1 -0
  141. package/utils/filter.d.ts +2 -2
  142. package/utils/filter.js +14 -53
  143. package/utils/filter.js.map +1 -1
  144. package/utils/get.d.ts +8 -6
  145. package/utils/get.js +14 -13
  146. package/utils/get.js.map +1 -1
  147. package/utils/index.d.ts +17 -0
  148. package/utils/index.js +19 -0
  149. package/utils/index.js.map +1 -0
  150. package/utils/listResponse.js +7 -15
  151. package/utils/listResponse.js.map +1 -1
  152. package/utils/put.d.ts +12 -0
  153. package/utils/put.js +12 -0
  154. package/utils/put.js.map +1 -0
  155. package/utils/query.d.ts +20 -14
  156. package/utils/query.js +66 -39
  157. package/utils/query.js.map +1 -1
  158. package/utils/scan.d.ts +38 -0
  159. package/utils/scan.js +82 -0
  160. package/utils/scan.js.map +1 -0
  161. package/utils/sort.d.ts +1 -1
  162. package/utils/sort.js +9 -31
  163. package/utils/sort.js.map +1 -1
  164. package/utils/table/Table.d.ts +10 -0
  165. package/utils/table/Table.js +27 -0
  166. package/utils/table/Table.js.map +1 -0
  167. package/utils/table/TableReadBatch.d.ts +19 -0
  168. package/utils/table/TableReadBatch.js +54 -0
  169. package/utils/table/TableReadBatch.js.map +1 -0
  170. package/utils/table/TableWriteBatch.d.ts +21 -0
  171. package/utils/table/TableWriteBatch.js +61 -0
  172. package/utils/table/TableWriteBatch.js.map +1 -0
  173. package/utils/table/index.d.ts +4 -0
  174. package/utils/table/index.js +6 -0
  175. package/utils/table/index.js.map +1 -0
  176. package/utils/table/types.d.ts +39 -0
  177. package/utils/table/types.js +3 -0
  178. package/utils/table/types.js.map +1 -0
  179. package/utils/update.d.ts +12 -0
  180. package/utils/update.js +12 -0
  181. package/utils/update.js.map +1 -0
  182. package/BatchProcess.d.ts +0 -47
  183. package/BatchProcess.js +0 -198
  184. package/BatchProcess.js.map +0 -1
  185. package/QueryGenerator.d.ts +0 -21
  186. package/QueryGenerator.js +0 -78
  187. package/QueryGenerator.js.map +0 -1
  188. package/operators/comparison/beginsWith.d.ts +0 -3
  189. package/operators/comparison/beginsWith.js +0 -24
  190. package/operators/comparison/beginsWith.js.map +0 -1
  191. package/operators/comparison/between.d.ts +0 -3
  192. package/operators/comparison/between.js +0 -30
  193. package/operators/comparison/between.js.map +0 -1
  194. package/operators/comparison/eq.d.ts +0 -3
  195. package/operators/comparison/eq.js +0 -34
  196. package/operators/comparison/eq.js.map +0 -1
  197. package/operators/comparison/gt.d.ts +0 -3
  198. package/operators/comparison/gt.js +0 -24
  199. package/operators/comparison/gt.js.map +0 -1
  200. package/operators/comparison/gte.d.ts +0 -3
  201. package/operators/comparison/gte.js +0 -24
  202. package/operators/comparison/gte.js.map +0 -1
  203. package/operators/comparison/lt.d.ts +0 -3
  204. package/operators/comparison/lt.js +0 -24
  205. package/operators/comparison/lt.js.map +0 -1
  206. package/operators/comparison/lte.d.ts +0 -3
  207. package/operators/comparison/lte.js +0 -24
  208. package/operators/comparison/lte.js.map +0 -1
  209. package/operators/index.d.ts +0 -12
  210. package/operators/index.js +0 -39
  211. package/operators/index.js.map +0 -1
  212. package/operators/logical/and.d.ts +0 -3
  213. package/operators/logical/and.js +0 -63
  214. package/operators/logical/and.js.map +0 -1
  215. package/operators/logical/or.d.ts +0 -3
  216. package/operators/logical/or.js +0 -63
  217. package/operators/logical/or.js.map +0 -1
  218. package/plugins/definitions/NumberTransformPlugin.d.ts +0 -4
  219. package/plugins/definitions/NumberTransformPlugin.js +0 -45
  220. package/plugins/definitions/NumberTransformPlugin.js.map +0 -1
  221. package/statements/createKeyConditionExpressionArgs.d.ts +0 -15
  222. package/statements/createKeyConditionExpressionArgs.js +0 -52
  223. package/statements/createKeyConditionExpressionArgs.js.map +0 -1
  224. package/statements/processStatement.d.ts +0 -3
  225. package/statements/processStatement.js +0 -42
  226. package/statements/processStatement.js.map +0 -1
  227. package/utils/batchRead.d.ts +0 -15
  228. package/utils/batchRead.js.map +0 -1
  229. package/utils/batchWrite.d.ts +0 -17
  230. package/utils/batchWrite.js +0 -30
  231. package/utils/batchWrite.js.map +0 -1
  232. package/utils/table.d.ts +0 -7
  233. package/utils/table.js +0 -27
  234. package/utils/table.js.map +0 -1
@@ -1,35 +1,21 @@
1
- /**
2
- * Remove this when no apps are using our internal db drivers anymore
3
- */
4
- import { DocumentClient } from "aws-sdk/clients/dynamodb";
5
- import BatchProcess from "./BatchProcess";
6
- import { DbDriver, Args, Result, ArgsBatch } from "@webiny/db";
7
- declare type ConstructorArgs = {
8
- documentClient?: DocumentClient;
9
- };
10
- interface ReadLogsParams {
11
- table: string;
1
+ import type { DynamoDBDocument } from "@webiny/aws-sdk/client-dynamodb/index.js";
2
+ import type { DbDriver, GetValueResult, GetValuesResult, IListValuesParams, ListValuesResult, RemoveValueResult, RemoveValuesResult, StorageKey, StoreValueResult, StoreValuesResult } from "@webiny/db";
3
+ import type { GenericRecord } from "@webiny/api/types.js";
4
+ interface ConstructorArgs {
5
+ documentClient: DynamoDBDocument;
12
6
  }
13
- interface CreateLogParams {
14
- id: string;
15
- operation: string;
16
- /**
17
- * TODO: determine the data type.
18
- */
19
- data: any;
20
- table: string;
21
- }
22
- declare class DynamoDbDriver implements DbDriver {
23
- batchProcesses: Record<string, BatchProcess>;
24
- documentClient: DocumentClient;
25
- constructor({ documentClient }?: ConstructorArgs);
26
- getClient(): DocumentClient;
27
- create({ table, data, meta, __batch: batch }: Args): Promise<Result>;
28
- update({ query, data, table, meta, __batch: batch }: Args): Promise<Result>;
29
- delete({ query, table, meta, __batch: batch }: Args): Promise<Result>;
30
- read<T>({ table, query, sort, limit, keys, meta, __batch: batch }: Args): Promise<Result<T[]>>;
31
- createLog({ id, operation, data, table }: CreateLogParams): Promise<Result>;
32
- readLogs<T>({ table }: ReadLogsParams): Promise<Result<T[]>>;
33
- getBatchProcess(__batch: ArgsBatch): BatchProcess;
7
+ declare class DynamoDbDriver implements DbDriver<DynamoDBDocument> {
8
+ readonly documentClient: DynamoDBDocument;
9
+ readonly table: import("./utils").ITable<string, "PK", "SK">;
10
+ readonly entity: import("./store/types").IStoreEntity;
11
+ constructor({ documentClient }: ConstructorArgs);
12
+ getClient(): DynamoDBDocument;
13
+ storeValue<V>(key: string, input: V): Promise<StoreValueResult<V>>;
14
+ storeValues<V extends GenericRecord<StorageKey>>(values: V): Promise<StoreValuesResult<V>>;
15
+ getValue<V>(key: StorageKey): Promise<GetValueResult<V>>;
16
+ getValues<V extends GenericRecord<StorageKey>>(input: (keyof V)[]): Promise<GetValuesResult<V>>;
17
+ listValues<V extends GenericRecord<StorageKey>>(params?: IListValuesParams): Promise<ListValuesResult<V>>;
18
+ removeValue<V>(key: StorageKey): Promise<RemoveValueResult<V>>;
19
+ removeValues<V extends GenericRecord<StorageKey>>(input: (keyof V)[]): Promise<RemoveValuesResult<V>>;
34
20
  }
35
21
  export default DynamoDbDriver;
package/DynamoDbDriver.js CHANGED
@@ -1,274 +1,233 @@
1
- "use strict";
2
-
3
- var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault").default;
4
-
5
- Object.defineProperty(exports, "__esModule", {
6
- value: true
7
- });
8
- exports.default = void 0;
9
-
10
- var _objectSpread2 = _interopRequireDefault(require("@babel/runtime/helpers/objectSpread2"));
11
-
12
- var _defineProperty2 = _interopRequireDefault(require("@babel/runtime/helpers/defineProperty"));
13
-
14
- var _dynamodb = require("aws-sdk/clients/dynamodb");
15
-
16
- var _BatchProcess = _interopRequireDefault(require("./BatchProcess"));
17
-
18
- var _QueryGenerator = _interopRequireDefault(require("./QueryGenerator"));
19
-
20
- /**
21
- * Remove this when no apps are using our internal db drivers anymore
22
- */
23
- // @ts-nocheck
24
- const LOG_KEYS = [{
25
- primary: true,
26
- unique: true,
27
- name: "primary",
28
- fields: [{
29
- name: "PK"
30
- }, {
31
- name: "SK"
32
- }]
33
- }];
34
-
1
+ import { createTable } from "./utils/createTable.js";
2
+ import { createEntity } from "./store/entity.js";
3
+ import { createPartitionKey, createSortKey, createType } from "./store/keys.js";
35
4
  class DynamoDbDriver {
36
5
  constructor({
37
6
  documentClient
38
- } = {}) {
39
- (0, _defineProperty2.default)(this, "batchProcesses", void 0);
40
- (0, _defineProperty2.default)(this, "documentClient", void 0);
41
- this.batchProcesses = {};
42
- this.documentClient = documentClient || new _dynamodb.DocumentClient();
7
+ }) {
8
+ this.documentClient = documentClient;
9
+ this.table = createTable({
10
+ name: process.env.DB_TABLE,
11
+ documentClient
12
+ });
13
+ this.entity = createEntity({
14
+ table: this.table
15
+ });
43
16
  }
44
-
45
17
  getClient() {
46
18
  return this.documentClient;
47
19
  }
48
-
49
- async create({
50
- table,
51
- data,
52
- meta,
53
- __batch: batch
54
- }) {
55
- if (!batch) {
56
- const result = await this.documentClient.put({
57
- TableName: table,
58
- Item: data,
59
- ReturnConsumedCapacity: meta ? "TOTAL" : "NONE"
60
- }).promise();
61
- return [true, {
62
- response: result.$response
63
- }];
20
+ async storeValue(key, input) {
21
+ let value;
22
+ try {
23
+ value = JSON.stringify(input);
24
+ } catch (ex) {
25
+ return {
26
+ key,
27
+ error: ex
28
+ };
64
29
  }
65
-
66
- const batchProcess = this.getBatchProcess(batch);
67
- batchProcess.addBatchWrite({
68
- table,
69
- data
70
- });
71
-
72
- if (batchProcess.allOperationsAdded()) {
73
- batchProcess.startExecution();
74
- } else {
75
- await batchProcess.waitStartExecution();
30
+ try {
31
+ await this.entity.put({
32
+ PK: createPartitionKey(),
33
+ SK: createSortKey({
34
+ key
35
+ }),
36
+ TYPE: createType(),
37
+ data: {
38
+ key,
39
+ value
40
+ }
41
+ });
42
+ return {
43
+ key,
44
+ data: input
45
+ };
46
+ } catch (ex) {
47
+ return {
48
+ key,
49
+ error: ex
50
+ };
76
51
  }
77
-
78
- await batchProcess.waitExecution();
79
- return [true, {
80
- response: batchProcess.response
81
- }];
82
52
  }
83
-
84
- async update({
85
- query,
86
- data,
87
- table,
88
- meta,
89
- __batch: batch
90
- }) {
91
- if (!batch) {
92
- const update = {
93
- UpdateExpression: "SET ",
94
- ExpressionAttributeNames: {},
95
- ExpressionAttributeValues: {}
96
- };
97
- const updateExpression = [];
98
-
99
- for (const key in data) {
100
- updateExpression.push(`#${key} = :${key}`);
101
- update.ExpressionAttributeNames[`#${key}`] = key;
102
- update.ExpressionAttributeValues[`:${key}`] = data[key];
53
+ async storeValues(values) {
54
+ const keys = Object.keys(values);
55
+ const batchWrite = this.entity.createEntityWriter();
56
+ try {
57
+ for (const key of keys) {
58
+ const input = values[key];
59
+ let value;
60
+ try {
61
+ value = JSON.stringify(input);
62
+ } catch (ex) {
63
+ throw ex;
64
+ }
65
+ batchWrite.put({
66
+ PK: createPartitionKey(),
67
+ SK: createSortKey({
68
+ key
69
+ }),
70
+ TYPE: createType(),
71
+ data: {
72
+ key,
73
+ value
74
+ }
75
+ });
103
76
  }
104
-
105
- update.UpdateExpression += updateExpression.join(", ");
106
- const result = await this.documentClient.update((0, _objectSpread2.default)({
107
- TableName: table,
108
- Key: query,
109
- ReturnConsumedCapacity: meta ? "TOTAL" : "NONE"
110
- }, update)).promise();
111
- return [true, {
112
- response: result.$response
113
- }];
114
- }
115
-
116
- const batchProcess = this.getBatchProcess(batch);
117
- batchProcess.addBatchWrite({
118
- table,
119
- data
120
- });
121
-
122
- if (batchProcess.allOperationsAdded()) {
123
- batchProcess.startExecution();
124
- } else {
125
- await batchProcess.waitStartExecution();
77
+ await batchWrite.execute();
78
+ return {
79
+ keys,
80
+ data: values
81
+ };
82
+ } catch (ex) {
83
+ return {
84
+ keys,
85
+ error: ex
86
+ };
126
87
  }
127
-
128
- await batchProcess.waitExecution();
129
- return [true, {
130
- response: batchProcess.response
131
- }];
132
88
  }
133
-
134
- async delete({
135
- query,
136
- table,
137
- meta,
138
- __batch: batch
139
- }) {
140
- if (!batch) {
141
- const result = await this.documentClient.delete({
142
- TableName: table,
143
- Key: query,
144
- ReturnConsumedCapacity: meta ? "TOTAL" : "NONE"
145
- }).promise();
146
- return [true, {
147
- response: result.$response
148
- }];
89
+ async getValue(key) {
90
+ try {
91
+ const result = await this.entity.get({
92
+ PK: createPartitionKey(),
93
+ SK: createSortKey({
94
+ key
95
+ })
96
+ });
97
+ return {
98
+ key,
99
+ data: result?.data?.value ? JSON.parse(result.data.value) : null
100
+ };
101
+ } catch (ex) {
102
+ return {
103
+ key,
104
+ error: ex
105
+ };
149
106
  }
150
-
151
- const batchProcess = this.getBatchProcess(batch);
152
- batchProcess.addBatchDelete({
153
- table,
154
- query
107
+ }
108
+ async getValues(input) {
109
+ const keys = [...input];
110
+ const batchRead = this.entity.createEntityReader({
111
+ read: keys.map(key => {
112
+ return {
113
+ PK: createPartitionKey(),
114
+ SK: createSortKey({
115
+ key
116
+ })
117
+ };
118
+ })
155
119
  });
156
-
157
- if (batchProcess.allOperationsAdded()) {
158
- batchProcess.startExecution();
159
- } else {
160
- await batchProcess.waitStartExecution();
120
+ try {
121
+ const results = await batchRead.execute();
122
+ const data = keys.reduce((collection, initialKey) => {
123
+ const key = initialKey;
124
+ const result = results.find(item => {
125
+ return item.PK === createPartitionKey() && item.SK === createSortKey({
126
+ key
127
+ });
128
+ });
129
+ if (!result?.data?.value) {
130
+ collection[key] = null;
131
+ return collection;
132
+ }
133
+ try {
134
+ collection[key] = JSON.parse(result.data.value);
135
+ } catch {
136
+ collection[key] = null;
137
+ }
138
+ return collection;
139
+ }, {});
140
+ return {
141
+ keys,
142
+ data
143
+ };
144
+ } catch (ex) {
145
+ return {
146
+ keys,
147
+ error: ex
148
+ };
161
149
  }
162
-
163
- await batchProcess.waitExecution();
164
- return [true, {
165
- response: batchProcess.response
166
- }];
167
150
  }
168
-
169
- async read({
170
- table,
171
- query,
172
- sort,
173
- limit,
174
- keys,
175
- meta,
176
- __batch: batch
177
- }) {
178
- if (!batch) {
179
- const queryGenerator = new _QueryGenerator.default();
180
- const queryParams = queryGenerator.generate({
181
- query,
182
- keys,
183
- sort,
184
- limit,
185
- tableName: table
151
+ async listValues(params) {
152
+ try {
153
+ const partitionKey = createPartitionKey();
154
+ const options = {
155
+ ...params
156
+ };
157
+ const results = await this.entity.queryAll({
158
+ partitionKey,
159
+ options
186
160
  });
187
- const response = await this.documentClient.query((0, _objectSpread2.default)((0, _objectSpread2.default)({}, queryParams), {}, {
188
- ReturnConsumedCapacity: meta ? "TOTAL" : "NONE"
189
- })).promise();
190
-
191
- if (Array.isArray(response.Items)) {
192
- return [response.Items, {
193
- response: response.$response
194
- }];
195
- }
196
-
197
- return [[], {
198
- response: response.$response
199
- }];
200
- } // DynamoDb doesn't support batch queries, so we can immediately assume the GetRequest operation.
201
-
202
-
203
- const batchProcess = this.getBatchProcess(batch);
204
- const getResult = batchProcess.addBatchGet({
205
- table,
206
- query
207
- });
208
-
209
- if (batchProcess.allOperationsAdded()) {
210
- batchProcess.startExecution();
211
- } else {
212
- await batchProcess.waitStartExecution();
213
- }
214
-
215
- await batchProcess.waitExecution();
216
- const result = getResult();
217
-
218
- if (result) {
219
- return [[result], {
220
- response: batchProcess.response
221
- }];
161
+ const data = results.reduce((collection, item) => {
162
+ const key = item.data.key;
163
+ try {
164
+ collection[key] = JSON.parse(item.data.value);
165
+ } catch {
166
+ collection[key] = null;
167
+ }
168
+ return collection;
169
+ }, {});
170
+ return {
171
+ keys: Object.keys(data),
172
+ data
173
+ };
174
+ } catch (ex) {
175
+ return {
176
+ error: ex
177
+ };
222
178
  }
223
-
224
- return [[], {
225
- response: batchProcess.response
226
- }];
227
179
  }
228
-
229
- async createLog({
230
- id,
231
- operation,
232
- data,
233
- table
234
- }) {
235
- await this.create({
236
- table: table,
237
- keys: LOG_KEYS,
238
- data: (0, _objectSpread2.default)({
239
- PK: "log",
240
- SK: id,
241
- id,
242
- operation
243
- }, data)
244
- });
245
- return [true, {}];
180
+ async removeValue(key) {
181
+ const result = await this.getValue(key);
182
+ if (result.error) {
183
+ return {
184
+ key,
185
+ error: result.error
186
+ };
187
+ }
188
+ try {
189
+ await this.entity.delete({
190
+ PK: createPartitionKey(),
191
+ SK: createSortKey({
192
+ key
193
+ })
194
+ });
195
+ return {
196
+ key,
197
+ data: result.data
198
+ };
199
+ } catch (ex) {
200
+ return {
201
+ key,
202
+ error: ex
203
+ };
204
+ }
246
205
  }
247
-
248
- async readLogs({
249
- table
250
- }) {
251
- return this.read({
252
- table,
253
- keys: LOG_KEYS,
254
- query: {
255
- PK: "log",
256
- SK: {
257
- $gte: " "
258
- }
259
- }
206
+ async removeValues(input) {
207
+ const keys = [...input];
208
+ const batchDelete = this.entity.createEntityWriter({
209
+ delete: keys.map(key => {
210
+ return {
211
+ PK: createPartitionKey(),
212
+ SK: createSortKey({
213
+ key
214
+ })
215
+ };
216
+ })
260
217
  });
261
- }
262
-
263
- getBatchProcess(__batch) {
264
- if (!this.batchProcesses[__batch.instance.id]) {
265
- this.batchProcesses[__batch.instance.id] = new _BatchProcess.default(__batch.instance, this.documentClient);
218
+ try {
219
+ await batchDelete.execute();
220
+ return {
221
+ keys
222
+ };
223
+ } catch (ex) {
224
+ return {
225
+ keys,
226
+ error: ex
227
+ };
266
228
  }
267
-
268
- return this.batchProcesses[__batch.instance.id];
269
229
  }
270
-
271
230
  }
231
+ export default DynamoDbDriver;
272
232
 
273
- var _default = DynamoDbDriver;
274
- exports.default = _default;
233
+ //# sourceMappingURL=DynamoDbDriver.js.map
@@ -1 +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"}
1
+ {"version":3,"names":["createTable","createEntity","createPartitionKey","createSortKey","createType","DynamoDbDriver","constructor","documentClient","table","name","process","env","DB_TABLE","entity","getClient","storeValue","key","input","value","JSON","stringify","ex","error","put","PK","SK","TYPE","data","storeValues","values","keys","Object","batchWrite","createEntityWriter","execute","getValue","result","get","parse","getValues","batchRead","createEntityReader","read","map","results","reduce","collection","initialKey","find","item","listValues","params","partitionKey","options","queryAll","removeValue","delete","removeValues","batchDelete"],"sources":["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"],"mappings":"AAaA,SAASA,WAAW;AAEpB,SAASC,YAAY;AACrB,SAASC,kBAAkB,EAAEC,aAAa,EAAEC,UAAU;AAMtD,MAAMC,cAAc,CAAuC;EAMvDC,WAAWA,CAAC;IAAEC;EAAgC,CAAC,EAAE;IAC7C,IAAI,CAACA,cAAc,GAAGA,cAAc;IACpC,IAAI,CAACC,KAAK,GAAGR,WAAW,CAAC;MACrBS,IAAI,EAAEC,OAAO,CAACC,GAAG,CAACC,QAAkB;MACpCL;IACJ,CAAC,CAAC;IACF,IAAI,CAACM,MAAM,GAAGZ,YAAY,CAAC;MACvBO,KAAK,EAAE,IAAI,CAACA;IAChB,CAAC,CAAC;EACN;EAEOM,SAASA,CAAA,EAAG;IACf,OAAO,IAAI,CAACP,cAAc;EAC9B;EAEA,MAAaQ,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,IAAI,CAACR,MAAM,CAACU,GAAG,CAAC;QAClBC,EAAE,EAAEtB,kBAAkB,CAAC,CAAC;QACxBuB,EAAE,EAAEtB,aAAa,CAAC;UAAEa;QAAI,CAAC,CAAC;QAC1BU,IAAI,EAAEtB,UAAU,CAAC,CAAC;QAClBuB,IAAI,EAAE;UACFX,GAAG;UACHE;QACJ;MACJ,CAAC,CAAC;MAEF,OAAO;QACHF,GAAG;QACHW,IAAI,EAAEV;MACV,CAAC;IACL,CAAC,CAAC,OAAOI,EAAE,EAAE;MACT,OAAO;QACHL,GAAG;QACHM,KAAK,EAAED;MACX,CAAC;IACL;EACJ;EACA,MAAaO,WAAWA,CACpBC,MAAS,EACoB;IAC7B,MAAMC,IAAI,GAAGC,MAAM,CAACD,IAAI,CAACD,MAAM,CAAC;IAEhC,MAAMG,UAAU,GAAG,IAAI,CAACnB,MAAM,CAACoB,kBAAkB,CAAC,CAAC;IACnD,IAAI;MACA,KAAK,MAAMjB,GAAG,IAAIc,IAAI,EAAE;QACpB,MAAMb,KAAK,GAAGY,MAAM,CAACb,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;QACAW,UAAU,CAACT,GAAG,CAAC;UACXC,EAAE,EAAEtB,kBAAkB,CAAC,CAAC;UACxBuB,EAAE,EAAEtB,aAAa,CAAC;YAAEa;UAAI,CAAC,CAAC;UAC1BU,IAAI,EAAEtB,UAAU,CAAC,CAAC;UAClBuB,IAAI,EAAE;YACFX,GAAG;YACHE;UACJ;QACJ,CAAC,CAAC;MACN;MAEA,MAAMc,UAAU,CAACE,OAAO,CAAC,CAAC;MAC1B,OAAO;QACHJ,IAAI;QACJH,IAAI,EAAEE;MACV,CAAC;IACL,CAAC,CAAC,OAAOR,EAAE,EAAE;MACT,OAAO;QACHS,IAAI;QACJR,KAAK,EAAED;MACX,CAAC;IACL;EACJ;EACA,MAAac,QAAQA,CAAInB,GAAe,EAA8B;IAClE,IAAI;MACA,MAAMoB,MAAM,GAAG,MAAM,IAAI,CAACvB,MAAM,CAACwB,GAAG,CAAC;QACjCb,EAAE,EAAEtB,kBAAkB,CAAC,CAAC;QACxBuB,EAAE,EAAEtB,aAAa,CAAC;UAAEa;QAAI,CAAC;MAC7B,CAAC,CAAC;MACF,OAAO;QACHA,GAAG;QACHW,IAAI,EAAES,MAAM,EAAET,IAAI,EAAET,KAAK,GAAGC,IAAI,CAACmB,KAAK,CAACF,MAAM,CAACT,IAAI,CAACT,KAAK,CAAC,GAAG;MAChE,CAAC;IACL,CAAC,CAAC,OAAOG,EAAE,EAAE;MACT,OAAO;QACHL,GAAG;QACHM,KAAK,EAAED;MACX,CAAC;IACL;EACJ;EACA,MAAakB,SAASA,CAClBtB,KAAkB,EACS;IAC3B,MAAMa,IAAI,GAAG,CAAC,GAAGb,KAAK,CAAa;IAEnC,MAAMuB,SAAS,GAAG,IAAI,CAAC3B,MAAM,CAAC4B,kBAAkB,CAAC;MAC7CC,IAAI,EAAEZ,IAAI,CAACa,GAAG,CAAC3B,GAAG,IAAI;QAClB,OAAO;UACHQ,EAAE,EAAEtB,kBAAkB,CAAC,CAAC;UACxBuB,EAAE,EAAEtB,aAAa,CAAC;YAAEa;UAAI,CAAC;QAC7B,CAAC;MACL,CAAC;IACL,CAAC,CAAC;IAEF,IAAI;MACA,MAAM4B,OAAO,GAAG,MAAMJ,SAAS,CAACN,OAAO,CAAC,CAAC;MACzC,MAAMP,IAAI,GAAGG,IAAI,CAACe,MAAM,CAAC,CAACC,UAAU,EAAEC,UAAU,KAAK;QACjD,MAAM/B,GAAG,GAAG+B,UAAqB;QACjC,MAAMX,MAAM,GAAGQ,OAAO,CAACI,IAAI,CAACC,IAAI,IAAI;UAChC,OAAOA,IAAI,CAACzB,EAAE,KAAKtB,kBAAkB,CAAC,CAAC,IAAI+C,IAAI,CAACxB,EAAE,KAAKtB,aAAa,CAAC;YAAEa;UAAI,CAAC,CAAC;QACjF,CAAC,CAAC;QACF,IAAI,CAACoB,MAAM,EAAET,IAAI,EAAET,KAAK,EAAE;UACtB4B,UAAU,CAAC9B,GAAG,CAAC,GAAG,IAAI;UACtB,OAAO8B,UAAU;QACrB;QACA,IAAI;UACAA,UAAU,CAAC9B,GAAG,CAAC,GAAGG,IAAI,CAACmB,KAAK,CAACF,MAAM,CAACT,IAAI,CAACT,KAAK,CAAC;QACnD,CAAC,CAAC,MAAM;UACJ4B,UAAU,CAAC9B,GAAG,CAAC,GAAG,IAAI;QAC1B;QAEA,OAAO8B,UAAU;MACrB,CAAC,EAAE,CAAC,CAAkB,CAAC;MACvB,OAAO;QACHhB,IAAI;QACJH;MACJ,CAAC;IACL,CAAC,CAAC,OAAON,EAAE,EAAE;MACT,OAAO;QACHS,IAAI;QACJR,KAAK,EAAED;MACX,CAAC;IACL;EACJ;EACA,MAAa6B,UAAUA,CACnBC,MAA0B,EACE;IAC5B,IAAI;MACA,MAAMC,YAAY,GAAGlD,kBAAkB,CAAC,CAAC;MACzC,MAAMmD,OAAO,GAAG;QACZ,GAAGF;MACP,CAAC;MACD,MAAMP,OAAO,GAAG,MAAM,IAAI,CAAC/B,MAAM,CAACyC,QAAQ,CAAC;QACvCF,YAAY;QACZC;MACJ,CAAC,CAAC;MAEF,MAAM1B,IAAI,GAAGiB,OAAO,CAACC,MAAM,CAAC,CAACC,UAAU,EAAEG,IAAI,KAAK;QAC9C,MAAMjC,GAAG,GAAGiC,IAAI,CAACtB,IAAI,CAACX,GAAc;QACpC,IAAI;UACA8B,UAAU,CAAC9B,GAAG,CAAC,GAAGG,IAAI,CAACmB,KAAK,CAACW,IAAI,CAACtB,IAAI,CAACT,KAAK,CAAC;QACjD,CAAC,CAAC,MAAM;UACJ4B,UAAU,CAAC9B,GAAG,CAAC,GAAG,IAAI;QAC1B;QAEA,OAAO8B,UAAU;MACrB,CAAC,EAAE,CAAC,CAAkB,CAAC;MAEvB,OAAO;QACHhB,IAAI,EAAEC,MAAM,CAACD,IAAI,CAACH,IAAI,CAAC;QACvBA;MACJ,CAAC;IACL,CAAC,CAAC,OAAON,EAAE,EAAE;MACT,OAAO;QACHC,KAAK,EAAED;MACX,CAAC;IACL;EACJ;EAEA,MAAakC,WAAWA,CAAIvC,GAAe,EAAiC;IACxE,MAAMoB,MAAM,GAAG,MAAM,IAAI,CAACD,QAAQ,CAAInB,GAAG,CAAC;IAC1C,IAAIoB,MAAM,CAACd,KAAK,EAAE;MACd,OAAO;QACHN,GAAG;QACHM,KAAK,EAAEc,MAAM,CAACd;MAClB,CAAC;IACL;IACA,IAAI;MACA,MAAM,IAAI,CAACT,MAAM,CAAC2C,MAAM,CAAC;QACrBhC,EAAE,EAAEtB,kBAAkB,CAAC,CAAC;QACxBuB,EAAE,EAAEtB,aAAa,CAAC;UAAEa;QAAI,CAAC;MAC7B,CAAC,CAAC;MACF,OAAO;QACHA,GAAG;QACHW,IAAI,EAAES,MAAM,CAACT;MACjB,CAAC;IACL,CAAC,CAAC,OAAON,EAAE,EAAE;MACT,OAAO;QACHL,GAAG;QACHM,KAAK,EAAED;MACX,CAAC;IACL;EACJ;EAEA,MAAaoC,YAAYA,CACrBxC,KAAkB,EACY;IAC9B,MAAMa,IAAI,GAAG,CAAC,GAAGb,KAAK,CAAa;IAEnC,MAAMyC,WAAW,GAAG,IAAI,CAAC7C,MAAM,CAACoB,kBAAkB,CAAC;MAC/CuB,MAAM,EAAE1B,IAAI,CAACa,GAAG,CAAC3B,GAAG,IAAI;QACpB,OAAO;UACHQ,EAAE,EAAEtB,kBAAkB,CAAC,CAAC;UACxBuB,EAAE,EAAEtB,aAAa,CAAC;YAAEa;UAAI,CAAC;QAC7B,CAAC;MACL,CAAC;IACL,CAAC,CAAC;IAEF,IAAI;MACA,MAAM0C,WAAW,CAACxB,OAAO,CAAC,CAAC;MAC3B,OAAO;QACHJ;MACJ,CAAC;IACL,CAAC,CAAC,OAAOT,EAAE,EAAE;MACT,OAAO;QACHS,IAAI;QACJR,KAAK,EAAED;MACX,CAAC;IACL;EACJ;AACJ;AAEA,eAAehB,cAAc","ignoreList":[]}