Skip to content

Commit d022059

Browse files
committed
Adjust to new extension method names
1 parent c42a1e3 commit d022059

File tree

5 files changed

+11
-11
lines changed

5 files changed

+11
-11
lines changed

src/middleware/app.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@ impl Middleware for AppMiddleware {
2121
}
2222

2323
fn after(&self, req: &mut dyn RequestExt, res: AfterResult) -> AfterResult {
24-
req.mut_extensions().pop::<Arc<App>>().unwrap();
24+
req.mut_extensions().remove::<Arc<App>>().unwrap();
2525
res
2626
}
2727
}
@@ -33,6 +33,6 @@ pub trait RequestApp {
3333

3434
impl<T: RequestExt + ?Sized> RequestApp for T {
3535
fn app(&self) -> &Arc<App> {
36-
self.extensions().find::<Arc<App>>().expect("Missing app")
36+
self.extensions().get::<Arc<App>>().expect("Missing app")
3737
}
3838
}

src/middleware/block_traffic.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@ impl AroundMiddleware for BlockTraffic {
3737

3838
impl Handler for BlockTraffic {
3939
fn call(&self, req: &mut dyn RequestExt) -> AfterResult {
40-
let app = req.extensions().find::<Arc<App>>().expect("Missing app");
40+
let app = req.extensions().get::<Arc<App>>().expect("Missing app");
4141
let domain_name = app.config.domain_name.clone();
4242

4343
let has_blocked_value = req

src/middleware/log_request.rs

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@ struct CustomMetadata {
4343
}
4444

4545
pub fn add_custom_metadata<V: Display>(req: &mut dyn RequestExt, key: &'static str, value: V) {
46-
if let Some(metadata) = req.mut_extensions().find_mut::<CustomMetadata>() {
46+
if let Some(metadata) = req.mut_extensions().get_mut::<CustomMetadata>() {
4747
metadata.entries.push((key, value.to_string()));
4848
} else {
4949
let mut metadata = CustomMetadata {
@@ -78,7 +78,7 @@ fn report_to_sentry(req: &dyn RequestExt, res: &AfterResult) {
7878
scope.set_tag("response.status", status.as_str());
7979
}
8080

81-
if let Some(response_time) = req.extensions().find::<ResponseTime>() {
81+
if let Some(response_time) = req.extensions().get::<ResponseTime>() {
8282
scope.set_extra("Response time [ms]", response_time.as_millis().into());
8383
}
8484
});
@@ -87,7 +87,7 @@ fn report_to_sentry(req: &dyn RequestExt, res: &AfterResult) {
8787
#[cfg(test)]
8888
pub(crate) fn get_log_message(req: &dyn RequestExt, key: &'static str) -> String {
8989
// Unwrap shouldn't panic as no other code has access to the private struct to remove it
90-
for (k, v) in &req.extensions().find::<CustomMetadata>().unwrap().entries {
90+
for (k, v) in &req.extensions().get::<CustomMetadata>().unwrap().entries {
9191
if key == *k {
9292
return v.clone();
9393
}
@@ -131,14 +131,14 @@ impl Display for RequestLine<'_> {
131131

132132
line.add_quoted_field("fwd", request_header(self.req, "x-real-ip"))?;
133133

134-
let response_time = self.req.extensions().find::<ResponseTime>();
134+
let response_time = self.req.extensions().get::<ResponseTime>();
135135
if let Some(response_time) = response_time {
136136
line.add_field("service", response_time)?;
137137
}
138138
line.add_field("status", status.as_str())?;
139139
line.add_quoted_field("user_agent", request_header(self.req, header::USER_AGENT))?;
140140

141-
if let Some(metadata) = self.req.extensions().find::<CustomMetadata>() {
141+
if let Some(metadata) = self.req.extensions().get::<CustomMetadata>() {
142142
for (key, value) in &metadata.entries {
143143
line.add_quoted_field(key, value)?;
144144
}
@@ -164,7 +164,7 @@ impl<'a> Display for FullPath<'a> {
164164
fn fmt(&self, f: &mut Formatter<'_>) -> fmt::Result {
165165
let request = self.0;
166166

167-
let original_path = request.extensions().find::<OriginalPath>();
167+
let original_path = request.extensions().get::<OriginalPath>();
168168
let path = original_path
169169
.map(|p| p.0.as_str())
170170
.unwrap_or_else(|| request.path());

src/middleware/update_metrics.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@ impl Middleware for UpdateMetrics {
2222

2323
let endpoint = req
2424
.extensions()
25-
.find::<RoutePattern>()
25+
.get::<RoutePattern>()
2626
.map(|p| p.pattern())
2727
.unwrap_or("<unknown>");
2828
metrics

src/router.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -151,7 +151,7 @@ struct C(pub fn(&mut dyn RequestExt) -> EndpointResult);
151151

152152
impl Handler for C {
153153
fn call(&self, req: &mut dyn RequestExt) -> HandlerResult {
154-
if let Some(pattern) = req.extensions().find::<RoutePattern>() {
154+
if let Some(pattern) = req.extensions().get::<RoutePattern>() {
155155
let pattern = pattern.pattern();
156156

157157
// Configure the Sentry `transaction` field *before* we handle the request,

0 commit comments

Comments
 (0)