diff --git a/labgrid/remote/coordinator.py b/labgrid/remote/coordinator.py index 3c6eb78bf..3d283add3 100644 --- a/labgrid/remote/coordinator.py +++ b/labgrid/remote/coordinator.py @@ -195,7 +195,7 @@ def __init__(self) -> None: self.load() self.loop = asyncio.get_running_loop() - self.poll_task = self.loop.create_task(self.poll()) + self.poll_task = self.loop.create_task(self.poll(), name="coordinator-poll") async def _poll_step(self): # save changes @@ -298,6 +298,7 @@ async def request_task(): name = in_msg.startup.name session = self.clients[peer] = ClientSession(self, peer, name, out_msg_queue, version) logging.debug("Received startup from %s with %s", name, version) + asyncio.current_task().set_name(f"client-{peer}-rx/started-{name}") elif kind == "subscribe": if in_msg.subscribe.all_places: session.subscribe_places() @@ -309,7 +310,8 @@ async def request_task(): except Exception: logging.exception("error in client message handler") - running_request_task = self.loop.create_task(request_task()) + asyncio.current_task().set_name(f"client-{peer}-tx") + running_request_task = self.loop.create_task(request_task(), name=f"client-{peer}-rx/init") try: async for out_msg in queue_as_aiter(out_msg_queue): @@ -389,6 +391,7 @@ async def request_task(): session = self.exporters[peer] = ExporterSession(self, peer, name, command_queue, version) logging.debug("Exporters: %s", self.exporters) logging.debug("Received startup from %s with %s", name, version) + asyncio.current_task().set_name(f"exporter-{peer}-rx/started-{name}") elif kind == "resource": logging.debug("Received resource from %s with %s", name, in_msg.resource) action, _ = session.set_resource( @@ -405,7 +408,8 @@ async def request_task(): except Exception: logging.exception("error in exporter message handler") - running_request_task = self.loop.create_task(request_task()) + asyncio.current_task().set_name(f"exporter-{peer}-tx") + running_request_task = self.loop.create_task(request_task(), name=f"exporter-{peer}-rx/init") try: async for cmd in queue_as_aiter(command_queue): @@ -958,6 +962,7 @@ async def GetReservations(self, request: labgrid_coordinator_pb2.GetReservations async def serve(listen, cleanup) -> None: + asyncio.current_task().set_name(f"coordinator-serve") # It seems since https://github.com/grpc/grpc/pull/34647, the # ping_timeout_ms default of 60 seconds overrides keepalive_timeout_ms, # so set it as well.