mirror of
https://github.com/makayabou/asg-server.git
synced 2026-05-02 17:43:36 +02:00
34 lines
511 B
Go
34 lines
511 B
Go
package db
|
|
|
|
import (
|
|
"go.uber.org/fx"
|
|
"go.uber.org/zap"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type CommandMigrateParams struct {
|
|
fx.In
|
|
|
|
Logger *zap.Logger
|
|
DB *gorm.DB
|
|
Shut fx.Shutdowner
|
|
}
|
|
|
|
func AutoMigrate(params CommandMigrateParams) error {
|
|
err := params.DB.Transaction(func(tx *gorm.DB) error {
|
|
for _, v := range migrations {
|
|
if err := v(tx); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
params.Logger.Info("Migrations completed")
|
|
|
|
return params.Shut.Shutdown()
|
|
}
|
