Specify process group for child processes

src/commands/config/apply.rs

@@ -5,6 +5,7 @@ use std::{
     collections::HashMap,
     fs,
     io::Write,
+    os::unix::process::CommandExt,
     path::Path,
     process::{Command, Stdio},
 };
@@ -36,6 +37,7 @@ fn apply_service_config(
             )
             .stdout(Stdio::null())
             .stdin(Stdio::piped())
+            .process_group(0)
             .spawn()
             .context("failed to run picodata admin")?;
 

src/commands/lib.rs

@@ -1,5 +1,6 @@
 use anyhow::{bail, Context, Result};
 use std::io::{BufRead, BufReader, Read};
+use std::os::unix::process::CommandExt;
 use std::process::{Command, Stdio};
 
 pub enum BuildType {
@@ -17,6 +18,7 @@ pub fn cargo_build(build_type: BuildType) -> Result<()> {
     let mut child = Command::new("cargo")
         .args(args)
         .stdout(Stdio::piped())
+        .process_group(0)
         .spawn()
         .context("running cargo build")?;
 

src/commands/run.rs

@@ -1,9 +1,10 @@
 use anyhow::{bail, Context, Result};
 use lib::cargo_build;
 use log::{error, info};
 use serde::Deserialize;
 use std::fs::File;
 use std::io::Write;
+use std::os::unix::process::CommandExt;
 use std::path::Path;
 use std::process::{exit, Child, Command, Stdio};
 use std::sync::{Arc, Mutex, OnceLock};
@@ -114,6 +115,7 @@
             .arg("admin")
             .arg(admin_soket.to_str().unwrap())
             .stdin(Stdio::piped())
+            .process_group(0)
             .spawn()
             .context("failed to spawn child proccess of picodata admin")?;
 
@@ -151,17 +153,6 @@
     }
 }
 
-fn kill_picodata_instances() -> Result<()> {
-    let processes_lock = Arc::clone(&get_picodata_processes());
-    let mut processes = processes_lock.lock().unwrap();
-
-    for mut process in processes.drain(..) {
-        process.kill()?;
-    }
-
-    Ok(())
-}
-
 #[allow(clippy::too_many_arguments)]
 pub fn cluster(
     topology_path: &PathBuf,
@@ -224,6 +215,7 @@
                     "--tier",
                     tier_name,
                 ])
+                .process_group(0)
                 .spawn()
                 .context(format!("failed to start picodata instance: {instance_id}"))?;
             thread::sleep(Duration::from_secs(5));
@@ -242,11 +234,6 @@
 
     if !disable_plugin_install {
         enable_plugins(topology, data_dir, picodata_path, &plugins_dir)
-            .inspect_err(|_| {
-                kill_picodata_instances().unwrap_or_else(|e| {
-                    error!("failed to kill picodata instances: {:#}", e);
-                });
-            })
             .context("failed to enable plugins")?;
     };
 
@@ -270,9 +257,6 @@
         ctrlc::set_handler(move || {
             info!("{}", "received Ctrl+C. Shutting down ...");
 
-            kill_picodata_instances()
-                .unwrap_or_else(|e| error!("failed to kill picodata instances: {:#}", e));
-
             exit(0);
         })
         .context("failed to set Ctrl+c handler")?;

tests/helpers/mod.rs

@@ -2,6 +2,7 @@ use constcat::concat;
 use log::info;
 use std::ffi::OsStr;
 use std::io::{BufRead, BufReader, Write};
+use std::os::unix::process::CommandExt;
 use std::thread;
 use std::{
     fs::{self},
@@ -131,6 +132,7 @@ where
         .arg("pike")
         .args(args)
         .current_dir(current_dir)
+        .process_group(0)
         .spawn()
 }
 
@@ -168,6 +170,7 @@ pub fn await_picodata_admin(timeout: Duration) -> Result<Child, std::io::Error>
             .arg(PLUGIN_DIR.to_string() + "tmp/cluster/i_1/admin.sock")
             .stdin(Stdio::piped())
             .stdout(Stdio::piped())
+            .process_group(0)
             .spawn();
 
         match picodata_admin {