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.
138 lines
5.0 KiB
138 lines
5.0 KiB
"use strict"; |
|
const AbstractConnectionManager = require("../abstract/connection-manager"); |
|
const AsyncQueue = require("./async-queue").default; |
|
const { logger } = require("../../utils/logger"); |
|
const sequelizeErrors = require("../../errors"); |
|
const DataTypes = require("../../data-types").mssql; |
|
const parserStore = require("../parserStore")("mssql"); |
|
const debug = logger.debugContext("connection:mssql"); |
|
const debugTedious = logger.debugContext("connection:mssql:tedious"); |
|
class ConnectionManager extends AbstractConnectionManager { |
|
constructor(dialect, sequelize) { |
|
sequelize.config.port = sequelize.config.port || 1433; |
|
super(dialect, sequelize); |
|
this.lib = this._loadDialectModule("tedious"); |
|
this.refreshTypeParser(DataTypes); |
|
} |
|
_refreshTypeParser(dataType) { |
|
parserStore.refresh(dataType); |
|
} |
|
_clearTypeParser() { |
|
parserStore.clear(); |
|
} |
|
async connect(config) { |
|
const connectionConfig = { |
|
server: config.host, |
|
authentication: { |
|
type: "default", |
|
options: { |
|
userName: config.username || void 0, |
|
password: config.password || void 0 |
|
} |
|
}, |
|
options: { |
|
port: parseInt(config.port, 10), |
|
database: config.database, |
|
trustServerCertificate: true |
|
} |
|
}; |
|
if (config.dialectOptions) { |
|
if (config.dialectOptions.options && config.dialectOptions.options.instanceName) { |
|
delete connectionConfig.options.port; |
|
} |
|
if (config.dialectOptions.authentication) { |
|
Object.assign(connectionConfig.authentication, config.dialectOptions.authentication); |
|
} |
|
Object.assign(connectionConfig.options, config.dialectOptions.options); |
|
} |
|
try { |
|
return await new Promise((resolve, reject) => { |
|
const connection = new this.lib.Connection(connectionConfig); |
|
if (connection.state === connection.STATE.INITIALIZED) { |
|
connection.connect(); |
|
} |
|
connection.queue = new AsyncQueue(); |
|
connection.lib = this.lib; |
|
const connectHandler = (error) => { |
|
connection.removeListener("end", endHandler); |
|
connection.removeListener("error", errorHandler); |
|
if (error) |
|
return reject(error); |
|
debug("connection acquired"); |
|
resolve(connection); |
|
}; |
|
const endHandler = () => { |
|
connection.removeListener("connect", connectHandler); |
|
connection.removeListener("error", errorHandler); |
|
reject(new Error("Connection was closed by remote server")); |
|
}; |
|
const errorHandler = (error) => { |
|
connection.removeListener("connect", connectHandler); |
|
connection.removeListener("end", endHandler); |
|
reject(error); |
|
}; |
|
connection.once("error", errorHandler); |
|
connection.once("end", endHandler); |
|
connection.once("connect", connectHandler); |
|
connection.on("error", (error) => { |
|
switch (error.code) { |
|
case "ESOCKET": |
|
case "ECONNRESET": |
|
this.pool.destroy(connection); |
|
} |
|
}); |
|
if (config.dialectOptions && config.dialectOptions.debug) { |
|
connection.on("debug", debugTedious.log.bind(debugTedious)); |
|
} |
|
}); |
|
} catch (error) { |
|
if (!error.code) { |
|
throw new sequelizeErrors.ConnectionError(error); |
|
} |
|
switch (error.code) { |
|
case "ESOCKET": |
|
if (error.message.includes("connect EHOSTUNREACH")) { |
|
throw new sequelizeErrors.HostNotReachableError(error); |
|
} |
|
if (error.message.includes("connect ENETUNREACH")) { |
|
throw new sequelizeErrors.HostNotReachableError(error); |
|
} |
|
if (error.message.includes("connect EADDRNOTAVAIL")) { |
|
throw new sequelizeErrors.HostNotReachableError(error); |
|
} |
|
if (error.message.includes("getaddrinfo ENOTFOUND")) { |
|
throw new sequelizeErrors.HostNotFoundError(error); |
|
} |
|
if (error.message.includes("connect ECONNREFUSED")) { |
|
throw new sequelizeErrors.ConnectionRefusedError(error); |
|
} |
|
throw new sequelizeErrors.ConnectionError(error); |
|
case "ER_ACCESS_DENIED_ERROR": |
|
case "ELOGIN": |
|
throw new sequelizeErrors.AccessDeniedError(error); |
|
case "EINVAL": |
|
throw new sequelizeErrors.InvalidConnectionError(error); |
|
default: |
|
throw new sequelizeErrors.ConnectionError(error); |
|
} |
|
} |
|
} |
|
async disconnect(connection) { |
|
if (connection.closed) { |
|
return; |
|
} |
|
connection.queue.close(); |
|
return new Promise((resolve) => { |
|
connection.on("end", resolve); |
|
connection.close(); |
|
debug("connection closed"); |
|
}); |
|
} |
|
validate(connection) { |
|
return connection && (connection.loggedIn || connection.state.name === "LoggedIn"); |
|
} |
|
} |
|
module.exports = ConnectionManager; |
|
module.exports.ConnectionManager = ConnectionManager; |
|
module.exports.default = ConnectionManager; |
|
//# sourceMappingURL=connection-manager.js.map
|
|
|