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
use axum::response::{IntoResponse, Response};
use std::ops::{Deref, DerefMut};
/// A simple wrapper around a `tonic::Status` usable in axum middleware.
///
/// ## Example
/// ```
/// use axum::{middleware::{Next, from_fn}, response::Response, Router};
/// use axum_tonic::GrpcStatus;
/// use hyper::Request;
///
/// async fn tonic_middleware<B>(
/// req: Request<B>,
/// next: Next<B>
/// ) -> Result<Response, GrpcStatus> {
/// if is_auth(&req) {
/// Ok(next.run(req).await)
/// } else {
/// Err(
/// tonic::Status::permission_denied("Not authenticated").into()
/// )
/// }
/// }
///
/// fn is_auth<B>(req: &Request<B>) -> bool {
/// true // or other logic
/// }
///
/// let router: Router<()> = Router::new()
/// .layer(from_fn(tonic_middleware));
/// ```
#[derive(Debug)]
pub struct GrpcStatus(pub tonic::Status);
impl From<tonic::Status> for GrpcStatus {
fn from(s: tonic::Status) -> Self {
Self(s)
}
}
impl Deref for GrpcStatus {
type Target = tonic::Status;
fn deref(&self) -> &Self::Target {
&self.0
}
}
impl DerefMut for GrpcStatus {
fn deref_mut(&mut self) -> &mut Self::Target {
&mut self.0
}
}
impl IntoResponse for GrpcStatus {
fn into_response(self) -> Response {
self.0.to_http().into_response()
}
}