[model-gateway] add back router worker health metric and fix init state (#15622)

This commit is contained in:
fzyzcjy
2025-12-23 18:50:38 -08:00
committed by GitHub
parent 4d64f15086
commit 99d3bcdfed
3 changed files with 19 additions and 2 deletions
+5 -2
View File
@@ -9,7 +9,7 @@ use super::{
WorkerType,
},
};
use crate::routers::grpc::client::GrpcClient;
use crate::{observability::metrics::Metrics, routers::grpc::client::GrpcClient};
/// Builder for creating BasicWorker instances with fluent API
pub struct BasicWorkerBuilder {
@@ -187,11 +187,14 @@ impl BasicWorkerBuilder {
None => OnceCell::new(),
});
let healthy = true;
Metrics::set_worker_health(&self.url, healthy);
BasicWorker {
metadata,
load_counter: Arc::new(AtomicUsize::new(0)),
processed_counter: Arc::new(AtomicUsize::new(0)),
healthy: Arc::new(AtomicBool::new(true)),
healthy: Arc::new(AtomicBool::new(healthy)),
consecutive_failures: Arc::new(AtomicUsize::new(0)),
consecutive_successes: Arc::new(AtomicUsize::new(0)),
circuit_breaker: CircuitBreaker::with_config_and_label(