use std::sync::atomic::{AtomicU32, Ordering};
use std::sync::Arc;
use std::time::Duration;
use mytheclipse::{
aggregate_error::AggregateError,
auto_metrics_service::AutoMetricsServiceBuilder,
parallel_map::parallel_map,
pool::{AutoReconnectPool, Pool, Reconnectable, SemaphorePool},
retry_ext::RetryExt,
runtime_auto::RuntimeConfig,
service_builder::ServiceConfig,
shutdown_guard::ShutdownGuard,
};
#[tokio::main]
async fn main() {
println!("== mytheclipse high-level abstractions demo ==");
let cfg = RuntimeConfig::auto();
println!(
"1. RuntimeConfig::auto() -> worker={}, blocking={}, compute={}, io={}",
cfg.worker_threads, cfg.max_blocking_threads, cfg.compute_threads, cfg.io_threads
);
let results: Result<Vec<i32>, AggregateError> =
parallel_map(vec![1, 2, 3, 4, 5], 2, |x| async move {
tokio::time::sleep(Duration::from_millis(10)).await;
Ok::<_, std::io::Error>(x * 10)
})
.await;
println!("2. parallel_map -> {:?}", results.unwrap());
let attempts = Arc::new(AtomicU32::new(0));
let a = Arc::clone(&attempts);
let cfg = mytheclipse::retry::RetryConfig {
max_attempts: 4,
base_delay: Duration::from_millis(1),
..Default::default()
};
let fut = async { Err::<u32, String>("boom".into()) };
let op = move || {
let a = Arc::clone(&a);
async move {
let n = a.fetch_add(1, Ordering::SeqCst);
if n < 3 {
Err::<u32, String>("transient".into())
} else {
Ok(42u32)
}
}
};
let value = fut.retry(cfg, |_: &String| true, op).await.unwrap();
println!(
"3. RetryExt with {} attempts -> {value}",
attempts.load(Ordering::SeqCst)
);
let fired = Arc::new(AtomicU32::new(0));
let f = Arc::clone(&fired);
{
let _guard = ShutdownGuard::new(move || {
f.fetch_add(1, Ordering::SeqCst);
});
} println!(
"4. ShutdownGuard fired {}x (exactly-once, even on unwind)",
fired.load(Ordering::SeqCst)
);
let auto = AutoReconnectPool::new(SemaphorePool::new(vec![1u32, 2u32]), Probe { dead: 1 });
let first = auto.acquire().await.unwrap().resource;
println!("5. AutoReconnectPool first acquire -> {first}");
let mut svc_cfg = ServiceConfig::default();
svc_cfg.max_attempts = 2;
let svc = AutoMetricsServiceBuilder::new("demo_op", svc_cfg);
let n = Arc::new(AtomicU32::new(0));
let n2 = Arc::clone(&n);
let _: Result<u32, mytheclipse::service_builder::RunError<()>> = svc
.run(|| {
let n2 = Arc::clone(&n2);
Box::pin(async move {
tokio::time::sleep(Duration::from_millis(5)).await;
let v = n2.fetch_add(1, Ordering::SeqCst);
if v < 1 {
Err(())
} else {
Ok(7u32)
}
})
})
.await;
let snap = svc.collector().snapshot();
println!(
"6. AutoMetrics -> {} counters, {} histograms",
snap.counters.len(),
snap.histograms.len()
);
println!("== demo complete ==");
}
struct Probe {
dead: u32,
}
#[async_trait::async_trait]
impl Reconnectable for Probe {
type Item = u32;
fn is_healthy(&self, item: &Self::Item) -> bool {
*item != self.dead
}
async fn reconnect(&self) -> Result<Self::Item, Box<dyn std::error::Error + Send + Sync>> {
Ok(999)
}
}