qemu: Always how many notify messages we received

diff --git a/mkosi/qemu.py b/mkosi/qemu.py
index 516f21bc0ea3e29eea0c58a98bba4446717afe0c..2a26aeb134af8e1a0f754766a77f8778dbec0fb7 100644 (file)
--- a/mkosi/qemu.py
+++ b/mkosi/qemu.py
@@ -409,11 +409,12 @@ def vsock_notify_handler() -> Iterator[tuple[str, dict[str, str]]]:
                     messages[k] = v
 
         with AsyncioThread(notify()):
-            yield f"vsock-stream:{socket.VMADDR_CID_HOST}:{vsock.getsockname()[1]}", messages
-
-        logging.debug(f"Received {num_messages} notify messages totalling {format_bytes(num_bytes)} bytes")
-        for k, v in messages.items():
-            logging.debug(f"- {k}={v}")
+            try:
+                yield f"vsock-stream:{socket.VMADDR_CID_HOST}:{vsock.getsockname()[1]}", messages
+            finally:
+                logging.debug(f"Received {num_messages} notify messages totalling {format_bytes(num_bytes)} bytes")
+                for k, v in messages.items():
+                    logging.debug(f"- {k}={v}")
 
 
 @contextlib.contextmanager