From 5d3ecb0de7d017ba13cefa15379ac8f1bc454877 Mon Sep 17 00:00:00 2001
From: Vladislav Polyakov <me@polrk.com>
Date: Wed, 14 Aug 2024 09:04:10 +0300
Subject: [PATCH] feat: add support for using keyword

---
 src/discovery/discovery-service.ts | 4 ++++
 src/driver.ts                      | 7 +++----
 src/schema/scheme-client.ts        | 4 ++++
 src/table/table-client.ts          | 4 ++++
 tsconfig-cjs.json                  | 1 +
 tsconfig-esm.json                  | 1 +
 tsconfig.json                      | 1 +
 7 files changed, 18 insertions(+), 4 deletions(-)

diff --git a/src/discovery/discovery-service.ts b/src/discovery/discovery-service.ts
index 89e348f0..c73ea8a6 100644
--- a/src/discovery/discovery-service.ts
+++ b/src/discovery/discovery-service.ts
@@ -148,4 +148,8 @@ export default class DiscoveryService extends AuthenticatedService<DiscoveryServ
         }
         return endpoint;
     }
+
+    [Symbol.dispose]() {
+        return this.destroy();
+    }
 }
diff --git a/src/driver.ts b/src/driver.ts
index 264502fa..933bb278 100644
--- a/src/driver.ts
+++ b/src/driver.ts
@@ -126,8 +126,7 @@ export default class Driver {
         this.logger.debug('Driver has been destroyed.');
     }
 
-    // TODO: Upgrade project to TS 5.2+
-    // async [Symbol.asyncDispose]() {
-    //     return this.destroy();
-    // }
+    async [Symbol.asyncDispose]() {
+        return this.destroy();
+    }
 }
diff --git a/src/schema/scheme-client.ts b/src/schema/scheme-client.ts
index 47ea25f1..e5cb43c7 100644
--- a/src/schema/scheme-client.ts
+++ b/src/schema/scheme-client.ts
@@ -72,4 +72,8 @@ export default class SchemeClient extends EventEmitter {
     public async destroy() {
         return;
     }
+
+    [Symbol.dispose]() {
+        return this.destroy();
+    }
 }
diff --git a/src/table/table-client.ts b/src/table/table-client.ts
index 1d7769b9..455be01f 100644
--- a/src/table/table-client.ts
+++ b/src/table/table-client.ts
@@ -51,4 +51,8 @@ export class TableClient extends EventEmitter {
     public async destroy(ctx: Context): Promise<void> {
         await this.pool.destroy(ctx);
     }
+
+    async [Symbol.asyncDispose]() {
+        return this.destroy();
+    }
 }
diff --git a/tsconfig-cjs.json b/tsconfig-cjs.json
index 57618278..7b5c494c 100644
--- a/tsconfig-cjs.json
+++ b/tsconfig-cjs.json
@@ -1,6 +1,7 @@
 {
   "extends": "./tsconfig-base.json",
   "compilerOptions": {
+    "lib": ["ES2022","ESNext.Disposable"],
     "target": "es2017",
     "module": "commonjs",
     "outDir": "build/cjs"
diff --git a/tsconfig-esm.json b/tsconfig-esm.json
index 2cedbf4e..778fff02 100644
--- a/tsconfig-esm.json
+++ b/tsconfig-esm.json
@@ -1,6 +1,7 @@
 {
   "extends": "./tsconfig-base.json",
   "compilerOptions": {
+    "lib": ["ES2022","ESNext.Disposable"],
     "module": "commonjs",
     "outDir": "build/esm",
     "target": "esnext"
diff --git a/tsconfig.json b/tsconfig.json
index 57618278..7b5c494c 100644
--- a/tsconfig.json
+++ b/tsconfig.json
@@ -1,6 +1,7 @@
 {
   "extends": "./tsconfig-base.json",
   "compilerOptions": {
+    "lib": ["ES2022","ESNext.Disposable"],
     "target": "es2017",
     "module": "commonjs",
     "outDir": "build/cjs"