Remove loop from infrequent polling

Fixes #90
temporalio · Nov 9, 2023 · f961e13 · f961e13
1 parent 8664134
commit f961e13
Showing 1 changed file with 3 additions and 7 deletions.
diff --git a/polling/infrequent/activities.py b/polling/infrequent/activities.py
@@ -1,4 +1,3 @@
-import asyncio
 from dataclasses import dataclass
 
 from temporalio import activity
@@ -15,9 +14,6 @@ class ComposeGreetingInput:
 @activity.defn
 async def compose_greeting(input: ComposeGreetingInput) -> str:
     test_service = TestService()
-    while True:
-        try:
-            result = test_service.get_service_result(input)
-            return result
-        except Exception:
-            activity.heartbeat("Invoking activity")
+    # If this raises an exception because it's not done yet, the activity will
+    # continually be scheduled for retry
+    return await test_service.get_service_result(input)