Skip to content

Commit 61b9c42

Browse files
committed
add optional parameter timeout in worker
1 parent 7119d38 commit 61b9c42

1 file changed

Lines changed: 5 additions & 3 deletions

File tree

cadence/workerfactory.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
1-
from dataclasses import dataclass, field
21
from typing import List
32

3+
from dataclasses import dataclass, field
4+
45
from cadence.worker import Worker, WorkerOptions
56

67

@@ -17,8 +18,9 @@ class WorkerFactory:
1718
options: WorkerFactoryOptions = None
1819
workers: List[Worker] = field(default_factory=list)
1920

20-
def new_worker(self, task_list: str, worker_options: WorkerOptions = None) -> Worker:
21-
worker = Worker(host=self.host, port=self.port, domain=self.domain, task_list=task_list, options=worker_options)
21+
def new_worker(self, task_list: str, timeout: int = 120, worker_options: WorkerOptions = None) -> Worker:
22+
worker = Worker(host=self.host, port=self.port, domain=self.domain, task_list=task_list, timeout=timeout,
23+
options=worker_options)
2224
self.workers.append(worker)
2325
return worker
2426

0 commit comments

Comments
 (0)