zhangzengfei
2024-12-09 366e2ff546092d9be26411fb698b3ddd8e834a11
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
33
34
35
36
37
38
39
package db
 
import (
    "fmt"
 
    "sdkCompare/config"
 
    "basic.com/valib/logger.git"
    "gorm.io/driver/mysql"
    "gorm.io/gorm"
)
 
var db *gorm.DB
 
func ConnectDB() error {
    var err error
 
    dsn := fmt.Sprintf("%s:%s@tcp(%s)/%s?charset=utf8mb4",
        config.MainConf.Username,
        config.MainConf.Password,
        config.MainConf.MysqlAddr,
        config.MainConf.Database)
 
    db, err = gorm.Open(mysql.Open(dsn), &gorm.Config{
        // 禁用外键(指定外键时不会在mysql创建真实的外键约束)
        DisableForeignKeyConstraintWhenMigrating: true,
        //// 指定表前缀
        //NamingStrategy: schema.NamingStrategy{
        //    TablePrefix: config.Conf.Mysql.TablePrefix + "_",
        //},
    })
 
    if err != nil {
        logger.Error("mysql database open err: %s", err.Error())
        return err
    }
 
    return nil
}