orange-orm/src/tedious/pool/newGenericPool.js

Object Relational Mapper

// @ts-nocheck
/* eslint-disable no-prototype-builtins */

var defaults = require('../../poolDefaults');
var genericPool = require('../../generic-pool');
var tedious;
var parseConnectionString = require('./parseConnectionString');

function newGenericPool(connectionString, poolOptions) {
	if (typeof connectionString === 'string')
		connectionString = parseConnectionString(connectionString);
	if (typeof connectionString === 'object')
		connectionString.options = { ...connectionString.options, ...{ useColumnNames: true } };
	poolOptions = poolOptions || {};
	var pool = genericPool.Pool({
		min: poolOptions.min || 0,
		max: poolOptions.size || poolOptions.poolSize || defaults.poolSize,
		idleTimeoutMillis: poolOptions.idleTimeout || defaults.poolIdleTimeout,
		reapIntervalMillis: poolOptions.reapIntervalMillis || defaults.reapIntervalMillis,
		log: poolOptions.log || defaults.poolLog,
		create: async function(cb) {
			try {
				if (!tedious)
					tedious = await import('tedious');
			} catch (err) {
				return cb(err, null);
			}
			var client = new tedious.Connection(connectionString);
			client.on('connect', onConnected);
			client.connect();

			function onConnected(err) {
				if (err) {
					if (err.errors)
						return cb(err.errors[0], null);
					else
						return cb(err, null);
				}
				client.poolCount = 0;
				return cb(null, client);
			}
		},

		destroy: function(client) {
			client.poolCount = undefined;
			client.close();
		}
	});
	//monkey-patch with connect method
	pool.connect = function(cb) {
		pool.acquire(function(err, client) {
			if (err) return cb(err, null, function() {/*NOOP*/ });
			client.poolCount++;
			cb(null, client, function(err) {
				if (err) {
					pool.destroy(client);
				} else {
					pool.release(client);
				}
			});
		});
	};
	return pool;
}

module.exports = newGenericPool;