实现远程日志

build.rs

@@ -0,0 +1,65 @@
+use std::error::Error;
+use std::fs;
+use std::path::{Path, PathBuf};
+use std::str::FromStr;
+
+fn get_filenames(dir: &str) -> Result<Vec<String>, std::io::Error> {
+    println!("cargo:rerun-if-changed=proto");
+    let entries = fs::read_dir(Path::new(dir))?;
+    let filenames: Vec<String> = entries
+        .filter_map(|entry| {
+            let path = entry.ok()?.path();
+            if path.is_file() {
+                path.file_name()?.to_str().map(|s| s.to_owned())
+            } else {
+                None
+            }
+        })
+        .collect();
+    Ok(filenames)
+}
+
+fn main() -> Result<(), Box<dyn Error>> {
+    let out_dir = "src/protobuf";
+    if PathBuf::from_str(out_dir)?.exists() {
+        fs::remove_dir_all(out_dir).expect("清空输出目录出错");
+    }
+    fs::create_dir_all(out_dir).expect("创建输出目录出错");
+
+    let proto_list = get_filenames("proto")
+        .expect("读取Protobuf定义文件列表失败")
+        .iter()
+        .filter_map(|file| {
+            if file.ends_with(".proto") {
+                Some(file.clone())
+            } else {
+                None
+            }
+        })
+        .collect::<Vec<_>>();
+
+    prost_build::Config::new()
+        .out_dir(out_dir)
+        .compile_protos(&proto_list, &["proto/".to_string()])?;
+
+    //tonic_prost_build::configure()
+    //    .out_dir(out_dir)
+    //    .compile_protos(&proto_list, &["proto/".to_string()])?;
+
+    let mod_file = format!("{}/mod.rs", out_dir);
+    let mod_content = get_filenames(out_dir).expect("无法列出目录内容");
+    let mod_content = mod_content
+        .iter()
+        .map(|file| format!("pub mod {};", file.trim_end_matches(".rs")))
+        .collect::<Vec<_>>()
+        .join("\n");
+    fs::write(&mod_file, mod_content).expect("写出文件失败");
+
+    fs::write(
+        format!("{}/README.md", out_dir),
+        "# Warning！\n\n[protobuf](../protobuf) 中的文件由构建脚本生成，重新生成时将会**删除**此目录内的**全部文件**\
+        ，请不要在此目录内工作，否则可能丢失工作进度！\n\n## 你被警告了！",
+    )?;
+
+    Ok(())
+}