rust_wheel 0.1.10

A project to define some public component.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34

use std::future::Future;

use log::error;
use log::info;
use tokio::task_local;

use crate::model::user::login_user_info::LoginUserInfo;

task_local! {
    static CURRENT_USER: LoginUserInfo;
}

pub struct ContextUtil;

impl ContextUtil {

    pub async fn with_user<F, T>(user: LoginUserInfo, f: F) -> T 
    where
        F: Future<Output = T>,
    {
        info!("set user to context: {}", serde_json::to_string(&user).unwrap());
        CURRENT_USER.scope(user, f).await
    }
    
    pub fn current_user() -> Result<LoginUserInfo, &'static str> {
        match CURRENT_USER.try_with(|user| user.clone()) {
            Ok(user) => Ok(user),
            Err(err) => {
                error!("ContextUtil::current_user failed: user context not found, err={:?}", err);
                Err("User context not found")
            }
        }
    }
}