dynamo-document-builder 0.6.0 → 0.7.0

package/dist/commands/table-batch-get.cjs

@@ -0,0 +1,123 @@
+'use strict';
+
+var tslib_es6 = require('../node_modules/tslib/tslib.es6.cjs');
+var errors = require('../errors.cjs');
+var libDynamodb = require('@aws-sdk/lib-dynamodb');
+var internalConstants = require('../internal-constants.cjs');
+var pMap = require('p-map');
+
+var _TableBatchGet_config;
+/**
+ * Table-level command to perform a batch get across multiple entity types.
+ *
+ * Unlike `BatchGet` which operates on a single entity type, this command accepts
+ * get operations from multiple entities via `entity.prepare(new BatchGet({ keys: [...] }))`
+ * and returns results grouped by entity in a tuple structure.
+ *
+ * Unprocessed keys are returned per entity in a tuple matching the input order.
+ *
+ * @example
+ * ```typescript
+ * const { items, unprocessedKeys } = await table.send(new TableBatchGet({
+ *   gets: [
+ *     userEntity.prepare(new BatchGet({
+ *       keys: [{ userId: 'u1' }, { userId: 'u2' }],
+ *     })),
+ *     orderEntity.prepare(new BatchGet({
+ *       keys: [{ orderId: 'o1' }],
+ *     })),
+ *   ],
+ *   consistent: true, // Apply strongly consistent reads to all groups
+ * }))
+ *
+ * const [users, orders] = items
+ * // users: User[]
+ * // orders: Order[]
+ * ```
+ */
+// biome-ignore lint/suspicious/noExplicitAny: gets span multiple heterogeneous entity schemas
+class TableBatchGet {
+    constructor(config) {
+        _TableBatchGet_config.set(this, void 0);
+        tslib_es6.__classPrivateFieldSet(this, _TableBatchGet_config, config, "f");
+    }
+    execute(table) {
+        return tslib_es6.__awaiter(this, void 0, void 0, function* () {
+            var _a, _b, _c, _d, _e, _f, _g, _h, _j;
+            for (const group of tslib_es6.__classPrivateFieldGet(this, _TableBatchGet_config, "f").gets) {
+                if (group.entity.table !== table) {
+                    throw new errors.DocumentBuilderError(`Entity table "${group.entity.table.tableName}" does not match the executing table "${table.tableName}"`);
+                }
+            }
+            // Command-level consistent overrides all group-level settings.
+            // If not specified, fall back to the per-group OR logic (any group consistent → all consistent).
+            // biome-ignore lint/suspicious/noExplicitAny: heterogeneous group schemas require any
+            const consistent = (_a = tslib_es6.__classPrivateFieldGet(this, _TableBatchGet_config, "f").consistent) !== null && _a !== void 0 ? _a : tslib_es6.__classPrivateFieldGet(this, _TableBatchGet_config, "f").gets.some((g) => g.consistent);
+            // Aggregate all keys from every entity group into a single list
+            // biome-ignore lint/suspicious/noExplicitAny: heterogeneous group schemas require any
+            const allKeys = tslib_es6.__classPrivateFieldGet(this, _TableBatchGet_config, "f").gets.flatMap((g) => g.keys);
+            const batchGet = new libDynamodb.BatchGetCommand({
+                RequestItems: {
+                    [table.tableName]: {
+                        Keys: allKeys,
+                        ConsistentRead: consistent,
+                    },
+                },
+                ReturnConsumedCapacity: tslib_es6.__classPrivateFieldGet(this, _TableBatchGet_config, "f").returnConsumedCapacity,
+            });
+            const result = yield table.documentClient.send(batchGet, {
+                abortSignal: (_b = tslib_es6.__classPrivateFieldGet(this, _TableBatchGet_config, "f").abortController) === null || _b === void 0 ? void 0 : _b.signal,
+                requestTimeout: tslib_es6.__classPrivateFieldGet(this, _TableBatchGet_config, "f").timeoutMs,
+            });
+            const rawItems = ((_d = (_c = result.Responses) === null || _c === void 0 ? void 0 : _c[table.tableName]) !== null && _d !== void 0 ? _d : []);
+            // Match returned items back to their entity groups by comparing primary keys.
+            // DynamoDB does not guarantee order in batch get responses, so we must match by key.
+            const groupedItems = tslib_es6.__classPrivateFieldGet(this, _TableBatchGet_config, "f").gets.map(() => []);
+            for (const item of rawItems) {
+                for (let i = 0; i < tslib_es6.__classPrivateFieldGet(this, _TableBatchGet_config, "f").gets.length; i++) {
+                    // biome-ignore lint/suspicious/noExplicitAny: heterogeneous group schemas require any
+                    const group = tslib_es6.__classPrivateFieldGet(this, _TableBatchGet_config, "f").gets[i];
+                    if (group.matchItem(item)) {
+                        groupedItems[i].push(item);
+                        break;
+                    }
+                }
+            }
+            // Parse each group's matched items
+            const parsedItems = yield pMap(tslib_es6.__classPrivateFieldGet(this, _TableBatchGet_config, "f").gets, 
+            // biome-ignore lint/suspicious/noExplicitAny: heterogeneous group schemas require any
+            (group, i) => tslib_es6.__awaiter(this, void 0, void 0, function* () { var _a; return group.parseResults(groupedItems[i], (_a = tslib_es6.__classPrivateFieldGet(this, _TableBatchGet_config, "f").skipValidation) !== null && _a !== void 0 ? _a : false); }), {
+                concurrency: internalConstants.BATCH_GET_VALIDATION_CONCURRENCY,
+                signal: (_e = tslib_es6.__classPrivateFieldGet(this, _TableBatchGet_config, "f").abortController) === null || _e === void 0 ? void 0 : _e.signal,
+            });
+            // Map unprocessed keys back to per-entity groups
+            const rawUnprocessedKeys = 
+            // biome-ignore lint/suspicious/noExplicitAny: DynamoDB SDK returns untyped key objects
+            ((_h = (_g = (_f = result.UnprocessedKeys) === null || _f === void 0 ? void 0 : _f[table.tableName]) === null || _g === void 0 ? void 0 : _g.Keys) !== null && _h !== void 0 ? _h : []);
+            const unprocessedKeys = tslib_es6.__classPrivateFieldGet(this, _TableBatchGet_config, "f").gets.map(() => undefined);
+            for (const key of rawUnprocessedKeys) {
+                for (let i = 0; i < tslib_es6.__classPrivateFieldGet(this, _TableBatchGet_config, "f").gets.length; i++) {
+                    // biome-ignore lint/suspicious/noExplicitAny: heterogeneous group schemas require any
+                    const group = tslib_es6.__classPrivateFieldGet(this, _TableBatchGet_config, "f").gets[i];
+                    const matched = group.matchUnprocessedKey(key);
+                    if (matched !== undefined) {
+                        if (!unprocessedKeys[i])
+                            unprocessedKeys[i] = [];
+                        unprocessedKeys[i].push(matched);
+                        break;
+                    }
+                }
+            }
+            return {
+                responseMetadata: result.$metadata,
+                consumedCapacity: (_j = result.ConsumedCapacity) === null || _j === void 0 ? void 0 : _j[0],
+                items: parsedItems,
+                unprocessedKeys: unprocessedKeys,
+            };
+        });
+    }
+}
+_TableBatchGet_config = new WeakMap();
+
+exports.TableBatchGet = TableBatchGet;
+//# sourceMappingURL=table-batch-get.cjs.map

