pingora_core/modules/http/
grpc_web.rs1use super::*;
16use crate::protocols::http::bridge::grpc_web::GrpcWebCtx;
17use std::ops::{Deref, DerefMut};
18
19#[derive(Default)]
22pub struct GrpcWebBridge(GrpcWebCtx);
23
24impl Deref for GrpcWebBridge {
25 type Target = GrpcWebCtx;
26
27 fn deref(&self) -> &Self::Target {
28 &self.0
29 }
30}
31
32impl DerefMut for GrpcWebBridge {
33 fn deref_mut(&mut self) -> &mut Self::Target {
34 &mut self.0
35 }
36}
37
38#[async_trait]
39impl HttpModule for GrpcWebBridge {
40 fn as_any(&self) -> &dyn std::any::Any {
41 self
42 }
43
44 fn as_any_mut(&mut self) -> &mut dyn std::any::Any {
45 self
46 }
47
48 async fn request_header_filter(&mut self, req: &mut RequestHeader) -> Result<()> {
49 self.0.request_header_filter(req);
50 Ok(())
51 }
52
53 async fn response_header_filter(
54 &mut self,
55 resp: &mut ResponseHeader,
56 _end_of_stream: bool,
57 ) -> Result<()> {
58 self.0.response_header_filter(resp);
59 Ok(())
60 }
61
62 fn response_trailer_filter(
63 &mut self,
64 trailers: &mut Option<Box<HeaderMap>>,
65 ) -> Result<Option<Bytes>> {
66 if let Some(trailers) = trailers {
67 return self.0.response_trailer_filter(trailers);
68 }
69 Ok(None)
70 }
71}
72
73pub struct GrpcWeb;
75
76impl HttpModuleBuilder for GrpcWeb {
77 fn init(&self) -> Module {
78 Box::new(GrpcWebBridge::default())
79 }
80}