Add Docker images.

Author: artema

.gitignore

@@ -1,3 +1,7 @@
+.DS_Store
+
 build
 dist
-node_modules
+node_modules
+
+docker/*.js

config.ts

@@ -1,6 +1,7 @@
 import * as aws from "@pulumi/aws";
 import * as pulumi from "@pulumi/pulumi";
 
+export const account = aws.getCallerIdentity().then(x => x.accountId);
 export const project = pulumi.getProject();
 export const region = aws.config.requireRegion().toString();
 export const stack = pulumi.getStack();

docker.sh

@@ -0,0 +1,29 @@
+#!/bin/bash
+
+NAME=lambda-benchmarks
+REGION=us-east-1
+STAGE=main
+
+AWS_ACCOUNT_ID=`aws sts get-caller-identity --query "Account" --output text`
+
+ECR=$AWS_ACCOUNT_ID.dkr.ecr.$REGION.amazonaws.com
+
+npx tsc --project ./docker
+
+cd docker
+
+docker build -t $NAME-x64:latest -f Dockerfile.x64 .
+docker build -t $NAME-arm64:latest -f Dockerfile.arm64 .
+
+docker tag $NAME-x64:latest $ECR/$NAME-$STAGE-x64:latest
+docker tag $NAME-arm64:latest $ECR/$NAME-$STAGE-arm64:latest
+
+cd ..
+
+echo aws configure set default.region $REGION
+aws configure set default.region $REGION
+
+aws ecr get-login-password --region $REGION | docker login --username AWS --password-stdin $ECR
+
+docker push $ECR/$NAME-$STAGE-x64:latest
+docker push $ECR/$NAME-$STAGE-arm64:latest

docker/Dockerfile.arm64

@@ -0,0 +1,24 @@
+FROM public.ecr.aws/lambda/nodejs:14-arm64
+
+RUN yum -y groupinstall "Development Tools" \
+  && yum -y install wget \
+  && yum -y install glib2 glib2-devel \
+  && yum -y install expat expat-devel
+RUN yum -y install libjpeg-devel
+RUN wget https://github.com/libvips/libvips/releases/download/v8.11.3/vips-8.11.3.tar.gz \
+  && tar xf vips-8.11.3.tar.gz \
+  && cd vips-8.11.3 \
+  && ./configure \
+  && make \
+  && make install \
+  && echo "/usr/local/lib" > /etc/ld.so.conf.d/usrlocal.conf \
+  && /sbin/ldconfig -v \
+  && cd .. \
+  && rm -rf vips-8.11.3 \
+  && rm vips-8.11.3.tar.gz
+
+COPY index.js image.jpeg package.json ${LAMBDA_TASK_ROOT}
+
+RUN npm install --production && npm run init:arm64
+
+CMD [ "index.handler" ]

docker/Dockerfile.x64

@@ -0,0 +1,24 @@
+FROM public.ecr.aws/lambda/nodejs:14-x86_64
+
+RUN yum -y groupinstall "Development Tools" \
+  && yum -y install wget \
+  && yum -y install glib2 glib2-devel \
+  && yum -y install expat expat-devel
+RUN yum -y install libjpeg-devel
+RUN wget https://github.com/libvips/libvips/releases/download/v8.11.3/vips-8.11.3.tar.gz \
+  && tar xf vips-8.11.3.tar.gz \
+  && cd vips-8.11.3 \
+  && ./configure \
+  && make \
+  && make install \
+  && echo "/usr/local/lib" > /etc/ld.so.conf.d/usrlocal.conf \
+  && /sbin/ldconfig -v \
+  && cd .. \
+  && rm -rf vips-8.11.3 \
+  && rm vips-8.11.3.tar.gz
+
+COPY index.js image.jpeg package.json ${LAMBDA_TASK_ROOT}
+
+RUN npm install --production && npm run init:x64
+
+CMD [ "index.handler" ]

docker/image.jpeg

@@ -0,0 +1,69 @@
+import axios from 'axios';
+import { Handler } from 'aws-lambda';
+import * as b from 'benny';
+import * as express from 'express';
+import * as os from 'os';
+import * as sharp from 'sharp';
+
+type Request = {
+  time?:number;
+};
+
+export const handler:Handler<Request> = (event, context, callback) => {
+  const time = event.time || 10;
+
+  const threads = os.cpus().length;
+
+  sharp.cache(false);
+  sharp.concurrency(threads);
+
+  const app = express();
+
+  app.post('/', (req, res) => {
+    res.json(req.body);
+  });
+  
+  const server = app.listen();
+
+  console.log(`Using ${threads} threads`);
+
+  const suite = b.suite('Lambda benchmark', 
+    b.add('express', () => {
+      const runner = async () => {
+        await axios.post(`http://localhost:${server.address()!['port']}/`, {});
+      };
+
+      return runner;
+    }, {
+      maxTime: time,
+    }),
+    b.add('sharp', () => {
+      const runner = async () => {
+        if (parseInt(context.memoryLimitInMB) < 256) {
+          return;
+        }
+
+        await sharp('./image.jpeg')
+          .resize(640, 400)
+          .jpeg({ quality: 80 })
+          .toBuffer();
+      };
+
+      return runner;
+    }, {
+      maxTime: time,
+    }),
+    b.cycle(),
+  );
+  
+  suite
+    .then((s) => {
+      callback(null, s);
+    })
+    .catch((err) => {
+      callback(err);
+    })
+    .finally(() => {
+      server.close();
+    });
+};