package/dist/commands/table-batch-get.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"table-batch-get.cjs","sources":["../../src/commands/table-batch-get.ts"],"sourcesContent":[null],"names":["__classPrivateFieldSet","__classPrivateFieldGet","DocumentBuilderError","BatchGetCommand","__awaiter","BATCH_GET_VALIDATION_CONCURRENCY"],"mappings":";;;;;;;;;AA+DA;;;;;;;;;;;;;;;;;;;;;;;;;;;AA2BG;AACH;MACa,aAAa,CAAA;AAKxB,IAAA,WAAA,CAAY,MAAiC,EAAA;QAF7C,qBAAA,CAAA,GAAA,CAAA,IAAA,EAAA,MAAA,CAAA;AAGE,QAAAA,gCAAA,CAAA,IAAI,EAAA,qBAAA,EAAW,MAAM,EAAA,GAAA,CAAA;IACvB;AAEa,IAAA,OAAO,CAAC,KAAkB,EAAA;;;YACrC,KAAK,MAAM,KAAK,IAAIC,gCAAA,CAAA,IAAI,EAAA,qBAAA,EAAA,GAAA,CAAQ,CAAC,IAAI,EAAE;gBACrC,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,KAAK,KAAK,EAAE;AAChC,oBAAA,MAAM,IAAIC,2BAAoB,CAC5B,CAAA,cAAA,EAAiB,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,SAAS,CAAA,sCAAA,EAAyC,KAAK,CAAC,SAAS,CAAA,CAAA,CAAG,CACzG;gBACH;YACF;;;;YAKA,MAAM,UAAU,GACd,CAAA,EAAA,GAAAD,gCAAA,CAAA,IAAI,EAAA,qBAAA,EAAA,GAAA,CAAQ,CAAC,UAAU,MAAA,IAAA,IAAA,EAAA,KAAA,MAAA,GAAA,EAAA,GAAIA,gCAAA,CAAA,IAAI,EAAA,qBAAA,EAAA,GAAA,CAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAwB,KAAK,CAAC,CAAC,UAAU,CAAC;;;AAI/F,YAAA,MAAM,OAAO,GAAGA,gCAAA,CAAA,IAAI,EAAA,qBAAA,EAAA,GAAA,CAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAwB,KAAK,CAAC,CAAC,IAAI,CAAC;AAE/E,YAAA,MAAM,QAAQ,GAAG,IAAIE,2BAAe,CAAC;AACnC,gBAAA,YAAY,EAAE;AACZ,oBAAA,CAAC,KAAK,CAAC,SAAS,GAAG;AACjB,wBAAA,IAAI,EAAE,OAAO;AACb,wBAAA,cAAc,EAAE,UAAU;AAC3B,qBAAA;AACF,iBAAA;AACD,gBAAA,sBAAsB,EAAEF,gCAAA,CAAA,IAAI,EAAA,qBAAA,EAAA,GAAA,CAAQ,CAAC,sBAAsB;AAC5D,aAAA,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,cAAc,CAAC,IAAI,CAAC,QAAQ,EAAE;gBACvD,WAAW,EAAE,MAAAA,gCAAA,CAAA,IAAI,6BAAQ,CAAC,eAAe,0CAAE,MAAM;AACjD,gBAAA,cAAc,EAAEA,gCAAA,CAAA,IAAI,EAAA,qBAAA,EAAA,GAAA,CAAQ,CAAC,SAAS;AACvC,aAAA,CAAC;AAEF,YAAA,MAAM,QAAQ,IAAI,CAAA,EAAA,GAAA,CAAA,EAAA,GAAA,MAAM,CAAC,SAAS,MAAA,IAAA,IAAA,EAAA,KAAA,MAAA,GAAA,MAAA,GAAA,EAAA,CAAG,KAAK,CAAC,SAAS,CAAC,MAAA,IAAA,IAAA,EAAA,KAAA,MAAA,GAAA,EAAA,GAAI,EAAE,CAA8B;;;AAIzF,YAAA,MAAM,YAAY,GAAqBA,gCAAA,CAAA,IAAI,6BAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,EAAE,CAAC;AAEtE,YAAA,KAAK,MAAM,IAAI,IAAI,QAAQ,EAAE;AAC3B,gBAAA,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAGA,gCAAA,CAAA,IAAI,EAAA,qBAAA,EAAA,GAAA,CAAQ,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE;;oBAEjD,MAAM,KAAK,GAAGA,gCAAA,CAAA,IAAI,EAAA,qBAAA,EAAA,GAAA,CAAQ,CAAC,IAAI,CAAC,CAAC,CAA0B;AAC3D,oBAAA,IAAI,KAAK,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE;wBACzB,YAAY,CAAC,CAAC,CAAE,CAAC,IAAI,CAAC,IAAI,CAAC;wBAC3B;oBACF;gBACF;YACF;;YAGA,MAAM,WAAW,GAAG,MAAM,IAAI,CAC5BA,gCAAA,CAAA,IAAI,EAAA,qBAAA,EAAA,GAAA,CAAQ,CAAC,IAAI;;YAEjB,CAAO,KAA4B,EAAE,CAAS,KAAIG,mBAAA,CAAA,IAAA,EAAA,MAAA,EAAA,MAAA,EAAA,aAAA,EAAA,IAAA,EAAA,CAAA,CAChD,OAAA,KAAK,CAAC,YAAY,CAAC,YAAY,CAAC,CAAC,CAAE,EAAE,CAAA,EAAA,GAAAH,gCAAA,CAAA,IAAI,EAAA,qBAAA,EAAA,GAAA,CAAQ,CAAC,cAAc,MAAA,IAAA,IAAA,EAAA,KAAA,MAAA,GAAA,EAAA,GAAI,KAAK,CAAC,CAAA,CAAA,CAAA,CAAA,EAC5E;AACE,gBAAA,WAAW,EAAEI,kDAAgC;gBAC7C,MAAM,EAAE,MAAAJ,gCAAA,CAAA,IAAI,6BAAQ,CAAC,eAAe,0CAAE,MAAM;AAC7C,aAAA,CACF;;AAGD,YAAA,MAAM,kBAAkB;;AAEtB,aAAC,MAAA,CAAA,EAAA,GAAA,CAAA,EAAA,GAAA,MAAM,CAAC,eAAe,MAAA,IAAA,IAAA,EAAA,KAAA,MAAA,GAAA,MAAA,GAAA,EAAA,CAAG,KAAK,CAAC,SAAS,CAAC,MAAA,IAAA,IAAA,EAAA,KAAA,MAAA,GAAA,MAAA,GAAA,EAAA,CAAE,IAAI,MAAA,IAAA,IAAA,EAAA,KAAA,MAAA,GAAA,EAAA,GAAI,EAAE,CAAU;AAElE,YAAA,MAAM,eAAe,GAAsCA,gCAAA,CAAA,IAAI,6BAAQ,CAAC,IAAI,CAAC,GAAG,CAC9E,MAAM,SAAS,CAChB;AAED,YAAA,KAAK,MAAM,GAAG,IAAI,kBAAkB,EAAE;AACpC,gBAAA,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAGA,gCAAA,CAAA,IAAI,EAAA,qBAAA,EAAA,GAAA,CAAQ,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE;;oBAEjD,MAAM,KAAK,GAAGA,gCAAA,CAAA,IAAI,EAAA,qBAAA,EAAA,GAAA,CAAQ,CAAC,IAAI,CAAC,CAAC,CAA0B;oBAC3D,MAAM,OAAO,GAAG,KAAK,CAAC,mBAAmB,CAAC,GAAG,CAAC;AAC9C,oBAAA,IAAI,OAAO,KAAK,SAAS,EAAE;AACzB,wBAAA,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC;AAAE,4BAAA,eAAe,CAAC,CAAC,CAAC,GAAG,EAAE;wBAChD,eAAe,CAAC,CAAC,CAAE,CAAC,IAAI,CAAC,OAAO,CAAC;wBACjC;oBACF;gBACF;YACF;YAEA,OAAO;gBACL,gBAAgB,EAAE,MAAM,CAAC,SAAS;AAClC,gBAAA,gBAAgB,EAAE,CAAA,EAAA,GAAA,MAAM,CAAC,gBAAgB,MAAA,IAAA,IAAA,EAAA,KAAA,MAAA,GAAA,MAAA,GAAA,EAAA,CAAG,CAAC,CAAC;AAC9C,gBAAA,KAAK,EAAE,WAAuC;AAC9C,gBAAA,eAAe,EAAE,eAAqD;aACvE;QACH,CAAC,CAAA;AAAA,IAAA;AACF;;;;;"}

