Merge pull request #2053 from wellwelwel/release-connection

fix: releaseConnection types and promise

Author: wellwelwel

index.d.ts

@@ -1,5 +1,6 @@
 import {
   Connection as PromiseConnection,
+  Pool as PromisePool,
   PoolConnection as PromisePoolConnection,
 } from './promise';
 
@@ -83,7 +84,7 @@ export interface Connection extends mysql.Connection {
 }
 
 export interface PoolConnection extends mysql.PoolConnection {
-  promise(promiseImpl?: PromiseConstructor): PromisePoolConnection;
+  promise(promiseImpl?: PromiseConstructor): PromisePool;
 }
 
 export interface Pool extends mysql.Connection {
@@ -152,13 +153,14 @@ export interface Pool extends mysql.Connection {
   getConnection(
     callback: (err: NodeJS.ErrnoException, connection: PoolConnection) => any
   ): void;
+  releaseConnection(connection: PoolConnection | PromisePoolConnection): void;
   on(event: 'connection', listener: (connection: PoolConnection) => any): this;
   on(event: 'acquire', listener: (connection: PoolConnection) => any): this;
   on(event: 'release', listener: (connection: PoolConnection) => any): this;
   on(event: 'enqueue', listener: () => any): this;
   unprepare(sql: string): mysql.PrepareStatementInfo;
   prepare(sql: string, callback?: (err: mysql.QueryError | null, statement: mysql.PrepareStatementInfo) => any): mysql.Prepare;
-  promise(promiseImpl?: PromiseConstructor): PromisePoolConnection;
+  promise(promiseImpl?: PromiseConstructor): PromisePool;
   config: mysql.PoolOptions;
 }
 

promise.d.ts

@@ -83,12 +83,11 @@ export interface Connection extends EventEmitter {
 }
 
 export interface PoolConnection extends Connection {
-  connection: Connection;
-  getConnection(): Promise<PoolConnection>;
   release(): void;
+  connection: Connection;
 }
 
-export interface Pool extends EventEmitter {
+export interface Pool extends EventEmitter, Connection {
   query<T extends RowDataPacket[][] | RowDataPacket[] | OkPacket | OkPacket[] | ResultSetHeader>(
     sql: string
   ): Promise<[T, FieldPacket[]]>;
@@ -128,6 +127,7 @@ export interface Pool extends EventEmitter {
   ): Promise<[T, FieldPacket[]]>;
 
   getConnection(): Promise<PoolConnection>;
+  releaseConnection(connection: PoolConnection): void;
   on(event: 'connection', listener: (connection: PoolConnection) => any): this;
   on(event: 'acquire', listener: (connection: PoolConnection) => any): this;
   on(event: 'release', listener: (connection: PoolConnection) => any): this;
@@ -138,7 +138,7 @@ export interface Pool extends EventEmitter {
   escapeId(value: string): string;
   escapeId(values: string[]): string;
   format(sql: string, values?: any | any[] | { [param: string]: any }): string;
-
+  
   pool: CorePool;
 }
 
@@ -153,4 +153,3 @@ export interface PreparedStatementInfo {
   close(): Promise<void>;
   execute(parameters: any[]): Promise<[RowDataPacket[][] | RowDataPacket[] | OkPacket | OkPacket[] | ResultSetHeader, FieldPacket[]]>;
 }
-

promise.js

@@ -346,6 +346,10 @@ class PromisePool extends EventEmitter {
     });
   }
 
+  releaseConnection(connection) {
+    if (connection instanceof PromisePoolConnection) connection.release();
+  }
+
   query(sql, args) {
     const corePool = this.pool;
     const localErr = new Error();

test/integration/promise-wrappers/test-promise-wrappers.js

@@ -207,6 +207,16 @@ function testEventsConnect() {
 
 function testBasicPool() {
   const pool = createPool(config);
+  const promiseConn = pool.getConnection();
+
+  promiseConn
+    .then(connResolved => {
+      pool.releaseConnection(connResolved);
+    })
+    .catch(err => {
+      throw err;
+    });
+
   pool
     .query('select 1+2 as ttt')
     .then(result1 => {

test/tsc-build/mysql/createPool/callbacks/connection.ts

@@ -0,0 +1,15 @@
+import { mysql } from '../../../index';
+import { access } from '../../baseConnection';
+
+const pool = mysql.createPool(access);
+
+pool.getConnection((err, conn) => {
+	conn.connection;
+   
+   try {
+      // @ts-expect-error: The pool can't be a connection itself
+      pool.connection;
+   } catch (err) {
+      console.log('This error is expected', err);
+   }
+});

test/tsc-build/mysql/createPool/callbacks/getConnection.ts

@@ -0,0 +1,13 @@
+import { mysql } from '../../../index';
+import { access } from '../../baseConnection';
+
+const pool = mysql.createPool(access);
+
+pool.getConnection((err, conn) => {
+  try {
+    // @ts-expect-error: The connection can't get another connection
+    conn.getConnection();
+  } catch (err) {
+    console.log('This error is expected', err);
+  }
+});

test/tsc-build/mysql/createPool/callbacks/release.ts

@@ -0,0 +1,15 @@
+import { mysql } from '../../../index';
+import { access } from '../../baseConnection';
+
+const pool = mysql.createPool(access);
+
+pool.getConnection((err, conn) => {
+	conn.release();
+	
+	try {
+		// @ts-expect-error: The pool isn't a connection itself, so it doesn't have the connection methods
+		pool.release();
+   } catch (err) {
+      console.log('This error is expected', err);
+   }
+});

test/tsc-build/mysql/createPool/callbacks/releaseConnection.ts

@@ -0,0 +1,8 @@
+import { mysql } from '../../../index';
+import { access } from '../../baseConnection';
+
+const pool = mysql.createPool(access);
+
+pool.getConnection((err, conn) => {
+  pool.releaseConnection(conn);
+});

test/tsc-build/mysql/createPool/promise/connection.ts

@@ -0,0 +1,16 @@
+import { mysql } from '../../../index';
+import { access } from '../../baseConnection';
+
+(async () => {
+  const pool = mysql.createPool(access);
+  const conn = await pool.promise().getConnection();
+
+  conn.connection;
+  
+  try {
+    // @ts-expect-error: The pool can't be a connection itself
+    pool.connection;
+  } catch (err) {
+    console.log('This error is expected', err);
+  }
+})();

test/tsc-build/mysql/createPool/promise/getConnection.ts

@@ -0,0 +1,14 @@
+import { mysql } from '../../../index';
+import { access } from '../../baseConnection';
+
+(async () => {
+  const pool = mysql.createPool(access);
+  const conn = await pool.promise().getConnection();
+  
+  try {
+    // @ts-expect-error: The connection can't get another connection
+    conn.getConnection();
+  } catch (err) {
+    console.log('This error is expected', err);
+  }
+})();