Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
29 changes: 28 additions & 1 deletion core/integration/kafka/plugin.go
Original file line number Diff line number Diff line change
Expand Up @@ -31,6 +31,7 @@ import (
"encoding/json"
"errors"
"net/url"
"slices"
"strconv"
"strings"
"sync"
Expand Down Expand Up @@ -304,9 +305,36 @@ func (p *Plugin) GetRunningEnvList() []*kafkapb.EnvInfo {
for _, v := range p.envsInRunning {
array = append(array, v)
}

p.SortRunningEnvList(array)
p.SortDetectorsOfRunningEnvs(array)
return array
}

// OCTRL-932 we sort the active envs here are grafana cannot do it
// order is defined as:
// 1) runs with more than 2 detectors first
// 2) if there are more than 2 runs with more than 2 detectors, those with ITS should be first
func (p *Plugin) SortRunningEnvList(activeEnvs []*kafkapb.EnvInfo) {
slices.SortStableFunc(activeEnvs, func(a, b *kafkapb.EnvInfo) int {
if len(a.Detectors) >= 2 && len(b.Detectors) >= 2 {
if slices.Contains(a.Detectors, "ITS") {
return -1
}
if slices.Contains(b.Detectors, "ITS") {
return 1
}
}
return len(b.Detectors) - len(a.Detectors)
})
}

func (p *Plugin) SortDetectorsOfRunningEnvs(activeEnvs []*kafkapb.EnvInfo) {
for _, env := range activeEnvs {
slices.Sort(env.Detectors)
}
}

func (p *Plugin) produceMessage(message []byte, topic string, envId string, call string) {
log.WithField("call", call).
WithField("partition", envId).
Expand All @@ -317,7 +345,6 @@ func (p *Plugin) produceMessage(message []byte, topic string, envId string, call
Topic: topic,
Value: message,
})

if err != nil {
log.WithField("call", call).
WithField("partition", envId).
Expand Down
65 changes: 65 additions & 0 deletions core/integration/kafka/plugin_test.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,65 @@
/*
* === This file is part of ALICE O² ===
*
* Copyright 2024 CERN and copyright holders of ALICE O².
* Author: Teo Mrnjavac <teo.mrnjavac@cern.ch>
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*
* In applying this license CERN does not waive the privileges and
* immunities granted to it by virtue of its status as an
* Intergovernmental Organization or submit itself to any jurisdiction.
*/

package kafka

import (
"testing"

kafkapb "github.com/AliceO2Group/Control/core/integration/kafka/protos"
)

func TestSortActiveRuns(t *testing.T) {
p := Plugin{}
p.SortRunningEnvList(nil)

var envs []*kafkapb.EnvInfo
envs = append(envs,
&kafkapb.EnvInfo{EnvironmentId: "1", Detectors: []string{"first"}},
&kafkapb.EnvInfo{EnvironmentId: "2", Detectors: []string{"second"}},
&kafkapb.EnvInfo{EnvironmentId: "3", Detectors: []string{"ITS", "first"}},
&kafkapb.EnvInfo{EnvironmentId: "4", Detectors: []string{"first", "second", "third"}})

p.SortRunningEnvList(envs)

if len(envs) != 4 {
t.Error("wrong number of environments")
}

if envs[0].EnvironmentId != "3" {
t.Errorf("first should have been environment 3, but is %s with dets %v", envs[0].EnvironmentId, envs[0].Detectors)
}

if envs[1].EnvironmentId != "4" {
t.Errorf("second should have been environment 4, but is %s with dets %v", envs[1].EnvironmentId, envs[1].Detectors)
}

if envs[2].EnvironmentId != "1" {
t.Errorf("third should have been environment 1, but is %s with dets %v", envs[2].EnvironmentId, envs[2].Detectors)
}

if envs[3].EnvironmentId != "2" {
t.Errorf("fourth should have been environment 2, but is %s with dets %v", envs[3].EnvironmentId, envs[3].Detectors)
}
}