package/dist/commands/table-batch-get.d.ts

@@ -0,0 +1,86 @@
+import { PreparedBatchGet, BaseConfig, TableCommand } from './index.js';
+import { EntitySchema } from '../core/index.js';
+import { DynamoTable } from '../core/table.js';
+import { ConsumedCapacity } from '@aws-sdk/client-dynamodb';
+import { ResponseMetadata } from '@aws-sdk/types';
+import { ZodObject } from 'zod/v4';
+
+/**
+ * Extracts the result item type from a PreparedBatchGet.
+ */
+type ExtractSchema<T> = T extends PreparedBatchGet<infer S> ? EntitySchema<S> : never;
+/**
+ * Maps an array of PreparedBatchGets to a tuple of their result item arrays.
+ */
+type TableBatchGetItems<Gets extends PreparedBatchGet<ZodObject>[]> = {
+    [K in keyof Gets]: Array<ExtractSchema<Gets[K]>>;
+};
+/**
+ * Maps an array of PreparedBatchGets to a tuple of their unprocessed key arrays.
+ */
+type TableBatchGetUnprocessedKeys<Gets extends PreparedBatchGet<ZodObject>[]> = {
+    [K in keyof Gets]: Array<Partial<ExtractSchema<Gets[K]>>> | undefined;
+};
+/**
+ * Configuration for the TableBatchGet command.
+ *
+ * @template Gets - Tuple of PreparedBatchGet types, one per entity group.
+ */
+type TableBatchGetConfig<Gets extends PreparedBatchGet<any>[]> = BaseConfig & {
+    gets: [...Gets];
+    /**
+     * If set, overrides the `consistent` setting on all individual prepared groups.
+     * When `true`, DynamoDB will use strongly consistent reads for all entity groups.
+     * When `false`, DynamoDB will use eventually consistent reads for all entity groups.
+     * When not set, the command falls back to the per-group `consistent` setting
+     * (any group with `consistent: true` makes the entire request consistent).
+     */
+    consistent?: boolean;
+};
+/**
+ * Result of the TableBatchGet command.
+ *
+ * @template Gets - Tuple of PreparedBatchGet types, one per entity group.
+ */
+type TableBatchGetResult<Gets extends PreparedBatchGet<any>[]> = {
+    responseMetadata?: ResponseMetadata;
+    consumedCapacity?: ConsumedCapacity | undefined;
+    items: TableBatchGetItems<Gets>;
+    unprocessedKeys: TableBatchGetUnprocessedKeys<Gets>;
+};
+/**
+ * Table-level command to perform a batch get across multiple entity types.
+ *
+ * Unlike `BatchGet` which operates on a single entity type, this command accepts
+ * get operations from multiple entities via `entity.prepare(new BatchGet({ keys: [...] }))`
+ * and returns results grouped by entity in a tuple structure.
+ *
+ * Unprocessed keys are returned per entity in a tuple matching the input order.
+ *
+ * @example
+ * ```typescript
+ * const { items, unprocessedKeys } = await table.send(new TableBatchGet({
+ *   gets: [
+ *     userEntity.prepare(new BatchGet({
+ *       keys: [{ userId: 'u1' }, { userId: 'u2' }],
+ *     })),
+ *     orderEntity.prepare(new BatchGet({
+ *       keys: [{ orderId: 'o1' }],
+ *     })),
+ *   ],
+ *   consistent: true, // Apply strongly consistent reads to all groups
+ * }))
+ *
+ * const [users, orders] = items
+ * // users: User[]
+ * // orders: Order[]
+ * ```
+ */
+declare class TableBatchGet<Gets extends PreparedBatchGet<any>[]> implements TableCommand<TableBatchGetResult<Gets>> {
+    #private;
+    constructor(config: TableBatchGetConfig<Gets>);
+    execute(table: DynamoTable): Promise<TableBatchGetResult<Gets>>;
+}
+
+export { TableBatchGet };
+export type { TableBatchGetConfig, TableBatchGetResult };

package/dist/commands/table-batch-get.mjs

