Gorm 中如何跨文件共享 DB 实例？

gorm 跨文件共享 db 实例

在 gorm 中，全局数据库连接通常是必要的，特别是当需要多个文件共享相同的数据库实例时。然而，当项目被分成多个文件时，如何获取 db 对象却成了一个问题。

解决方案

为了解决这个问题，有两种常用的方法：

1. 定义模型包

创建一个新的包，称为 model，其中包含数据库模型和变量 db，如下所示：

// model/user.go
package model

import "gorm.io/gorm"

var db *gorm.db

type user struct {
    id       int
    mobile    string
}

在 main 文件中，导入 model 包并初始化 db 变量：

package main

import (
    "fmt"

    "github.com/gin-gonic/gin"
    "gorm.io/driver/mysql"
    "gorm.io/gorm"

    "model"
)

func main() {
    dsn := "root:nexon8888@tcp(127.0.0.1:3306)/iapplus?charset=utf8mb4&parsetime=true&loc=local"
    db, err := gorm.open(mysql.open(dsn), &gorm.config{})
    if err != nil {
        fmt.println("数据库连接错误")
        return
    }
    model.db = db

    // 下面代码与上面示例相同
}

2. 定义常量包

另一种方法是创建常量包，其中导出 db 变量，如下所示：

// constants/db.go
package constants

import (
    "github.com/gin-gonic/gin"
    "gorm.io/driver/mysql"
    "gorm.io/gorm"
)

var db *gorm.db

func init() {
    dsn := "root:nexon8888@tcp(127.0.0.1:3306)/iapplus?charset=utf8mb4&parsetime=true&loc=local"
    db, err := gorm.open(mysql.open(dsn), &gorm.config{})
    if err != nil {
        log.fatal("数据库连接错误")
    }
    db = db
}

在 main 文件中，导入常量包并使用 exported db 变量：

package main

import (
    "fmt"

    "github.com/gin-gonic/gin"

    "constants"
)

func main() {
    // 下面代码与上面示例相同
}

通过使用这些方法，可以在 gorm 中轻松跨文件共享 db 实例，从而提高可维护性和可扩展性。

以上就是Gorm 中如何跨文件共享 DB 实例？的详细内容，更多请关注其它相关文章！