在 `GoZero` 框架中实现数据库的单例连接可以通过以下步骤来完成。GoZero 使用 `gorm` 作为默认的数据库操作框架，接下来我会展示一个简单的单例模式实现。
### 1. 定义数据库连接的单例结构
首先，你需要定义一个数据库连接的结构体，并在初始化时保证只有一个连接。```go
package databaseimport ("fmt""log""sync""gorm.io/driver/mysql""gorm.io/gorm"
)var (db *gorm.DBonce sync.Once
)// InitDB 初始化数据库连接
func InitDB(dsn string) {once.Do(func() {var err errordb, 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 mainimport ("log""myapp/database"
)func main() {// 初始化数据库连接database.InitDB("root:password@tcp(localhost:3306)/mydb?charset=utf8mb4&parseTime=True&loc=Local")// 获取数据库连接db := database.GetDB()// 执行数据库操作，比如查询数据var user Userif err := db.First(&user).Error; err != nil {log.Fatalf("Error querying user: %v", err)}// 输出查询结果log.Printf("User: %+v", user)
}
```
### 4. 完整示例
假设你的 `User` 结构体是如下定义的：```go
package maintype User struct {ID uint `gorm:"primaryKey"`Name string `gorm:"size:255"`
}
```
通过这样的方式，你可以保证在应用程序的整个生命周期内，数据库连接仅会被创建一次，从而实现了数据库连接的单例模式。
### 总结
这种方式使用 `sync.Once` 来保证数据库连接在应用中只有一个实例，并提供 `InitDB` 和 `GetDB` 函数来初始化和获取数据库连接。这种方式非常适合于 GoZero 框架的项目，可以有效减少不必要的数据库连接创建。
