mirror of
https://github.com/hatchet-dev/hatchet.git
synced 2026-01-07 01:09:38 -06:00
6ea38a99f2
* feat: round robin queueing * feat: configurable max runs per worker * fix: address PR review * docs for max runs and group round robin
32 lines
776 B
Python
32 lines
776 B
Python
from hatchet_sdk import Hatchet
|
|
from dotenv import load_dotenv
|
|
|
|
load_dotenv()
|
|
|
|
hatchet = Hatchet(debug=True)
|
|
|
|
@hatchet.workflow(on_events=["concurrency-test"])
|
|
class ConcurrencyDemoWorkflow:
|
|
def __init__(self):
|
|
self.my_value = "test"
|
|
|
|
@hatchet.concurrency(max_runs=5)
|
|
def concurrency(self, context) -> str:
|
|
return "concurrency-key"
|
|
|
|
@hatchet.step()
|
|
def step1(self, context):
|
|
print("executed step1")
|
|
pass
|
|
|
|
@hatchet.step(parents=["step1"],timeout='4s')
|
|
def step2(self, context):
|
|
print("started step2")
|
|
context.sleep(1)
|
|
print("finished step2")
|
|
|
|
workflow = ConcurrencyDemoWorkflow()
|
|
worker = hatchet.worker('concurrency-demo-worker', max_runs=4)
|
|
worker.register_workflow(workflow)
|
|
|
|
worker.start() |