fix: async examples

grutt · Feb 3, 2024 · 4640e62 · 4640e62
1 parent b1e9bf0
commit 4640e62
Show file tree

Hide file tree

Showing 7 changed files with 37 additions and 17 deletions.
diff --git a/typescript-sdk/examples/concurrency/concurrency-event.ts b/typescript-sdk/examples/concurrency/concurrency-event.ts
@@ -0,0 +1,7 @@
+import Hatchet from '../../src/sdk';
+
+const hatchet = Hatchet.init();
+
+hatchet.event.push('concurrency:create', {
+  test: 'test',
+});
diff --git a/...script-sdk/examples/concurrency-worker.ts → ...xamples/concurrency/concurrency-worker.ts b/...script-sdk/examples/concurrency-worker.ts → ...xamples/concurrency/concurrency-worker.ts
@@ -1,5 +1,5 @@
-import Hatchet from '../src/sdk';
-import { Workflow } from '../src/workflow';
+import Hatchet from '../../src/sdk';
+import { Workflow } from '../../src/workflow';
 
 const hatchet = Hatchet.init();
 
@@ -9,10 +9,10 @@ const sleep = (ms: number) =>
   });
 
 const workflow: Workflow = {
-  id: 'example',
+  id: 'concurrency-example',
   description: 'test',
   on: {
-    event: 'user:create',
+    event: 'concurrency:create',
   },
   concurrency: {
     key: (ctx) => ctx.workflowInput().userId,
@@ -38,6 +38,10 @@ const workflow: Workflow = {
   ],
 };
 
-const worker = hatchet.worker('example-worker');
-worker.registerWorkflow(workflow);
-worker.start();
+async function main() {
+  const worker = await hatchet.worker('example-worker');
+  await worker.registerWorkflow(workflow);
+  worker.start();
+}
+
+main();
diff --git a/typescript-sdk/examples/dag-worker.ts b/typescript-sdk/examples/dag-worker.ts
@@ -52,6 +52,10 @@ const workflow: Workflow = {
   ],
 };
 
-const worker = hatchet.worker('example-worker');
-worker.registerWorkflow(workflow);
-worker.start();
+async function main() {
+  const worker = await hatchet.worker('example-worker');
+  await worker.registerWorkflow(workflow);
+  worker.start();
+}
+
+main();
diff --git a/typescript-sdk/examples/simple-worker.ts b/typescript-sdk/examples/simple-worker.ts
@@ -36,6 +36,10 @@ const workflow: Workflow = {
   ],
 };
 
-const worker = hatchet.worker('example-worker');
-worker.registerWorkflow(workflow);
-worker.start();
+async function main() {
+  const worker = await hatchet.worker('example-worker');
+  await worker.registerWorkflow(workflow);
+  worker.start();
+}
+
+main();
diff --git a/typescript-sdk/package.json b/typescript-sdk/package.json
@@ -29,7 +29,8 @@
     "example:event": "npm run exec -- ./examples/example-event.ts",
     "worker:simple": "npm run exec -- ./examples/simple-worker.ts",
     "worker:dag": "npm run exec -- ./examples/dag-worker.ts",
-    "worker:concurrency": "npm run exec -- ./examples/concurrency-worker.ts"
+    "worker:concurrency": "npm run exec -- ./examples/concurrency/concurrency-worker.ts",
+    "event:concurrency": "npm run exec -- ./examples/concurrency/concurrency-event.ts"
   },
   "keywords": [],
   "author": "",

diff --git a/typescript-sdk/src/clients/hatchet-client/hatchet-client.ts b/typescript-sdk/src/clients/hatchet-client/hatchet-client.ts
@@ -120,14 +120,14 @@ export class HatchetClient {
     return worker;
   }
 
-  worker(workflow: string | Workflow): Worker {
+  async worker(workflow: string | Workflow): Promise<Worker> {
     const name = typeof workflow === 'string' ? workflow : workflow.id;
     const worker = new Worker(this, {
       name,
     });
 
     if (typeof workflow !== 'string') {
-      worker.registerWorkflow(workflow);
+      await worker.registerWorkflow(workflow);
       return worker;
     }
 

diff --git a/typescript-sdk/src/clients/worker/worker.ts b/typescript-sdk/src/clients/worker/worker.ts
@@ -45,7 +45,7 @@ export class Worker {
     this.logger = new Logger(`Worker/${this.name}`, this.client.config.log_level);
   }
 
-  async registerWorkflow(workflow: Workflow, options?: { autoVersion?: boolean }) {
+  async registerWorkflow(workflow: Workflow) {
     try {
       const concurrency: WorkflowConcurrencyOpts | undefined = undefined;