如何在另一个函数中使用已连接到 mongodb 和集合的句柄？

来到golang学习网的大家，相信都是编程学习爱好者，希望在这里学习Golang相关编程知识。下面本篇文章就来带大家聊聊《如何在另一个函数中使用已连接到 mongodb 和集合的句柄？》，介绍一下，希望对大家的知识积累有所帮助，助力实战开发！

我在 mongo 上设置了一个数据库，其中包含一些我需要通过端点的 url 参数查询的数据。为了使用该库，我定义了一些句柄，并在单独的 setup() 函数中完成了数据库连接的整个设置，但我无法在其外部使用我需要的句柄。

package main

import (
    "context"
    "encoding/json"
    "fmt"
    "log"
    "net/http"
    "time"

    "github.com/gorilla/mux"
    "go.mongodb.org/mongo-driver/mongo"
    "go.mongodb.org/mongo-driver/mongo/options"
    "go.mongodb.org/mongo-driver/mongo/readpref"
)

func setup() {
    clientoptions := options.client().
        applyuri("mongodb+srv://:@cluster0.um5qb.mongodb.net/?retrywrites=true&w=majority")
    ctx, cancel := context.withtimeout(context.background(), 10*time.second)
    defer cancel()
    client, err := mongo.connect(ctx, clientoptions)
    if err != nil {
        log.fatal(err)
    }
    err = client.ping(ctx, readpref.primary())
    if err != nil {
        log.fatal(err)
    }
    defer client.disconnect(ctx)
    // db := client.database("cities-nighthack")
    // cities := db.collection("city")

}

// model for user endpoint
type user struct {
    email string `json:"email"`
}

// fake db to temp store users
var users []user

// checks if json is empty or not
func (u *user) isempty() bool {
    return u.email == ""
}

type app struct {
    mongo *mongo.client
}

func main() {
    setup()
    r := mux.newrouter()
    r.handlefunc("/user", createuser).methods("post")
    // r.handlefunc("/suggest?city_name={city}", searchcity).methods("get")

    fmt.println("server running at port 8080")
    log.fatal(http.listenandserve(":8080", r))

}

func createuser(w http.responsewriter, r *http.request) {
    w.header().set("content-type", "application/json")
    if r.body == nil {
        json.newencoder(w).encode("must send data")
    }

    var user user
    err := json.newdecoder(r.body).decode(&user)
    if err != nil {
        log.fatal(err)
    }
    if user.isempty() {
        json.newencoder(w).encode("invalid! enter user email.")
        return
    }
    users = append(users, user)
    json.newencoder(w).encode(user)

}

func (a *app) searchcity(w http.responsewriter, r *http.request) {

    vars := mux.vars(r)
    city := vars["city"]
    
}

我想我可以像这样简单地传递处理程序：

func searchCity(city *mongo.Collection) (w http.ResponseWriter, r *http.Request) {
...
}

但是，gmux 不允许您这样做，因为它隐式传入 http.responsewriter 和 *http.request。因此，任何输入都不能出现在参数中。我尝试在全球范围内声明它们，但这不起作用，建议不要这样做。有人告诉我可以尝试使用闭包或结构来传递它，但我也不太明白如何去做。

正确答案

一种方法是这样的，首先添加服务器类型

type server struct {
    router *mux.router
    cities *mongo.collection
}

将路由包装器添加到服务器

func (s *server) routes() {
    s.router.handlefunc("/base", s.handleindex()).methods("get")
}

处理函数

func (s *server) handleindex() http.handlerfunc {
    return func(w http.responsewriter, r *http.request) {
        cities := s.cities.find(...) // something like that
        // write your response, etc
    }
}

然后在main中

func main() {
    sr := &server{
        router: mux.NewRouter(),
        cities: getMongoDBCollection('cities') // implement this one :) should return a *mongo.Collection...
    }
    sr.routes()
...
}

以上就是本文的全部内容了，是否有顺利帮助你解决问题？若是能给你带来学习上的帮助，请大家多多支持golang学习网！更多关于Golang的相关知识，也可关注golang学习网公众号。