Super tiny remove sgl_router_active_workers (#14891)
This commit is contained in:
@@ -359,10 +359,6 @@ impl RouterMetrics {
|
|||||||
.increment(1);
|
.increment(1);
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn set_active_workers(count: usize) {
|
|
||||||
gauge!("sgl_router_active_workers").set(count as f64);
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn set_worker_health(worker_url: &str, healthy: bool) {
|
pub fn set_worker_health(worker_url: &str, healthy: bool) {
|
||||||
gauge!("sgl_router_worker_health",
|
gauge!("sgl_router_worker_health",
|
||||||
"worker" => worker_url.to_string()
|
"worker" => worker_url.to_string()
|
||||||
@@ -928,7 +924,6 @@ mod tests {
|
|||||||
RouterMetrics::record_request_error("/generate", "timeout");
|
RouterMetrics::record_request_error("/generate", "timeout");
|
||||||
RouterMetrics::record_retry("/generate");
|
RouterMetrics::record_retry("/generate");
|
||||||
|
|
||||||
RouterMetrics::set_active_workers(5);
|
|
||||||
RouterMetrics::set_worker_health("http://worker1", true);
|
RouterMetrics::set_worker_health("http://worker1", true);
|
||||||
RouterMetrics::set_worker_load("http://worker1", 10);
|
RouterMetrics::set_worker_load("http://worker1", 10);
|
||||||
RouterMetrics::record_processed_request("http://worker1");
|
RouterMetrics::record_processed_request("http://worker1");
|
||||||
@@ -1081,9 +1076,6 @@ mod tests {
|
|||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_extreme_metric_values() {
|
fn test_extreme_metric_values() {
|
||||||
RouterMetrics::set_active_workers(0);
|
|
||||||
RouterMetrics::set_active_workers(usize::MAX);
|
|
||||||
|
|
||||||
RouterMetrics::set_worker_load("worker", 0);
|
RouterMetrics::set_worker_load("worker", 0);
|
||||||
RouterMetrics::set_worker_load("worker", usize::MAX);
|
RouterMetrics::set_worker_load("worker", usize::MAX);
|
||||||
|
|
||||||
|
|||||||
@@ -54,16 +54,6 @@ pub struct Router {
|
|||||||
impl Router {
|
impl Router {
|
||||||
/// Create a new router with injected policy and client
|
/// Create a new router with injected policy and client
|
||||||
pub async fn new(ctx: &Arc<crate::app_context::AppContext>) -> Result<Self, String> {
|
pub async fn new(ctx: &Arc<crate::app_context::AppContext>) -> Result<Self, String> {
|
||||||
let workers = ctx.worker_registry.get_workers_filtered(
|
|
||||||
None, // any model
|
|
||||||
Some(WorkerType::Regular),
|
|
||||||
Some(ConnectionMode::Http),
|
|
||||||
None, // any runtime type
|
|
||||||
false, // include all workers
|
|
||||||
);
|
|
||||||
|
|
||||||
RouterMetrics::set_active_workers(workers.len());
|
|
||||||
|
|
||||||
Ok(Router {
|
Ok(Router {
|
||||||
worker_registry: ctx.worker_registry.clone(),
|
worker_registry: ctx.worker_registry.clone(),
|
||||||
policy_registry: ctx.policy_registry.clone(),
|
policy_registry: ctx.policy_registry.clone(),
|
||||||
|
|||||||
Reference in New Issue
Block a user