@jakub.knejzlik/ts-query-server
Version:
TypeScript library for executing SQL queries built with ts-query across diverse server environments.
97 lines (96 loc) • 3.85 kB
JavaScript
;
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
if (k2 === undefined) k2 = k;
var desc = Object.getOwnPropertyDescriptor(m, k);
if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
desc = { enumerable: true, get: function() { return m[k]; } };
}
Object.defineProperty(o, k2, desc);
}) : (function(o, m, k, k2) {
if (k2 === undefined) k2 = k;
o[k2] = m[k];
}));
var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
Object.defineProperty(o, "default", { enumerable: true, value: v });
}) : function(o, v) {
o["default"] = v;
});
var __importStar = (this && this.__importStar) || function (mod) {
if (mod && mod.__esModule) return mod;
var result = {};
if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
__setModuleDefault(result, mod);
return result;
};
var __rest = (this && this.__rest) || function (s, e) {
var t = {};
for (var p in s) if (Object.prototype.hasOwnProperty.call(s, p) && e.indexOf(p) < 0)
t[p] = s[p];
if (s != null && typeof Object.getOwnPropertySymbols === "function")
for (var i = 0, p = Object.getOwnPropertySymbols(s); i < p.length; i++) {
if (e.indexOf(p[i]) < 0 && Object.prototype.propertyIsEnumerable.call(s, p[i]))
t[p[i]] = s[p[i]];
}
return t;
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.executeQueries = exports.QueryRouterClientMySQL = void 0;
const ts_query_1 = require("@jakub.knejzlik/ts-query");
const mysql = __importStar(require("mysql2/promise"));
const url_1 = require("url");
const client_1 = require("./client");
class QueryRouterClientMySQL extends client_1.QueryRouterClient {
constructor(_a) {
var { flavor } = _a, opts = __rest(_a, ["flavor"]);
super(Object.assign({ flavor: flavor !== null && flavor !== void 0 ? flavor : ts_query_1.Q.flavors.mysql }, opts));
}
async executeRawQueries(queries) {
const { databaseUrl, databaseName, poolOptions } = this.opts;
const res = await (0, exports.executeQueries)({
databaseUrl,
databaseName,
sqls: queries,
poolOptions,
});
return res.results;
}
}
exports.QueryRouterClientMySQL = QueryRouterClientMySQL;
// Caches connection pools by MySQL hostname
const poolCache = {};
function createPoolKey(databaseUrl) {
const { hostname } = new url_1.URL(databaseUrl);
return hostname || "";
}
function getPool(poolKey, databaseUrl, poolOptions) {
if (!poolCache[poolKey]) {
poolCache[poolKey] = mysql.createPool(Object.assign(Object.assign({ uri: databaseUrl }, poolOptions), { multipleStatements: true }));
}
return poolCache[poolKey];
}
const executeQueries = async (opts) => {
const { poolOptions, databaseName, sqls } = opts;
if (sqls.length === 0)
return { results: [] };
const databaseUrl = opts.databaseUrl
.replace("{{MYSQL_USER}}", process.env.MYSQL_USER || "")
.replace("{{MYSQL_PASSWORD}}", process.env.MYSQL_PASSWORD || "");
const poolKey = createPoolKey(databaseUrl);
const pool = getPool(poolKey, databaseUrl, poolOptions);
let connection = null;
try {
connection = await pool.getConnection();
const sql = `USE \`${databaseName}\`;\n` + sqls.join(";\n");
const [results] = await connection.query({ sql });
return { results: results.slice(1) };
}
catch (error) {
console.error(error);
throw error;
}
finally {
if (connection)
connection.release();
}
};
exports.executeQueries = executeQueries;