Go 语言中通过依赖注入管理数据库连接配置的最佳实践-Golang-PHP中文网

Go 语言中通过依赖注入管理数据库连接配置的最佳实践

本文介绍如何在 go 中避免硬编码数据库连接字符串，通过配置集中化、连接池复用和依赖注入方式安全高效地管理 mysql 数据库连接。

本文介绍如何在 go 中避免硬编码数据库连接字符串，通过配置集中化、连接池复用和依赖注入方式安全高效地管理 mysql 数据库连接。

在 Go 应用开发中，将数据库连接参数（如驱动名、DSN）分散在多个函数中不仅违反单一职责原则，还带来维护风险——一旦密码或地址变更，需全局搜索替换，极易遗漏。真正的解决方案不是简单提取为全局常量，而是结合 Go 的并发模型与 database/sql 包的设计哲学，采用一次初始化 + 依赖注入的方式实现可维护、线程安全、资源可控的数据库访问。

✅ 推荐做法：单例连接池 + 显式依赖传递

database/sql.DB 本身是并发安全的连接池句柄，官方明确建议 只调用一次 sql.Open()，并长期复用该实例（而非每次请求新建连接）。因此，最佳实践是：

百度AI开放平台

百度AI开放平台是一款提供语音、图像、自然语言处理和行业 AI 能力的技术开放平台。

下载

在应用启动时初始化全局 *sql.DB 实例；
将该实例作为依赖注入到需要数据库操作的函数、方法或结构体中；
绝不频繁 Close() —— db.Close() 应仅在程序退出前调用一次。

// config/database.go
package config

import (
    "database/sql"
    "fmt"
    "log"
    _ "github.com/go-sql-driver/mysql" // MySQL 驱动
)

type DBConfig struct {
    Driver   string
    User     string
    Password string
    Host     string
    Port     string
    Name     string
}

var DefaultDBConfig = DBConfig{
    Driver:   "mysql",
    User:     "user",
    Password: "password",
    Host:     "127.0.0.1",
    Port:     "3306",
    Name:     "hello",
}

// NewDB 创建并验证数据库连接池
func NewDB(cfg DBConfig) (*sql.DB, error) {
    dsn := fmt.Sprintf("%s:%s@tcp(%s:%s)/%s", 
        cfg.User, cfg.Password, cfg.Host, cfg.Port, cfg.Name)

    db, err := sql.Open(cfg.Driver, dsn)
    if err != nil {
        return nil, fmt.Errorf("failed to open database: %w", err)
    }

    // 可选：设置连接池参数（提升稳定性）
    db.SetMaxOpenConns(25)
    db.SetMaxIdleConns(25)
    db.SetConnMaxLifetime(5 * 60 * time.Second) // 5分钟

    // 验证连接是否可用
    if err = db.Ping(); err != nil {
        db.Close()
        return nil, fmt.Errorf("failed to ping database: %w", err)
    }

    return db, nil
}

// main.go
package main

import (
    "log"
    "yourapp/config"
    "yourapp/repository"
)

func main() {
    db, err := config.NewDB(config.DefaultDBConfig)
    if err != nil {
        log.Fatal("Failed to initialize DB:", err)
    }
    defer db.Close() // 程序退出时关闭（仅此处一次）

    // 注入依赖：将 *sql.DB 传给业务逻辑层
    repo := repository.NewUserRepository(db)
    users, err := repo.ReadAll()
    if err != nil {
        log.Fatal("Read failed:", err)
    }
    log.Printf("Fetched %d users", len(users))
}

// repository/user_repository.go
package repository

import (
    "database/sql"
    "log"
)

type UserRepository struct {
    db *sql.DB
}

func NewUserRepository(db *sql.DB) *UserRepository {
    return &UserRepository{db: db}
}

func (r *UserRepository) ReadAll() ([]string, error) {
    rows, err := r.db.Query("SELECT name FROM users")
    if err != nil {
        return nil, err
    }
    defer rows.Close()

    var names []string
    for rows.Next() {
        var name string
        if err := rows.Scan(&name); err != nil {
            return nil, err
        }
        names = append(names, name)
    }
    return names, rows.Err()
}