Only call send_eof on supported types

jumpstarter-dev · Aug 1, 2024 · 1474937 · 1474937
1 parent e150fa6
commit 1474937
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/jumpstarter/common/streams.py b/jumpstarter/common/streams.py
@@ -2,6 +2,7 @@
 
 import grpc
 from anyio import BrokenResourceError, ClosedResourceError, create_memory_object_stream, create_task_group
+from anyio.abc import ByteStream, ObjectStream
 from anyio.streams.stapled import StapledObjectStream
 
 from jumpstarter.v1 import router_pb2, router_pb2_grpc
@@ -24,7 +25,8 @@ async def decapsulate_stream(tx, rx, tg):
                 case router_pb2.FRAME_TYPE_DATA:
                     await tx.send(frame.payload)
                 case router_pb2.FRAME_TYPE_GOAWAY:
-                    await tx.send_eof()
+                    if isinstance(tx, ObjectStream) or isinstance(tx, ByteStream):
+                        await tx.send_eof()
                 case _:
                     pass
     # ignore peer disconnet