如何在 Go Web 服务中正确传递数据库连接至 HTTP 处理函数-Golang-PHP中文网

如何在 Go Web 服务中正确传递数据库连接至 HTTP 处理函数

本文详解 Go 中因变量遮蔽（shadowing）导致全局 *sql.DB 未被初始化，从而在 HTTP handler 中触发 nil pointer panic 的根本原因，并提供符合 Go 工程实践的修复方案与项目结构建议。

本文详解 go 中因变量遮蔽（shadowing）导致全局 `*sql.db` 未被初始化，从而在 http handler 中触发 nil pointer panic 的根本原因，并提供符合 go 工程实践的修复方案与项目结构建议。

问题核心在于：局部变量遮蔽（variable shadowing）破坏了全局数据库连接的初始化。

在你的 main() 函数中，这一行代码：

db, err := sql.Open("mysql", DefaultDSN)

使用短变量声明 := 创建了一个新的局部变量 db，而非为包级全局变量 db *sql.DB 赋值。由于该局部变量作用域仅限于 main 函数内部，一旦 main 执行完毕，该局部 db 即被销毁，而包级全局 db 仍保持其零值 nil。当 booksIndex 尝试调用 db.Query(...) 时，实际操作的是 nil 指针，引发运行时 panic：

panic: runtime error: invalid memory address or nil pointer dereference

✅ 正确写法：使用赋值而非短声明

将 main() 中的初始化改为显式赋值（使用 =），确保修改的是全局变量：

func main() {
    flag.Parse()

    var err error
    db, err = sql.Open("mysql", dsn) // ← 关键修正：去掉冒号，使用 = 赋值
    if err != nil {
        log.Fatal("failed to open database:", err)
    }
    if err = db.Ping(); err != nil {
        log.Fatal("failed to ping database:", err)
    }

    // 设置连接池参数（推荐）
    db.SetMaxOpenConns(20)
    db.SetMaxIdleConns(10)

    http.HandleFunc("/", homepage)
    http.HandleFunc("/books", booksIndex)

    log.Printf("HTTP server started on %s", httpAddr)
    log.Fatal(http.ListenAndServe(httpAddr, nil))
}

? 注意：dsn 参数已从硬编码 DefaultDSN 改为使用命令行解析后的 dsn 变量，使配置更灵活。

Golang Uber Dig
使用 uber-go/dig 在 Golang 中实现依赖注入——基于反射容器，提供 Provide/Invoke、dig.In/dig.Out 参数与结果对象，支持命名依赖等

下载

?️ 进阶建议：避免全局状态，采用依赖注入

虽然修复后程序可运行，但依赖全局变量会降低可测试性与可维护性。更符合 Go 工程规范的做法是将 *sql.DB 显式注入到 handler 中：

// 定义带 DB 依赖的 handler 结构体
type BookHandler struct {
    db *sql.DB
}

func (h *BookHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
    if r.Method != "GET" {
        http.Error(w, http.StatusText(405), 405)
        return
    }

    rows, err := h.db.Query("SELECT id, title, author FROM books")
    if err != nil {
        http.Error(w, "database query failed", http.StatusInternalServerError)
        log.Printf("DB query error: %v", err)
        return
    }
    defer rows.Close()

    var books []Book // 使用导出字段便于 JSON 序列化
    for rows.Next() {
        var bk Book
        if err := rows.Scan(&bk.ID, &bk.Title, &bk.Author); err != nil {
            http.Error(w, "scan error", http.StatusInternalServerError)
            log.Printf("Scan error: %v", err)
            return
        }
        books = append(books, bk)
    }
    if err := rows.Err(); err != nil {
        http.Error(w, "row iteration error", http.StatusInternalServerError)
        log.Printf("Rows error: %v", err)
        return
    }

    for _, bk := range books {
        fmt.Fprintf(w, "%d, %s, %s\n", bk.ID, bk.Title, bk.Author)
    }
}

// 在 main 中注册 handler
func main() {
    flag.Parse()

    db, err := sql.Open("mysql", dsn)
    if err != nil {
        log.Fatal("DB open failed:", err)
    }
    if err = db.Ping(); err != nil {
        log.Fatal("DB ping failed:", err)
    }
    defer db.Close() // 程序退出前关闭连接池

    bookHandler := &BookHandler{db: db}
    http.Handle("/books", bookHandler)
    http.HandleFunc("/", homepage)

    log.Printf("Server listening on %s", httpAddr)
    log.Fatal(http.ListenAndServe(httpAddr, nil))
}

? 推荐项目结构（面向生产）

为提升可维护性，建议按职责拆分文件：

librarian/
├── main.go               # 入口，配置解析、DB 初始化、路由注册
├── handler/
│   ├── book_handler.go   # /books 相关逻辑
│   └── home_handler.go   # / 相关逻辑
├── model/
│   └── book.go           # Book 结构体与数据库操作方法（如 FindAll）
├── db/
│   └── db.go             # DB 初始化与连接池配置封装
└── go.mod

同时，务必在程序生命周期结束前调用 db.Close() 释放资源；对高并发场景，合理设置 SetMaxOpenConns 和 SetMaxIdleConns 避免连接耗尽。

✅ 总结：