liujiandao
2023-11-16 d20acf38c36c11ee4428c3e74a17f5870dc61b51
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
package internal
 
import (
    "gorm.io/gorm"
    "gorm.io/gorm/logger"
    "gorm.io/gorm/schema"
    "moul.io/zapgorm2"
    "srm/global"
)
 
type DBBASE interface {
    GetLogMode() string
}
 
var Gorm = new(_gorm)
 
type _gorm struct{}
 
// Config gorm 自定义配置
// Author [SliverHorn](https://github.com/SliverHorn)
func (g *_gorm) Config(prefix string, singular bool) *gorm.Config {
    config := &gorm.Config{
        NamingStrategy: schema.NamingStrategy{
            TablePrefix:   prefix,
            SingularTable: singular,
        },
        DisableForeignKeyConstraintWhenMigrating: true,
    }
    dbLogger := zapgorm2.New(global.GVA_LOG).LogMode(logger.Info)
    config.Logger = dbLogger
    return config
}