feat: add names filter (#5865)

kubeshop · Sep 25, 2024 · 894db7e · 894db7e
1 parent 4567a1a
commit 894db7e
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 0 deletions.
diff --git a/pkg/repository/testworkflow/filter.go b/pkg/repository/testworkflow/filter.go
@@ -8,6 +8,7 @@ import (
 
 type FilterImpl struct {
 	FName          string
+	FNames         []string
 	FLastNDays     int
 	FStartDate     *time.Time
 	FEndDate       *time.Time
@@ -30,6 +31,11 @@ func (f *FilterImpl) WithName(name string) *FilterImpl {
 	return f
 }
 
+func (f *FilterImpl) WithNames(names []string) *FilterImpl {
+	f.FNames = names
+	return f
+}
+
 func (f *FilterImpl) WithLastNDays(days int) *FilterImpl {
 	f.FLastNDays = days
 	return f
@@ -91,6 +97,14 @@ func (f FilterImpl) NameDefined() bool {
 	return f.FName != ""
 }
 
+func (f FilterImpl) Names() []string {
+	return f.FNames
+}
+
+func (f FilterImpl) NamesDefined() bool {
+	return len(f.FNames) > 0
+}
+
 func (f FilterImpl) LastNDaysDefined() bool {
 	return f.FLastNDays > 0
 }

diff --git a/pkg/repository/testworkflow/interface.go b/pkg/repository/testworkflow/interface.go
@@ -24,6 +24,8 @@ const PageDefaultLimit int = 100
 type Filter interface {
 	Name() string
 	NameDefined() bool
+	Names() []string
+	NamesDefined() bool
 	LastNDays() int
 	LastNDaysDefined() bool
 	StartDate() time.Time

diff --git a/pkg/repository/testworkflow/mongo.go b/pkg/repository/testworkflow/mongo.go
@@ -315,6 +315,10 @@ func composeQueryAndOpts(filter Filter) (bson.M, *options.FindOptions) {
 		query["workflow.name"] = filter.Name()
 	}
 
+	if filter.NamesDefined() {
+		query["workflow.name"] = bson.M{"$in": filter.Names()}
+	}
+
 	if filter.TextSearchDefined() {
 		query["name"] = bson.M{"$regex": primitive.Regex{Pattern: filter.TextSearch(), Options: "i"}}
 	}