Ensure truncate command is executed on stage

RedHatInsights · Aug 12, 2024 · 4e7722a · 4e7722a
1 parent f4245a6
commit 4e7722a
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/cmd/migrate/manual/000_truncate_packages.go b/cmd/migrate/manual/000_truncate_packages.go
@@ -1,17 +1,19 @@
 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"
 )
 
 func init() {
-	registerMigration("truncate packages if enabled (000)", truncatePackages)
+	//registerMigration("truncate packages if enabled (000)", truncatePackages)
 }
 
 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