ehlxr
/
poem_demo
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

update at 2022-02-21 17:44:08 by ehlxr

master
ehlxr 2022-02-21 17:44:08 +08:00
parent 7d46777f99
commit 88b9d511dd
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/main.rs
View File

@ -5,7 +5,7 @@ use poem_openapi::OpenApiService;
mod user; mod user;
use time::{format_description, macros::offset}; use time::{format_description, macros::offset};
use tokio::sync::Mutex; use tokio::sync::RwLock;
use tracing::{debug, Level}; use tracing::{debug, Level};
use tracing_appender::non_blocking::WorkerGuard; use tracing_appender::non_blocking::WorkerGuard;
use tracing_subscriber::{ use tracing_subscriber::{
@ -20,7 +20,7 @@ const FORMAT_STR: &str = "[year]-[month]-[day] [hour]:[minute]:[second]";
extern crate lazy_static; extern crate lazy_static;
lazy_static! { lazy_static! {
static ref CACHE: Mutex<HashMap<String, String>> = Mutex::new(HashMap::new()); static ref CACHE: RwLock<HashMap<String, String>> = RwLock::new(HashMap::new());
} }
#[tokio::main] #[tokio::main]
@ -85,7 +85,7 @@ async fn refresh_token() {
interval = 5; interval = 5;
CACHE CACHE
.lock() .write()
.await .await
.insert("token".to_string(), format!("token{}", count)); .insert("token".to_string(), format!("token{}", count));
} }

2
src/user/api.rs
View File

@ -22,7 +22,7 @@ impl Api {
let Token(token) = data.0; let Token(token) = data.0;
assert_eq!(token, "token123"); assert_eq!(token, "token123");
let m_guard = CACHE.lock().await; let m_guard = CACHE.read().await;
let rt = m_guard.get(&"token".to_string()).unwrap(); let rt = m_guard.get(&"token".to_string()).unwrap();
info!("refresh_token is {}", rt); info!("refresh_token is {}", rt);