当前位置：首页 > news >正文

gorm使用gen自动生成模型和查询文件

news 2026/5/12 19:43:23

项目使用的gin，整合的gorm，可以利用gen工具快速生成模型和查询文件

package databaseimport ("fmt""sync""time""github.com/spf13/viper""gorm.io/driver/mysql""gorm.io/gen""gorm.io/gorm"
)var (instance *gorm.DBonce     sync.Once
)func Instance() *gorm.DB {once.Do(func() {var err errordsn := fmt.Sprintf("%s:%s@tcp(%s:%d)/%s?charset=utf8mb4&parseTime=True&loc=Local",viper.GetString("database.user"),viper.GetString("database.pass"),viper.GetString("database.host"),viper.GetInt("database.port"),viper.GetString("database.name"))instance, err = gorm.Open(mysql.Open(dsn), &gorm.Config{})if err != nil {panic("数据库连接失败")}})if viper.GetBool("database.debug") {instance = instance.Debug()}generator := gen.NewGenerator(gen.Config{OutPath: "./app/dao",Mode: gen.WithDefaultQuery | gen.WithQueryInterface,})generator.UseDB(instance)// 非常重要，不加不生成文件！generator.ApplyBasic(generator.GenerateAllTable()...)generator.Execute()db, _ := instance.DB()db.SetMaxIdleConns(10)db.SetMaxOpenConns(100)db.SetConnMaxLifetime(time.Hour)db.SetConnMaxIdleTime(30 * time.Minute)return instance
}

需要注意以下几点:

如果你想定义model目录的路径，目前只支持绝对路径（如写成./app/models这样的相对路径是不行的）或者通过设置OutPath和ModelPkgPath，可以实现自定义模型路径
model生成位置:取OutPath路径中的目录(如OutPath是./app/dao，则目录是./app)，再拼接上ModelPkgPath(如不设置ModelPkgPath，默认就叫model)，拼接后最终model会生成在./app/model中
generator.ApplyBasic(generator.GenerateAllTable()...)必须有这句，否则不生成任何model和query文件