refactor(pkg/oomstore): modify FeatureList.GroupIDs()

oom-ai · Jan 14, 2022 · 3ce0794 · 3ce0794
1 parent aa2a757
commit 3ce0794
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/pkg/oomstore/types/feature.go b/pkg/oomstore/types/feature.go
@@ -96,8 +96,13 @@ func (l FeatureList) Find(find func(*Feature) bool) *Feature {
 }
 
 func (l FeatureList) GroupIDs() (ids []int) {
-	for _, f := range l {
-		ids = append(ids, f.GroupID)
+	groupIDMap := make(map[int]struct{})
+	for _, r := range l {
+		groupIDMap[r.GroupID] = struct{}{}
+	}
+	groupIDs := make([]int, 0, len(groupIDMap))
+	for id := range groupIDMap {
+		groupIDs = append(groupIDs, id)
 	}
-	return ids
+	return groupIDs
 }