Function enhanced_disconnect_monitor_fn

api_utils/queue_worker.py:273–299

()

                                comp_ev = cast(Event, completion_event)

                                async def enhanced_disconnect_monitor_fn():
                                    nonlocal client_disconnected_early
                                    disco_count = 0
                                    while not comp_ev.is_set():
                                        if GlobalState.IS_SHUTTING_DOWN.is_set():
                                            comp_ev.set()
                                            break
                                        if (
                                            GlobalState.IS_QUOTA_EXCEEDED
                                            and not GlobalState.IS_RECOVERING
                                        ):
                                            # Abort if quota exceeded and not recovering
                                            client_disconnected_early = True
                                            comp_ev.set()
                                            break

                                        if not await _test_client_connection(
                                            req_id, http_request
                                        ):
                                            disco_count += 1
                                            if disco_count >= 3:
                                                client_disconnected_early = True
                                                comp_ev.set()
                                                break
                                        else:
                                            disco_count = 0
                                        await asyncio.sleep(0.2)

                                disconnect_monitor_task = asyncio.create_task(
                                    enhanced_disconnect_monitor_fn()