added trends and rankings methods in dispatchers

This commit is contained in:
gezimbll
2024-11-11 17:01:35 +01:00
committed by Dan Christian Bogos
parent 0854ef0bbb
commit fc6aff8484
7 changed files with 294 additions and 1 deletions

View File

@@ -243,6 +243,18 @@ func newDispatcherServiceMap(val *dispatchers.DispatcherService) (engine.IntServ
}
srvMap[srv.Name] = srv
srv, err = birpc.NewService(v1.NewDispatcherRankingSv1(val), utils.RankingSv1, true)
if err != nil {
return nil, err
}
srvMap[srv.Name] = srv
srv, err = birpc.NewService(v1.NewDispatcherTrendSv1(val), utils.TrendSv1, true)
if err != nil {
return nil, err
}
srvMap[srv.Name] = srv
srv, err = birpc.NewService(v1.NewDispatcherResponder(val),
utils.Responder, true)
if err != nil {

View File

@@ -103,7 +103,7 @@ func TestNewDispatcherServiceMap(t *testing.T) {
if srvMap == nil {
t.Fatal("Expected non-nil map, but got nil")
}
expectedLength := 20
expectedLength := 22
if len(srvMap) != expectedLength {
t.Fatalf("Expected map length %d, but got %d", expectedLength, len(srvMap))
}
@@ -128,6 +128,8 @@ func TestNewDispatcherServiceMap(t *testing.T) {
utils.SchedulerSv1,
utils.SessionSv1,
utils.StatSv1,
utils.RankingSv1,
utils.TrendSv1,
}
for _, name := range expectedServiceNames {
if _, ok := srvMap[name]; !ok {