Adds middleware for setting callbackWaitsForEmptyEventLoop false

src/doNotWaitForEmptyEventLoop/index.spec.ts

@@ -0,0 +1,21 @@
+import { compose, types } from '../core';
+import { doNotWaitForEmptyEventLoop } from './index';
+import { inject } from '../inject';
+
+let handler;
+
+beforeEach(() => {
+  handler = compose(
+    types<{}, any>(),
+    inject({}),
+    doNotWaitForEmptyEventLoop(),
+  )(async () => {});
+});
+
+it('should set callbackWaitsForEmptyEventLoop to false', async () => {
+  const context = {};
+  await handler({}, context);
+  expect(context).toEqual({
+    callbackWaitsForEmptyEventLoop: false
+  });
+});

src/doNotWaitForEmptyEventLoop/index.ts

@@ -0,0 +1,10 @@
+import { Context } from 'aws-lambda';
+import { Middleware } from '../core';
+
+export const doNotWaitForEmptyEventLoop =
+  <T, R>(): Middleware<T, T, R, R> =>
+  (next) =>
+  (event, context: Context, ...args) => {
+    context.callbackWaitsForEmptyEventLoop = false;
+    return next(event, context, ...args);
+  };