Skip to content

Commit 27f368e

Browse files
authored
fix: list entitlement by event subject (#3748)
1 parent 82cbd1c commit 27f368e

File tree

1 file changed

+13
-5
lines changed

1 file changed

+13
-5
lines changed

openmeter/entitlement/adapter/entitlement.go

Lines changed: 13 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -260,24 +260,32 @@ func (a *entitlementDBAdapter) ListEntitlementsAffectedByIngestEvents(ctx contex
260260

261261
result := make([]balanceworker.ListAffectedEntitlementsResponse, 0)
262262

263+
now := clock.Now()
264+
263265
for _, pair := range eventFilters {
264266
entities, err := repo.db.Entitlement.Query().
265267
Where(
266268
db_entitlement.Namespace(pair.Namespace),
267269
db_entitlement.HasCustomerWith(
268270
customerdb.Namespace(pair.Namespace),
269-
customerNotDeletedAt(clock.Now()),
270-
customerdb.HasSubjectsWith(
271-
customersubjectsdb.SubjectKey(pair.EventSubject),
272-
customersubjectsdb.DeletedAtIsNil(),
271+
customerNotDeletedAt(now),
272+
customerdb.Or(
273+
customerdb.Key(pair.EventSubject),
274+
customerdb.HasSubjectsWith(
275+
customersubjectsdb.SubjectKey(pair.EventSubject),
276+
customersubjectsdb.Or(
277+
customersubjectsdb.DeletedAtIsNil(),
278+
customersubjectsdb.DeletedAtGT(now),
279+
),
280+
),
273281
),
274282
),
275283
db_entitlement.HasFeatureWith(db_feature.MeterSlugIn(pair.MeterSlugs...)),
276284
).
277285
WithFeature().
278286
WithCustomer(
279287
func(q *db.CustomerQuery) {
280-
customeradapter.WithSubjects(q, clock.Now())
288+
customeradapter.WithSubjects(q, now)
281289
},
282290
).
283291
All(ctx)

0 commit comments

Comments
 (0)