check in aclose

Author: longcw

livekit-rtc/livekit/rtc/data_stream.py

@@ -217,8 +217,6 @@ async def _send_chunk(self, chunk: proto_DataStream.Chunk):
             raise ConnectionError(cb.send_stream_chunk.error)
 
     async def _send_trailer(self, trailer: proto_DataStream.Trailer):
-        if self._closed:
-            raise RuntimeError(f"Cannot send trailer after stream is closed: {trailer}")
         req = proto_ffi.FfiRequest(
             send_stream_trailer=proto_room.SendStreamTrailerRequest(
                 trailer=trailer,
@@ -240,11 +238,12 @@ async def _send_trailer(self, trailer: proto_DataStream.Trailer):
         if cb.send_stream_chunk.error:
             raise ConnectionError(cb.send_stream_trailer.error)
 
-        self._closed = True
-
     async def aclose(
         self, *, reason: str = "", attributes: Optional[Dict[str, str]] = None
     ):
+        if self._closed:
+            raise RuntimeError("Stream already closed")
+        self._closed = True
         await self._send_trailer(
             trailer=proto_DataStream.Trailer(
                 stream_id=self._header.stream_id, reason=reason, attributes=attributes