@openstax/ts-utils 1.9.0 → 1.9.1

package/dist/cjs/services/documentStore/unversioned/dynamodb.d.ts

@@ -7,6 +7,7 @@ export declare const dynamoUnversionedDocumentStore: <C extends string = "dynamo
     tableName: import("../../../config").ConfigValueProvider<string>;
 }; }) => <K extends keyof T>(_: {}, hashKey: K) => {
     loadAllDocumentsTheBadWay: () => Promise<T[]>;
+    batchGetItem: (ids: T[K][]) => Promise<T[]>;
     getItem: (id: T[K]) => Promise<T | undefined>;
     putItem: (item: T) => Promise<T>;
 };

package/dist/cjs/services/documentStore/unversioned/dynamodb.js

@@ -13,10 +13,10 @@ const dynamoUnversionedDocumentStore = (initializer) => () => (configProvider) =
     return {
         loadAllDocumentsTheBadWay: async () => {
             const loadAllResults = async (ExclusiveStartKey) => {
-                var _a;
+                var _a, _b;
                 const cmd = new client_dynamodb_1.ScanCommand({ TableName: await tableName(), ExclusiveStartKey });
                 const result = await dynamodb().send(cmd);
-                const resultItems = ((_a = result.Items) === null || _a === void 0 ? void 0 : _a.map(dynamoEncoding_1.decodeDynamoDocument)) || [];
+                const resultItems = (_b = (_a = result.Items) === null || _a === void 0 ? void 0 : _a.map((item) => (0, dynamoEncoding_1.decodeDynamoDocument)(item))) !== null && _b !== void 0 ? _b : [];
                 if (result.LastEvaluatedKey) {
                     return [...resultItems, ...await loadAllResults(result.LastEvaluatedKey)];
                 }
@@ -24,19 +24,26 @@ const dynamoUnversionedDocumentStore = (initializer) => () => (configProvider) =
             };
             return loadAllResults();
         },
+        batchGetItem: async (ids) => {
+            const table = await tableName();
+            const key = hashKey.toString();
+            const getBatches = async (requestItems) => {
+                const cmd = new client_dynamodb_1.BatchGetItemCommand({
+                    RequestItems: requestItems !== null && requestItems !== void 0 ? requestItems : { [table]: { Keys: ids.map((id) => ({ [key]: (0, dynamoEncoding_1.encodeDynamoAttribute)(id) })) } },
+                });
+                const response = await dynamodb().send(cmd);
+                const currentResponses = response.Responses ?
+                    response.Responses[table].map(response => (0, dynamoEncoding_1.decodeDynamoDocument)(response)) : [];
+                return currentResponses.concat(response.UnprocessedKeys ? await getBatches(response.UnprocessedKeys) : []);
+            };
+            return getBatches();
+        },
         getItem: async (id) => {
-            const cmd = new client_dynamodb_1.QueryCommand({
+            const cmd = new client_dynamodb_1.GetItemCommand({
+                Key: { [hashKey.toString()]: (0, dynamoEncoding_1.encodeDynamoAttribute)(id) },
                 TableName: await tableName(),
-                KeyConditionExpression: '#hk = :hkv',
-                ExpressionAttributeNames: { '#hk': hashKey.toString() },
-                ExpressionAttributeValues: { ':hkv': (0, dynamoEncoding_1.encodeDynamoAttribute)(id) },
-                ScanIndexForward: false,
-                Limit: 1
-            });
-            return dynamodb().send(cmd).then(result => {
-                var _a;
-                return (_a = result.Items) === null || _a === void 0 ? void 0 : _a.map(dynamoEncoding_1.decodeDynamoDocument)[0];
             });
+            return dynamodb().send(cmd).then(result => result.Item ? (0, dynamoEncoding_1.decodeDynamoDocument)(result.Item) : undefined);
         },
         /* saves a new version of a document with the given data */
         putItem: async (item) => {

package/dist/cjs/services/documentStore/unversioned/file-system.d.ts

@@ -9,6 +9,7 @@ export declare const fileSystemUnversionedDocumentStore: <C extends string = "fi
     tableName: import("../../../config").ConfigValueProvider<string>;
 }; }) => <K extends keyof T>(_: {}, hashKey: K) => {
     loadAllDocumentsTheBadWay: () => Promise<T[]>;
+    batchGetItem: (ids: T[K][]) => Promise<Exclude<Awaited<T>, undefined>[]>;
     getItem: (id: T[K]) => Promise<T | undefined>;
     putItem: (item: T) => Promise<T>;
 };

package/dist/cjs/services/documentStore/unversioned/file-system.js

@@ -68,6 +68,10 @@ const fileSystemUnversionedDocumentStore = (initializer) => () => (configProvide
                 Promise.all(files.map((file) => load(file)))
                     .then((allData) => resolve(allData.filter(guards_1.isDefined)), (err) => reject(err))));
         },
+        batchGetItem: async (ids) => {
+            const items = await Promise.all(ids.map((id) => load(hashFilename(id))));
+            return items.filter(guards_1.isDefined);
+        },
         getItem: (id) => load(hashFilename(id)),
         putItem: async (item) => {
             const path = await filePath(hashFilename(item[hashKey]));