@openstax/ts-utils 1.9.0 → 1.9.1

package/dist/esm/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/esm/services/documentStore/unversioned/dynamodb.js

@@ -1,4 +1,4 @@
-import { DynamoDB, PutItemCommand, QueryCommand, ScanCommand } from '@aws-sdk/client-dynamodb';
+import { BatchGetItemCommand, DynamoDB, GetItemCommand, PutItemCommand, ScanCommand } from '@aws-sdk/client-dynamodb';
 import { once } from '../../..';
 import { resolveConfigValue } from '../../../config';
 import { ifDefined } from '../../../guards';
@@ -10,10 +10,10 @@ export const dynamoUnversionedDocumentStore = (initializer) => () => (configProv
     return {
         loadAllDocumentsTheBadWay: async () => {
             const loadAllResults = async (ExclusiveStartKey) => {
-                var _a;
+                var _a, _b;
                 const cmd = new ScanCommand({ TableName: await tableName(), ExclusiveStartKey });
                 const result = await dynamodb().send(cmd);
-                const resultItems = ((_a = result.Items) === null || _a === void 0 ? void 0 : _a.map(decodeDynamoDocument)) || [];
+                const resultItems = (_b = (_a = result.Items) === null || _a === void 0 ? void 0 : _a.map((item) => decodeDynamoDocument(item))) !== null && _b !== void 0 ? _b : [];
                 if (result.LastEvaluatedKey) {
                     return [...resultItems, ...await loadAllResults(result.LastEvaluatedKey)];
                 }
@@ -21,19 +21,26 @@ export const dynamoUnversionedDocumentStore = (initializer) => () => (configProv
             };
             return loadAllResults();
         },
+        batchGetItem: async (ids) => {
+            const table = await tableName();
+            const key = hashKey.toString();
+            const getBatches = async (requestItems) => {
+                const cmd = new BatchGetItemCommand({
+                    RequestItems: requestItems !== null && requestItems !== void 0 ? requestItems : { [table]: { Keys: ids.map((id) => ({ [key]: encodeDynamoAttribute(id) })) } },
+                });
+                const response = await dynamodb().send(cmd);
+                const currentResponses = response.Responses ?
+                    response.Responses[table].map(response => decodeDynamoDocument(response)) : [];
+                return currentResponses.concat(response.UnprocessedKeys ? await getBatches(response.UnprocessedKeys) : []);
+            };
+            return getBatches();
+        },
         getItem: async (id) => {
-            const cmd = new QueryCommand({
+            const cmd = new GetItemCommand({
+                Key: { [hashKey.toString()]: encodeDynamoAttribute(id) },
                 TableName: await tableName(),
-                KeyConditionExpression: '#hk = :hkv',
-                ExpressionAttributeNames: { '#hk': hashKey.toString() },
-                ExpressionAttributeValues: { ':hkv': 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(decodeDynamoDocument)[0];
             });
+            return dynamodb().send(cmd).then(result => result.Item ? decodeDynamoDocument(result.Item) : undefined);
         },
         /* saves a new version of a document with the given data */
         putItem: async (item) => {

package/dist/esm/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/esm/services/documentStore/unversioned/file-system.js

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