Add terminate server

Author: tombertrand

server/sentry.js

@@ -11,22 +11,6 @@ Sentry.init({
   },
 });
 
-process.on("uncaughtException", err => {
-  console.log("Error", err);
-  Sentry.captureException(err);
-});
-
-process.on("exit", code => {
-  Sentry.captureException(new Error("Exiting with code"), { extra: { code } });
-  process.exit(code);
-});
-
-process.on("unhandledRejection", (reason, promise) => {
-  Sentry.captureException(new Error("Unhandled promise rejection"), {
-    extra: { reason, promise },
-  });
-});
-
 module.exports = {
   Sentry,
 };

server/server.js

@@ -60,6 +60,7 @@ const { getTelemetry } = require("./api/telemetry");
 const { getRepresentative, getAllRepresentatives } = require("./api/representative");
 const { Sentry } = require("./sentry");
 const { isValidAccountAddress } = require("./utils");
+const { terminate } = require("./terminate");
 
 const app = express();
 app.use(
@@ -316,4 +317,10 @@ app.get("*", (req, res, next) => {
 const server = app.listen(process.env.SERVER_PORT);
 server.timeout = 20000;
 
+const exitHandler = terminate(server);
+process.on("uncaughtException", exitHandler(1, "Unexpected Error"));
+process.on("unhandledRejection", exitHandler(1, "Unhandled Promise"));
+process.on("SIGTERM", exitHandler(0, "SIGTERM"));
+process.on("SIGINT", exitHandler(0, "SIGINT"));
+
 console.log(`Server started on http://localhost:${process.env.SERVER_PORT}`);

server/terminate.js

@@ -0,0 +1,21 @@
+const { Sentry } = require("./sentry");
+
+function terminate(server, options = { coredump: false, timeout: 500 }) {
+  const exit = code => {
+    options.coredump ? process.abort() : process.exit(code);
+  };
+
+  return (code, reason) => (err, promise) => {
+    if (err && err instanceof Error) {
+      Sentry.captureException(err);
+    }
+
+    // Attempt a graceful shutdown
+    server.close(exit);
+    setTimeout(exit, options.timeout).unref();
+  };
+}
+
+module.exports = {
+  terminate,
+};