feat: expose lookback time for recalculation

openmeterio · Aug 9, 2024 · 79ef225 · 79ef225
1 parent 09a4c82
commit 79ef225
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/internal/entitlement/balanceworker/recalculate.go b/internal/entitlement/balanceworker/recalculate.go
@@ -21,9 +21,9 @@ import (
 )
 
 const (
-	// defaultIncludeDeletedDuration is the default duration for which deleted entitlements are included in recalculation.
+	// DefaultIncludeDeletedDuration is the default duration for which deleted entitlements are included in recalculation.
 	// This ensures that the recent deleted snapshot events are also resent.
-	defaultIncludeDeletedDuration = 24 * time.Hour
+	DefaultIncludeDeletedDuration = 24 * time.Hour
 
 	defaultLRUCacheSize = 10_000
 )
@@ -66,7 +66,7 @@ func (r *Recalculator) Recalculate(ctx context.Context) error {
 		entitlement.ListEntitlementsParams{
 			Namespaces:          []string{r.opts.Namespace},
 			IncludeDeleted:      true,
-			IncludeDeletedAfter: time.Now().Add(-defaultIncludeDeletedDuration),
+			IncludeDeletedAfter: time.Now().Add(-DefaultIncludeDeletedDuration),
 		})
 	if err != nil {
 		return err

diff --git a/openmeter/entitlement/balanceworker/worker.go b/openmeter/entitlement/balanceworker/worker.go
@@ -18,6 +18,10 @@ type (
 	RecalculatorOptions = balanceworker.RecalculatorOptions
 )
 
+const (
+	DefaultIncludeDeletedDuration = balanceworker.DefaultIncludeDeletedDuration
+)
+
 func NewRecalculator(opts RecalculatorOptions) (*Recalculator, error) {
 	return balanceworker.NewRecalculator(opts)
 }