asclepius-standard/index.js at develop · ordermentum/asclepius-standard · GitHub

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
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
const asclepius = require('asclepius');
const nullLogger = require('null-logger');

const defaultTimeout = 500;

const sequelizeHealthcheck = (sequelize, logger = nullLogger, timeout = defaultTimeout) => asclepius.healthcheck(
  'psql',
  () =>
    sequelize
      .query('select current_timestamp')
      .then(() => {})
      .catch(err => {
        logger.error('PSQL Healthcheck Failed. Reason:', err);
        return Promise.reject(err.name);
      }),
  timeout
);

const redisHealthcheck = (redis, logger = nullLogger, timeout = defaultTimeout) => asclepius.healthcheck(
  'redis',
  () =>
    new Promise((resolve, reject) => {
      redis.ping(err => {
        if (err) {
          logger.error('Redis Healthcheck Failed. Reason:', err);
          return reject(err);
        }
        return resolve();
      });
    }),
  timeout
);

const elasticsearchHealthcheck = (elasticsearch, logger = nullLogger, timeout = defaultTimeout) => asclepius.healthcheck(
  'elasticsearch',
  () =>
    elasticsearch
      .ping()
      .then(() => {})
      .catch(err => {
        logger.error('Elasticsearch Healthcheck Failed. Reason:', err);
        return Promise.reject('Ping Failed');
      }),
  timeout
);

const processHealthcheck = asclepius.healthcheck(
  'process',
  () => Promise.resolve(),
  defaultTimeout
);

const _buildHealthCheckArray = ({ sequelize = null, redis = null, elasticsearch = null, logger = nullLogger, timeout = defaultTimeout } = {}) => {
    const healthchecks = [processHealthcheck];

    if (sequelize) healthchecks.push(sequelizeHealthcheck(sequelize, logger, timeout));
    if (elasticsearch) healthchecks.push(elasticsearchHealthcheck(elasticsearch, logger, timeout));
    if (redis) healthchecks.push(redisHealthcheck(redis, logger, timeout));

    return healthchecks;
}

module.exports = {
  setup: ({ sequelize = null, redis = null, elasticsearch = null, logger = nullLogger, timeout = defaultTimeout } = {}) => {
    const healthchecks = _buildHealthCheckArray({ sequelize, redis, elasticsearch, logger, timeout });
    return asclepius.makeRoute(healthchecks);
  },
  runner: ({ sequelize = null, redis = null, elasticsearch = null, logger = nullLogger, timeout = defaultTimeout } = {}) => {
    const healthchecks = _buildHealthCheckArray({ sequelize, redis, elasticsearch, logger, timeout });
    return asclepius.makeRunner(healthchecks);
  }
};