Method _handle_connection

relay/server.py:189–224

Handle a new connection (Unity or CLI)

(
        self,
        reader: asyncio.StreamReader,
        writer: asyncio.StreamWriter,
    )

            logger.info("Relay Server stopped")

    async def _handle_connection(
        self,
        reader: asyncio.StreamReader,
        writer: asyncio.StreamWriter,
    ) -> None:
        """Handle a new connection (Unity or CLI)"""
        peername = writer.get_extra_info("peername")
        logger.debug(f"New connection from {peername}")

        try:
            # Read first message to determine connection type
            first_msg = await asyncio.wait_for(read_frame(reader), timeout=10.0)
            msg_type = first_msg.get("type")

            if msg_type == MessageType.REGISTER.value:
                await self._handle_unity_connection(reader, writer, first_msg)
            elif msg_type in (
                MessageType.REQUEST.value,
                MessageType.LIST_INSTANCES.value,
                MessageType.SET_DEFAULT.value,
            ):
                await self._handle_cli_message(writer, first_msg)
                # CLI connections are one-shot
            else:
                logger.warning(f"Unknown message type: {msg_type}")

        except TimeoutError:
            logger.warning(f"Connection timeout from {peername}")
        except asyncio.IncompleteReadError:
            logger.debug(f"Connection closed by {peername}")
        except Exception as e:
            logger.error(f"Error handling connection from {peername}: {e}")
        finally:
            if not writer.is_closing():
                writer.close()
                await writer.wait_closed()

    # ===== Unity Connection Handling =====