utils.ts 1.34 KB
Newer Older
Rayyyyy's avatar
Rayyyyy committed
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
import winston from 'winston';

import config from './config';

export class TimeoutError extends Error {}

const logLevel = config.LOG_LEVEL;

export const logger = winston.createLogger({
  level: logLevel,
  format: winston.format.combine(
    winston.format.colorize(),
    winston.format.printf(info => {
      return `${info.level}: ${info.message}`;
    }),
  ),
  transports: [new winston.transports.Console()],
});

console.log('LOG_LEVEL', logLevel);

export const parseHrtimeToMillisecond = (hrtime: [number, number]): number => {
    return (hrtime[0] + hrtime[1] / 1e9) * 1000;
  };

export const promiseWithTime = <T>(
    promise: Promise<T>
  ): Promise<{
    value: T;
    time: number;
  }> => {
    return new Promise((resolve, reject) => {
      const startTime = process.hrtime();
      promise
        .then(value => {
          resolve({
            value: value,
            time: parseHrtimeToMillisecond(process.hrtime(startTime))
          });
        })
        .catch(err => reject(err));
    });
  };

export const withTimeout = <T>(
    millis: number,
    promise: Promise<T>
  ): Promise<{
    value: T;
    time: number;
  }> => {
    const timeout = new Promise<{ value: T; time: number }>((_, reject) =>
      setTimeout(() => reject(new TimeoutError()), millis)
    );
    return Promise.race([promiseWithTime(promise), timeout]);
Rayyyyy's avatar
Rayyyyy committed
56
  };