gozero实现数据库MySQL单例模式连接
在 `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 框架的项目，可以有效减少不必要的数据库连接创建。