Merge pull request #123 from pipeless-ai/fixes

Minor fixes to improve UX

Author: miguelaeh

pipeless/Cargo.lock

@@ -46,6 +46,7 @@ num_cpus = "1.16.0"
 gstreamer-rtsp = "0.21.0"
 inquire = "0.6.2"
 tabled = "0.15.0"
+ctrlc = "3.4.2"
 
 [dependencies.uuid]
 version = "1.4.1"

pipeless/Cargo.toml

@@ -4,10 +4,16 @@ use tokio::sync::RwLock;
 use gstreamer as gst;
 use glib;
 use tokio;
+use ctrlc;
 
 use crate as pipeless;
 
 pub fn start_pipeless_node(stages_dir: &str) {
+    ctrlc::set_handler(|| {
+        println!("Exiting...");
+        std::process::exit(0);
+    }).expect("Error setting Ctrl+C handler");
+
     pipeless::setup_logger();
     pyo3::prepare_freethreaded_python(); // Setup Pyo3
 
@@ -34,4 +40,4 @@ pub fn start_pipeless_node(stages_dir: &str) {
     });
 
     glib_main_loop.run();
-}
+}

pipeless/src/cli/start.rs

@@ -507,8 +507,18 @@ impl Pipeline {
                         gst_buffer_mut.set_duration(duration);
                     }
 
+
                     if let Err(err) = appsrc.push_buffer(gst_buffer) {
-                        return Err(OutputPipelineError::new(&format!("Failed to send the output buffer: {}", err)));
+                        match err {
+                            gst::FlowError::Eos => {
+                                // Do not log error or warn because confuses users
+                                debug!("Unable to send frame, output pipeline is EOS.");
+                                return Ok(());
+                            }
+                            _ => {
+                                return Err(OutputPipelineError::new(&format!("Failed to send the output buffer: {}", err)));
+                            }
+                        }
                     }
                 }
             }