Skip to content

Commit cbb3b23

Browse files
authored
chore: Clean up CacheHeaderInterceptor (#33)
1 parent b38f461 commit cbb3b23

File tree

1 file changed

+7
-14
lines changed

1 file changed

+7
-14
lines changed

src/grpc/cache_header_interceptor.rs

Lines changed: 7 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -4,27 +4,20 @@ pub struct CacheHeaderInterceptor {
44
}
55

66
impl tonic::service::Interceptor for CacheHeaderInterceptor {
7-
fn call(&mut self, request: tonic::Request<()>) -> Result<tonic::Request<()>, tonic::Status> {
8-
let request_metadata = request.metadata().clone();
9-
let mut result = {
10-
let message = request.into_inner();
11-
tonic::Request::new(message)
12-
};
13-
result.metadata_mut().insert(
7+
fn call(
8+
&mut self,
9+
mut request: tonic::Request<()>,
10+
) -> Result<tonic::Request<()>, tonic::Status> {
11+
request.metadata_mut().insert(
1412
"authorization",
1513
tonic::metadata::AsciiMetadataValue::from_str(self.auth_key.as_str()).unwrap(),
1614
);
1715
// for reasons unknown, tonic seems to be stripping out the content-type. So we need to add this as
1816
// a workaround so that the requests are successful
19-
result.metadata_mut().insert(
17+
request.metadata_mut().insert(
2018
"content-type",
2119
tonic::metadata::AsciiMetadataValue::from_str("application/grpc").unwrap(),
2220
);
23-
24-
let cache_name = request_metadata.get("cache").unwrap();
25-
26-
// need to re-add our `cache` header back into the interceptor or it will be stripped out
27-
result.metadata_mut().insert("cache", cache_name.clone());
28-
Ok(result)
21+
Ok(request)
2922
}
3023
}

0 commit comments

Comments
 (0)