108 lines
3.0 KiB
JavaScript
108 lines
3.0 KiB
JavaScript
'use strict';
|
|
|
|
const Connection = require('./connection');
|
|
const PoolBase = require('./pool-base');
|
|
const Errors = require('./misc/errors');
|
|
const util = require('util');
|
|
|
|
function PoolPromise(options) {
|
|
const processTaskPromise = function (conn, sql, values, isBatch) {
|
|
if (sql) {
|
|
const fct = isBatch ? conn.batch : conn.query;
|
|
return fct(sql, values)
|
|
.then((res) => {
|
|
conn.releaseWithoutError();
|
|
return Promise.resolve(res);
|
|
})
|
|
.catch((err) => {
|
|
conn.releaseWithoutError();
|
|
return Promise.reject(err);
|
|
});
|
|
}
|
|
return Promise.resolve(conn);
|
|
};
|
|
|
|
/**
|
|
* Add connection to pool.
|
|
*/
|
|
const createConnectionPoolPromise = function (pool) {
|
|
const conn = new Connection(options.connOptions);
|
|
return conn
|
|
.connect()
|
|
.then(() => {
|
|
if (pool.closed) {
|
|
conn
|
|
.end()
|
|
.then(() => {})
|
|
.catch(() => {});
|
|
return Promise.reject(
|
|
Errors.createError(
|
|
'Cannot create new connection to pool, pool closed',
|
|
true,
|
|
null,
|
|
'08S01',
|
|
Errors.ER_ADD_CONNECTION_CLOSED_POOL,
|
|
null
|
|
)
|
|
);
|
|
}
|
|
|
|
conn.releaseWithoutError = () => {
|
|
conn.release().catch(() => {});
|
|
};
|
|
|
|
conn.forceEnd = conn.end;
|
|
|
|
conn.release = () => {
|
|
if (pool.closed) {
|
|
pool._discardConnection(conn);
|
|
return Promise.resolve();
|
|
}
|
|
if (options.noControlAfterUse) {
|
|
pool._releaseConnection(conn);
|
|
return Promise.resolve();
|
|
}
|
|
|
|
//if server permit it, reset the connection, or rollback only if not
|
|
// COM_RESET_CONNECTION exist since mysql 5.7.3 and mariadb 10.2.4
|
|
// but not possible to use it with mysql waiting for https://bugs.mysql.com/bug.php?id=97633 correction.
|
|
// and mariadb only since https://jira.mariadb.org/browse/MDEV-18281
|
|
let revertFunction = conn.rollback;
|
|
if (
|
|
options.resetAfterUse &&
|
|
conn.info.isMariaDB() &&
|
|
((conn.info.serverVersion.minor === 2 && conn.info.hasMinVersion(10, 2, 22)) ||
|
|
conn.info.hasMinVersion(10, 3, 13))
|
|
) {
|
|
revertFunction = conn.reset;
|
|
}
|
|
|
|
return revertFunction()
|
|
.then(() => {
|
|
pool._releaseConnection(conn);
|
|
return Promise.resolve();
|
|
})
|
|
.catch((err) => {
|
|
//uncertain connection state.
|
|
// discard it
|
|
pool._discardConnection(conn);
|
|
return Promise.resolve();
|
|
});
|
|
};
|
|
conn.end = conn.release;
|
|
return Promise.resolve(conn);
|
|
})
|
|
.catch((err) => {
|
|
return Promise.reject(err);
|
|
});
|
|
};
|
|
|
|
PoolBase.call(this, options, processTaskPromise, createConnectionPoolPromise, (conn) =>
|
|
conn.ping(options.pingTimeout)
|
|
);
|
|
}
|
|
|
|
util.inherits(PoolPromise, PoolBase);
|
|
|
|
module.exports = PoolPromise;
|