zero4rs 2.0.0

zero4rs is a powerful, pragmatic, and extremely fast web framework for Rust
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

use crate::prelude2::*;

use std::collections::HashMap;

use futures::stream::StreamExt;

#[derive(Debug, serde::Serialize, serde::Deserialize)]
pub struct SimpileMessage {
    pub id: u32,
    pub message: String,
}

pub async fn publish_message(
    query: web::Query<HashMap<String, String>>,
    message: web::Json<SimpileMessage>,
    request: HttpRequest,
    app_state: web::Data<AppState>,
) -> impl Responder {
    let topic = query.get("topic").ok_or_else(|| {
        Errors::InvalidRequestError("Missing query string parameter: topic".to_string())
    })?;

    match app_state.redis().publish(topic, &message) {
        Ok(()) => request.json(200, R::success(message, String::from("message published"))),
        Err(e) => request.json(200, R::failed(500, e.to_string())),
    }
}

pub async fn subscribe_message(
    query: web::Query<HashMap<String, String>>,
    app_state: web::Data<AppState>,
) -> Result<HttpResponse, Errors> {
    let topic = query.get("topic").ok_or_else(|| {
        Errors::InvalidRequestError("Missing query string parameter: topic".to_string())
    })?;

    match app_state.redis_pub().subscribe(topic).await {
        Ok(rx) => Ok(HttpResponse::Ok()
            .append_header(("Content-Type", "text/event-stream; charset=utf-8"))
            .append_header(("Cache-Control", "no-store"))
            .append_header(("X-Accel-Buffering", "no"))
            .streaming(rx.map(
                |message| -> Result<actix_web::web::Bytes, std::convert::Infallible> {
                    log::info!("subscribe_message: message={}", message);
                    Ok(actix_web::web::Bytes::from(message))
                },
            ))),
        Err(e) => Ok(HttpResponse::Ok().json(R::failed(400, e.to_string()))),
    }
}

// pub async fn subscribe_message2(
//     query: web::Query<HashMap<String, String>>,
//     app_state: web::Data<AppState>,
// ) -> Result<HttpResponse, Errors> {
//     let topic = query.get("topic").ok_or_else(|| {
//         Errors::InvalidRequestError("Missing query string parameter: topic".to_string())
//     })?;

//     match app_state.redis_pub().subscribe2(topic).await {
//         Ok(()) => Ok(HttpResponse::Ok().json(R::ok(true))),
//         Err(e) => Ok(HttpResponse::Ok().json(R::failed(400, e.to_string()))),
//     }
// }