@@ -0,0 +1,121 @@
+import { __classPrivateFieldSet, __awaiter, __classPrivateFieldGet } from '../node_modules/tslib/tslib.es6.mjs';
+import { DocumentBuilderError } from '../errors.mjs';
+import { BatchGetCommand } from '@aws-sdk/lib-dynamodb';
+import { BATCH_GET_VALIDATION_CONCURRENCY } from '../internal-constants.mjs';
+import pMap from 'p-map';
+
+var _TableBatchGet_config;
+/**
+ * Table-level command to perform a batch get across multiple entity types.
+ *
+ * Unlike `BatchGet` which operates on a single entity type, this command accepts
+ * get operations from multiple entities via `entity.prepare(new BatchGet({ keys: [...] }))`
+ * and returns results grouped by entity in a tuple structure.
+ *
+ * Unprocessed keys are returned per entity in a tuple matching the input order.
+ *
+ * @example
+ * ```typescript
+ * const { items, unprocessedKeys } = await table.send(new TableBatchGet({
+ *   gets: [
+ *     userEntity.prepare(new BatchGet({
+ *       keys: [{ userId: 'u1' }, { userId: 'u2' }],
+ *     })),
+ *     orderEntity.prepare(new BatchGet({
+ *       keys: [{ orderId: 'o1' }],
+ *     })),
+ *   ],
+ *   consistent: true, // Apply strongly consistent reads to all groups
+ * }))
+ *
+ * const [users, orders] = items
+ * // users: User[]
+ * // orders: Order[]
+ * ```
+ */
+// biome-ignore lint/suspicious/noExplicitAny: gets span multiple heterogeneous entity schemas
+class TableBatchGet {
+    constructor(config) {
+        _TableBatchGet_config.set(this, void 0);
+        __classPrivateFieldSet(this, _TableBatchGet_config, config, "f");
+    }
+    execute(table) {
+        return __awaiter(this, void 0, void 0, function* () {
+            var _a, _b, _c, _d, _e, _f, _g, _h, _j;
+            for (const group of __classPrivateFieldGet(this, _TableBatchGet_config, "f").gets) {
+                if (group.entity.table !== table) {
+                    throw new DocumentBuilderError(`Entity table "${group.entity.table.tableName}" does not match the executing table "${table.tableName}"`);
+                }
+            }
+            // Command-level consistent overrides all group-level settings.
+            // If not specified, fall back to the per-group OR logic (any group consistent → all consistent).
+            // biome-ignore lint/suspicious/noExplicitAny: heterogeneous group schemas require any
+            const consistent = (_a = __classPrivateFieldGet(this, _TableBatchGet_config, "f").consistent) !== null && _a !== void 0 ? _a : __classPrivateFieldGet(this, _TableBatchGet_config, "f").gets.some((g) => g.consistent);
+            // Aggregate all keys from every entity group into a single list
+            // biome-ignore lint/suspicious/noExplicitAny: heterogeneous group schemas require any
+            const allKeys = __classPrivateFieldGet(this, _TableBatchGet_config, "f").gets.flatMap((g) => g.keys);
+            const batchGet = new BatchGetCommand({
+                RequestItems: {
+                    [table.tableName]: {
+                        Keys: allKeys,
+                        ConsistentRead: consistent,
+                    },
+                },
+                ReturnConsumedCapacity: __classPrivateFieldGet(this, _TableBatchGet_config, "f").returnConsumedCapacity,
+            });
+            const result = yield table.documentClient.send(batchGet, {
+                abortSignal: (_b = __classPrivateFieldGet(this, _TableBatchGet_config, "f").abortController) === null || _b === void 0 ? void 0 : _b.signal,
+                requestTimeout: __classPrivateFieldGet(this, _TableBatchGet_config, "f").timeoutMs,
+            });
+            const rawItems = ((_d = (_c = result.Responses) === null || _c === void 0 ? void 0 : _c[table.tableName]) !== null && _d !== void 0 ? _d : []);
+            // Match returned items back to their entity groups by comparing primary keys.
+            // DynamoDB does not guarantee order in batch get responses, so we must match by key.
+            const groupedItems = __classPrivateFieldGet(this, _TableBatchGet_config, "f").gets.map(() => []);
+            for (const item of rawItems) {
+                for (let i = 0; i < __classPrivateFieldGet(this, _TableBatchGet_config, "f").gets.length; i++) {
+                    // biome-ignore lint/suspicious/noExplicitAny: heterogeneous group schemas require any
+                    const group = __classPrivateFieldGet(this, _TableBatchGet_config, "f").gets[i];
+                    if (group.matchItem(item)) {
+                        groupedItems[i].push(item);
+                        break;
+                    }
+                }
+            }
+            // Parse each group's matched items
+            const parsedItems = yield pMap(__classPrivateFieldGet(this, _TableBatchGet_config, "f").gets, 
+            // biome-ignore lint/suspicious/noExplicitAny: heterogeneous group schemas require any
+            (group, i) => __awaiter(this, void 0, void 0, function* () { var _a; return group.parseResults(groupedItems[i], (_a = __classPrivateFieldGet(this, _TableBatchGet_config, "f").skipValidation) !== null && _a !== void 0 ? _a : false); }), {
+                concurrency: BATCH_GET_VALIDATION_CONCURRENCY,
+                signal: (_e = __classPrivateFieldGet(this, _TableBatchGet_config, "f").abortController) === null || _e === void 0 ? void 0 : _e.signal,
+            });
+            // Map unprocessed keys back to per-entity groups
+            const rawUnprocessedKeys = 
+            // biome-ignore lint/suspicious/noExplicitAny: DynamoDB SDK returns untyped key objects
+            ((_h = (_g = (_f = result.UnprocessedKeys) === null || _f === void 0 ? void 0 : _f[table.tableName]) === null || _g === void 0 ? void 0 : _g.Keys) !== null && _h !== void 0 ? _h : []);
+            const unprocessedKeys = __classPrivateFieldGet(this, _TableBatchGet_config, "f").gets.map(() => undefined);
+            for (const key of rawUnprocessedKeys) {
+                for (let i = 0; i < __classPrivateFieldGet(this, _TableBatchGet_config, "f").gets.length; i++) {
+                    // biome-ignore lint/suspicious/noExplicitAny: heterogeneous group schemas require any
+                    const group = __classPrivateFieldGet(this, _TableBatchGet_config, "f").gets[i];
+                    const matched = group.matchUnprocessedKey(key);
+                    if (matched !== undefined) {
+                        if (!unprocessedKeys[i])
+                            unprocessedKeys[i] = [];
+                        unprocessedKeys[i].push(matched);
+                        break;
+                    }
+                }
+            }
+            return {
+                responseMetadata: result.$metadata,
+                consumedCapacity: (_j = result.ConsumedCapacity) === null || _j === void 0 ? void 0 : _j[0],
+                items: parsedItems,
+                unprocessedKeys: unprocessedKeys,
+            };
+        });
+    }
+}
+_TableBatchGet_config = new WeakMap();
+
+export { TableBatchGet };
+//# sourceMappingURL=table-batch-get.mjs.map

