Fix: Skip random number generation for single endpoint

cloudwego · Oct 21, 2024 · 869b83b · 869b83b
1 parent 5d7198f
commit 869b83b
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/monolake-services/src/http/handlers/route.rs b/monolake-services/src/http/handlers/route.rs
@@ -178,6 +178,10 @@ where
             Ok(route) => {
                 let route = route.value;
                 tracing::info!("the route id: {}", route.id);
+                if route.upstreams.len() == 1 {
+                    rewrite_request(&mut request, &route.upstreams[0]);
+                    return self.inner.handle(request, ctx).await;
+                }
                 use rand::seq::SliceRandom;
                 let upstream = route
                     .upstreams