【Tokio】監聽停止事件
阿新 • • 發佈:2022-03-31
環境
- Time 2022-01-13
- Rust 1.57.0
- Tokio 1.15.0
概念
參考:https://tokio.rs/tokio/topics/shutdown
示例
server
use bytes::Bytes; use mini_redis::Command::{Get, Set}; use mini_redis::{Command, Connection, Frame, Result}; use std::collections::HashMap; use std::sync::{Arc, Mutex, MutexGuard}; use tokio::net::{TcpListener, TcpStream}; use tokio::signal; type DatabaseLock = Arc<Mutex<HashMap<String, Bytes>>>; type Database<'a> = MutexGuard<'a, HashMap<String, Bytes>>; #[tokio::main] async fn main() -> Result<()> { let listener = TcpListener::bind("127.0.0.1:6379").await?; let database: DatabaseLock = Arc::new(Mutex::new(HashMap::new())); match signal::ctrl_c().await { Ok(()) => { println!("shutdown"); return Ok(()); } Err(err) => { eprintln!("Unable to listen for shutdown signal: {}", err); } } loop { let (socket, address) = listener.accept().await?; println!("客戶端: {}", address); // 提交任務 tokio::spawn(process(socket, database.clone())).await??; } } async fn process(socket: TcpStream, db_lock: DatabaseLock) -> Result<()> { // Connection 是 mini redis 定義的內容 let mut client = Connection::new(socket); // 迴圈接收資料 while let Some(frame) = client.read_frame().await? { let response = match Command::from_frame(frame)? { Set(cmd) => { let mut database: Database = db_lock.lock().unwrap(); database.insert(cmd.key().to_string(), cmd.value().clone()); Frame::Simple("OK".to_string()) } Get(cmd) => { let database: Database = db_lock.lock().unwrap(); if let Some(value) = database.get(cmd.key()) { Frame::Bulk(value.clone()) } else { Frame::Null } } _ => Frame::Null, }; client.write_frame(&response).await.unwrap(); } Ok(()) }
總結
可以監聽停止事件,然後做優雅關閉操作。