package/dist/commands/table-batch-get.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"table-batch-get.mjs","sources":["../../src/commands/table-batch-get.ts"],"sourcesContent":[null],"names":[],"mappings":";;;;;;;AA+DA;;;;;;;;;;;;;;;;;;;;;;;;;;;AA2BG;AACH;MACa,aAAa,CAAA;AAKxB,IAAA,WAAA,CAAY,MAAiC,EAAA;QAF7C,qBAAA,CAAA,GAAA,CAAA,IAAA,EAAA,MAAA,CAAA;AAGE,QAAA,sBAAA,CAAA,IAAI,EAAA,qBAAA,EAAW,MAAM,EAAA,GAAA,CAAA;IACvB;AAEa,IAAA,OAAO,CAAC,KAAkB,EAAA;;;YACrC,KAAK,MAAM,KAAK,IAAI,sBAAA,CAAA,IAAI,EAAA,qBAAA,EAAA,GAAA,CAAQ,CAAC,IAAI,EAAE;gBACrC,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,KAAK,KAAK,EAAE;AAChC,oBAAA,MAAM,IAAI,oBAAoB,CAC5B,CAAA,cAAA,EAAiB,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,SAAS,CAAA,sCAAA,EAAyC,KAAK,CAAC,SAAS,CAAA,CAAA,CAAG,CACzG;gBACH;YACF;;;;YAKA,MAAM,UAAU,GACd,CAAA,EAAA,GAAA,sBAAA,CAAA,IAAI,EAAA,qBAAA,EAAA,GAAA,CAAQ,CAAC,UAAU,MAAA,IAAA,IAAA,EAAA,KAAA,MAAA,GAAA,EAAA,GAAI,sBAAA,CAAA,IAAI,EAAA,qBAAA,EAAA,GAAA,CAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAwB,KAAK,CAAC,CAAC,UAAU,CAAC;;;AAI/F,YAAA,MAAM,OAAO,GAAG,sBAAA,CAAA,IAAI,EAAA,qBAAA,EAAA,GAAA,CAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAwB,KAAK,CAAC,CAAC,IAAI,CAAC;AAE/E,YAAA,MAAM,QAAQ,GAAG,IAAI,eAAe,CAAC;AACnC,gBAAA,YAAY,EAAE;AACZ,oBAAA,CAAC,KAAK,CAAC,SAAS,GAAG;AACjB,wBAAA,IAAI,EAAE,OAAO;AACb,wBAAA,cAAc,EAAE,UAAU;AAC3B,qBAAA;AACF,iBAAA;AACD,gBAAA,sBAAsB,EAAE,sBAAA,CAAA,IAAI,EAAA,qBAAA,EAAA,GAAA,CAAQ,CAAC,sBAAsB;AAC5D,aAAA,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,cAAc,CAAC,IAAI,CAAC,QAAQ,EAAE;gBACvD,WAAW,EAAE,MAAA,sBAAA,CAAA,IAAI,6BAAQ,CAAC,eAAe,0CAAE,MAAM;AACjD,gBAAA,cAAc,EAAE,sBAAA,CAAA,IAAI,EAAA,qBAAA,EAAA,GAAA,CAAQ,CAAC,SAAS;AACvC,aAAA,CAAC;AAEF,YAAA,MAAM,QAAQ,IAAI,CAAA,EAAA,GAAA,CAAA,EAAA,GAAA,MAAM,CAAC,SAAS,MAAA,IAAA,IAAA,EAAA,KAAA,MAAA,GAAA,MAAA,GAAA,EAAA,CAAG,KAAK,CAAC,SAAS,CAAC,MAAA,IAAA,IAAA,EAAA,KAAA,MAAA,GAAA,EAAA,GAAI,EAAE,CAA8B;;;AAIzF,YAAA,MAAM,YAAY,GAAqB,sBAAA,CAAA,IAAI,6BAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,EAAE,CAAC;AAEtE,YAAA,KAAK,MAAM,IAAI,IAAI,QAAQ,EAAE;AAC3B,gBAAA,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,sBAAA,CAAA,IAAI,EAAA,qBAAA,EAAA,GAAA,CAAQ,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE;;oBAEjD,MAAM,KAAK,GAAG,sBAAA,CAAA,IAAI,EAAA,qBAAA,EAAA,GAAA,CAAQ,CAAC,IAAI,CAAC,CAAC,CAA0B;AAC3D,oBAAA,IAAI,KAAK,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE;wBACzB,YAAY,CAAC,CAAC,CAAE,CAAC,IAAI,CAAC,IAAI,CAAC;wBAC3B;oBACF;gBACF;YACF;;YAGA,MAAM,WAAW,GAAG,MAAM,IAAI,CAC5B,sBAAA,CAAA,IAAI,EAAA,qBAAA,EAAA,GAAA,CAAQ,CAAC,IAAI;;YAEjB,CAAO,KAA4B,EAAE,CAAS,KAAI,SAAA,CAAA,IAAA,EAAA,MAAA,EAAA,MAAA,EAAA,aAAA,EAAA,IAAA,EAAA,CAAA,CAChD,OAAA,KAAK,CAAC,YAAY,CAAC,YAAY,CAAC,CAAC,CAAE,EAAE,CAAA,EAAA,GAAA,sBAAA,CAAA,IAAI,EAAA,qBAAA,EAAA,GAAA,CAAQ,CAAC,cAAc,MAAA,IAAA,IAAA,EAAA,KAAA,MAAA,GAAA,EAAA,GAAI,KAAK,CAAC,CAAA,CAAA,CAAA,CAAA,EAC5E;AACE,gBAAA,WAAW,EAAE,gCAAgC;gBAC7C,MAAM,EAAE,MAAA,sBAAA,CAAA,IAAI,6BAAQ,CAAC,eAAe,0CAAE,MAAM;AAC7C,aAAA,CACF;;AAGD,YAAA,MAAM,kBAAkB;;AAEtB,aAAC,MAAA,CAAA,EAAA,GAAA,CAAA,EAAA,GAAA,MAAM,CAAC,eAAe,MAAA,IAAA,IAAA,EAAA,KAAA,MAAA,GAAA,MAAA,GAAA,EAAA,CAAG,KAAK,CAAC,SAAS,CAAC,MAAA,IAAA,IAAA,EAAA,KAAA,MAAA,GAAA,MAAA,GAAA,EAAA,CAAE,IAAI,MAAA,IAAA,IAAA,EAAA,KAAA,MAAA,GAAA,EAAA,GAAI,EAAE,CAAU;AAElE,YAAA,MAAM,eAAe,GAAsC,sBAAA,CAAA,IAAI,6BAAQ,CAAC,IAAI,CAAC,GAAG,CAC9E,MAAM,SAAS,CAChB;AAED,YAAA,KAAK,MAAM,GAAG,IAAI,kBAAkB,EAAE;AACpC,gBAAA,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,sBAAA,CAAA,IAAI,EAAA,qBAAA,EAAA,GAAA,CAAQ,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE;;oBAEjD,MAAM,KAAK,GAAG,sBAAA,CAAA,IAAI,EAAA,qBAAA,EAAA,GAAA,CAAQ,CAAC,IAAI,CAAC,CAAC,CAA0B;oBAC3D,MAAM,OAAO,GAAG,KAAK,CAAC,mBAAmB,CAAC,GAAG,CAAC;AAC9C,oBAAA,IAAI,OAAO,KAAK,SAAS,EAAE;AACzB,wBAAA,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC;AAAE,4BAAA,eAAe,CAAC,CAAC,CAAC,GAAG,EAAE;wBAChD,eAAe,CAAC,CAAC,CAAE,CAAC,IAAI,CAAC,OAAO,CAAC;wBACjC;oBACF;gBACF;YACF;YAEA,OAAO;gBACL,gBAAgB,EAAE,MAAM,CAAC,SAAS;AAClC,gBAAA,gBAAgB,EAAE,CAAA,EAAA,GAAA,MAAM,CAAC,gBAAgB,MAAA,IAAA,IAAA,EAAA,KAAA,MAAA,GAAA,MAAA,GAAA,EAAA,CAAG,CAAC,CAAC;AAC9C,gBAAA,KAAK,EAAE,WAAuC;AAC9C,gBAAA,eAAe,EAAE,eAAqD;aACvE;QACH,CAAC,CAAA;AAAA,IAAA;AACF;;;;;"}

