feat: subscribe arbitrary task

gelatodigital · Dec 4, 2023 · c4cea77 · c4cea77
1 parent 01fdf8b
commit c4cea77
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/src/index.ts b/src/index.ts
@@ -409,4 +409,20 @@ export class GelatoRelay {
   offError = (handler: (error: Error) => void): void => {
     this.#websocketHandler.offError(handler);
   };
+
+  /**
+   * @param {string} taskId - Task Id
+   *
+   */
+  subscribeTaskStatusUpdate = async (taskId: string): Promise<void> => {
+    await this.#websocketHandler.subscribe(taskId);
+  };
+
+  /**
+   * @param {string} taskId - Task Id
+   *
+   */
+  unsubscribeTaskStatusUpdate = async (taskId: string): Promise<void> => {
+    await this.#websocketHandler.unsubscribe(taskId);
+  };
 }