Pass a list of IDs for exact querying (#1288)

* feat: pass a list of IDs for exact querying * feat: support both id and key
openmeterio · Aug 2, 2024 · 4ded17e · 4ded17e
1 parent 782f142
commit 4ded17e
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/internal/productcatalog/feature_connector.go b/internal/productcatalog/feature_connector.go
@@ -58,6 +58,7 @@ func (f FeatureOrderBy) StrValues() []string {
 }
 
 type ListFeaturesParams struct {
+	IDsOrKeys       []string
 	Namespace       string
 	MeterSlugs      []string
 	IncludeArchived bool

diff --git a/internal/productcatalog/postgresadapter/feature.go b/internal/productcatalog/postgresadapter/feature.go
@@ -112,6 +112,12 @@ func (c *featureDBAdapter) ListFeatures(ctx context.Context, params productcatal
 		query.Where(db_feature.MeterSlugIn(params.MeterSlugs...))
 	}
 
+	if len(params.IDsOrKeys) > 0 {
+		for _, idOrKey := range params.IDsOrKeys {
+			query.Where(db_feature.Or(db_feature.Key(idOrKey), db_feature.ID(idOrKey)))
+		}
+	}
+
 	if !params.IncludeArchived {
 		query = query.Where(db_feature.Or(db_feature.ArchivedAtIsNil(), db_feature.ArchivedAtGT(clock.Now())))
 	}