Enable graceful agent termination

Task cancellation was being wrongly ignored.
albertofaria · Dec 18, 2021 · a578910 · a578910
1 parent 3eb38e0
commit a578910
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/image/pav/agent/common.py b/image/pav/agent/common.py
@@ -174,7 +174,7 @@ async def handle_volume_provisioning(
                 api_client, handler_node_name, handlers
             )
         except CancelledError:
-            pass
+            break
         except:
             log(format_exc())
             await sleep(AGENT_HANDLER_RETRY_DELAY.total_seconds())
@@ -405,7 +405,7 @@ async def handle_volume_staging(
                 api_client, handler_node_name, handlers
             )
         except CancelledError:
-            pass
+            break
         except:
             log(format_exc())
             await sleep(AGENT_HANDLER_RETRY_DELAY.total_seconds())