Deduplicate some code

Dobefu · Nov 27, 2024 · 4c8061c · 4c8061c
1 parent 4ddea70
commit 4c8061c
Showing 1 changed file with 20 additions and 20 deletions.
diff --git a/cmd/migrate_db/main.go b/cmd/migrate_db/main.go
@@ -64,18 +64,9 @@ func down() error {
 		}
 
 		logger.Info("Running migration: %s", name)
-
-		queryBytes, err := content.ReadFile(fmt.Sprintf("migrations/%s", name))
+		err = runMigration(name, migrationIndex)
 
 		if err != nil {
-			_ = setMigrationState(migrationIndex, true)
-			return err
-		}
-
-		_, err = database.DB.Exec(string(queryBytes))
-
-		if err != nil {
-			_ = setMigrationState(migrationIndex, true)
 			return err
 		}
 	}
@@ -117,18 +108,9 @@ func up() error {
 		}
 
 		logger.Info("Running migration: %s", name)
-
-		queryBytes, err := content.ReadFile(fmt.Sprintf("migrations/%s", name))
+		err = runMigration(name, migrationIndex)
 
 		if err != nil {
-			_ = setMigrationState(migrationIndex, true)
-			return err
-		}
-
-		_, err = database.DB.Exec(string(queryBytes))
-
-		if err != nil {
-			_ = setMigrationState(migrationIndex, true)
 			return err
 		}
 	}
@@ -212,3 +194,21 @@ func setMigrationState(version int, dirty bool) error {
 
 	return nil
 }
+
+func runMigration(filename string, index int) error {
+	queryBytes, err := content.ReadFile(fmt.Sprintf("migrations/%s", filename))
+
+	if err != nil {
+		_ = setMigrationState(index, true)
+		return err
+	}
+
+	_, err = database.DB.Exec(string(queryBytes))
+
+	if err != nil {
+		_ = setMigrationState(index, true)
+		return err
+	}
+
+	return nil
+}