From 140e9ebe1c7ad4078ce86a85b9509d748bdac2b6 Mon Sep 17 00:00:00 2001
From: licsber <admin@licsber.site>
Date: Thu, 19 Sep 2024 17:16:25 +0800
Subject: [PATCH] extract retry from main file.

---
 src/main.rs | 92 ++---------------------------------------------------
 1 file changed, 3 insertions(+), 89 deletions(-)

diff --git a/src/main.rs b/src/main.rs
index 5e1eb92..5c716fd 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,92 +1,6 @@
+mod retry;
+
 use std::env;
-use std::io::{BufRead, BufReader};
-use std::process::{Command, Stdio};
-use std::sync::mpsc;
-use std::thread;
-use std::time::Duration;
-
-fn read_stream(stream: &mut mpsc::Receiver<String>, output: &mut Vec<String>) {
-    loop {
-        match stream.try_recv() {
-            Ok(line) => {
-                output.push(line);
-            }
-            Err(mpsc::TryRecvError::Disconnected) => {
-                break;
-            }
-            Err(mpsc::TryRecvError::Empty) => {
-                thread::sleep(Duration::from_millis(100));
-            }
-        }
-    }
-}
-
-fn retry(cmd: &str, cmd_args: &[String]) {
-    loop {
-        let (stdout_sender, mut stdout_receiver) = mpsc::channel();
-        let (stderr_sender, stderr_receiver) = mpsc::channel();
-
-        let mut child = Command::new(cmd)
-            .args(cmd_args)
-            .stdout(Stdio::piped())
-            .stderr(Stdio::piped())
-            .spawn()
-            .expect("Failed to spawn command");
-
-        let status = child.wait().expect("Failed to wait for command");
-
-        if status.success() {
-            let stdout_vec: Vec<String> = {
-                let stdout_handle = thread::spawn(move || {
-                    let reader = BufReader::new(child.stdout.take().unwrap());
-                    for line in reader.lines() {
-                        let line = line.expect("Failed to read line from stdout");
-                        stdout_sender.send(line).expect("Failed to send stdout line");
-                    }
-                });
-
-                stdout_handle.join().unwrap();
-                stdout_receiver.try_iter().collect()
-            };
-
-            println!("Command succeeded.");
-            for line in stdout_vec {
-                println!("stdout: {}", line);
-            }
-
-            break;
-        } else {
-            let stderr_vec: Vec<String> = {
-                let stderr_handle = thread::spawn(move || {
-                    let reader = BufReader::new(child.stderr.take().unwrap());
-                    let mut lines = Vec::new();
-                    for line_result in reader.lines() {
-                        if let Ok(line) = line_result {
-                            // 克隆line后发送
-                            stderr_sender.send(line.clone()).expect("Failed to send stderr line");
-                            lines.push(line);
-                        } else {
-                            eprintln!("Error reading stderr line: {:?}", line_result);
-                        }
-                    }
-                    lines
-                });
-
-                stderr_handle.join().unwrap();
-                stderr_receiver.try_iter().collect()
-            };
-            eprintln!("Command failed with status: {}", status);
-            for line in stderr_vec {
-                eprintln!("stderr: {}", line);
-            }
-        }
-
-        let mut stdout_output: Vec<String> = Vec::new();
-        read_stream(&mut stdout_receiver, &mut stdout_output);
-
-        thread::sleep(Duration::from_secs(1));
-    }
-}
 
 fn main() {
     let args: Vec<String> = env::args().collect();
@@ -103,5 +17,5 @@ fn main() {
         &[]
     };
 
-    retry(cmd, cmd_args);
+    retry::retry(cmd, cmd_args);
 }