retun MigrateTool errors instead of logging them

infiniteloopcloud · Jul 10, 2024 · 37df4fc · 37df4fc
1 parent 9d0c6d8
commit 37df4fc
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/migration/migration.go b/migration/migration.go
@@ -100,6 +100,23 @@ func MigrationTool(conn string, down bool, opts Opts, embeds ...embed.FS) {
 	log.Info(ctx, "migration succeed")
 }
 
+func MigrationToolWithError(conn string, down bool, opts Opts, embeds ...embed.FS) error {
+	ctx := context.Background()
+	m, err := Get(ctx, conn, opts, embeds...)
+	if err != nil {
+		return err
+	}
+	if m == nil {
+		return errors.New("migration is nil")
+	}
+	if down {
+		err = m.Down()
+	} else {
+		err = m.Up()
+	}
+	return err
+}
+
 func PrepareCockroach(conn string) string {
 	return strings.Replace(conn, "postgres://", "cockroach://", 1)
 }