Reset stream in camera_stream if camera changes #305

viamrobotics · Nov 25, 2024 · c9bc4fa · c9bc4fa
1 parent 3de272a
commit c9bc4fa
Showing 1 changed file with 16 additions and 3 deletions.
diff --git a/lib/widgets/camera_stream.dart b/lib/widgets/camera_stream.dart
@@ -33,11 +33,18 @@ class _ViamCameraStreamViewState extends State<ViamCameraStreamView> {
     _startStream();
   }
 
+  @override
+  void didUpdateWidget(ViamCameraStreamView oldWidget) {
+    super.didUpdateWidget(oldWidget);
+    if (oldWidget.camera != widget.camera) {
+      _cleanupStream();
+      _startStream();
+    }
+  }
+
   @override
   void dispose() {
-    _renderer.dispose();
-    _streamSub.cancel();
-    widget.streamClient.closeStream();
+    _cleanupStream();
     super.dispose();
   }
 
@@ -70,6 +77,12 @@ class _ViamCameraStreamViewState extends State<ViamCameraStreamView> {
     });
   }
 
+  void _cleanupStream() {
+    _renderer.dispose();
+    _streamSub.cancel();
+    widget.streamClient.closeStream();
+  }
+
   @override
   Widget build(BuildContext context) {
     return (_error != null)