【Rust笔记】Rocket实现自定义的Responder

在Java项目中，我们通常会将响应包装一层来实现统一响应格式，在Rocket中，我们也可以通过实现Responder来返回统一的响应。

res.rs

use crate::api::err::Error;
use rocket::response::Responder;
use rocket::serde::json::json;
use rocket::serde::{Deserialize, Serialize};
use rocket::Request;#[derive(Serialize, Deserialize)]
pub struct Res<'a, T> {code: i32,msg: &'a str,data: Option<T>,
}impl<T> Res<T> {pub fn success(data: T) -> Self {Res {code: 0,msg: "",data: Some(data),}}pub fn error(msg: &str) -> Self {Res {code: 1,msg,data: None,}}pub fn from_error(error: Error) -> Self {Res {code: error.0,msg: error.1,data: None,}}
}impl<'r, 'o: 'r, T: Serialize> Responder<'r, 'o> for Res<T> {fn respond_to(self, request: &'r Request<'_>) -> rocket::response::Result<'o> {json!(&self).respond_to(request)}
}

err.rs

pub type Error<'a> = (i32, &'a str);pub const ANY: Error = (1, "system error");
pub const SOURCE_NOT_EMPTY: Error = (1001, "source is not empty");

然后就可以按具体处理结果返回统一的失败或者成功响应：

• 成功：

  Res::success(source)
  

• 失败：

  Res::error("失败原因")
  

• 自定义失败code：

  Res::from_error((10001,"失败原因"))
  

  或者在err.rs中添加错误类型后：

  Res::from_error(err::SOURCE_NOT_EMPTY)