当前位置： > > > > 如何在另一个函数中使用已连接到 mongodb 和集合的句柄？
来源：stackoverflow
2024-04-27 23:12:36
0浏览
收藏
来到的大家，相信都是编程学习爱好者，希望在这里学习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://<username>:<password>@cluster0.um5qb.mongodb.net/<db>?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的相关知识，也可关注公众号。
