feat: add createMany API to aws-dynamodb

tools-deno/tools-server/deno.json

@@ -1,6 +1,6 @@
 {
   "name": "@vighnesh153/tools-server",
-  "version": "0.1.1",
+  "version": "0.1.2",
   "exports": {
     "./aws_dynamodb": "./src/aws_dynamodb/mod.ts"
   },

tools-deno/tools-server/src/aws_dynamodb/dynamodb_table.ts

@@ -14,6 +14,10 @@ export type OptionalCreateOne = { error: null } | {
   error: { message: "CREATION_FAILED"; errorObject: unknown };
 };
 
+export type OptionalCreateMany = { error: null } | {
+  error: { message: "CREATION_FAILED"; errorObject: unknown };
+};
+
 export interface DynamoDBTable<T extends TableMetadata> {
   queryOne: <
     TKey extends keyof T["fields"],
@@ -33,6 +37,10 @@ export interface DynamoDBTable<T extends TableMetadata> {
     data: { [key in TField]: DynamoTypeMap[T["fields"][key]] };
   }): Promise<OptionalCreateOne>;
 
+  createMany<TField extends keyof T["fields"]>(params: {
+    data: { [key in TField]: DynamoTypeMap[T["fields"][key]] }[];
+  }): Promise<OptionalCreateOne>;
+
   scanOne: <
     TKey extends keyof T["fields"],
     TFilterBy extends keyof T["fields"],

tools-deno/tools-server/src/aws_dynamodb/dynamodb_table_impl.ts

@@ -1,4 +1,8 @@
-import { PutCommand, QueryCommand, ScanCommand } from "@aws-sdk/lib-dynamodb";
+import {
+  BatchWriteCommand,
+  QueryCommand,
+  ScanCommand,
+} from "@aws-sdk/lib-dynamodb";
 import type { DynamoTypeMap, TableMetadata } from "./table_metadata.ts";
 import type {
   DynamoDBTable,
@@ -48,14 +52,28 @@ export class DynamoDBTableImpl<T extends TableMetadata>
     return this.fetchOne(params, "scan");
   }
 
+  // deno-lint-ignore require-await
   async createOne<TField extends keyof T["fields"]>(params: {
     data: { [key in TField]: DynamoTypeMap[T["fields"][key]] };
+  }): Promise<OptionalCreateOne> {
+    return this.createMany({
+      data: [params.data],
+    });
+  }
+
+  async createMany<TField extends keyof T["fields"]>(params: {
+    data: { [key in TField]: DynamoTypeMap[T["fields"][key]] }[];
   }): Promise<OptionalCreateOne> {
     try {
       await this.client.send(
-        new PutCommand({
-          TableName: this.tableMetadata.tableName,
-          Item: params.data,
+        new BatchWriteCommand({
+          RequestItems: {
+            [this.tableMetadata.tableName]: params.data.map((item) => ({
+              PutRequest: {
+                Item: item,
+              },
+            })),
+          },
         }),
       );
       return { error: null };

tools-deno/tools-server/src/aws_dynamodb/fake_dynamodb_table.ts

@@ -1,5 +1,6 @@
 import type {
   DynamoDBTable,
+  OptionalCreateMany,
   OptionalCreateOne,
   OptionalGetOne,
 } from "./dynamodb_table.ts";
@@ -24,6 +25,9 @@ export class FakeDynamoDBTable<T extends TableMetadata>
   createOneResult: OptionalCreateOne = { error: null };
   createOneCalledTimes = 0;
 
+  createManyResult: OptionalCreateMany = { error: null };
+  createManyCalledTimes = 0;
+
   // deno-lint-ignore require-await
   async queryOne<TKey extends keyof T["fields"]>(): Promise<
     OptionalGetOne<{ [key in TKey]: DynamoTypeMap[T["fields"][key]] }>
@@ -45,4 +49,10 @@ export class FakeDynamoDBTable<T extends TableMetadata>
     this.createOneCalledTimes += 1;
     return this.createOneResult;
   }
+
+  // deno-lint-ignore require-await
+  async createMany(): Promise<OptionalCreateMany> {
+    this.createManyCalledTimes += 1;
+    return this.createManyResult;
+  }
 }

tools-deno/tools-server/src/aws_dynamodb/table_metadata.ts

@@ -1,3 +1,4 @@
+// Descriptors: https://docs.aws.amazon.com/amazondynamodb/latest/developerguide/HowItWorks.NamingRulesDataTypes.html#HowItWorks.DataTypeDescriptors
 export type DynamoTypeMap = {
   string: string;
   number: number;