implement close function in audio/video source

Author: typester

livekit-rtc/livekit/rtc/audio_source.py

@@ -118,7 +118,7 @@ async def capture_frame(self, frame: AudioFrame) -> None:
             Exception: If there is an error during frame capture.
         """
 
-        if frame.samples_per_channel == 0:
+        if frame.samples_per_channel == 0 or self._ffi_handle.disposed:
             return
 
         now = time.monotonic()
@@ -175,3 +175,11 @@ def _release_waiter(self) -> None:
         self._last_capture = 0.0
         self._q_size = 0.0
         self._join_fut = None
+
+    def close(self) -> None:
+        """Close the audio source
+
+        This method cleans up resources associated with the audio source.
+        """
+        self._ffi_handle.dispose()
+

livekit-rtc/livekit/rtc/video_source.py

@@ -42,3 +42,6 @@ def capture_frame(
         req.capture_video_frame.rotation = rotation
         req.capture_video_frame.timestamp_us = timestamp_us
         FfiClient.instance.request(req)
+
+    def close() -> None:
+        self._ffi_handle.dispose()