worker.py

import asyncio
from datetime import timedelta

from temporalio import activity, workflow
from temporalio.client import Client
from temporalio.worker import Worker


@activity.defn
async def say_hello(name: str) -> str:
    print(f"Hello, {name}!")
    return f"Hello, {name}!"


@activity.defn
async def say_goodbye(name: str) -> str:
    print(f"Goodbye, {name}!")
    return f"Goodbye, {name}!"


@workflow.defn
class HelloWorldWorkflow:
    @workflow.run
    async def run(self, name: str) -> str:
        return await workflow.execute_activity(
            say_hello,
            name,
            start_to_close_timeout=timedelta(seconds=10),
        )


@workflow.defn
class HelloWorldWorkflow2:
    @workflow.run
    async def run(self, name: str) -> str:
        greeting = await workflow.execute_activity(
            say_hello,
            name,
            start_to_close_timeout=timedelta(seconds=10),
        )
        farewell = await workflow.execute_activity(
            say_goodbye,
            name,
            start_to_close_timeout=timedelta(seconds=10),
        )
        return f"{greeting} | {farewell}"


@workflow.defn
class HelloWorldWorkflow3:
    @workflow.run
    async def run(self, name: str) -> str:
        result = await workflow.execute_child_workflow(
            HelloWorldWorkflow2.run,
            name,
            id=f"hello-world-2-child-{workflow.info().workflow_id}",
        )
        return f"Workflow3 called Workflow2 -> {result}"


@workflow.defn
class HelloWorldWorkflow4:
    def __init__(self):
        self._resumed = False

    @workflow.signal
    async def resume(self) -> None:
        self._resumed = True

    @workflow.run
    async def run(self, name: str) -> str:
        greeting = await workflow.execute_activity(
            say_hello,
            name,
            start_to_close_timeout=timedelta(seconds=10),
        )
        print(f"Workflow4 paused, waiting for resume signal...")
        await workflow.wait_condition(lambda: self._resumed)
        farewell = await workflow.execute_activity(
            say_goodbye,
            name,
            start_to_close_timeout=timedelta(seconds=10),
        )
        return f"{greeting} | {farewell}"


@workflow.defn
class HelloWorldWorkflow5:
    @workflow.run
    async def run(self, workflow4_id: str) -> str:
        handle = workflow.get_external_workflow_handle(workflow4_id)
        await handle.signal(HelloWorldWorkflow4.resume)
        return f"Sent resume signal to {workflow4_id}"


async def main():
    client = await Client.connect("localhost:7233")
    worker = Worker(
        client,
        task_queue="hello-world-task-queue",
        workflows=[HelloWorldWorkflow, HelloWorldWorkflow2, HelloWorldWorkflow3, HelloWorldWorkflow4, HelloWorldWorkflow5],
        activities=[say_hello, say_goodbye],
    )
    print("Worker started, listening on 'hello-world-task-queue'...")
    await worker.run()


if __name__ == "__main__":
    asyncio.run(main())