初始版本,工具基础类

examples/migration_example.go

@@ -0,0 +1,58 @@
+package main
+
+import (
+	"fmt"
+	"log"
+
+	"gorm.io/driver/mysql"
+	"gorm.io/gorm"
+	"github.com/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)
+	}
+}
+