INNER CODE UNIT · Python

wait_startup_complete

stacklok/codegate · src/codegate/cli.py:69

    async def wait_startup_complete(self) -> None:
        """Wait for the server to complete startup."""
        self.logger.debug("Waiting for server startup to complete")
        await self._startup_complete.wait()

    async def cleanup(self) -> None:
        """Cleanup server resources and ensure graceful shutdown."""
        self.logger.debug("Cleaning up server resources")
        if not self._should_exit:
            self._should_exit = True
            self.logger.debug("Initiating server shutdown")
            self._shutdown_event.set()

            if hasattr(self.server, "shutdown"):
                self.logger.debug("Shutting down server")
                await self.server.shutdown()

            # Ensure all connections are closed

View source record →