|
| 1 | +use { |
| 2 | + aws_sdk_cloudwatch::{ |
| 3 | + primitives::DateTime, |
| 4 | + types::{Dimension, MetricDatum, StandardUnit}, |
| 5 | + }, |
| 6 | + std::time::SystemTime, |
| 7 | + tracing_subscriber::{layer::SubscriberExt, util::SubscriberInitExt}, |
| 8 | + yttrium::sign::test_helpers::{probe_layer, test_sign_impl}, |
| 9 | +}; |
| 10 | + |
| 11 | +#[tokio::main] |
| 12 | +pub async fn main() { |
| 13 | + let probe_layer = probe_layer::ProbeLayer::new(); |
| 14 | + tracing_subscriber::fmt() |
| 15 | + .with_max_level(tracing::Level::DEBUG) |
| 16 | + .finish() |
| 17 | + .with(probe_layer.clone()) |
| 18 | + .try_init() |
| 19 | + .unwrap(); |
| 20 | + |
| 21 | + // let start = Instant::now(); |
| 22 | + let result = test_sign_impl().await; |
| 23 | + tracing::debug!(probe = "e2e"); |
| 24 | + // let e2e_latency = start.elapsed(); |
| 25 | + |
| 26 | + let config = aws_config::load_from_env().await; |
| 27 | + let cloudwatch_client = aws_sdk_cloudwatch::Client::new(&config); |
| 28 | + let region = config.region().unwrap().to_string(); |
| 29 | + let region_dimension = Dimension::builder() |
| 30 | + .name("Region".to_string()) |
| 31 | + .value(region.clone()) |
| 32 | + .build(); |
| 33 | + |
| 34 | + // TODO measure crypto operation latency |
| 35 | + // TODO measure storage operation latency |
| 36 | + // TODO measure client request latency |
| 37 | + |
| 38 | + println!("probe_layer: {:?}", probe_layer.accumulator()); |
| 39 | + // panic!(); |
| 40 | + |
| 41 | + let now = DateTime::from(SystemTime::now()); |
| 42 | + let metrics = probe_layer |
| 43 | + .accumulator() |
| 44 | + .iter() |
| 45 | + .map(|p| { |
| 46 | + let probe_dimension = Dimension::builder() |
| 47 | + .name("Probe".to_string()) |
| 48 | + .value(p.probe.clone()) |
| 49 | + .build(); |
| 50 | + let group_dimension = p.group.clone().map(|g| { |
| 51 | + Dimension::builder().name("Group".to_string()).value(g).build() |
| 52 | + }); |
| 53 | + vec![ |
| 54 | + { |
| 55 | + let mut metric = MetricDatum::builder() |
| 56 | + .metric_name(format!("probe_hit")) |
| 57 | + // .metric_name(format!("probe_{}_hit", p.probe.clone())) |
| 58 | + .dimensions(probe_dimension.clone()); |
| 59 | + if let Some(group_dimension) = group_dimension.clone() { |
| 60 | + metric = metric.dimensions(group_dimension); |
| 61 | + } |
| 62 | + metric |
| 63 | + .dimensions(region_dimension.clone()) |
| 64 | + .value(1.) |
| 65 | + .unit(StandardUnit::Count) |
| 66 | + .timestamp(now) |
| 67 | + .build() |
| 68 | + }, |
| 69 | + { |
| 70 | + let mut metric = MetricDatum::builder() |
| 71 | + .metric_name(format!( |
| 72 | + "probe_latency_seconds", |
| 73 | + // "probe_{}_latency_seconds", |
| 74 | + // p.probe.clone() |
| 75 | + )) |
| 76 | + .dimensions(probe_dimension.clone()); |
| 77 | + if let Some(group_dimension) = group_dimension.clone() { |
| 78 | + metric = metric.dimensions(group_dimension); |
| 79 | + } |
| 80 | + metric |
| 81 | + .dimensions(region_dimension.clone()) |
| 82 | + .value(p.elapsed_s) |
| 83 | + .unit(StandardUnit::Seconds) |
| 84 | + .timestamp(now) |
| 85 | + .build() |
| 86 | + }, |
| 87 | + ] |
| 88 | + }) |
| 89 | + .flatten() |
| 90 | + .collect::<Vec<_>>(); |
| 91 | + |
| 92 | + cloudwatch_client |
| 93 | + .put_metric_data() |
| 94 | + .namespace("dev2_Canary_RustSignClient") |
| 95 | + .set_metric_data(Some(metrics)) |
| 96 | + // .metric_data( |
| 97 | + // MetricDatum::builder() |
| 98 | + // .metric_name("e2e.success".to_string()) |
| 99 | + // .dimensions(dimensions.clone()) |
| 100 | + // .value(if result.is_ok() { 1. } else { 0. }) |
| 101 | + // .unit(StandardUnit::Count) |
| 102 | + // .timestamp(DateTime::from(SystemTime::now())) |
| 103 | + // .build(), |
| 104 | + // ) |
| 105 | + // .metric_data( |
| 106 | + // MetricDatum::builder() |
| 107 | + // .metric_name("e2e.latency".to_string()) |
| 108 | + // .dimensions(dimensions.clone()) |
| 109 | + // .value(e2e_latency.as_millis() as f64) |
| 110 | + // .unit(StandardUnit::Milliseconds) |
| 111 | + // .timestamp(DateTime::from(SystemTime::now())) |
| 112 | + // .build(), |
| 113 | + // ) |
| 114 | + .send() |
| 115 | + .await |
| 116 | + .unwrap(); |
| 117 | + |
| 118 | + if let Err(e) = result { |
| 119 | + panic!("Test failed: {e}"); |
| 120 | + } |
| 121 | +} |
0 commit comments