You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
118 lines
4.2 KiB
118 lines
4.2 KiB
"use strict"; |
|
var __defProp = Object.defineProperty; |
|
var __getOwnPropSymbols = Object.getOwnPropertySymbols; |
|
var __hasOwnProp = Object.prototype.hasOwnProperty; |
|
var __propIsEnum = Object.prototype.propertyIsEnumerable; |
|
var __defNormalProp = (obj, key, value) => key in obj ? __defProp(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value; |
|
var __spreadValues = (a, b) => { |
|
for (var prop in b || (b = {})) |
|
if (__hasOwnProp.call(b, prop)) |
|
__defNormalProp(a, prop, b[prop]); |
|
if (__getOwnPropSymbols) |
|
for (var prop of __getOwnPropSymbols(b)) { |
|
if (__propIsEnum.call(b, prop)) |
|
__defNormalProp(a, prop, b[prop]); |
|
} |
|
return a; |
|
}; |
|
const semver = require("semver"); |
|
const AbstractConnectionManager = require("../abstract/connection-manager"); |
|
const SequelizeErrors = require("../../errors"); |
|
const { logger } = require("../../utils/logger"); |
|
const DataTypes = require("../../data-types").mariadb; |
|
const momentTz = require("moment-timezone"); |
|
const debug = logger.debugContext("connection:mariadb"); |
|
const parserStore = require("../parserStore")("mariadb"); |
|
class ConnectionManager extends AbstractConnectionManager { |
|
constructor(dialect, sequelize) { |
|
sequelize.config.port = sequelize.config.port || 3306; |
|
super(dialect, sequelize); |
|
this.lib = this._loadDialectModule("mariadb"); |
|
this.refreshTypeParser(DataTypes); |
|
} |
|
static _typecast(field, next) { |
|
if (parserStore.get(field.type)) { |
|
return parserStore.get(field.type)(field, this.sequelize.options, next); |
|
} |
|
return next(); |
|
} |
|
_refreshTypeParser(dataType) { |
|
parserStore.refresh(dataType); |
|
} |
|
_clearTypeParser() { |
|
parserStore.clear(); |
|
} |
|
async connect(config) { |
|
let tzOffset = this.sequelize.options.timezone; |
|
tzOffset = /\//.test(tzOffset) ? momentTz.tz(tzOffset).format("Z") : tzOffset; |
|
const connectionConfig = __spreadValues({ |
|
host: config.host, |
|
port: config.port, |
|
user: config.username, |
|
password: config.password, |
|
database: config.database, |
|
timezone: tzOffset, |
|
typeCast: ConnectionManager._typecast.bind(this), |
|
bigNumberStrings: false, |
|
supportBigNumbers: true, |
|
foundRows: false |
|
}, config.dialectOptions); |
|
if (!this.sequelize.config.keepDefaultTimezone) { |
|
if (connectionConfig.initSql) { |
|
if (!Array.isArray(connectionConfig.initSql)) { |
|
connectionConfig.initSql = [connectionConfig.initSql]; |
|
} |
|
connectionConfig.initSql.push(`SET time_zone = '${tzOffset}'`); |
|
} else { |
|
connectionConfig.initSql = `SET time_zone = '${tzOffset}'`; |
|
} |
|
} |
|
try { |
|
const connection = await this.lib.createConnection(connectionConfig); |
|
this.sequelize.options.databaseVersion = semver.coerce(connection.serverVersion()).version; |
|
debug("connection acquired"); |
|
connection.on("error", (error) => { |
|
switch (error.code) { |
|
case "ESOCKET": |
|
case "ECONNRESET": |
|
case "EPIPE": |
|
case "PROTOCOL_CONNECTION_LOST": |
|
this.pool.destroy(connection); |
|
} |
|
}); |
|
return connection; |
|
} catch (err) { |
|
switch (err.code) { |
|
case "ECONNREFUSED": |
|
throw new SequelizeErrors.ConnectionRefusedError(err); |
|
case "ER_ACCESS_DENIED_ERROR": |
|
case "ER_ACCESS_DENIED_NO_PASSWORD_ERROR": |
|
throw new SequelizeErrors.AccessDeniedError(err); |
|
case "ENOTFOUND": |
|
throw new SequelizeErrors.HostNotFoundError(err); |
|
case "EHOSTUNREACH": |
|
case "ENETUNREACH": |
|
case "EADDRNOTAVAIL": |
|
throw new SequelizeErrors.HostNotReachableError(err); |
|
case "EINVAL": |
|
throw new SequelizeErrors.InvalidConnectionError(err); |
|
default: |
|
throw new SequelizeErrors.ConnectionError(err); |
|
} |
|
} |
|
} |
|
async disconnect(connection) { |
|
if (!connection.isValid()) { |
|
debug("connection tried to disconnect but was already at CLOSED state"); |
|
return; |
|
} |
|
return await connection.end(); |
|
} |
|
validate(connection) { |
|
return connection && connection.isValid(); |
|
} |
|
} |
|
module.exports = ConnectionManager; |
|
module.exports.ConnectionManager = ConnectionManager; |
|
module.exports.default = ConnectionManager; |
|
//# sourceMappingURL=connection-manager.js.map
|
|
|