poe lint

Author: nagl-temporal

resource_locking/lock_manager_workflow.py

@@ -1,5 +1,6 @@
 from dataclasses import dataclass
 from datetime import timedelta
+from typing import Optional
 
 from temporalio import workflow
 
@@ -163,7 +164,7 @@ async def handoff_resource(self, request: HandoffRequest):
         await requester.signal("assign_resource", AssignedResource(request.resource))
 
     @workflow.query
-    def get_current_holders(self) -> dict[str, AcquireRequest]:
+    def get_current_holders(self) -> dict[str, Optional[AcquireRequest]]:
         return {k: v[0] if v else None for k, v in self.resource_queues.items()}
 
     @workflow.run

resource_locking/resource_locking_workflow.py

@@ -82,7 +82,7 @@ async def run(self, input: ResourceLockingWorkflowInput):
             await sem_handle.signal(
                 "acquire_resource", AcquireRequest(info.workflow_id, info.run_id)
             )
-        else:
+        elif info.continued_run_id:
             # If we continued as new, we already have a resource. We need to transfer ownership from our predecessor to
             # ourselves.
             await sem_handle.signal(
@@ -94,6 +94,10 @@ async def run(self, input: ResourceLockingWorkflowInput):
                     info.run_id,
                 ),
             )
+        else:
+            raise FailWorkflowException(
+                f"Only set 'already_owned_resource' when using continue_as_new"
+            )
 
         # Both branches above should cause us to receive an "assign_resource" signal.
         await workflow.wait_condition(