EngineX-Hygon/sglang
5
0
Fork 0
Code Issues Pull Requests Actions 7 Projects Releases Wiki Activity

[router] grpc router bootstraps (#9759)

Browse Source
This commit is contained in:
Simo Lin
2025-08-28 12:07:06 -07:00
committed by GitHub
parent fce7ae33f8
commit 5343058875
14 changed files with 366 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
sgl-router/src/service_discovery.rs
View File

@@ -383,7 +383,7 @@ async fn handle_pod_event(
// Handle PD mode with specific pod types
let result = if pd_mode && pod_info.pod_type.is_some() {
// Need to import PDRouter type
use crate::routers::pd_router::PDRouter;
use crate::routers::http::pd_router::PDRouter;
// Try to downcast to PDRouter
if let Some(pd_router) = router.as_any().downcast_ref::<PDRouter>() {
@@ -453,7 +453,7 @@ async fn handle_pod_deletion(
// Handle PD mode removal
if pd_mode && pod_info.pod_type.is_some() {
use crate::routers::pd_router::PDRouter;
use crate::routers::http::pd_router::PDRouter;
// Try to downcast to PDRouter for PD-specific removal
if let Some(pd_router) = router.as_any().downcast_ref::<PDRouter>() {
@@ -581,7 +581,7 @@ mod tests {
async fn create_test_router() -> Arc<dyn RouterTrait> {
use crate::config::PolicyConfig;
use crate::policies::PolicyFactory;
use crate::routers::router::Router;
use crate::routers::http::router::Router;
let policy = PolicyFactory::create_from_config(&PolicyConfig::Random);
let router = Router::new(