chore: add benchmark app

.gitignore

@@ -1,4 +1,6 @@
 lib/
 node_modules/
 sdk.generated.ts
-.bundle
+.bundle
+*.tgz
+cdk.out

benchmark/functions/package.json

@@ -0,0 +1,10 @@
+{
+  "name": "@benchmark/functions",
+  "version": "0.0.0",
+  "private": true,
+  "type": "module",
+  "dependencies": {
+    "itty-aws": "workspace:^",
+    "@aws-sdk/client-dynamodb": "^3.245.0"
+  }
+}

benchmark/functions/src/itty-handler.ts

@@ -0,0 +1,19 @@
+import { AWS } from "itty-aws";
+
+const client = new AWS.DynamoDB();
+
+const TableName = process.env.TABLE_NAME!;
+
+export async function handler() {
+  await client.putItem({
+    TableName,
+    Item: {
+      pk: {
+        S: "pk",
+      },
+      prop: {
+        S: "value",
+      },
+    },
+  });
+}

benchmark/functions/src/v2-lambda.ts

@@ -0,0 +1,21 @@
+import { DynamoDB } from "aws-sdk";
+
+const client = new DynamoDB();
+
+const TableName = process.env.TABLE_NAME!;
+
+export async function handler() {
+  await client
+    .putItem({
+      TableName,
+      Item: {
+        pk: {
+          S: "pk",
+        },
+        prop: {
+          S: "value",
+        },
+      },
+    })
+    .promise();
+}

benchmark/functions/src/v3-lambda.ts

@@ -0,0 +1,21 @@
+import { DynamoDBClient, PutItemCommand } from "@aws-sdk/client-dynamodb";
+
+const client = new DynamoDBClient({});
+
+const TableName = process.env.TABLE_NAME!;
+
+export async function handler() {
+  await client.send(
+    new PutItemCommand({
+      TableName,
+      Item: {
+        pk: {
+          S: "pk",
+        },
+        prop: {
+          S: "value",
+        },
+      },
+    })
+  );
+}

benchmark/functions/tsconfig.json

@@ -0,0 +1,8 @@
+{
+  "extends": "../../tsconfig.base.json",
+  "include": ["src"],
+  "exclude": ["lib", "node_modules"],
+  "compilerOptions": {
+    "outDir": "lib"
+  }
+}

benchmark/infra/cdk.json

@@ -0,0 +1,3 @@
+{
+  "app": "ts-node ./src/index.ts"
+}

benchmark/infra/package.json

@@ -0,0 +1,21 @@
+{
+  "name": "@benchmark/infra",
+  "version": "0.0.0",
+  "private": true,
+  "scripts": {
+    "synth": "cdk synth",
+    "deploy": "cdk deploy",
+    "hotswap": "cdk deploy --hotswap"
+  },
+  "dependencies": {
+    "aws-cdk-lib": "^2.60.0",
+    "@benchmark/functions": "workspace:^"
+  },
+  "devDependencies": {
+    "@types/node": "^16",
+    "aws-cdk": "^2.60.0",
+    "constructs": "^10",
+    "ts-node": "^10.9.1",
+    "typescript": "^4.9.4"
+  }
+}

benchmark/infra/src/index.ts

@@ -0,0 +1,64 @@
+import {
+  App,
+  aws_dynamodb,
+  aws_lambda,
+  aws_lambda_nodejs,
+  Stack,
+} from "aws-cdk-lib";
+
+const app = new App();
+
+const stack = new Stack(app, "itty-benchmark");
+
+const table = new aws_dynamodb.Table(stack, "Table", {
+  partitionKey: {
+    name: "pk",
+    type: aws_dynamodb.AttributeType.STRING,
+  },
+  billingMode: aws_dynamodb.BillingMode.PAY_PER_REQUEST,
+});
+
+const props = {
+  environment: {
+    TABLE_NAME: table.tableName,
+  },
+  memorySize: 512,
+};
+
+const ittyFunc = new aws_lambda_nodejs.NodejsFunction(stack, "IttyFunc", {
+  functionName: "benchmark-itty",
+  runtime: aws_lambda.Runtime.NODEJS_18_X,
+  entry: require.resolve("@benchmark/functions/lib/itty-handler"),
+  bundling: {
+    externalModules: ["@aws-sdk/*"],
+  },
+  ...props,
+});
+
+const v3BundledFunc = new aws_lambda_nodejs.NodejsFunction(stack, "V3Bundled", {
+  functionName: "benchmark-v3-bundled",
+  runtime: aws_lambda.Runtime.NODEJS_18_X,
+  entry: require.resolve("@benchmark/functions/lib/itty-handler"),
+  bundling: {
+    externalModules: [],
+  },
+  ...props,
+});
+
+const v3ExcludedFunc = new aws_lambda_nodejs.NodejsFunction(
+  stack,
+  "V3Excluded",
+  {
+    functionName: "benchmark-v3-excluded",
+    runtime: aws_lambda.Runtime.NODEJS_18_X,
+    entry: require.resolve("@benchmark/functions/lib/itty-handler"),
+    bundling: {
+      externalModules: ["@aws-sdk/*"],
+    },
+    ...props,
+  }
+);
+
+table.grantReadWriteData(ittyFunc);
+table.grantReadWriteData(v3BundledFunc);
+table.grantReadWriteData(v3ExcludedFunc);

benchmark/infra/tsconfig.json

@@ -0,0 +1,11 @@
+{
+  "extends": "../../tsconfig.base.json",
+  "include": ["src"],
+  "exclude": ["lib", "node_modules"],
+  "compilerOptions": {
+    "outDir": "lib",
+    "module": "CommonJS",
+    "moduleResolution": "Node",
+    "target": "ES2022"
+  }
+}

package.json

@@ -1,24 +1,21 @@
 {
   "name": "itty-aws",
-  "version": "0.0.0",
+  "version": "0.0.1",
   "files": [
     "lib"
   ],
   "type": "module",
-  "main": "./lib/cjs/index.js",
-  "module": "./lib/esm/index.js",
-  "exports": {
-    ".": {
-      "import": "./lib/esm/index.js",
-      "require": "./lib/cjs/index.js"
-    }
-  },
+  "module": "./lib/index.js",
+  "types": "./lib/index.d.ts",
   "scripts": {
     "gen": "ts-node-esm ./scripts/gen-sdk-types.mts",
     "build": "tsc -b",
     "test": "NODE_OPTIONS=--experimental-vm-modules jest",
     "watch": "tsc -b -w",
-    "analyze:bundle": "sh ./scripts/analyze-bundle.sh"
+    "analyze:bundle": "sh ./scripts/analyze-bundle.sh",
+    "bench:synth": "pnpm -r --filter @benchmark/infra synth",
+    "bench:deploy": "pnpm -r --filter @benchmark/infra run deploy",
+    "bench:hotswap": "pnpm -r --filter @benchmark/infra run hotswap"
   },
   "dependencies": {
     "aws-sdk": "2.1295.0"