Fix missing shielding

5 years ago · 9aeefc3691
parent 9289d34df0
commit 9aeefc3691
1 changed files with 1 additions and 1 deletions
--- a/src/python/grpcio/grpc/_cython/_cygrpc/aio/server.pyx.pxi
+++ b/src/python/grpcio/grpc/_cython/_cygrpc/aio/server.pyx.pxi
@ -410,7 +410,7 @@ cdef class AioServer:
            await self._shutdown_completed
        else:
            try:
-                await asyncio.wait_for(self._shutdown_completed, timeout)
+                await asyncio.wait_for(asyncio.shield(self._shutdown_completed), timeout)
            except asyncio.TimeoutError:
                if self._crash_exception is not None:
                    raise self._crash_exception