feat: add app migrate command

This commit is contained in:
rogeecn
2025-02-28 12:00:13 +08:00
parent ad7f1b2c47
commit 94dd15b4e9
2 changed files with 62 additions and 0 deletions

View File

@@ -0,0 +1,60 @@
package migrate
import (
"context"
"database/sql"
"{{.ModuleName}}/app/service"
"{{.ModuleName}}/database"
"{{.ModuleName}}/providers/postgres"
"github.com/pressly/goose/v3"
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 {
return service.Default(container.Providers{
postgres.DefaultProvider(),
}...)
}
func Command() atom.Option {
return atom.Command(
atom.Name("migrate"),
atom.Short("run migrations"),
atom.RunE(Serve),
atom.Providers(defaultProviders()),
atom.Example("migrate [up|up-by-one|up-to|create|down|down-to|fix|redo|reset|status|version]"),
)
}
type Service struct {
dig.In
DB *sql.DB
}
// migrate
func Serve(cmd *cobra.Command, args []string) error {
return container.Container.Invoke(func(ctx context.Context, svc Service) error {
if len(args) == 0 {
args = append(args, "up")
}
if args[0] == "create" {
return nil
}
action, args := args[0], args[1:]
log.Infof("migration action: %s args: %+v", action, args)
goose.SetBaseFS(database.MigrationFS)
goose.SetTableName("migrations")
return goose.RunContext(context.Background(), action, svc.DB, "migrations", args...)
})
}

View File

@@ -2,6 +2,7 @@ package main
import (
"{{.ModuleName}}/app/service/http"
"{{.ModuleName}}/app/service/migrate"
log "github.com/sirupsen/logrus"
"go.ipao.vip/atom"
@@ -25,6 +26,7 @@ func main() {
opts := []atom.Option{
atom.Name("{{ .ProjectName }}"),
http.Command(),
migrate.Command(),
}
if err := atom.Serve(opts...); err != nil {