Delete operations-as-classes sample

Author: dandavison

hello_nexus/basic/handler/service_handler_with_operation_handler_classes.py

@@ -7,9 +7,6 @@
 
 from hello_nexus.basic.handler.db_client import MyDBClient
 from hello_nexus.basic.handler.service_handler import MyNexusServiceHandler
-from hello_nexus.basic.handler.service_handler_with_operation_handler_classes import (
-    MyNexusServiceHandlerUsingOperationHandlerClasses,
-)
 from hello_nexus.basic.handler.workflows import WorkflowStartedByNexusOperation
 
 interrupt_event = asyncio.Event()
@@ -18,12 +15,7 @@
 TASK_QUEUE = "my-handler-task-queue"
 
 
-async def main(
-    client: Optional[Client] = None,
-    # Change this to use the service handler defined in
-    # hello_nexus/basic/handler/service_handler_with_operation_handler_classes.py
-    use_operation_handler_classes: bool = False,
-):
+async def main(client: Optional[Client] = None):
     logging.basicConfig(level=logging.INFO)
 
     client = client or await Client.connect(
@@ -36,14 +28,6 @@ async def main(
     # requests. In this example we provide a database client object to the service hander.
     connected_db_client = MyDBClient.connect()
 
-    my_nexus_service_handler = (
-        MyNexusServiceHandlerUsingOperationHandlerClasses(
-            connected_db_client=connected_db_client
-        )
-        if use_operation_handler_classes
-        else MyNexusServiceHandler(connected_db_client=connected_db_client)
-    )
-
     # Start the worker, passing the Nexus service handler instance, in addition to the
     # workflow classes that are started by your nexus operations, and any activities
     # needed. This Worker will poll for both workflow tasks and Nexus tasks (this example
@@ -52,7 +36,9 @@ async def main(
         client,
         task_queue=TASK_QUEUE,
         workflows=[WorkflowStartedByNexusOperation],
-        nexus_service_handlers=[my_nexus_service_handler],
+        nexus_service_handlers=[
+            MyNexusServiceHandler(connected_db_client=connected_db_client)
+        ],
     ):
         logging.info("Worker started, ctrl+c to exit")
         await interrupt_event.wait()

hello_nexus/basic/handler/worker.py

@@ -16,27 +16,25 @@ async def test_nexus_service_basic(client: Client):
         client=client,
     )
     try:
-        for use_operation_handler_classes in [True, False]:
-            handler_worker_task = asyncio.create_task(
-                hello_nexus.basic.handler.worker.main(
-                    client,
-                    use_operation_handler_classes=use_operation_handler_classes,
-                )
-            )
-            await asyncio.sleep(1)
-            results = await hello_nexus.basic.caller.app.execute_caller_workflow(
+        handler_worker_task = asyncio.create_task(
+            hello_nexus.basic.handler.worker.main(
                 client,
             )
-            hello_nexus.basic.handler.worker.interrupt_event.set()
-            await handler_worker_task
-            hello_nexus.basic.handler.worker.interrupt_event.clear()
-            print("\n\n")
-            print([r.message for r in results])
-            print("\n\n")
-            assert [r.message for r in results] == [
-                "Hello world from sync operation!",
-                "Hello world from workflow run operation!",
-            ]
+        )
+        await asyncio.sleep(1)
+        results = await hello_nexus.basic.caller.app.execute_caller_workflow(
+            client,
+        )
+        hello_nexus.basic.handler.worker.interrupt_event.set()
+        await handler_worker_task
+        hello_nexus.basic.handler.worker.interrupt_event.clear()
+        print("\n\n")
+        print([r.message for r in results])
+        print("\n\n")
+        assert [r.message for r in results] == [
+            "Hello world from sync operation!",
+            "Hello world from workflow run operation!",
+        ]
     finally:
         await delete_nexus_endpoint(
             id=create_response.endpoint.id,