Ensure truncate command is executed on stage

RedHatInsights · Aug 12, 2024 · c397b7b · c397b7b
1 parent f4245a6
commit c397b7b
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/cmd/migrate/manual/000_truncate_packages.go b/cmd/migrate/manual/000_truncate_packages.go
@@ -1,8 +1,10 @@
 package manual
 
 import (
+	"strings"
+
+	"github.com/redhatinsights/edge-api/config"
 	"github.com/redhatinsights/edge-api/pkg/db"
-	feature "github.com/redhatinsights/edge-api/unleash/features"
 	log "github.com/sirupsen/logrus"
 )
 
@@ -11,7 +13,7 @@ func init() {
 }
 
 func truncatePackages() error {
-	if feature.TruncatePackages.IsEnabled() {
+	if strings.Contains(config.Get().EdgeAPIBaseURL, "stage") {
 		log.Info("Truncating packages table ...")
 		if err := db.DB.Exec("TRUNCATE TABLE commit_installed_packages, installed_packages").Error; err != nil {
 			return err