如何從Go lang中的數據庫中獲取數據

Question

package main

import (
    "encoding/json"
    "gorilla/mux"
    "log"
    "net/http"
    "database/sql"
    _ "go-sql-driver/mysql"
    )

func GetPeople(w http.ResponseWriter, r *http.Request, db *sql.DB) {
    json.NewEncoder(w).Encode(db.QueryRow("SELECT * from test.animal"))
}

func main() {

db, err := sql.Open("mysql", "root:mariadb@tcp(127.0.0.1:3306)/test")
if err != nil {
    panic(err.Error())
}
defer db.Close()
router := mux.NewRouter()

router.HandleFunc("/people", GetPeople).Methods("GET")

log.Fatal(http.ListenAndServe(":8000", router))
}

我需要使我的數據庫在GetPeople函數中被識別，但是不可能為該函數提供第三個參數（db * sql.DB）。

我該如何解決？ 謝謝

Answer 1

另一種方法是使用閉包，這可以防止全局使用變量：

func GetPeopleHandler(db *sql.DB) func (w http.ResponseWriter, r *http.Request) {
    return func (w http.ResponseWriter, r *http.Request) {
        json.NewEncoder(w).Encode(db.QueryRow("SELECT * from test.animal"))
    }
}

func main() {
    var err error
    db, err = sql.Open("mysql", "root:mariadb@tcp(127.0.0.1:3306)/test")
    if err != nil {
        panic(err.Error())
    }
    defer db.Close()
    router := mux.NewRouter()

    router.HandleFunc("/people", GetPeopleHandler(db)).Methods("GET")

    log.Fatal(http.ListenAndServe(":8000", router))
}

Answer 2

只需將您的db變量設置為全局變量，而不是在main()局部變量：

var db *sql.DB

func GetPeople(w http.ResponseWriter, r *http.Request) {
    json.NewEncoder(w).Encode(db.QueryRow("SELECT * from test.animal"))
}

func main() {
    var err error
    db, err = sql.Open("mysql", "root:mariadb@tcp(127.0.0.1:3306)/test")
    if err != nil {
        panic(err.Error())
    }
    defer db.Close()

    router := mux.NewRouter()
    router.HandleFunc("/people", GetPeople).Methods("GET")
    log.Fatal(http.ListenAndServe(":8000", router))
}

查看相關問題：

如何跨包中的文件使用全局變量？

Golang上每次全局數據庫連接與每次打開連接之間的性能差異