108 lines
3.2 KiB
JavaScript
108 lines
3.2 KiB
JavaScript
'use strict';
|
|
|
|
const fs = require('fs');
|
|
const path = require('path');
|
|
const process = require('process');
|
|
const express = require('express');
|
|
const app = express();
|
|
const bodyParser = require('body-parser');
|
|
const morgan = require('morgan');
|
|
|
|
// Command line opts
|
|
const argv = require('minimist')(process.argv.slice(1));
|
|
if (!argv.codepath) {
|
|
argv.codepath = "/userfunc/user";
|
|
console.log("Codepath defaulting to ", argv.codepath);
|
|
}
|
|
if (!argv.port) {
|
|
console.log("Port defaulting to 8888");
|
|
argv.port = 8888;
|
|
}
|
|
|
|
|
|
// Node resolves module paths according to a file's location. We load
|
|
// the file from argv.codepath, but tell users to put dependencies in
|
|
// the server's package.json; this means the function's dependencies
|
|
// are in /usr/src/app/node_modules. We could be smarter and have the
|
|
// function deps in the right place in argv.codepath; but for now we
|
|
// just symlink the function's node_modules to the server's
|
|
// node_modules.
|
|
fs.symlinkSync('/usr/src/app/node_modules', `${path.dirname(argv.codepath)}/node_modules`);
|
|
|
|
// User function. Starts out undefined.
|
|
let userFunction;
|
|
|
|
//
|
|
// Specialize this server to a given user function. The user function
|
|
// is read from argv.codepath; it's expected to be placed there by the
|
|
// fission runtime.
|
|
//
|
|
function specialize(req, res) {
|
|
// Make sure we're a generic container. (No reuse of containers.
|
|
// Once specialized, the container remains specialized.)
|
|
if (userFunction) {
|
|
res.status(400).send("Not a generic container");
|
|
return;
|
|
}
|
|
|
|
// Read and load the code. It's placed there securely by the fission runtime.
|
|
try {
|
|
var startTime = process.hrtime();
|
|
userFunction = require(argv.codepath);
|
|
var elapsed = process.hrtime(startTime);
|
|
console.log(`user code loaded in ${elapsed[0]}sec ${elapsed[1]/1000000}ms`);
|
|
} catch(e) {
|
|
console.error(`user code load error: ${e}`);
|
|
res.status(500).send(JSON.stringify(e));
|
|
return;
|
|
}
|
|
res.status(202).send();
|
|
}
|
|
|
|
|
|
// Request logger
|
|
app.use(morgan('combined'))
|
|
|
|
app.use(bodyParser.urlencoded({ extended: false }));
|
|
app.use(bodyParser.json());
|
|
app.use(bodyParser.raw());
|
|
|
|
app.post('/specialize', specialize);
|
|
|
|
// Generic route -- all http requests go to the user function.
|
|
app.all('/', function (req, res) {
|
|
if (!userFunction) {
|
|
res.status(500).send("Generic container: no requests supported");
|
|
return;
|
|
}
|
|
const context = {
|
|
request: req,
|
|
response: res
|
|
// TODO: context should also have: URL template params, query string
|
|
};
|
|
function callback(status, body, headers) {
|
|
if (!status)
|
|
return;
|
|
if (headers) {
|
|
for (let name of Object.keys(headers)) {
|
|
res.set(name, headers[name]);
|
|
}
|
|
}
|
|
res.status(status).send(body);
|
|
}
|
|
try {
|
|
//
|
|
// Customizing the request context
|
|
//
|
|
// If you want to modify the context to add anything to it,
|
|
// you can do that here by adding properties to the context.
|
|
//
|
|
userFunction(context, callback);
|
|
} catch(e) {
|
|
console.log(`Function error: ${e}`);
|
|
callback(500, "Internal server error")
|
|
}
|
|
});
|
|
|
|
app.listen(argv.port);
|