Merge pull request #41 from gelatodigital/feat/subscribe-arbitrary-task

Feat: Subscribe Arbitrary Task
gelatodigital · Dec 5, 2023 · 74c1e75 · 74c1e75
2 parents 01fdf8b + c4cea77
commit 74c1e75
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);
+  };
 }