如果你正在运行一个服务器，假设服务器需要从磁盘读取一些文件，比如证书或密钥。证书经常会发生变化，因此你的服务器必须重新加载它们。如何告诉服务器重新加载这些文件？
传统的方法是使用Unix信号，你的服务器侦听特定的信号，如SIGUSR1(用户定义的信号#1)或SIGHUP(挂起信号)，并且可以在接收到信号时执行你编写的任何代码。因此，你的服务器等待适当的信号，接收它，然后重新加载证书。
这种方法工作得很好，但是在实际应用中出现了一些可用性的问题。使用单独的一个http服务器来处理信号会更好。
下面我们先来看一下使用Unix信号的例子，然后我们使用服务器处理信号来改进这个例子。
首先，我们先创建一个Rust项目：
cargo new signals-servers在Cargo.toml文件中加入以下依赖项：
[dependencies]axum = "0.7.2"tokio = { version = "1.25.0", features = ["macros", "rt-multi-thread", "signal"] }在项目根目录下创建一个cert.pem文件，内容随便写，只是为了演示。
我们看一个完整的服务器侦听信号的示例，当你启动你的服务器时，也启动一个异步任务(或进程，或线程)来监听这个信号，当接收到信号时，重新加载证书。
创建一个src/bin/unix_signal.rs文件，代码如下：
use axum::{routing::get, Router};use std::process;use tokio::signal::unix::{signal, SignalKind};#[tokio::main]async fn main() { let _cert = std::fs::read_to_string("cert.pem"); println!("已加载证书，正在启动web服务器"); println!("PID: {}", process::id()); tokio::select! { _ = start_normal_server(8080) => { println!("Web服务器关闭") } _ = listen_for_reload(SignalKind::hangup()) => { println!("信号监听器停止") } }}async fn start_normal_server(port: u32) { // 构建我们的应用程序 let app = Router::new().route("/hello", get(|| async { "Hello, world!" })); let addr = format!("127.0.0.1:{port}"); let listener = tokio::net::TcpListener::bind(addr).await.unwrap(); axum::serve(listener, app).await.unwrap();}async fn listen_for_reload(signal_kind: SignalKind) -> Result<(), std::io::Error> { // 监听信号 let mut stream = signal(signal_kind)?; loop { stream.recv().await; match std::fs::read_to_string("cert.pem") { Ok(_) => eprintln!("重新加载证书成功"), Err(e) => eprintln!("无法重新加载证书: {e}"), } }}运行如下命令启动服务器：
cargo run --bin unix_signal已加载证书，正在启动web服务器PID: 41945然后打开一个新的终端，输入以下命令：
kill -s sighup 41945这是服务器的日志如下：
已加载证书，正在启动web服务器PID: 41945重新加载证书成功这是可行的，但对于发送信号的人来说，这不是一个很好的用户体验。假设你是SRE或系统管理员，当需要重新加载服务器证书时，首先查找进程的PID，并使用kill -s sighup pid发送信号。
服务器可能重新加载了，但也许它没有，可能出现了错误，例如新证书无效，或者服务器没有读取新证书的权限。系统管理员如何知道是否发生了这种情况？他们应该检查一下服务器的日志，但这需要切换窗口，或者打开一个不同的程序。
这不是一个很好的用户体验。通常，当你运行命令时，希望得到一些反馈。但是当你发送Unix信号时，终端不会给你任何响应。你必须查找服务器的日志并检查它们，以确保重新加载成功完成。阅读一个不熟悉的程序日志是很困难的，特别是当日志中有很多其他错误时。
Unix信号的主要问题是，它们让用户向进程发出信号，但程序不向用户发送响应。
因此，我们希望进程接受请求(“重新加载您的证书”)，并响应(“是的，它成功了”或“它失败了，原因如下”)。这听起来很熟悉——它只是一个普通的请求-响应协议。没有必要重新发明轮子——我们可以在这个过程中启动第二个小HTTP服务器。
创建一个src/bin/http_signal.rs文件，代码如下：
use axum::{ http::StatusCode, response::IntoResponse, routing::{get, post}, Router,};#[tokio::main]async fn main() { let _cert = std::fs::read_to_string("cert.pem"); println!("已加载证书，正在启动web服务器"); tokio::select! { _ = start_normal_server(8080) => { println!("Web服务器关闭") } _ = start_control_server(3000) => { println!("信号服务器关闭") } }}async fn start_normal_server(port: u32) { // 构建我们的应用程序 let app = Router::new().route("/hello", get(|| async { "Hello, world!" })); let addr = format!("127.0.0.1:{port}"); let listener = tokio::net::TcpListener::bind(addr).await.unwrap(); axum::serve(listener, app).await.unwrap();}async fn start_control_server(port: u32) { // 构建信号控制服务器 let app: Router = Router::new().route( "/reload_certs", post(|| async { println!("重新加载证书"); match std::fs::read_to_string("cert.pem") { Ok(_) => "重新加载证书成功".into_response(), Err(e) => { let error = format!("无法重新加载证书: {e}"); eprintln!("{error}"); let resp = (StatusCode::INTERNAL_SERVER_ERROR, error); resp.into_response() } } }), ); let addr = format!("127.0.0.1:{port}"); let listener = tokio::net::TcpListener::bind(addr).await.unwrap(); axum::serve(listener, app).await.unwrap();}对于SRE或系统管理员来说，这是一个更好的用户体验。使用如下命令重新加载证书：
$ curl -X POST 0.0.0.0:3000/reload_certs重新加载证书成功%如果没有找到证书，会立即得到有关错误的反馈：
$ curl -X POST 0.0.0.0:3000/reload_certs无法重新加载证书: No such file or directory (os error 2)如果你的程序不需要HTTP或网络，那么引入一个完整的HTTP框架来监听信号可能有点多余。因此，根据程序的大小，以及系统管理员的需求或SRE团队的大小，来决定是否添加HTTP服务器，因为这对于管理流程的人员和软件来说，它有更好的用户体验。
本文链接：http://www.28at.com/showinfo-26-80844-0.htmlRust中的信号处理：Unix信号 vs 信号服务器
声明：本网页内容旨在传播知识，若有侵权等问题请及时与本网联系，我们将在第一时间删除处理。邮件：2376512515@qq.com
下一篇： 我们一起聊聊如何三分钟学会异步任务基础
Copyright © 2016-2023 天津谷骐科技有限公司 版权所有 sitemap.xml
违法及侵权请联系：2376512515@qq.com 津ICP备18001702号
津公网安备 12010102000574号