feat(seatbelt): ResilienceContext is now ThreadAware (#290)

I found a need for it. Minimal changes so far, until more scenarios are
uncovered. We can then start putting these on top of middleware
services.

Author: martintmk

Cargo.lock

@@ -36,7 +36,7 @@ layered = { path = "crates/layered", default-features = false, version = "0.3.0"
 ohno = { path = "crates/ohno", default-features = false, version = "0.3.1" }
 ohno_macros = { path = "crates/ohno_macros", default-features = false, version = "0.3.0" }
 recoverable = { path = "crates/recoverable", default-features = false, version = "0.1.0" }
-seatbelt = { path = "crates/seatbelt", default-features = false, version = "0.3.0" }
+seatbelt = { path = "crates/seatbelt", default-features = false, version = "0.3.1" }
 testing_aids = { path = "crates/testing_aids", default-features = false }
 thread_aware = { path = "crates/thread_aware", default-features = false, version = "0.6.2" }
 thread_aware_macros = { path = "crates/thread_aware_macros", default-features = false, version = "0.6.1" }

Cargo.toml

@@ -1,5 +1,11 @@
 # Changelog
 
+## [0.3.1] - 2026-02-27
+
+- ✨ Features
+
+  - ResilienceContext is now ThreadAware
+
 ## [0.3.0] - 2026-02-17
 
 - ✨ Features

crates/seatbelt/CHANGELOG.md

@@ -4,7 +4,7 @@
 [package]
 name = "seatbelt"
 description = "Resilience and recovery mechanisms for fallible operations."
-version = "0.3.0"
+version = "0.3.1"
 readme = "README.md"
 keywords = ["oxidizer", "resilience", "layered", "recovery", "retry"]
 categories = ["data-structures"]
@@ -27,6 +27,9 @@ allowed_external_types = [
     "tick::clock::Clock",
     "tower_layer::Layer",
     "tower_service::Service",
+    "thread_aware::core::ThreadAware",
+    "thread_aware::affinity::MemoryAffinity",
+    "thread_aware::affinity::PinnedAffinity",
 ]
 
 [package.metadata.docs.rs]
@@ -46,6 +49,7 @@ fastrand = { workspace = true, optional = true }
 layered = { workspace = true }
 opentelemetry = { workspace = true, optional = true }
 recoverable = { workspace = true }
+thread_aware = { workspace = true }
 tick = { workspace = true }
 tower-service = { workspace = true, optional = true }
 tracing = { workspace = true, optional = true }

crates/seatbelt/Cargo.toml

@@ -140,19 +140,19 @@ This crate provides several optional features that can be enabled in your `Cargo
 This crate was developed as part of <a href="../..">The Oxidizer Project</a>. Browse this crate's <a href="https://github.com/microsoft/oxidizer/tree/main/crates/seatbelt">source code</a>.
 </sub>
 
- [__cargo_doc2readme_dependencies_info]: ggGkYW0CYXSEGy4k8ldDFPOhG2VNeXtD5nnKG6EPY6OfW5wBG8g18NOFNdxpYXKEG4XTWg5D20iXG5-sZfM55RcnG_PRlQIWXOiaG6Oh8f0TxP2DYWSFgmdsYXllcmVkZTAuMy4wgmtyZWNvdmVyYWJsZWUwLjEuMIJoc2VhdGJlbHRlMC4zLjCCZHRpY2tlMC4yLjCCbXRvd2VyX3NlcnZpY2VlMC4zLjM
+ [__cargo_doc2readme_dependencies_info]: ggGkYW0CYXSEGy4k8ldDFPOhG2VNeXtD5nnKG6EPY6OfW5wBG8g18NOFNdxpYXKEG4XTWg5D20iXG5-sZfM55RcnG_PRlQIWXOiaG6Oh8f0TxP2DYWSFgmdsYXllcmVkZTAuMy4wgmtyZWNvdmVyYWJsZWUwLjEuMIJoc2VhdGJlbHRlMC4zLjGCZHRpY2tlMC4yLjCCbXRvd2VyX3NlcnZpY2VlMC4zLjM
  [__link0]: https://crates.io/crates/layered/0.3.0
  [__link1]: https://docs.rs/layered/0.3.0/layered/?search=Stack
- [__link10]: https://docs.rs/seatbelt/0.3.0/seatbelt/breaker/index.html
- [__link11]: https://docs.rs/seatbelt/0.3.0/seatbelt/timeout/index.html
- [__link12]: https://docs.rs/seatbelt/0.3.0/seatbelt/retry/index.html
- [__link13]: https://docs.rs/seatbelt/0.3.0/seatbelt/breaker/index.html
+ [__link10]: https://docs.rs/seatbelt/0.3.1/seatbelt/breaker/index.html
+ [__link11]: https://docs.rs/seatbelt/0.3.1/seatbelt/timeout/index.html
+ [__link12]: https://docs.rs/seatbelt/0.3.1/seatbelt/retry/index.html
+ [__link13]: https://docs.rs/seatbelt/0.3.1/seatbelt/breaker/index.html
  [__link14]: https://docs.rs/tower_service/0.3.3/tower_service/?search=Service
  [__link2]: https://docs.rs/tick/0.2.0/tick/?search=Clock
  [__link3]: https://crates.io/crates/tick/0.2.0
- [__link4]: https://docs.rs/seatbelt/0.3.0/seatbelt/?search=ResilienceContext
- [__link5]: https://docs.rs/seatbelt/0.3.0/seatbelt/?search=ResilienceContext
+ [__link4]: https://docs.rs/seatbelt/0.3.1/seatbelt/?search=ResilienceContext
+ [__link5]: https://docs.rs/seatbelt/0.3.1/seatbelt/?search=ResilienceContext
  [__link6]: https://docs.rs/recoverable/0.1.0/recoverable/?search=RecoveryInfo
  [__link7]: https://docs.rs/recoverable/0.1.0/recoverable/?search=Recovery
- [__link8]: https://docs.rs/seatbelt/0.3.0/seatbelt/timeout/index.html
- [__link9]: https://docs.rs/seatbelt/0.3.0/seatbelt/retry/index.html
+ [__link8]: https://docs.rs/seatbelt/0.3.1/seatbelt/timeout/index.html
+ [__link9]: https://docs.rs/seatbelt/0.3.1/seatbelt/retry/index.html

crates/seatbelt/README.md

@@ -3,6 +3,10 @@
 
 use std::borrow::Cow;
 
+use thread_aware::{
+    ThreadAware,
+    affinity::{MemoryAffinity, PinnedAffinity},
+};
 use tick::Clock;
 
 pub(crate) const DEFAULT_CONTEXT_NAME: &str = "default";
@@ -23,6 +27,15 @@ pub struct ResilienceContext<In, Out> {
     _out: std::marker::PhantomData<fn() -> Out>,
 }
 
+impl<In, Out> ThreadAware for ResilienceContext<In, Out> {
+    fn relocated(mut self, source: MemoryAffinity, destination: PinnedAffinity) -> Self {
+        // Only clock is thread-aware for now. At some point, we also want
+        // telemetry to be tread-aware too.
+        self.clock = self.clock.relocated(source, destination);
+        self
+    }
+}
+
 impl<In, Out> ResilienceContext<In, Out> {
     /// Create a context with a clock. Initializes with `name = "default"`.
     pub fn new(clock: impl AsRef<Clock>) -> Self {
@@ -113,8 +126,14 @@ impl<In, Out> Clone for ResilienceContext<In, Out> {
 #[cfg(test)]
 mod tests {
 
+    use std::fmt::Debug;
+
+    use thread_aware::affinity::pinned_affinities;
+
     use super::*;
 
+    static_assertions::assert_impl_all!(ResilienceContext<(), ()>: Send, Sync, ThreadAware, Debug, Clone);
+
     #[test]
     fn test_new_with_clock_sets_default_pipeline_name() {
         let clock = tick::Clock::new_frozen();
@@ -156,6 +175,14 @@ mod tests {
         assert!(dump.contains('3'));
     }
 
+    #[test]
+    fn relocate_ok() {
+        let ctx = ResilienceContext::<(), ()>::new(tick::Clock::new_frozen());
+        let affinites = pinned_affinities(&[2]);
+
+        _ = ctx.relocated(affinites[0].into(), affinites[1]);
+    }
+
     #[cfg(not(miri))]
     fn test_meter_provider() -> (
         opentelemetry_sdk::metrics::SdkMeterProvider,