fix migration issues

templates/project/app/srv/migrate/migrate.go.tpl

@@ -14,6 +14,15 @@ import (
 	"go.ipao.vip/atom"
 	"go.ipao.vip/atom/container"
 	"go.uber.org/dig"
+
+	"github.com/pressly/goose/v3"
+	"github.com/riverqueue/river/riverdriver/riverdatabasesql"
+	"github.com/riverqueue/river/rivermigrate"
+	log "github.com/sirupsen/logrus"
+	"github.com/spf13/cobra"
+	"go.ipao.vip/atom"
+	"go.ipao.vip/atom/container"
+	"go.uber.org/dig"
 )
 
 func defaultProviders() container.Providers {
@@ -54,7 +63,33 @@ func Serve(cmd *cobra.Command, args []string) error {
 
 		goose.SetBaseFS(database.MigrationFS)
 		goose.SetTableName("migrations")
+		goose.AddNamedMigrationNoTxContext("0001_river_job.go", RiverUp, RiverDown)
 
 		return goose.RunContext(context.Background(), action, svc.DB, "migrations", args...)
 	})
 }
+
+func RiverUp(ctx context.Context, db *sql.DB) error {
+	migrator, err := rivermigrate.New(riverdatabasesql.New(db), nil)
+	if err != nil {
+		return err
+	}
+
+	// Migrate up. An empty MigrateOpts will migrate all the way up, but
+	// best practice is to specify a specific target version.
+	_, err = migrator.Migrate(ctx, rivermigrate.DirectionUp, &rivermigrate.MigrateOpts{})
+	return err
+}
+
+func RiverDown(ctx context.Context, db *sql.DB) error {
+	migrator, err := rivermigrate.New(riverdatabasesql.New(db), nil)
+	if err != nil {
+		return err
+	}
+
+	// TargetVersion -1 removes River's schema completely.
+	_, err = migrator.Migrate(ctx, rivermigrate.DirectionDown, &rivermigrate.MigrateOpts{
+		TargetVersion: -1,
+	})
+	return err
+}