如何在中间件二次处理response中的数据

[z91300]

中间件代码示例：

#[handler]
pub async fn logger_mid_fn(
    req: &mut Request,
    res: &mut Response,
    ctrl: &mut FlowCtrl,
    depot: &mut Depot,
) {

    async move {
	let mut chunks = VecDeque::new();

        // 需要新增的数据
        // { "request_id": "01HHW89K6T39RQTRA5WFHNJ1PG" }
        let json_data = json!({
            "request_id":  req.header::<String>("x-request-id"),
        });
        let json_bytes = Bytes::from(serde_json::to_string(&json_data).unwrap());
        let body = ResBody::Once(json_bytes);
        println!("{:?}", body);

        // 获取 response 中的数据
        // { "code": 0, "msg": "ok"}
        let body2: ResBody = res.body_mut().take();
        println!("{:?}", body2);

        if let ResBody::Once(bytes) = body {
            chunks.push_back(bytes);
        }
        if let ResBody::Once(bytes) = body2 {
            chunks.push_back(bytes);
        }
        let combined_body = ResBody::Chunks(chunks);

        res.replace_body(combined_body);
   }
}

期望返回：

{
    "request_id": "01HHW89K6T39RQTRA5WFHNJ1PG",
    "code": 0,
    "msg": "ok"
}

实际返回

{
    "code": 0,
    "msg": "ok"
}{
    "request_id": "01HHW89K6T39RQTRA5WFHNJ1PG",
}

求助应该怎么写才正确？

[chrislearn]

这个跟中间件无关，你应该序列化数据，然后修改，再放入 body。

[z91300]

感谢答复！

还是想问一下，有办法通过 res.replace_body 或 res.render 二次处理吗，类似上述。

因为目前代码已实现很多,统一走的类似下方这种，想在在原有逻辑上变更也没有明确思路。

impl<T: ToSchema> ResObj<T> {
    pub fn ok(data: Option<T>) -> Self {
        Self {
            code: 0,
            msg: "ok".to_string(),
            data,
        }
    }
    pub fn custom_code(code: i32, msg: String) -> Self {
        Self {
            code,
            msg,
            data: None,
        }
    }

    pub fn err(err: String) -> Self {
        Self {
            code: 500,
            msg: err,
            data: None,
        }
    }
}

#[allow(dead_code)]
pub fn res_json_ok<T: ToSchema>(data: Option<T>) -> Json<ResObj<T>> {
    Json(ResObj::ok(data))
}

res.render(res_json_custom::<()>(404, "没有对应的接口".to_string()));

[chrislearn]

我没很看懂你的代码。

但是肯定是可以再处理的。本身 request response 就是可写可读的对象， 你任何一个中间件想写就写，想去就去里面的数据。本身没问题， 只是可能你代码逻辑本身有问题。

[z91300]

如果想要在中间件中二次处理 response ，在原有的响应中增加request_id字段，能否提供下思路或者示例代码? 非常感谢！

use salvo_core::http::ResBody;

#[handler]
pub async fn logger_mid_fn(
    req: &mut Request,
    res: &mut Response,
    ctrl: &mut FlowCtrl,
    depot: &mut Depot,
) {
    let request_id = req.header::<String>("x-request-id");
   
    async move {
        let now = Instant::now();
        ctrl.call_next(req, depot, res).await;

        // 获取原有响应 { "code": 0, "msg": "ok"}
        let body: ResBody = res.body_mut().take();

        // 输出新响应
        // res.replace_body(xxx);
        // res.write_body(xxx);
        // res.render(xxx);

        return ctrl.skip_rest();
    }

}

[chrislearn]

你要把拿到的 request 的 body take 了，然后反序列化成某个结构体或者直接 serde_json 里的 Value， 然后往里面写入 request_id, 然后，再重写写入 body.

[z91300]

想要实现的效果：
程序原有响应：

{
    "code": 0,
    "msg": "ok"
}

增加中间件后的响应：

{
    "request_id": "01HHW89K6T39RQTRA5WFHNJ1PG",
    "code": 0,
    "msg": "ok"
}