59 lines
1.2 KiB
Go
59 lines
1.2 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
|
|
"gorm.io/driver/mysql"
|
|
"gorm.io/gorm"
|
|
"git.toowon.com/jimmy/go-common/migration"
|
|
)
|
|
|
|
func main() {
|
|
// 初始化数据库连接
|
|
dsn := "user:password@tcp(localhost:3306)/dbname?charset=utf8mb4&parseTime=True&loc=Local"
|
|
db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{})
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
// 创建迁移器
|
|
migrator := migration.NewMigrator(db)
|
|
|
|
// 添加迁移
|
|
migrator.AddMigration(migration.Migration{
|
|
Version: "20240101000001",
|
|
Description: "create_users_table",
|
|
Up: func(db *gorm.DB) error {
|
|
return db.Exec(`
|
|
CREATE TABLE IF NOT EXISTS users (
|
|
id INT PRIMARY KEY AUTO_INCREMENT,
|
|
name VARCHAR(255) NOT NULL,
|
|
email VARCHAR(255) UNIQUE NOT NULL,
|
|
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
|
|
)
|
|
`).Error
|
|
},
|
|
Down: func(db *gorm.DB) error {
|
|
return db.Exec("DROP TABLE IF EXISTS users").Error
|
|
},
|
|
})
|
|
|
|
// 执行迁移
|
|
if err := migrator.Up(); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
// 查看迁移状态
|
|
status, err := migrator.Status()
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
for _, s := range status {
|
|
fmt.Printf("Version: %s, Description: %s, Applied: %v\n",
|
|
s.Version, s.Description, s.Applied)
|
|
}
|
|
}
|
|
|