Merge pull request #1319 from openmeterio/fix/subject-id-resolving-error

fix: subject ID resolving error
openmeterio · Aug 8, 2024 · 626c54c · 626c54c
2 parents 2bd4286 + d45ea47
commit 626c54c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/internal/entitlement/balanceworker/entitlementhandler.go b/internal/entitlement/balanceworker/entitlementhandler.go
@@ -112,7 +112,7 @@ func (w *Worker) createSnapshotEvent(ctx context.Context, entitlementID Namespac
 
 	subjectID := ""
 	if w.opts.SubjectIDResolver != nil {
-		subjectID, err = w.opts.SubjectIDResolver.GetSubjectIDByKey(ctx, entitlementID.Namespace, entitlementID.ID)
+		subjectID, err = w.opts.SubjectIDResolver.GetSubjectIDByKey(ctx, entitlementID.Namespace, entitlement.SubjectKey)
 		if err != nil {
 			return nil, fmt.Errorf("failed to get subject ID: %w", err)
 		}

diff --git a/openmeter/entitlement/balanceworker/worker.go b/openmeter/entitlement/balanceworker/worker.go
@@ -6,6 +6,8 @@ type (
 	Worker           = balanceworker.Worker
 	WorkerOptions    = balanceworker.WorkerOptions
 	WorkerDLQOptions = balanceworker.WorkerDLQOptions
+
+	SubjectIDResolver = balanceworker.SubjectIDResolver
 )
 
 func New(opts WorkerOptions) (*Worker, error) {