gitea_admin/sientia-dataops-scouter_temporal
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

SIENTIAPDE-1441

Browse Source 
Refactor prepare_worker function to return a single Worker instance instead of a list, streamlining worker configuration and improving code clarity.
This commit is contained in:
vitor-aignosi
2025-12-16 15:43:01 -03:00
parent f1cdf7471d
commit 5e2403090b
1 changed files with 21 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

44
scouter/worker/prepare_worker.py
View File

@@ -25,7 +25,7 @@ def prepare_worker(
other_workflows: Sequence[Type], other_workflows: Sequence[Type],
activities: Sequence[Any], activities: Sequence[Any],
temporal_client: Client, temporal_client: Client,
): ) -> Worker:
main_workflow_name = main_workflow.__name__.upper() main_workflow_name = main_workflow.__name__.upper()
@@ -36,25 +36,23 @@ def prepare_worker(
os.getenv(main_workflow_name + '_' + parameter[0], parameter[1])) os.getenv(main_workflow_name + '_' + parameter[0], parameter[1]))
return [ return Worker(
Worker( temporal_client,
temporal_client, task_queue='scouter-queue',
task_queue='scouter-queue', workflows=[main_workflow, *other_workflows],
workflows=[main_workflow, *other_workflows], activities=[*activities],
activities=[*activities], max_concurrent_workflow_tasks=local_workflow_parameters['MAX_CONCURRENT_WORKFLOW_TASKS'],
max_concurrent_workflow_tasks=local_workflow_parameters['MAX_CONCURRENT_WORKFLOW_TASKS'], max_concurrent_activities=local_workflow_parameters['MAX_CONCURRENT_ACTIVITIES'],
max_concurrent_activities=local_workflow_parameters['MAX_CONCURRENT_ACTIVITIES'], max_concurrent_local_activities=local_workflow_parameters['MAX_CONCURRENT_LOCAL_ACTIVITIES'],
max_concurrent_local_activities=local_workflow_parameters['MAX_CONCURRENT_LOCAL_ACTIVITIES'], max_cached_workflows=local_workflow_parameters['MAX_CACHED_WORKFLOWS'],
max_cached_workflows=local_workflow_parameters['MAX_CACHED_WORKFLOWS'], workflow_task_poller_behavior=PollerBehaviorAutoscaling(
workflow_task_poller_behavior=PollerBehaviorAutoscaling( minimum=local_workflow_parameters['WORKFLOW_POLLER_BEHAVIUR_MINIMUM'],
minimum=local_workflow_parameters['WORKFLOW_POLLER_BEHAVIUR_MINIMUM'], initial=local_workflow_parameters['WORKFLOW_POLLER_BEHAVIUR_INITIAL'],
initial=local_workflow_parameters['WORKFLOW_POLLER_BEHAVIUR_INITIAL'], maximum=local_workflow_parameters['WORKFLOW_POLLER_BEHAVIUR_MAXIMUM'],
maximum=local_workflow_parameters['WORKFLOW_POLLER_BEHAVIUR_MAXIMUM'], ),
), activity_task_poller_behavior=PollerBehaviorAutoscaling(
activity_task_poller_behavior=PollerBehaviorAutoscaling( minimum=local_workflow_parameters['ACTIVITY_POLLER_BEHAVIUR_MINIMUM'],
minimum=local_workflow_parameters['ACTIVITY_POLLER_BEHAVIUR_MINIMUM'], initial=local_workflow_parameters['ACTIVITY_POLLER_BEHAVIUR_INITIAL'],
initial=local_workflow_parameters['ACTIVITY_POLLER_BEHAVIUR_INITIAL'], maximum=local_workflow_parameters['ACTIVITY_POLLER_BEHAVIUR_MAXIMUM'],
maximum=local_workflow_parameters['ACTIVITY_POLLER_BEHAVIUR_MAXIMUM'], ),
), )
)
]