Skip to content

Commit

Permalink
refactor(app): Outline route label extraction
Browse files Browse the repository at this point in the history
this addresses the `TODO` comment, moving the `RouteLabelExtract` type
out of the policy route layer function.

this defines a method on `MatchedRoute` which may be called to retrieve
a `RouteLabelExtract`. that type holds references to the
contruction-time information about the route, and can be used to later
inspect an outbound request at call-time, returning a `labels::Route`
set of labels.

this is a helpful piece of reusable glue that is broadly useful in
instrumenting our route-level middleware.

Signed-off-by: katelyn martin <[email protected]>
  • Loading branch information
cratelyn committed Nov 10, 2024
1 parent ec8eb29 commit a75492c
Show file tree
Hide file tree
Showing 2 changed files with 22 additions and 11 deletions.
13 changes: 2 additions & 11 deletions linkerd/app/outbound/src/http/logical/policy/route.rs
Original file line number Diff line number Diff line change
Expand Up @@ -124,18 +124,9 @@ where
.push_on_service(svc::LoadShed::layer())
.push(filters::NewApplyFilters::<Self, _, _>::layer())
.push({
// TODO(kate): extracting route labels like this should ideally live somewhere
// else, like e.g. the `SetExtensions` middleware.
let mk_extract = |rt: &Self| {
let Route {
parent_ref,
route_ref,
..
} = &rt.params;
metrics::labels::RouteLabelExtract(parent_ref.clone(), route_ref.clone())
};
let mk = Self::label_extractor;
let metrics = metrics.retry.clone();
retry::NewHttpRetry::layer_via_mk(mk_extract, metrics)
retry::NewHttpRetry::layer_via_mk(mk, metrics)
})
.check_new::<Self>()
.check_new_service::<Self, http::Request<http::BoxBody>>()
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -215,6 +215,26 @@ impl EncodeLabelSet for GrpcRsp {
}
}

// === impl MatchedRoute ===

impl<T, M, F, P> super::super::MatchedRoute<T, M, F, P> {
/// Returns a [`RouteLabelExtract`].
///
/// The extractor returned by this function provides a [`ExtractParam<P, T>`] implementation
/// that can be used to acquire the route-level labels corresponding to a given outbound
/// request.
pub(crate) fn label_extractor(&self) -> RouteLabelExtract {
use super::super::Route;
let Route {
parent_ref,
route_ref,
..
} = &self.params;

RouteLabelExtract(parent_ref.clone(), route_ref.clone())
}
}

// === impl RouteLabelExtract ===

impl<B> ExtractParam<Route, http::Request<B>> for RouteLabelExtract {
Expand Down

0 comments on commit a75492c

Please sign in to comment.