Wait for stream close before returning from callback.

lsst-dm · Sep 12, 2024 · d9da01a · d9da01a
1 parent b5e72fa
commit d9da01a
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/python/s3daemon/s3daemon.py b/python/s3daemon/s3daemon.py
@@ -73,6 +73,9 @@ async def handle_client(client, reader, writer):
             writer.write(b"Success")
         except Exception as e:
             writer.write(bytes(repr(e), "UTF-8"))
+    await writer.drain()
+    writer.close()
+    await writer.wait_closed()
     log.info("%f %f sec", start, time.time() - start)