webhttp 0.1.12

webhttp
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
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97

use super::{super::AppState, wsconn::WsConn};
use actix_web::{get, web, HttpRequest, HttpResponse, Responder, Scope};
use actix_web_actors::ws;
use time::macros::offset;
use tracing::{debug, warn};

#[get("/health")]
pub async fn ws_health(_data: web::Data<AppState>) -> impl Responder {
    debug!("health is called");
    let now = time::OffsetDateTime::now_utc().to_offset(offset!(+8));
    let return_info = format!("running: {:?}", now,);
    HttpResponse::Ok().body(return_info)
}

/// connid can be uuid, actor is role
/// connid can be in multi client, if by this way, they are same group, the use actor to differ them
#[get("/{business}/{actor}/{connid}")]
pub async fn ws_entry(
    req: HttpRequest,
    stream: web::Payload,
    params: web::Path<(String, String, String)>,
    appdata: web::Data<AppState>,
) -> HttpResponse {
    // info!("req: {:?} params: {:?}", req, params);
    let (business, actor, connid) = params.into_inner();
    let ip = req.peer_addr().unwrap().to_string();

    let headers = req.headers();
    let token = if !headers.contains_key(actix_http::header::SEC_WEBSOCKET_PROTOCOL) {
        warn!("@@@@@ token is null for this websocket connection @@@@@");
        "".to_string()
    } else {
        let token = headers
            .get(actix_http::header::SEC_WEBSOCKET_PROTOCOL)
            .unwrap()
            .to_str()
            .unwrap()
            .to_string();
        token
    };

    // let token = if !headers.contains_key(actix_http::header::AUTHORIZATION) {
    //     warn!("@@@@@ token is null for this websocket connection @@@@@");
    //     "".to_string()
    // } else {
    //     let token = headers
    //         .get(actix_http::header::AUTHORIZATION)
    //         .unwrap()
    //         .to_str()
    //         .unwrap()
    //         .to_string();
    //     if token.len() < 7 {
    //         return HttpResponse::BadRequest().body(format!("Header中鉴权信息bearer的格式不对"));
    //     } else {
    //         let token = token.split_at(7).1.to_string();
    //         token
    //     }
    // };
    // let token = if headers.contains_key("token") {
    //     headers.get("token").unwrap().to_str().unwrap().to_string()
    // } else {
    //     warn!("@@@@@ token is null for this websocket connection @@@@@");
    //     "".to_string()
    // };

    let wsconn = WsConn::new(
        ip,
        business,
        connid,
        actor,
        token.clone(),
        appdata.get_ref().clone(),
    );
    // let resp = ws::start(wsconn, &req, stream);
    let resp = if token.is_empty() {
        ws::WsResponseBuilder::new(wsconn, &req, stream)
            .frame_size(1024 * 1024 * 64)
            .start()
    } else {
        ws::WsResponseBuilder::new(wsconn, &req, stream)
            .protocols(&[token.as_str()])
            .frame_size(1024 * 1024 * 64)
            .start()
    };

    debug!("{:?}", resp);
    if let Ok(resp) = resp {
        resp
    } else {
        HttpResponse::BadRequest()
            .body(format!("start ws conn with error: {}", resp.err().unwrap()))
    }
}

pub fn ws_api() -> Scope {
    web::scope("").service(ws_entry).service(ws_health)
}