src/util/config.ts


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37

import winston from "winston";
import { MongoClient } from "mongodb";
import { Server } from "http";

import loggerConfig from "./logger";

const logger = winston.createLogger(loggerConfig("ROOT"));

export const configOrError = (varName: string): string => {
  if (!process.env[varName]) {
    logger.error("%s must be defined", varName);
    process.exit(1);
    return "";
  } else {
    return process.env[varName] || "";
  }
};

export const handleShutdown = (mongo: MongoClient, server: Server) => (
  signal: NodeJS.Signals
): void => {
  logger.warn("%s signal received. Closing shop.", signal);

  mongo
    .close()
    .then(() => {
      logger.info("MongoDB client connection closed.");
      return new Promise((res, rej) =>
        server.close(err => {
          logger.info("Express down.");
          (err ? rej : res)(err);
        })
      );
    })
    .then(() => process.exit(0))
    .catch(() => process.exit(1));
};