migration.go 395 B

1234567891011121314151617181920212223
  1. package database
  2. import (
  3. "github.com/go-nunu/nunu-layout/internal/model"
  4. "github.com/go-nunu/nunu-layout/pkg/log"
  5. "gorm.io/gorm"
  6. )
  7. type Migrate struct {
  8. db *gorm.DB
  9. log *log.Logger
  10. }
  11. func NewMigrate(db *gorm.DB, log *log.Logger) *Migrate {
  12. return &Migrate{
  13. db: db,
  14. log: log,
  15. }
  16. }
  17. func (m *Migrate) Run() {
  18. m.db.AutoMigrate(&model.User{})
  19. m.log.Logger.Info("AutoMigrate end")
  20. }