db-mysql.js
3.58 KB
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
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
const HEALTHY_CHECK_INTERVAL_MS = 10000;
const MODULE_NAME = require('path').basename(__filename);
const mysql = require('mysql');
const config = require('komodo-sdk/config');
const logger = require('tektrans-logger');
const connectionLimit = config.mysql && config.mysql.pool_connection_limit
? config.mysql.pool_connection_limit
: 0;
const ERROR_POOL_NOT_READY = new Error(`${MODULE_NAME}: pool is not ready`);
exports.ERROR_POOL_NOT_READY = ERROR_POOL_NOT_READY;
const pool = config.mysql ? mysql.createPool({
connectionLimit,
host: config.mysql.host || 'localhost',
database: config.mysql.database || 'komodo',
user: config.mysql.user || 'komodo',
password: config.mysql.password,
timezone: config.mysql.timezone,
}) : null;
exports.pool = pool;
exports.query = (query, values, cb) => {
// pool.query.apply(null, arguments);
if (!pool || !pool.query) {
logger.warn(`${MODULE_NAME}: ${ERROR_POOL_NOT_READY.toString()}`);
if (typeof cb === 'function') {
cb(ERROR_POOL_NOT_READY);
}
return;
}
pool.query(query, values, cb);
};
exports.format = (sql, values, cb) => new Promise((resolve, reject) => {
if (!pool) {
reject(ERROR_POOL_NOT_READY);
if (typeof cb === 'function') cb(ERROR_POOL_NOT_READY);
return;
}
pool.getConnection((err, connection) => {
if (err) {
reject(err);
if (typeof cb === 'function') cb(err);
return;
}
const formatted = connection.format(sql, values);
connection.release();
resolve(formatted);
if (typeof cb === 'function') cb(null, formatted);
});
});
exports.beginConnection = (cb) => new Promise((resolve) => {
pool.getConnection((errGetConnection, connection) => {
if (errGetConnection) {
resolve([errGetConnection]);
if (typeof cb === 'function') cb(errGetConnection);
return;
}
connection.beginTransaction((errBeginTransaction) => {
if (errBeginTransaction) {
resolve([errBeginTransaction]);
if (typeof cb === 'function') cb(errBeginTransaction);
return;
}
resolve([null, connection]);
if (typeof cb === 'function') cb(null, connection);
});
});
});
exports.getBy = (tableName, fieldName, value, cb) => new Promise((resolve) => {
const query = 'SELECT * FROM ?? WHERE ?? = ? LIMIT 1';
const values = [tableName, fieldName, value];
pool.query(query, values, (err, results) => {
const result = results && results[0];
resolve([err, result || null]);
if (typeof cb === 'function') cb(err, result || null);
});
});
function healthyCheck() {
const query = 'SELECT 1';
const values = [];
if (!pool) {
logger.warn(`${MODULE_NAME}: Skip healthy check on undefined pool (ERR-EB9E5C08)`);
return;
}
if (!pool.query) {
logger.warn(`${MODULE_NAME}: Skip healthy check on undefined pool.query (ERR-D10F70F3)`);
return;
}
pool.query(query, values, (err) => {
if (err) {
logger.warn(`${MODULE_NAME}: Error on healthy check (ERR-38EC9B78)`, { err });
}
});
}
setInterval(() => {
const randomMs = Math.floor(Math.random() * HEALTHY_CHECK_INTERVAL_MS * 0.3);
setTimeout(() => {
try {
healthyCheck();
} catch (err) {
logger.warn(`${MODULE_NAME}: Exception on periodic healthy check (ERR-2D137502)`, { err });
}
}, randomMs);
}, HEALTHY_CHECK_INTERVAL_MS);