GoZero实现数据库MySQL单例模式连接的简单示例

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
}

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)
}

package main

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