package/dist/commands/table-batch-write.cjs

@@ -0,0 +1,119 @@
+'use strict';
+
+var tslib_es6 = require('../node_modules/tslib/tslib.es6.cjs');
+var errors = require('../errors.cjs');
+var libDynamodb = require('@aws-sdk/lib-dynamodb');
+var internalConstants = require('../internal-constants.cjs');
+var pMap = require('p-map');
+
+var _TableBatchWrite_config;
+/**
+ * Table-level command to perform a batch write across multiple entity types.
+ *
+ * Unlike `BatchWrite` which operates on a single entity type, this command accepts
+ * write operations from multiple entities via `entity.prepare(new BatchWrite({ ... }))`
+ * and executes them in a single DynamoDB BatchWrite request.
+ *
+ * Unprocessed puts and deletes are returned per entity in a tuple matching the input order.
+ *
+ * @example
+ * ```typescript
+ * const { unprocessedPuts, unprocessedDeletes } = await table.send(new TableBatchWrite({
+ *   writes: [
+ *     userEntity.prepare(new BatchWrite({
+ *       items: [{ userId: 'u1', name: 'Alice' }],
+ *       deletes: [{ userId: 'u2', name: 'Bob' }],
+ *     })),
+ *     orderEntity.prepare(new BatchWrite({
+ *       items: [{ orderId: 'o1', status: 'pending', total: 99 }],
+ *     })),
+ *   ],
+ * }))
+ *
+ * const [userUnprocessedPuts, orderUnprocessedPuts] = unprocessedPuts
+ * const [userUnprocessedDeletes, orderUnprocessedDeletes] = unprocessedDeletes
+ * ```
+ */
+// biome-ignore lint/suspicious/noExplicitAny: writes span multiple heterogeneous entity schemas
+class TableBatchWrite {
+    constructor(config) {
+        _TableBatchWrite_config.set(this, void 0);
+        tslib_es6.__classPrivateFieldSet(this, _TableBatchWrite_config, config, "f");
+    }
+    execute(table) {
+        return tslib_es6.__awaiter(this, void 0, void 0, function* () {
+            var _a, _b, _c, _d, _e, _f, _g;
+            for (const group of tslib_es6.__classPrivateFieldGet(this, _TableBatchWrite_config, "f").writes) {
+                if (group.entity.table !== table) {
+                    throw new errors.DocumentBuilderError(`Entity table "${group.entity.table.tableName}" does not match the executing table "${table.tableName}"`);
+                }
+            }
+            // Build all requests across all entity groups, preserving group index for unprocessed mapping
+            // biome-ignore lint/suspicious/noExplicitAny: heterogeneous group schemas require any
+            const allRequests = (yield pMap(tslib_es6.__classPrivateFieldGet(this, _TableBatchWrite_config, "f").writes, (group) => {
+                var _a, _b;
+                return group.buildRequests((_a = tslib_es6.__classPrivateFieldGet(this, _TableBatchWrite_config, "f").skipValidation) !== null && _a !== void 0 ? _a : false, (_b = tslib_es6.__classPrivateFieldGet(this, _TableBatchWrite_config, "f").abortController) === null || _b === void 0 ? void 0 : _b.signal);
+            }, {
+                concurrency: internalConstants.BATCH_WRITE_VALIDATION_CONCURRENCY,
+                signal: (_a = tslib_es6.__classPrivateFieldGet(this, _TableBatchWrite_config, "f").abortController) === null || _a === void 0 ? void 0 : _a.signal,
+            })).flat();
+            const batchWrite = new libDynamodb.BatchWriteCommand({
+                RequestItems: {
+                    [table.tableName]: allRequests,
+                },
+                ReturnConsumedCapacity: tslib_es6.__classPrivateFieldGet(this, _TableBatchWrite_config, "f").returnConsumedCapacity,
+                ReturnItemCollectionMetrics: tslib_es6.__classPrivateFieldGet(this, _TableBatchWrite_config, "f").returnItemCollectionMetrics,
+            });
+            const result = yield table.documentClient.send(batchWrite, {
+                abortSignal: (_b = tslib_es6.__classPrivateFieldGet(this, _TableBatchWrite_config, "f").abortController) === null || _b === void 0 ? void 0 : _b.signal,
+                requestTimeout: tslib_es6.__classPrivateFieldGet(this, _TableBatchWrite_config, "f").timeoutMs,
+            });
+            const unprocessedItems = (_d = (_c = result.UnprocessedItems) === null || _c === void 0 ? void 0 : _c[table.tableName]) !== null && _d !== void 0 ? _d : [];
+            // Map unprocessed items back to their respective entity groups
+            const unprocessedPuts = tslib_es6.__classPrivateFieldGet(this, _TableBatchWrite_config, "f").writes.map(() => undefined);
+            const unprocessedDeletes = tslib_es6.__classPrivateFieldGet(this, _TableBatchWrite_config, "f").writes.map(() => undefined);
+            for (const unprocessedItem of unprocessedItems) {
+                if ((_e = unprocessedItem.PutRequest) === null || _e === void 0 ? void 0 : _e.Item) {
+                    const item = unprocessedItem.PutRequest.Item;
+                    for (let i = 0; i < tslib_es6.__classPrivateFieldGet(this, _TableBatchWrite_config, "f").writes.length; i++) {
+                        // biome-ignore lint/suspicious/noExplicitAny: heterogeneous group schemas require any
+                        const group = tslib_es6.__classPrivateFieldGet(this, _TableBatchWrite_config, "f").writes[i];
+                        const matched = group.matchUnprocessedPut(item);
+                        if (matched !== undefined) {
+                            if (!unprocessedPuts[i])
+                                unprocessedPuts[i] = [];
+                            unprocessedPuts[i].push(matched);
+                            break;
+                        }
+                    }
+                }
+                else if ((_f = unprocessedItem.DeleteRequest) === null || _f === void 0 ? void 0 : _f.Key) {
+                    // biome-ignore lint/suspicious/noExplicitAny: DynamoDB SDK returns untyped key objects
+                    const key = unprocessedItem.DeleteRequest.Key;
+                    for (let i = 0; i < tslib_es6.__classPrivateFieldGet(this, _TableBatchWrite_config, "f").writes.length; i++) {
+                        // biome-ignore lint/suspicious/noExplicitAny: heterogeneous group schemas require any
+                        const group = tslib_es6.__classPrivateFieldGet(this, _TableBatchWrite_config, "f").writes[i];
+                        const matched = group.matchUnprocessedDelete(key);
+                        if (matched !== undefined) {
+                            if (!unprocessedDeletes[i])
+                                unprocessedDeletes[i] = [];
+                            unprocessedDeletes[i].push(matched);
+                            break;
+                        }
+                    }
+                }
+            }
+            return {
+                responseMetadata: result.$metadata,
+                consumedCapacity: (_g = result.ConsumedCapacity) === null || _g === void 0 ? void 0 : _g[0],
+                itemCollectionMetrics: result.ItemCollectionMetrics,
+                unprocessedPuts: unprocessedPuts,
+                unprocessedDeletes: unprocessedDeletes,
+            };
+        });
+    }
+}
+_TableBatchWrite_config = new WeakMap();
+
+exports.TableBatchWrite = TableBatchWrite;
+//# sourceMappingURL=table-batch-write.cjs.map

