在 gin web 框架中管理数据库连接的方法主要有 3 种：单例模式：创建一个全局变量 db，并在 gin 中间件中将其设置为上下文中可用的键。连接池：使用支持连接池功能的驱动程序，配置连接池的最大空闲连接数和最大打开连接数。上下文键：将数据库连接作为上下文键传递，在 gin.context 中设置和获取 db 实例。

如何在 Go 框架中管理数据库连接

在开发 Go 应用程序时，管理数据库连接至关重要。本文将介绍在 Gin Web 框架中管理数据库连接的有效技术，并提供实战案例。

单例模式

单例模式是一种确保只有一个数据库连接实例的方法。这是 Gin 中管理数据库连接的最简单方法：

import (
    "context"
    "database/sql"

    "github.com/gin-gonic/gin"
)

var db *sql.DB

func main() {
    r := gin.Default()

    db, err := sql.Open("postgres", "user=root password=mypass host=localhost port=5432 dbname=mydb sslmode=disable")
    if err != nil {
        panic(err)
    }

    r.Use(func(c *gin.Context) {
        c.Set("db", db)
    })

    // ... 定义路由 ...
}

在上面代码中，我们创建了一个 db 全局变量，并在 Gin 中间件中将其设置为上下文中可用的键。现在，可以在路由处理程序中使用 c.MustGet("db").(*sql.DB) 获取 db 实例。

立即学习“go语言免费学习笔记（深入）”；

连接池

连接池是一个包含预先建立的数据库连接集合，这有助于提高应用程序的性能。go-sql-driver/mysql 等驱动程序支持连接池功能。

import (
    "context"
    "database/sql"

    "github.com/gin-gonic/gin"
    "github.com/go-sql-driver/mysql"
)

func main() {
    r := gin.Default()

    cfg := mysql.Config{
        User:   "root",
        Passwd: "mypass",
        Net:    "tcp",
        Addr:   "localhost:5432",
        DBName: "mydb",
    }
    db, err := sql.Open("mysql", cfg.FormatDSN())
    if err != nil {
        panic(err)
    }

    db.SetMaxIdleConns(5)
    db.SetMaxOpenConns(10)

    r.Use(func(c *gin.Context) {
        c.Set("db", db)
    })

    // ... 定义路由 ...
}

在这里，我们配置了一个最大空闲连接数为 5、最大打开连接数为 10 的连接池。

上下文键

上下文键用于在不同的中间件和处理程序之间传递值。一个常见的做法是将数据库连接作为上下文键传递。

import (
    "context"
    "database/sql"

    "github.com/gin-gonic/gin"
)

var dbKey = "db"

func main() {
    r := gin.Default()

    db, err := sql.Open("postgres", "user=root password=mypass host=localhost port=5432 dbname=mydb sslmode=disable")
    if err != nil {
        panic(err)
    }

    r.Use(func(c *gin.Context) {
        ctx := context.WithValue(c.Request.Context(), dbKey, db)
        c.Request = c.Request.WithContext(ctx)
    })

    // ... 定义路由 ...
}

在上面代码中，我们使用 context.WithValue 将 db 设置为上下文键。然后，我们可以在路由处理程序中使用 c.Request.Context().Value(dbKey) 获取 db 实例。