如何从 Golang 中的函数访问打开的数据库连接？

如何从 Golang 中的函数引用打开的数据库连接

在许多 Golang 应用程序中，需要与数据库进行交互。通常，数据库连接是在主函数中打开的。但是，在定义需要访问数据库的函数时，您可能会遇到引用打开的数据库连接的问题。

要解决此问题，有几种方法可用：

选项 1：全局数据库连接

一种解决方案是将数据库连接声明为全局变量，使其在整个程序中（包括在函数内）可访问。此方法很简单，但对于具有多个数据库连接的复杂应用程序来说可能并不理想。

选项 2：函数参数

另一个选项是将数据库连接作为函数的参数。这种方法提供了对使用哪个数据库连接的显式控制，并允许创建可以使用不同数据库的函数。

选项 3：函数方法

或者，您可以将该函数定义为保存数据库连接的结构中的方法。这种技术提供了一种更加面向对象的方法，并确保函数始终可以访问正确的数据库连接。

示例代码：

全局数据库连接:

<code class="go">var db *sql.DB

func main() {
    db = sql.Open("sqlite3", "./house.db")
    room := Room{}
    err := addRoom(room)
    if err != nil {
        // Error handling
    }
}

func addRoom(room Room) error {
    stmt, err := db.Prepare("INSERT INTO Rooms (Name, Size, WindowCount, WallDecorationType, Floor) VALUES(?, ?, ?, ?, ?)")
    if err != nil {
        return err
    }

    _, err = stmt.Exec(room.Name, room.Size, room.WindowCount, room.WallDecorationType, room.Floor)
    return err
}</code>

函数参数：

<code class="go">func addRow(db *sql.DB, row Room) error {
    stmt, err := db.Prepare("INSERT INTO Rooms (Name, Size, WindowCount, WallDecorationType, Floor) VALUES(?, ?, ?, ?, ?)")
    if err != nil {
        return err
    }

    _, err = stmt.Exec(row.Name, row.Size, row.WindowCount, row.WallDecorationType, row.Floor)
    return err
}</code>

函数方法：

<code class="go">type dbConn struct {
    db *sql.DB
}

func (conn dbConn) addRow(row Room) error {
    stmt, err := conn.db.Prepare("INSERT INTO Rooms (Name, Size, WindowCount, WallDecorationType, Floor) VALUES(?, ?, ?, ?, ?)")
    if err != nil {
        return err
    }

    _, err = stmt.Exec(row.Name, row.Size, row.WindowCount, row.WallDecorationType, row.Floor)
    return err
}</code>

最佳方法取决于您的应用程序的具体要求以及所需的灵活性和控制级别。这些选项提供了从 Golang 程序中的函数访问打开的数据库连接的各种方法。

以上是如何从 Golang 中的函数访问打开的数据库连接？的详细内容。更多信息请关注PHP中文网其他相关文章！