Merge pull request #1185 from openmeterio/feat/allow-searching-by-fea…

…ture-key feat: allow searching by feature key
openmeterio · Jul 10, 2024 · 2f05b2c · 2f05b2c
2 parents 6c684cc + f52b8f0
commit 2f05b2c
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/internal/entitlement/connector.go b/internal/entitlement/connector.go
@@ -23,6 +23,7 @@ type ListEntitlementsParams struct {
 	Namespaces     []string
 	SubjectKey     string
 	FeatureIDs     []string
+	FeatureKeys    []string
 	Limit          int
 	Offset         int
 	OrderBy        ListEntitlementsOrderBy

diff --git a/internal/entitlement/postgresadapter/entitlement.go b/internal/entitlement/postgresadapter/entitlement.go
@@ -162,6 +162,10 @@ func (a *entitlementDBAdapter) ListEntitlements(ctx context.Context, params enti
 		query = query.Where(db_entitlement.FeatureIDIn(params.FeatureIDs...))
 	}
 
+	if len(params.FeatureKeys) > 0 {
+		query = query.Where(db_entitlement.FeatureKeyIn(params.FeatureKeys...))
+	}
+
 	if !params.IncludeDeleted {
 		query = query.Where(db_entitlement.Or(db_entitlement.DeletedAtGT(clock.Now()), db_entitlement.DeletedAtIsNil()))
 	}