[model-gateway] add back router worker health metric and fix init state (#15622)
This commit is contained in:
@@ -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(
|
||||
|
||||
Reference in New Issue
Block a user