@model-ts/dynamodb 3.0.1 → 3.0.3

package/src/client.ts

@@ -741,7 +741,7 @@ export class Client {
         ? E.left({ ...state, rollbackSuccessful: true })
         : E.right(state)
 
-    const [currentBatch, remaining] = A.splitAt(25)(operations)
+    const [currentBatch, remaining] = A.splitAt(100)(operations)
 
     try {
       const transactItems = currentBatch

package/src/sandbox.ts

@@ -5,17 +5,29 @@ import diff from "snapshot-diff"
 import { Client } from "./client"
 import { GSI_NAMES } from "./gsi"
 
+/**
+ * Returns a random endpoint from the LOCAL_ENDPOINT environment variable.
+ */
+function getEndpoint() {
+  if (!process.env.LOCAL_ENDPOINT) throw new Error("LOCAL_ENDPOINT is not set")
+
+  const endpoints = process.env.LOCAL_ENDPOINT.split(",")
+  return endpoints[Math.floor(Math.random() * endpoints.length)]
+}
+
+const endpoint = getEndpoint()
+
 const ddb = new DynamoDB({
   accessKeyId: "xxx",
   secretAccessKey: "xxx",
-  endpoint: process.env.LOCAL_ENDPOINT,
+  endpoint,
   region: "local",
 })
 
 const docClient = new DynamoDB.DocumentClient({
   accessKeyId: "xxx",
   secretAccessKey: "xxx",
-  endpoint: process.env.LOCAL_ENDPOINT,
+  endpoint,
   region: "local",
 })
 
@@ -125,23 +137,25 @@ export const createSandbox = async (client: Client): Promise<Sandbox> => {
     seed: async (...args: Array<{ [key: string]: any }>) => {
       const chunks = chunksOf(25)(args)
 
-      for (const chunk of chunks) {
-        const items = chunk.map((i) =>
-          typeof i?._model?.__dynamoDBEncode === "function"
-            ? i._model.__dynamoDBEncode(i)
-            : typeof i.encode === "function"
-            ? i.encode()
-            : i
-        )
-
-        await client.documentClient
-          .batchWrite({
-            RequestItems: {
-              [tableName]: items.map((i) => ({ PutRequest: { Item: i } })),
-            },
-          })
-          .promise()
-      }
+      await Promise.all(
+        chunks.map(async (chunk) => {
+          const items = chunk.map((i) =>
+            typeof i?._model?.__dynamoDBEncode === "function"
+              ? i._model.__dynamoDBEncode(i)
+              : typeof i.encode === "function"
+              ? i.encode()
+              : i
+          )
+
+          return client.documentClient
+            .batchWrite({
+              RequestItems: {
+                [tableName]: items.map((i) => ({ PutRequest: { Item: i } })),
+              },
+            })
+            .promise()
+        })
+      )
     },
     get: (pk: string, sk: string) =>
       client.documentClient