@quillsql/node 0.6.1 → 0.6.2

package/dist/db/DatabaseHelper.d.ts

@@ -19,6 +19,7 @@ export interface QuillQueryResults {
 export declare function getDatabaseCredentials(databaseType: "postgresql" | "snowflake" | "bigquery" | "mysql", connectionString: string): PostgresConnectionConfig | SnowflakeConnectionConfig | BigQueryConfig | MysqlConnectionConfig;
 export declare function connectToDatabase(databaseType: "postgresql" | "snowflake" | "bigquery" | "mysql", config: PostgresConnectionConfig | SnowflakeConnectionConfig | BigQueryConfig | MysqlConnectionConfig): DatabaseConnection;
 export declare function withConnection<T>(databaseType: "postgresql" | "snowflake" | "bigquery" | "mysql", connectionString: string, callback: (connection: DatabaseConnection) => T): Promise<T>;
+export declare function withPool<T>(databaseType: "postgresql" | "snowflake" | "bigquery" | "mysql", pool: DatabaseConnection, callback: (connection: DatabaseConnection) => T): Promise<T>;
 export declare function runQueryByDatabase(databaseType: "postgresql" | "snowflake" | "bigquery" | "mysql", connection: DatabaseConnection, sql: string): Promise<QuillQueryResults> | undefined;
 export declare function connectAndRunQuery(databaseType: "postgresql" | "snowflake" | "bigquery" | "mysql", connectionString: string, sql: string): Promise<QuillQueryResults | undefined>;
 export declare function disconnectFromDatabase(databaseType: "postgresql" | "snowflake" | "bigquery" | "mysql", database: DatabaseConnection): void | Promise<void>;

package/dist/db/DatabaseHelper.js

@@ -9,7 +9,7 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
     });
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.getColumnInfoBySchemaByDatabase = exports.getForiegnKeysByDatabase = exports.getColumnsByTableByDatabase = exports.getTablesBySchemaByDatabase = exports.getSchemasByDatabase = exports.disconnectFromDatabase = exports.connectAndRunQuery = exports.runQueryByDatabase = exports.withConnection = exports.connectToDatabase = exports.getDatabaseCredentials = void 0;
+exports.getColumnInfoBySchemaByDatabase = exports.getForiegnKeysByDatabase = exports.getColumnsByTableByDatabase = exports.getTablesBySchemaByDatabase = exports.getSchemasByDatabase = exports.disconnectFromDatabase = exports.connectAndRunQuery = exports.runQueryByDatabase = exports.withPool = exports.withConnection = exports.connectToDatabase = exports.getDatabaseCredentials = void 0;
 const Postgres_1 = require("./Postgres");
 const Snowflake_1 = require("./Snowflake");
 const BigQuery_1 = require("./BigQuery");
@@ -61,6 +61,17 @@ function withConnection(databaseType, connectionString, callback) {
     });
 }
 exports.withConnection = withConnection;
+function withPool(databaseType, pool, callback) {
+    return __awaiter(this, void 0, void 0, function* () {
+        try {
+            return yield callback(pool);
+        }
+        finally {
+            yield disconnectFromDatabase(databaseType, pool);
+        }
+    });
+}
+exports.withPool = withPool;
 function runQueryByDatabase(databaseType, connection, sql) {
     switch (databaseType.toLowerCase()) {
         case "postgres":

package/dist/index.js

@@ -245,6 +245,8 @@ const requireQuill = ({ privateKey, databaseConnectionString, databaseConfig, ca
 module.exports = requireQuill;
 module.exports.default = requireQuill;
 module.exports.withConnection = DatabaseHelper_1.withConnection;
+module.exports.withPool = DatabaseHelper_1.withPool;
+module.exports.connectToDatabase = DatabaseHelper_1.connectToDatabase;
 module.exports.connectAndRunQuery = DatabaseHelper_1.connectAndRunQuery;
 module.exports.Quill = Quill;
 module.exports.getTablesBySchemaByDatabase = DatabaseHelper_1.getTablesBySchemaByDatabase;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@quillsql/node",
-  "version": "0.6.1",
+  "version": "0.6.2",
   "description": "Quill Server SDK for Node.js",
   "main": "dist/index.js",
   "module": "./dist/index.js",

package/src/db/DatabaseHelper.ts

@@ -124,6 +124,18 @@ export async function withConnection<T>(
   }
 }
 
+export async function withPool<T>(
+  databaseType: "postgresql" | "snowflake" | "bigquery" | "mysql",
+  pool: DatabaseConnection,
+  callback: (connection: DatabaseConnection) => T,
+): Promise<T> {
+  try {
+    return await callback(pool);
+  } finally {
+    await disconnectFromDatabase(databaseType, pool);
+  }
+}
+
 export function runQueryByDatabase(
   databaseType: "postgresql" | "snowflake" | "bigquery" | "mysql",
   connection: DatabaseConnection,

package/src/index.ts

@@ -9,6 +9,7 @@ import axios from "axios";
 import "dotenv/config";
 import { mapQueries, removeFields } from "./utils/RunQueryProcesses";
 import {
+  connectToDatabase,
   connectAndRunQuery,
   getColumnInfoBySchemaByDatabase,
   getColumnsByTableByDatabase,
@@ -18,6 +19,7 @@ import {
   getTablesBySchemaByDatabase,
   runQueryByDatabase,
   withConnection,
+  withPool,
 } from "./db/DatabaseHelper";
 import { convertTypeToPostgres } from "./utils/schemaConversion";
 
@@ -360,6 +362,8 @@ const requireQuill = ({
 module.exports = requireQuill;
 module.exports.default = requireQuill;
 module.exports.withConnection = withConnection;
+module.exports.withPool = withPool;
+module.exports.connectToDatabase = connectToDatabase;
 module.exports.connectAndRunQuery = connectAndRunQuery;
 module.exports.Quill = Quill;
 module.exports.getTablesBySchemaByDatabase = getTablesBySchemaByDatabase;