API/api.medcify.app/node_modules/sequelize/lib/dialects/mssql/connection-manager.js

139 lines
5.0 KiB
JavaScript
Raw Normal View History

2022-09-26 06:11:44 +00:00
"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