storse: add index creation to migration code

SiaFoundation · Nov 10, 2023 · 83bbfe1 · 83bbfe1
1 parent 6ea8a05
commit 83bbfe1
Showing 1 changed file with 17 additions and 4 deletions.
diff --git a/stores/migrations.go b/stores/migrations.go
@@ -1122,19 +1122,32 @@ func performMigration00024_slabIndices(txn *gorm.DB, logger *zap.SugaredLogger)
 
 func performMigration00025_contractState(txn *gorm.DB, logger *zap.SugaredLogger) error {
 	logger.Info("performing migration 00025_contractState")
+	// create column
 	if !txn.Migrator().HasColumn(&dbContract{}, "State") {
 		if err := txn.Migrator().AddColumn(&dbContract{}, "State"); err != nil {
 			return err
 		}
-		if err := txn.Model(&dbContract{}).Where("TRUE").Update("State", contractStateActive).Error; err != nil {
-			return err
-		}
 	}
 	if !txn.Migrator().HasColumn(&dbArchivedContract{}, "State") {
 		if err := txn.Migrator().AddColumn(&dbArchivedContract{}, "State"); err != nil {
 			return err
 		}
-		if err := txn.Model(&dbArchivedContract{}).Where("TRUE").Update("State", contractStateComplete).Error; err != nil {
+	}
+	// update column
+	if err := txn.Model(&dbContract{}).Where("TRUE").Update("State", contractStateActive).Error; err != nil {
+		return err
+	}
+	if err := txn.Model(&dbArchivedContract{}).Where("TRUE").Update("State", contractStateComplete).Error; err != nil {
+		return err
+	}
+	// create index
+	if !txn.Migrator().HasIndex(&dbContract{}, "State") {
+		if err := txn.Migrator().CreateIndex(&dbContract{}, "State"); err != nil {
+			return err
+		}
+	}
+	if !txn.Migrator().HasIndex(&dbArchivedContract{}, "State") {
+		if err := txn.Migrator().CreateIndex(&dbArchivedContract{}, "State"); err != nil {
 			return err
 		}
 	}