任务三

cmd/gen/main.go

@@ -0,0 +1,51 @@
+package main
+
+import (
+	"fmt"
+	"path/filepath"
+	"runtime"
+
+	"gorm.io/driver/mysql"
+	"gorm.io/gen"
+	"gorm.io/gorm"
+)
+
+func main() {
+	// 连接数据库
+	dsn := "root:Xu950329.@tcp(localhost:3306)/spa?charset=utf8mb4&parseTime=True&loc=Local"
+	db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{})
+	if err != nil {
+		panic(fmt.Errorf("connect db fail: %w", err))
+	}
+
+	// 获取项目根目录
+	_, filename, _, _ := runtime.Caller(0)
+	projectRoot := filepath.Dir(filepath.Dir(filepath.Dir(filename)))
+
+	// 创建gen实例
+	g := gen.NewGenerator(gen.Config{
+		OutPath:           filepath.Join(projectRoot, "internal/data/query"),
+		ModelPkgPath:      filepath.Join(projectRoot, "internal/data/model"),
+		Mode:              gen.WithDefaultQuery | gen.WithQueryInterface,
+		FieldNullable:     true,
+		FieldCoverable:    false,
+		FieldSignable:     false,
+		FieldWithIndexTag: true,
+		FieldWithTypeTag:  true,
+	})
+
+	g.UseDB(db)
+
+	// 字段名映射，避免与方法名冲突
+	fieldOpts := []gen.ModelOpt{
+		gen.FieldRename("table_name", "TblName"),
+		gen.FieldRename("table", "TblName"), // 避免与Table()方法冲突
+	}
+
+	// 生成所有表的model和query
+	allTables := g.GenerateAllTable(fieldOpts...)
+	g.ApplyBasic(allTables...)
+
+	// 执行生成
+	g.Execute()
+}