在 `GoZero` 框架中实现数据库的单例连接可以通过以下步骤来完成。GoZero 使用 `gorm` 作为默认的数据库操作框架,接下来我会展示一个简单的单例模式实现。

### 1. 定义数据库连接的单例结构

首先,你需要定义一个数据库连接的结构体,并在初始化时保证只有一个连接。```go

package database

import (
    "fmt"
    "log"
    "sync"

    "gorm.io/driver/mysql"
    "gorm.io/gorm"
)

var (
    db   *gorm.DB
    once sync.Once
)

// InitDB 初始化数据库连接
func InitDB(dsn string) {
    once.Do(func() {
        var err error
        db, err = gorm.Open(mysql.Open(dsn), &gorm.Config{})
        if err != nil {
            log.Fatalf("failed to connect database: %v", err)
        }
        fmt.Println("Database connected successfully")
    })
}

// GetDB 获取数据库连接
func GetDB() *gorm.DB {
    if db == nil {
        log.Fatal("Database is not initialized")
    }
    return db
}


```

### 2. 说明

- `once.Do`:保证 `InitDB` 函数只会执行一次,即使多次调用也只会初始化数据库连接一次。
- `dsn`:你需要传入数据库连接的 DSN(Data Source Name),通常是类似 `username:password@tcp(host:port)/dbname?charset=utf8mb4&parseTime=True&loc=Local` 这样的格式。
- `GetDB()`:返回数据库的连接,如果数据库未初始化,会触发错误。

### 3. 如何使用

在你的业务代码中,只需要调用 `InitDB` 初始化数据库连接,然后使用 `GetDB` 获取到数据库连接进行操作。```go

package main

import (
    "log"
    "myapp/database"
)

func main() {
    // 初始化数据库连接
    database.InitDB("root:password@tcp(localhost:3306)/mydb?charset=utf8mb4&parseTime=True&loc=Local")
    
    // 获取数据库连接
    db := database.GetDB()

    // 执行数据库操作,比如查询数据
    var user User
    if err := db.First(&user).Error; err != nil {
        log.Fatalf("Error querying user: %v", err)
    }

    // 输出查询结果
    log.Printf("User: %+v", user)
}


```

### 4. 完整示例

假设你的 `User` 结构体是如下定义的:```go

package main

type User struct {
    ID   uint   `gorm:"primaryKey"`
    Name string `gorm:"size:255"`
}


```

通过这样的方式,你可以保证在应用程序的整个生命周期内,数据库连接仅会被创建一次,从而实现了数据库连接的单例模式。

### 总结

这种方式使用 `sync.Once` 来保证数据库连接在应用中只有一个实例,并提供 `InitDB` 和 `GetDB` 函数来初始化和获取数据库连接。这种方式非常适合于 GoZero 框架的项目,可以有效减少不必要的数据库连接创建。

更多推荐