是否需要调用 db.Close()？

Question

到目前为止，Go 最难的部分是理解如何组织代码。表面上看起来非常简单，但每次我尝试做任何事情时，我都会遇到循环导入或“exported func Start 返回未导出的类型 models.dbStore，使用起来很烦人”之类的东西。

使用以下代码如何调用db.Close() 或者我真的不明白我应该如何为我的模型提供数据库。这是我得到的：

App.go

package app

import (
    "database/sql"

    // Comment
    _ "github.com/mattn/go-sqlite3"
)

var (
    // DB The database connection
    db *sql.DB
)

// Setup Sets up the many many app settings
func Setup() {
    d, err := sql.Open("sqlite3", "./foo.db")
    if err != nil {
        panic(err)
    }

    // TODO: How does the DB get closed?
    // defer db.Close()
    db = d
}

// GetDB Returns a reference to the database
func GetDB() *sql.DB {
    return db
}

Users.go

package models

import (
    "github.com/proj/org/app"
)

// User struct
type User struct {
    ID int
}

// CreateUser Creates a user
func (u *User) CreateUser() (int64, error) {

    // For the sake of brevity just make sure you can
    // "connect" to the database
    if err := app.GetDB().Ping(); err != nil {
        panic(err)
    }

    return 1234, nil
}

main.go

package main

import (
    "fmt"
    "net/http"

    _ "github.com/mattn/go-sqlite3"
    "github.com/proj/org/app"
    "github.com/proj/org/models"
)

func homeHandler(w http.ResponseWriter, r *http.Request) {
    fmt.Fprintf(w, "You are home")
}

func subscribeHandler(w http.ResponseWriter, r *http.Request) {
    fmt.Fprintf(w, "Subscribing...")
    u := models.User{}

    u.CreateUser()

}

func main() {
    fmt.Println("Running")

    app.Setup()

    http.HandleFunc("/", homeHandler)
    http.HandleFunc("/subscribe", subscribeHandler)

    err := http.ListenAndServe(":9090", nil)
    if err != nil {
        panic(err)
    }
}

我想过做一个app.Shutdown()，但这不适用于我最常见的用例CTRL-C。如果我不关闭数据库，似乎数据库连接只会增长......只是试图理解。

Answer 1

不需要关闭数据库。

返回的数据库对于多个 goroutine 并发使用是安全的，并且 维护自己的空闲连接池。因此，打开功能 应该只调用一次。很少需要关闭数据库。

发件人：https://golang.org/pkg/database/sql/#Open

当您的程序退出时，任何打开的连接都将关闭，它不会在以太中的某处保持打开状态，等待您的程序重新启动，因此不要担心当您 CTRL-C 应用程序时连接“增长”。

---

但是，如果您仍然想关闭它，那么您可以像使用 GetDB 一样导出 CloseDB 函数。

App.go

// ...

func CloseDB() error {
    return db.Close()
}

main.go

// ...

func main() {
    // ...

    app.Setup()
    defer app.CloseDB()

    // ...

}