From 8d6d939f75ad0407e58d7dd2697d1062797c39d3 Mon Sep 17 00:00:00 2001
From: karishmas6 <carishmashukla28@gmail.com>
Date: Sat, 8 Jun 2024 00:18:06 +0530
Subject: [PATCH] feat: update socket

---
 server/src/workflow-management/classes/Generator.ts | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/server/src/workflow-management/classes/Generator.ts b/server/src/workflow-management/classes/Generator.ts
index e2faf92c..9368b2f8 100644
--- a/server/src/workflow-management/classes/Generator.ts
+++ b/server/src/workflow-management/classes/Generator.ts
@@ -357,5 +357,15 @@ export class WorkflowGenerator {
     }
   };
 
+  /**
+   * Updates the socket used for communication with the client.
+   * @param socket The socket to be used for communication.
+   * @returns void
+   */
+  public updateSocket = (socket: Socket) : void => {
+    this.socket = socket;
+    this.registerEventHandlers(socket);
+  };
+
   
 }