package/dist/commands/table-batch-write.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"table-batch-write.cjs","sources":["../../src/commands/table-batch-write.ts"],"sourcesContent":[null],"names":["__classPrivateFieldSet","__classPrivateFieldGet","DocumentBuilderError","BATCH_WRITE_VALIDATION_CONCURRENCY","BatchWriteCommand"],"mappings":";;;;;;;;;AA6DA;;;;;;;;;;;;;;;;;;;;;;;;;;AA0BG;AACH;MACa,eAAe,CAAA;AAK1B,IAAA,WAAA,CAAY,MAAqC,EAAA;QAFjD,uBAAA,CAAA,GAAA,CAAA,IAAA,EAAA,MAAA,CAAA;AAGE,QAAAA,gCAAA,CAAA,IAAI,EAAA,uBAAA,EAAW,MAAM,EAAA,GAAA,CAAA;IACvB;AAEa,IAAA,OAAO,CAAC,KAAkB,EAAA;;;YACrC,KAAK,MAAM,KAAK,IAAIC,gCAAA,CAAA,IAAI,EAAA,uBAAA,EAAA,GAAA,CAAQ,CAAC,MAAM,EAAE;gBACvC,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,KAAK,KAAK,EAAE;AAChC,oBAAA,MAAM,IAAIC,2BAAoB,CAC5B,CAAA,cAAA,EAAiB,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,SAAS,CAAA,sCAAA,EAAyC,KAAK,CAAC,SAAS,CAAA,CAAA,CAAG,CACzG;gBACH;YACF;;;AAIA,YAAA,MAAM,WAAW,GAAG,CAClB,MAAM,IAAI,CACRD,gCAAA,CAAA,IAAI,EAAA,uBAAA,EAAA,GAAA,CAAQ,CAAC,MAAM,EACnB,CAAC,KAA8B,KAAI;;gBACjC,OAAA,KAAK,CAAC,aAAa,CACjB,MAAAA,gCAAA,CAAA,IAAI,EAAA,uBAAA,EAAA,GAAA,CAAQ,CAAC,cAAc,MAAA,IAAA,IAAA,EAAA,KAAA,MAAA,GAAA,EAAA,GAAI,KAAK,EACpC,CAAA,EAAA,GAAAA,iCAAA,IAAI,EAAA,uBAAA,EAAA,GAAA,CAAQ,CAAC,eAAe,MAAA,IAAA,IAAA,EAAA,KAAA,MAAA,GAAA,MAAA,GAAA,EAAA,CAAE,MAAM,CACrC;aAAA,EACH;AACE,gBAAA,WAAW,EAAEE,oDAAkC;gBAC/C,MAAM,EAAE,MAAAF,gCAAA,CAAA,IAAI,+BAAQ,CAAC,eAAe,0CAAE,MAAM;AAC7C,aAAA,CACF,EACD,IAAI,EAAE;AAER,YAAA,MAAM,UAAU,GAAG,IAAIG,6BAAiB,CAAC;AACvC,gBAAA,YAAY,EAAE;AACZ,oBAAA,CAAC,KAAK,CAAC,SAAS,GAAG,WAAW;AAC/B,iBAAA;AACD,gBAAA,sBAAsB,EAAEH,gCAAA,CAAA,IAAI,EAAA,uBAAA,EAAA,GAAA,CAAQ,CAAC,sBAAsB;AAC3D,gBAAA,2BAA2B,EAAEA,gCAAA,CAAA,IAAI,EAAA,uBAAA,EAAA,GAAA,CAAQ,CAAC,2BAA2B;AACtE,aAAA,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,cAAc,CAAC,IAAI,CAAC,UAAU,EAAE;gBACzD,WAAW,EAAE,MAAAA,gCAAA,CAAA,IAAI,+BAAQ,CAAC,eAAe,0CAAE,MAAM;AACjD,gBAAA,cAAc,EAAEA,gCAAA,CAAA,IAAI,EAAA,uBAAA,EAAA,GAAA,CAAQ,CAAC,SAAS;AACvC,aAAA,CAAC;AAEF,YAAA,MAAM,gBAAgB,GAAG,CAAA,EAAA,GAAA,CAAA,EAAA,GAAA,MAAM,CAAC,gBAAgB,MAAA,IAAA,IAAA,EAAA,KAAA,MAAA,GAAA,MAAA,GAAA,EAAA,CAAG,KAAK,CAAC,SAAS,CAAC,MAAA,IAAA,IAAA,EAAA,KAAA,MAAA,GAAA,EAAA,GAAI,EAAE;;AAGzE,YAAA,MAAM,eAAe,GAAsCA,gCAAA,CAAA,IAAI,+BAAQ,CAAC,MAAM,CAAC,GAAG,CAChF,MAAM,SAAS,CAChB;AACD,YAAA,MAAM,kBAAkB,GAAsCA,gCAAA,CAAA,IAAI,+BAAQ,CAAC,MAAM,CAAC,GAAG,CACnF,MAAM,SAAS,CAChB;AAED,YAAA,KAAK,MAAM,eAAe,IAAI,gBAAgB,EAAE;AAC9C,gBAAA,IAAI,MAAA,eAAe,CAAC,UAAU,MAAA,IAAA,IAAA,EAAA,KAAA,MAAA,GAAA,MAAA,GAAA,EAAA,CAAE,IAAI,EAAE;AACpC,oBAAA,MAAM,IAAI,GAAG,eAAe,CAAC,UAAU,CAAC,IAA+B;AACvE,oBAAA,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAGA,gCAAA,CAAA,IAAI,EAAA,uBAAA,EAAA,GAAA,CAAQ,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE;;wBAEnD,MAAM,KAAK,GAAGA,gCAAA,CAAA,IAAI,EAAA,uBAAA,EAAA,GAAA,CAAQ,CAAC,MAAM,CAAC,CAAC,CAA4B;wBAC/D,MAAM,OAAO,GAAG,KAAK,CAAC,mBAAmB,CAAC,IAAI,CAAC;AAC/C,wBAAA,IAAI,OAAO,KAAK,SAAS,EAAE;AACzB,4BAAA,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC;AAAE,gCAAA,eAAe,CAAC,CAAC,CAAC,GAAG,EAAE;4BAChD,eAAe,CAAC,CAAC,CAAE,CAAC,IAAI,CAAC,OAAO,CAAC;4BACjC;wBACF;oBACF;gBACF;AAAO,qBAAA,IAAI,MAAA,eAAe,CAAC,aAAa,MAAA,IAAA,IAAA,EAAA,KAAA,MAAA,GAAA,MAAA,GAAA,EAAA,CAAE,GAAG,EAAE;;AAE7C,oBAAA,MAAM,GAAG,GAAG,eAAe,CAAC,aAAa,CAAC,GAAU;AACpD,oBAAA,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAGA,gCAAA,CAAA,IAAI,EAAA,uBAAA,EAAA,GAAA,CAAQ,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE;;wBAEnD,MAAM,KAAK,GAAGA,gCAAA,CAAA,IAAI,EAAA,uBAAA,EAAA,GAAA,CAAQ,CAAC,MAAM,CAAC,CAAC,CAA4B;wBAC/D,MAAM,OAAO,GAAG,KAAK,CAAC,sBAAsB,CAAC,GAAG,CAAC;AACjD,wBAAA,IAAI,OAAO,KAAK,SAAS,EAAE;AACzB,4BAAA,IAAI,CAAC,kBAAkB,CAAC,CAAC,CAAC;AAAE,gCAAA,kBAAkB,CAAC,CAAC,CAAC,GAAG,EAAE;4BACtD,kBAAkB,CAAC,CAAC,CAAE,CAAC,IAAI,CAAC,OAAO,CAAC;4BACpC;wBACF;oBACF;gBACF;YACF;YAEA,OAAO;gBACL,gBAAgB,EAAE,MAAM,CAAC,SAAS;AAClC,gBAAA,gBAAgB,EAAE,CAAA,EAAA,GAAA,MAAM,CAAC,gBAAgB,MAAA,IAAA,IAAA,EAAA,KAAA,MAAA,GAAA,MAAA,GAAA,EAAA,CAAG,CAAC,CAAC;gBAC9C,qBAAqB,EAAE,MAAM,CAAC,qBAAqB;AACnD,gBAAA,eAAe,EAAE,eAAyD;AAC1E,gBAAA,kBAAkB,EAAE,kBAA+D;aACpF;QACH,CAAC,CAAA;AAAA,IAAA;AACF;;;;;"}

