fix(wire): entitlement disabled

openmeterio · Dec 13, 2024 · d4d2c93 · d4d2c93
1 parent 328b5d7
commit d4d2c93
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/app/common/entitlement.go b/app/common/entitlement.go
@@ -33,10 +33,6 @@ func NewEntitlementRegistry(
 		return nil
 	}
 
-	if !entitlementConfig.Enabled {
-		return nil
-	}
-
 	return registrybuilder.GetEntitlementRegistry(registrybuilder.EntitlementOptions{
 		DatabaseClient:     db,
 		StreamingConnector: streamingConnector,

diff --git a/app/common/productcatalog.go b/app/common/productcatalog.go
@@ -41,7 +41,12 @@ func NewFeatureConnector(logger *slog.Logger, db *entdb.Client, meterRepo meter.
 	return feature.NewFeatureConnector(featureRepo, meterRepo)
 }
 
-func NewPlanService(logger *slog.Logger, db *entdb.Client, productCatalogConf config.ProductCatalogConfiguration, featureConnector feature.FeatureConnector) (plan.Service, error) {
+func NewPlanService(
+	logger *slog.Logger,
+	db *entdb.Client,
+	productCatalogConf config.ProductCatalogConfiguration,
+	featureConnector feature.FeatureConnector,
+) (plan.Service, error) {
 	// TODO: remove this check after enabled by default
 	if db == nil {
 		return nil, nil