快捷導(dǎo)航

GoZero實現(xiàn)數(shù)據(jù)庫MySQL單例模式連接的簡單示例

更新時間：2025年02月17日 08:24:51 作者：360_go_php

在 GoZero 框架中實現(xiàn)數(shù)據(jù)庫的單例連接可以通過以下步驟來完成,GoZero 使用 gorm 作為默認(rèn)的數(shù)據(jù)庫操作框架,接下來我會展示一個簡單的單例模式實現(xiàn),需要的朋友可以參考下

1. 定義數(shù)據(jù)庫連接的單例結(jié)構(gòu)

首先，你需要定義一個數(shù)據(jù)庫連接的結(jié)構(gòu)體，并在初始化時保證只有一個連接。

package database

import (
	"fmt"
	"log"
	"sync"

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

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

// InitDB 初始化數(shù)據(jù)庫連接
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 獲取數(shù)據(jù)庫連接
func GetDB() *gorm.DB {
	if db == nil {
		log.Fatal("Database is not initialized")
	}
	return db
}

2. 說明

once.Do：保證 InitDB 函數(shù)只會執(zhí)行一次，即使多次調(diào)用也只會初始化數(shù)據(jù)庫連接一次。
dsn：你需要傳入數(shù)據(jù)庫連接的 DSN（Data Source Name），通常是類似 username:password@tcp(host:port)/dbname?charset=utf8mb4&parseTime=True&loc=Local 這樣的格式。
GetDB()：返回數(shù)據(jù)庫的連接，如果數(shù)據(jù)庫未初始化，會觸發(fā)錯誤。

3. 如何使用

在你的業(yè)務(wù)代碼中，只需要調(diào)用 InitDB 初始化數(shù)據(jù)庫連接，然后使用 GetDB 獲取到數(shù)據(jù)庫連接進(jìn)行操作。

package main

import (
	"log"
	"myapp/database"
)

func main() {
	// 初始化數(shù)據(jù)庫連接
	database.InitDB("root:password@tcp(localhost:3306)/mydb?charset=utf8mb4&parseTime=True&loc=Local")
	
	// 獲取數(shù)據(jù)庫連接
	db := database.GetDB()

	// 執(zhí)行數(shù)據(jù)庫操作，比如查詢數(shù)據(jù)
	var user User
	if err := db.First(&user).Error; err != nil {
		log.Fatalf("Error querying user: %v", err)
	}

	// 輸出查詢結(jié)果
	log.Printf("User: %+v", user)
}

4. 完整示例

假設(shè)你的 User 結(jié)構(gòu)體是如下定義的：

package main

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

通過這樣的方式，你可以保證在應(yīng)用程序的整個生命周期內(nèi)，數(shù)據(jù)庫連接僅會被創(chuàng)建一次，從而實現(xiàn)了數(shù)據(jù)庫連接的單例模式。

總結(jié)

這種方式使用 sync.Once 來保證數(shù)據(jù)庫連接在應(yīng)用中只有一個實例，并提供 InitDB 和 GetDB 函數(shù)來初始化和獲取數(shù)據(jù)庫連接。這種方式非常適合于 GoZero 框架的項目，可以有效減少不必要的數(shù)據(jù)庫連接創(chuàng)建。

到此這篇關(guān)于GoZero實現(xiàn)數(shù)據(jù)庫MySQL單例模式連接的簡單示例的文章就介紹到這了,更多相關(guān)GoZero MySQL單例模式連接內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: