tina-core 0.0.2

Tina platform
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
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58

//! Session 扩展

use std::sync::{atomic::AtomicBool, Arc};

use crate::{
    app_system_error,
    tina::{
        data::AppResult,
        server::session::{Session, SessionInner},
    },
};

use super::request_ext::RequestExt;

impl Session {
    /// 构建
    pub async fn from_http_request<Req: RequestExt>(req: &Req) -> AppResult<Self> {
        req.get_session().ok_or_else(|| app_system_error!("no Session found in request"))
    }
    /// 构建
    pub async fn from_http_request_mut<Req: RequestExt>(req: &mut Req) -> AppResult<Self> {
        let (session, is_new) = {
            match req.get_session() {
                None => {
                    let locale = Arc::new(req.get_locale());
                    let application = req.get_application()?;
                    let route_config = match req.get_route_config() {
                        Ok(v) => v,
                        Err(err) => return Err(err),
                    };
                    let request_time = req.get_request_time()?;
                    let enable_token = route_config.enable_token;
                    let security_config = application.get_security_config()?;
                    let token_key = security_config.token_header_name.as_str();
                    let token = req.get_request_header(token_key)?.map(|v| v.into_owned());

                    let session = Session::new(SessionInner {
                        application,
                        enable_token,
                        token,
                        locale,
                        request_time,
                        route_config: Some(route_config),
                        transaction_flag: AtomicBool::new(false),
                        extension: Default::default(),
                        attributes: Arc::new(Default::default()),
                    });
                    (session, true)
                }
                Some(session) => (session, false),
            }
        };
        if is_new {
            req.set_session(session.clone());
        }
        Ok(session)
    }
}