文盘Rust -- 把程序作为守护进程启动 _生活百科

当我们写完一个服务端程序，需要上线部署的时候，或多或少都会和操作系统的守护进程打交道，毕竟谁也不希望shell关闭既停服。今天我们就来聊聊这个事儿。
最早大家部署应用的通常操作是 “nohup xxxx &”，别说像weblogic 或者其他java 容器有启动脚本，里面其实也差不多；很喜欢 nginx的 -d 参数，或者像redis 配置文件里可以指定是否以守护进程启动。看起来很优雅。
那么，使用rust 写一个服务端程序能不能优雅的使用一个参数指定应用 daemon 模式启动，同时使用stop 方式优雅的停机呢？我们通过一个例子来说说基本的实现方式。
实例代码依然集成在[interactcli-rs](https://github.com/jiashiwen/interactcli-rs)工程中。
首先来模拟一个启动的服务进程 /src/server/server.rs
pub fn start(prefix: String) {for i in 0..1000 {println!("{}", prefix.clone() + &i.to_string());thread::sleep(Duration::from_secs(1));}}程序每秒输出一个字符串，持续999秒，这个时间足够验证实验结果了。
后台启动有两个实现，分别是利用[fork](github.com/immortal/fork) 或 [daemonize](github.com/knsd/daemonize),这两个crate 实现原理类似，但在使用上稍有不同。
/src/cmd/cmdserver.rs,构建了两个启动子命令，分别来调用 fork 和 daemonize的守护进程启动实现.
pub fn new_server_cmd() -> Command {clap::Command::new("server").about("server").subcommand(server_start_byfork()).subcommand(server_start_bydaemonize())}pub fn server_start_byfork() -> Command {clap::Command::new("byfork").about("start daemon by fork crate").arg(Arg::new("daemon").short('d').long("daemon").action(ArgAction::SetTrue).help("start as daemon").required(false),)}pub fn server_start_bydaemonize() -> Command {clap::Command::new("bydaemonize").about("start daemon by daemonize crate").arg(Arg::new("daemon").short('d').long("daemon").action(ArgAction::SetTrue).help("start as daemon").required(false),)}【文盘Rust -- 把程序作为守护进程启动】server 的子命令 byfork 启动通过 fork 实现的功能，bydaemonize 则调用通过 daemonize 的功能实现。
命令解析的代码在 /src/cmd/rootcmd.rs 文件中。
先来看看基于 fork 的实现:
if let Some(startbyfork) = server.subcommand_matches("byfork") {println!("start by fork");if startbyfork.get_flag("daemon") {let args: Vec<String> = env::args().collect();if let Ok(Fork::Child) = daemon(true, false) {// 启动子进程let mut cmd = Command::new(&args[0])for idx in 1..args.len() {let arg = args.get(idx).expect("get cmd arg error!");// 去除后台启动参数,避免重复启动if arg.eq("-d") || arg.eq("-daemon") {continue;}cmd.arg(arg);let child = cmd.spawn().expect("Child process failed to start.");fs::write("pid", child.id().to_string()).unwrap();println!("process id is:{}", std::process::id());println!("child id is:{}", child.id());}println!("{}", "daemon mod");process::exit(0);}start("by_fork:".to_string());}首先，通过 Fork::daemon 函数派生出一个子进程；然后解析一下当前命令，去掉 -d 参数，构建一个启动命令，子命令启动，退出父进程。这基本符合操作系统创建守护进程的过程 -- 两次 fork 。
再来看看基于 daemonize 的实现:
if let Some(startbydaemonize) = server.subcommand_matches("bydaemonize") {println!("start by daemonize");let base_dir = env::current_dir().unwrap();if startbydaemonize.get_flag("daemon") {let stdout = File::create("/tmp/daemon.out").unwrap();let stderr = File::create("/tmp/daemon.err").unwrap();println!("{:?}", base_dir);let daemonize = Daemonize::new().pid_file("/tmp/test.pid") // Every method except `new` and `start`.chown_pid_file(true) // is optional, see `Daemonize` documentation.working_directory(base_dir.as_path()) // for default behaviour..umask(0o777) // Set umask, `0o027` by default..stdout(stdout) // Redirect stdout to `/tmp/daemon.out`..stderr(stderr) // Redirect stderr to `/tmp/daemon.err`..privileged_action(|| "Executed before drop privileges");match daemonize.start() {Ok(_) => {println!("Success, daemonized");}Err(e) => eprintln!("Error, {}", e),}}println!("pid is:{}", std::process::id());fs::write("pid", process::id().to_string()).unwrap();start("by_daemonize:".to_string());}

文盘Rust -- 把程序作为守护进程启动

经验总结扩展阅读

墨鱼自蔽和什么寓言故事类似

山楂糕不凝固能回锅重熬吗

多人运动事件是什么梗

美白的真谛是什么

妙奇星球新手初阶怎么玩

男子结婚前夕|男子结婚前夕，妻子消失无踪，6年后逛街偶遇，他终于明白了

哪种面膜是你的菜

会计学专业就业方向与就业前景怎么样

五月一号怎么说早安五月一号早安句子精选

荔枝放冰箱能保存多久荔枝放冰箱能保存多长时间

秋天来了，女性不想肚子鼓鼓，推荐以下食物，排毒减肥，抗衰老！

1、两个人在一起|总有一个人会改变自己

发型不要再披头散发了，脸上有这5个特征的女生，披头发真的很丑

玉和翡翠假的怎么分辨

水培蔬菜有哪些,适合水培的蔬菜和水果

疯人愿原唱是谁疯了疯了伴奏

因为42岁温峥嵘近照瘦脱相，皮包骨似大头娃娃，面对蛋糕一口不吃

头条怎么才能涨粉

智商|“以油养肤”是在交智商税？最全网红护肤油测评，拿走不谢！

家用饭碗可以放烤箱吗