Spoke/lambda.js at main · campaigncode/Spoke · GitHub

History

116 lines (108 loc) · 3.68 KB

Raw

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

"use strict";

const AWS = require("aws-sdk");

const awsServerlessExpress = require("aws-serverless-express");

let app, server, jobs, dispatcher;

let invocationContext = {};

let invocationEvent = {};

if (process.env.DOWNTIME_NO_DB) {

app = require("./build/server/server/downtime");

server = awsServerlessExpress.createServer(app.default);

jobs = {};

} else {

try {

app = require("./build/server/server/index");

server = awsServerlessExpress.createServer(app.default);

jobs = require("./build/server/workers/job-processes");

dispatcher = require("./build/server/extensions/job-runners/lambda-async/handler");

app.default.set("awsContextGetter", function(req, res) {

return [invocationEvent, invocationContext];

});

} catch (err) {

if (!global.TEST_ENVIRONMENT) {

console.error(`Unable to load built server: ${err}`);

}

app = require("./src/server/index");

server = awsServerlessExpress.createServer(app.default);

jobs = require("./src/workers/job-processes");

}

// NOTE: the downside of loading above is environment variables are initially loaded immediately,

// so changing them means that the code must test environment variable inline (rather than use a const set on-load)

// We should NOT load app and server inside the handler, or all connection pools and state are re-instantiated per-request:

// See: http://docs.aws.amazon.com/lambda/latest/dg/best-practices.html#function-code

// "Separate the Lambda handler (entry point) from your core logic"

function cleanHeaders(event) {

// X-Twilio-Body can contain unicode and disallowed chars by aws-serverless-express like "'"

// We don't need it anyway

if (event.headers) {

delete event.headers["X-Twilio-Body"];

}

if (event.multiValueHeaders) {

delete event.multiValueHeaders["X-Twilio-Body"];

}

exports.handler = async (event, context) => {

// Note: When lambda is called with invoke() we MUST return with success

// or Lambda will re-run/re-try the invocation twice:

// https://docs.aws.amazon.com/lambda/latest/dg/retries-on-errors.html

if (process.env.LAMBDA_DEBUG_LOG) {

console.log("LAMBDA EVENT", event);

}

if (

event.type &&

(event.type === "JOB" || event.type === "TASK") &&

dispatcher

) {

return await dispatcher.handler(event, context);

}

if (!event.command) {

// default web server stuff

const startTime = context.getRemainingTimeInMillis

? context.getRemainingTimeInMillis()

: 0;

invocationEvent = event;

invocationContext = context;

cleanHeaders(event);

const webResponse = awsServerlessExpress.proxy(

server,

event,

context,

"PROMISE"

).promise;

if (process.env.DEBUG_SCALING) {

const endTime = context.getRemainingTimeInMillis

? context.getRemainingTimeInMillis()

: 0;

if (endTime - startTime > 3000) {

//3 seconds

console.log(

"SLOW_RESPONSE milliseconds:",

endTime - startTime,

event,

webResponse

);

}

return webResponse;

} else {

// handle a custom command sent as an event

const functionName = context.functionName;

if (event.env) {

for (var a in event.env) {

process.env[a] = event.env[a];

}

console.log("Running " + event.command);

if (event.command in jobs) {

const job = jobs[event.command];

// behavior and arguments documented here:

// https://docs.aws.amazon.com/AWSJavaScriptSDK/latest/AWS/Lambda.html#invoke-property

const result = await job(event, context);

return result;

} else {

console.error("Unfound command sent as a Lambda event: " + event.command);

}

};

Spoke/lambda.js at main · campaigncode/Spoke · GitHub

Uh oh!

Expand file tree

Breadcrumbs

lambda.js

Latest commit

History

Breadcrumbs

lambda.js

File metadata and controls