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

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

1. package database
3. import (
4.         "fmt"
5.         "sync"
6.         "time"
8.         "github.com/spf13/viper"
9.         "gorm.io/driver/mysql"
10.         "gorm.io/gen"
11.         "gorm.io/gorm"
12. )
14. var (
15.         instance *gorm.DB
16.         once     sync.Once
17. )
19. func Instance() *gorm.DB {
20.         once.Do(func() {
21.                 var err error
22.                 dsn := fmt.Sprintf("%s:%s@tcp(%s:%d)/%s?charset=utf8mb4&parseTime=True&loc=Local",
23.                         viper.GetString("database.user"),
24.                         viper.GetString("database.pass"),
25.                         viper.GetString("database.host"),
26.                         viper.GetInt("database.port"),
27.                         viper.GetString("database.name"))
28.                   instance, err = gorm.Open(mysql.Open(dsn), &gorm.Config{})
29.                   if err != nil {
30.                           panic("数据库连接失败")
31.                   }
32.         })
34.         if viper.GetBool("database.debug") {
35.                 instance = instance.Debug()
36.         }
38.         generator := gen.NewGenerator(gen.Config{
39.                 OutPath: "./app/dao",
40.                 Mode: gen.WithDefaultQuery | gen.WithQueryInterface,
41.         })
42.         generator.UseDB(instance)
43.     // 非常重要，不加不生成文件！
44.         generator.ApplyBasic(generator.GenerateAllTable()...)
45.         generator.Execute()
46.         db, _ := instance.DB()
47.         db.SetMaxIdleConns(10)
48.         db.SetMaxOpenConns(100)
49.         db.SetConnMaxLifetime(time.Hour)
50.         db.SetConnMaxIdleTime(30 * time.Minute)
51.         return instance
52. }

复制代码

需要注意以下几点:

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

来源：程序园用户自行投稿发布，如果侵权，请联系站长删除
免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！

yyds。多谢分享

感谢分享，学习下。

谢谢分享，试用一下

谢谢楼主提供！

过来提前占个楼

新版吗？好像是停更了吧。

yyds。多谢分享

鼓励转贴优秀软件安全工具和文档！

感谢分享

东西不错很实用谢谢分享