package/dist/commands/table-batch-write.d.ts

@@ -0,0 +1,79 @@
+import { PreparedBatchWrite, BaseConfig, TableCommand } from './index.js';
+import { DynamoTable } from '../core/table.js';
+import { EntitySchema } from '../core/index.js';
+import { ReturnItemCollectionMetrics, ConsumedCapacity, ItemCollectionMetrics } from '@aws-sdk/client-dynamodb';
+import { ResponseMetadata } from '@aws-sdk/types';
+import { ZodObject } from 'zod/v4';
+
+/**
+ * Extracts the entity schema type from a PreparedBatchWrite.
+ */
+type ExtractSchema<T> = T extends PreparedBatchWrite<infer S> ? EntitySchema<S> : never;
+/**
+ * Maps an array of PreparedBatchWrites to a tuple of their unprocessed put arrays.
+ */
+type TableBatchWriteUnprocessedPuts<Writes extends PreparedBatchWrite<ZodObject>[]> = {
+    [K in keyof Writes]: Array<ExtractSchema<Writes[K]>> | undefined;
+};
+/**
+ * Maps an array of PreparedBatchWrites to a tuple of their unprocessed delete arrays.
+ */
+type TableBatchWriteUnprocessedDeletes<Writes extends PreparedBatchWrite<ZodObject>[]> = {
+    [K in keyof Writes]: Array<Partial<ExtractSchema<Writes[K]>>> | undefined;
+};
+/**
+ * Configuration for the TableBatchWrite command.
+ *
+ * @template Writes - Tuple of PreparedBatchWrite types, one per entity group.
+ */
+type TableBatchWriteConfig<Writes extends PreparedBatchWrite<any>[]> = BaseConfig & {
+    writes: [...Writes];
+    returnItemCollectionMetrics?: ReturnItemCollectionMetrics;
+};
+/**
+ * Result of the TableBatchWrite command.
+ *
+ * @template Writes - Tuple of PreparedBatchWrite types, one per entity group.
+ */
+type TableBatchWriteResult<Writes extends PreparedBatchWrite<any>[]> = {
+    responseMetadata?: ResponseMetadata;
+    consumedCapacity?: ConsumedCapacity | undefined;
+    itemCollectionMetrics?: ItemCollectionMetrics;
+    unprocessedPuts: TableBatchWriteUnprocessedPuts<Writes>;
+    unprocessedDeletes: TableBatchWriteUnprocessedDeletes<Writes>;
+};
+/**
+ * Table-level command to perform a batch write across multiple entity types.
+ *
+ * Unlike `BatchWrite` which operates on a single entity type, this command accepts
+ * write operations from multiple entities via `entity.prepare(new BatchWrite({ ... }))`
+ * and executes them in a single DynamoDB BatchWrite request.
+ *
+ * Unprocessed puts and deletes are returned per entity in a tuple matching the input order.
+ *
+ * @example
+ * ```typescript
+ * const { unprocessedPuts, unprocessedDeletes } = await table.send(new TableBatchWrite({
+ *   writes: [
+ *     userEntity.prepare(new BatchWrite({
+ *       items: [{ userId: 'u1', name: 'Alice' }],
+ *       deletes: [{ userId: 'u2', name: 'Bob' }],
+ *     })),
+ *     orderEntity.prepare(new BatchWrite({
+ *       items: [{ orderId: 'o1', status: 'pending', total: 99 }],
+ *     })),
+ *   ],
+ * }))
+ *
+ * const [userUnprocessedPuts, orderUnprocessedPuts] = unprocessedPuts
+ * const [userUnprocessedDeletes, orderUnprocessedDeletes] = unprocessedDeletes
+ * ```
+ */
+declare class TableBatchWrite<Writes extends PreparedBatchWrite<any>[]> implements TableCommand<TableBatchWriteResult<Writes>> {
+    #private;
+    constructor(config: TableBatchWriteConfig<Writes>);
+    execute(table: DynamoTable): Promise<TableBatchWriteResult<Writes>>;
+}
+
+export { TableBatchWrite };
+export type { TableBatchWriteConfig, TableBatchWriteResult };