go gorm gen 根据model，生成数据库表和查询

下载
go get gorm.io/gen

需要自己定义model

type UserDetail struct {
    ID         int64     `gorm:"column:id;type:bigint;primaryKey;autoIncrement:true" json:"id"`
    UserName   string    `gorm:"column:user_name;type:varchar(255);not null" json:"user_name"`
    NickName   string    `gorm:"column:nick_name;type:varchar(255);not null" json:"nick_name"`
    CreateTime time.Time `gorm:"column:create_time;type:datetime;not null;default:CURRENT_TIMESTAMP" json:"create_time"`
    UpdateTime time.Time `gorm:"column:update_time;type:datetime;not null;default:CURRENT_TIMESTAMP" json:"update_time"`
}

// TableName 表名
func (*UserDetail) TableName() string {
    return "tb_user"
}

定义查询

type UserMethod interface {
    // FindByUserName where("user_name=@userName")
    FindByUserName(userName string) (gen.T, error)

    // GetById where("id=@id")
    GetById(id int32) (gen.T, error)
}

main

func main() {
    db := manager.NewMySQL()
    //字段类型的修改需要单独处理，先删除然后再加上
    //db.Migrator().DropColumn(&UserDetail{}, "user_name")
    //db.Migrator().AddColumn(&model.UserDetail{}, "user_name")
    err := db.AutoMigrate(&model.UserDetail{}) // 自动同步数据库表

    if err != nil {
        fmt.Println(err)
        return
    }

    g := gen.NewGenerator(gen.Config{
        OutPath: "./dal/query",
        Mode:    gen.WithDefaultQuery | gen.WithQueryInterface,
    })

    g.UseDB(db)
    g.ApplyInterface(func(method method.UserMethod) {}, model.UserDetail{})
    g.Execute()

}