connection-manager.js 5.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147
  1. "use strict";
  2. var __defProp = Object.defineProperty;
  3. var __getOwnPropSymbols = Object.getOwnPropertySymbols;
  4. var __hasOwnProp = Object.prototype.hasOwnProperty;
  5. var __propIsEnum = Object.prototype.propertyIsEnumerable;
  6. var __defNormalProp = (obj, key, value) => key in obj ? __defProp(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
  7. var __spreadValues = (a, b) => {
  8. for (var prop in b || (b = {}))
  9. if (__hasOwnProp.call(b, prop))
  10. __defNormalProp(a, prop, b[prop]);
  11. if (__getOwnPropSymbols)
  12. for (var prop of __getOwnPropSymbols(b)) {
  13. if (__propIsEnum.call(b, prop))
  14. __defNormalProp(a, prop, b[prop]);
  15. }
  16. return a;
  17. };
  18. var __markAsModule = (target) => __defProp(target, "__esModule", { value: true });
  19. var __export = (target, all) => {
  20. __markAsModule(target);
  21. for (var name in all)
  22. __defProp(target, name, { get: all[name], enumerable: true });
  23. };
  24. __export(exports, {
  25. OracleConnectionManager: () => OracleConnectionManager
  26. });
  27. const AbstractConnectionManager = require("../abstract/connection-manager");
  28. const SequelizeErrors = require("../../errors");
  29. const parserStore = require("../parserStore")("oracle");
  30. const { logger } = require("../../utils/logger");
  31. const semver = require("semver");
  32. const debug = logger.debugContext("connection:oracle");
  33. const DataTypes = require("../../data-types").oracle;
  34. const { promisify } = require("util");
  35. class OracleConnectionManager extends AbstractConnectionManager {
  36. constructor(dialect, sequelize) {
  37. super(dialect, sequelize);
  38. this.sequelize = sequelize;
  39. this.sequelize.config.port = this.sequelize.config.port || 1521;
  40. this.lib = this._loadDialectModule("oracledb");
  41. this.extendLib();
  42. this.refreshTypeParser(DataTypes);
  43. }
  44. extendLib() {
  45. if (this.sequelize.config && "dialectOptions" in this.sequelize.config) {
  46. const dialectOptions = this.sequelize.config.dialectOptions;
  47. if (dialectOptions && "maxRows" in dialectOptions) {
  48. this.lib.maxRows = this.sequelize.config.dialectOptions.maxRows;
  49. }
  50. if (dialectOptions && "fetchAsString" in dialectOptions) {
  51. this.lib.fetchAsString = this.sequelize.config.dialectOptions.fetchAsString;
  52. } else {
  53. this.lib.fetchAsString = [this.lib.CLOB];
  54. }
  55. }
  56. this.lib.fetchAsBuffer = [this.lib.BLOB];
  57. }
  58. buildConnectString(config) {
  59. if (!config.host || config.host.length === 0)
  60. return config.database;
  61. let connectString = config.host;
  62. if (config.port && config.port > 0) {
  63. connectString += `:${config.port}`;
  64. } else {
  65. connectString += ":1521";
  66. }
  67. if (config.database && config.database.length > 0) {
  68. connectString += `/${config.database}`;
  69. }
  70. return connectString;
  71. }
  72. _refreshTypeParser(dataType) {
  73. parserStore.refresh(dataType);
  74. }
  75. _clearTypeParser() {
  76. parserStore.clear();
  77. }
  78. async connect(config) {
  79. const connectionConfig = __spreadValues({
  80. user: config.username,
  81. password: config.password,
  82. externalAuth: config.externalAuth,
  83. stmtCacheSize: 0,
  84. connectString: this.buildConnectString(config)
  85. }, config.dialectOptions);
  86. try {
  87. const connection = await this.lib.getConnection(connectionConfig);
  88. this.sequelize.options.databaseVersion = semver.coerce(connection.oracleServerVersionString).version;
  89. debug("connection acquired");
  90. connection.on("error", (error) => {
  91. switch (error.code) {
  92. case "ESOCKET":
  93. case "ECONNRESET":
  94. case "EPIPE":
  95. case "PROTOCOL_CONNECTION_LOST":
  96. this.pool.destroy(connection);
  97. }
  98. });
  99. return connection;
  100. } catch (err) {
  101. let errorCode = err.message.split(":");
  102. errorCode = errorCode[0];
  103. switch (errorCode) {
  104. case "ORA-12560":
  105. case "ORA-12154":
  106. case "ORA-12505":
  107. case "ORA-12514":
  108. case "NJS-511":
  109. case "NJS-516":
  110. case "NJS-517":
  111. case "NJS-520":
  112. throw new SequelizeErrors.ConnectionRefusedError(err);
  113. case "ORA-28000":
  114. case "ORA-28040":
  115. case "ORA-01017":
  116. case "NJS-506":
  117. throw new SequelizeErrors.AccessDeniedError(err);
  118. case "ORA-12541":
  119. case "NJS-503":
  120. case "NJS-508":
  121. case "NJS-507":
  122. throw new SequelizeErrors.HostNotReachableError(err);
  123. case "NJS-512":
  124. case "NJS-515":
  125. case "NJS-518":
  126. case "NJS-519":
  127. throw new SequelizeErrors.InvalidConnectionError(err);
  128. case "ORA-12170":
  129. case "NJS-510":
  130. throw new SequelizeErrors.ConnectionTimedOutError(err);
  131. default:
  132. throw new SequelizeErrors.ConnectionError(err);
  133. }
  134. }
  135. }
  136. async disconnect(connection) {
  137. if (!connection.isHealthy()) {
  138. debug("connection tried to disconnect but was already at CLOSED state");
  139. return;
  140. }
  141. return await promisify((callback) => connection.close(callback))();
  142. }
  143. validate(connection) {
  144. return connection && connection.isHealthy();
  145. }
  146. }
  147. //# sourceMappingURL=connection-manager.js.map