🐛 Close udp socket after request is done

GiyoMoon · Jan 3, 2022 · 97d7f06 · 97d7f06
1 parent 46311b1
commit 97d7f06
Show file tree

Hide file tree

Showing 5 changed files with 15 additions and 3 deletions.
diff --git a/package-lock.json b/package-lock.json
diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "steam-server-query",
-  "version": "1.1.0",
+  "version": "1.1.1",
   "description": "Module which implements the Master Server Query Protocol and Game Server Queries.",
   "main": "lib/index.js",
   "types": "lib/index.d.ts",

diff --git a/src/gameServer/gameServer.ts b/src/gameServer/gameServer.ts
@@ -84,6 +84,8 @@ class GameServerQuery {
       }
     }
 
+    this._promiseSocket.closeSocket();
+
     const parsedInfoBuffer = this._parseInfoBuffer(resultBuffer);
     return parsedInfoBuffer as InfoResponse;
   }
@@ -105,6 +107,8 @@ class GameServerQuery {
       throw new Error(err);
     }
 
+    this._promiseSocket.closeSocket();
+
     const parsedPlayerBuffer = this._parsePlayerBuffer(resultBuffer);
     return parsedPlayerBuffer;
   }
@@ -126,6 +130,8 @@ class GameServerQuery {
       throw new Error(err);
     }
 
+    this._promiseSocket.closeSocket();
+
     const parsedRulesBuffer = this._parseRulesBuffer(resultBuffer);
     return parsedRulesBuffer;
   }

diff --git a/src/masterServer/masterServer.ts b/src/masterServer/masterServer.ts
@@ -57,6 +57,8 @@ class MasterServerQuery {
       }
     } while (this._seedId !== ZERO_IP);
 
+    this._promiseSocket.closeSocket();
+
     // remove ZERO_IP from end of host list
     this._hosts.pop();
     return this._hosts;

diff --git a/src/promiseSocket.ts b/src/promiseSocket.ts
@@ -35,6 +35,10 @@ export class PromiseSocket {
     });
   }
 
+  public closeSocket() {
+    this._socket.close();
+  }
+
   private _socketSend(buffer: Buffer, host: string, port: number, timeout: number): Promise<Buffer> {
     return new Promise((resolve, reject) => {
       this._socket.send(buffer, port, host, (err) => {