i/Aether
1
0
Fork 0
mirror of https://github.com/fawney19/Aether.git synced 2026-09-12 06:00:20 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix PR 599 check regressions

Browse Source
This commit is contained in:
fawney19
2026-05-29 12:30:56 +08:00
parent 3e4dfd2bac
commit 9fb600df1b
4 changed files with 126 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files
crates/aether-data/src/repository/proxy_nodes/sqlite.rs
+74 -10
View File
@@ -1566,6 +1566,33 @@ VALUES ('node-1', 'registered', 'ok', 3)
.await
.expect("node should register");
let now = super::current_unix_secs();
repository
.apply_heartbeat(&ProxyNodeHeartbeatMutation {
node_id: registered.id.clone(),
heartbeat_interval: Some(30),
active_connections: Some(0),
total_requests_delta: None,
avg_latency_ms: None,
failed_requests_delta: None,
dns_failures_delta: None,
stream_errors_delta: None,
proxy_metadata: Some(json!({
"tunnel_metrics": {
"connect_errors": 0,
"disconnects": 0,
"error_events_total": 0,
"ws_in_bytes": 0,
"ws_out_bytes": 0,
"ws_in_frames": 0,
"ws_out_frames": 0,
"heartbeat_rtt_last_ms": 0
}
})),
proxy_version: Some("1.0.0".to_string()),
})
.await
.expect("baseline heartbeat should apply")
.expect("node should exist");
repository
.apply_heartbeat(&ProxyNodeHeartbeatMutation {
node_id: registered.id.clone(),
@@ -1609,13 +1636,44 @@ VALUES ('node-1', 'registered', 'ok', 3)
)
.await
.expect("metrics should list");
assert_eq!(metrics.len(), 1);
assert_eq!(metrics[0].samples, 1);
assert_eq!(metrics[0].uptime_samples, 1);
assert_eq!(metrics[0].active_connections_max, 5);
assert_eq!(metrics[0].heartbeat_rtt_ms_sum, 33);
assert_eq!(metrics[0].connect_errors_delta, 4);
assert_eq!(metrics[0].ws_out_frames_delta, 6);
assert!(!metrics.is_empty());
assert_eq!(metrics.iter().map(|bucket| bucket.samples).sum::<i64>(), 2);
assert_eq!(
metrics
.iter()
.map(|bucket| bucket.uptime_samples)
.sum::<i64>(),
2
);
assert_eq!(
metrics
.iter()
.map(|bucket| bucket.active_connections_max)
.max()
.unwrap_or_default(),
5
);
assert_eq!(
metrics
.iter()
.map(|bucket| bucket.heartbeat_rtt_ms_sum)
.sum::<i64>(),
33
);
assert_eq!(
metrics
.iter()
.map(|bucket| bucket.connect_errors_delta)
.sum::<i64>(),
4
);
assert_eq!(
metrics
.iter()
.map(|bucket| bucket.ws_out_frames_delta)
.sum::<i64>(),
6
);
let fleet = repository
.list_proxy_fleet_metrics(
@@ -1626,9 +1684,15 @@ VALUES ('node-1', 'registered', 'ok', 3)
)
.await
.expect("fleet metrics should list");
assert_eq!(fleet.len(), 1);
assert_eq!(fleet[0].samples, 1);
assert_eq!(fleet[0].error_events_delta, 1);
assert!(!fleet.is_empty());
assert_eq!(fleet.iter().map(|bucket| bucket.samples).sum::<i64>(), 2);
assert_eq!(
fleet
.iter()
.map(|bucket| bucket.error_events_delta)
.sum::<i64>(),
1
);
let events = repository
.list_proxy_node_events_filtered(