在golang中使用mongodb官方驱动用ID查询的一个小坑

Posted 2021-02-20 飞天艾特小猪

简单的配置什么的，可以取官网看，连接地址 https://docs.mongodb.com/drivers/go ；

普通的查询很好说，几乎一看就明白，但是如果想要根据  _id 来查询的话，在python中，会有一个ObjectId()的类；在golang中其实也差不多，不过变成了

primitive.ObjectIDFromHex()；返回类型为 ObjectId,err；传入参数为string；这样的话，将返回的OID传入filter中，即可使用ID查询想要的数据
完整代码如下

	if oid ,err:= primitive.ObjectIDFromHex(id);err==nil{
		result := make(map[string]interface{})
		db.Find("user", bson.M{"_id":oid}, &result)
		_, _ = ctx.JSON(result)
	}

　　其中  db.Find 我自己做了一下封装，如下

package db

import (
	"context"
	"fmt"
	"go.mongodb.org/mongo-driver/mongo"
	"go.mongodb.org/mongo-driver/mongo/options"
	"log"
)

var DB  *mongo.Client
var  Database *mongo.Database
func Init()  {
	// 设置客户端连接配置
	clientOptions := options.Client().ApplyURI("mongodb://localhost:27017")

	// 连接到MongoDB
	client, err := mongo.Connect(context.TODO(), clientOptions)
	if err != nil {
		log.Fatal(err)
	}

	// 检查连接
	err = client.Ping(context.TODO(), nil)
	if err != nil {
		log.Fatal(err)
	}
	fmt.Println("Connected to MongoDB!")
	DB=client
	Database = client.Database("safety")

}

func Find(table string, filter interface{}, result interface{}) {
	//var result models.User
	err := Database.Collection(table).FindOne(context.TODO(), filter).Decode(result)
	if err != nil {
		fmt.Println(err)
	}
	fmt.Printf("Found a single document: %+v\n", result)
	//Database.Collection(table).UpdateOne(context.TODO(),bson.M{"_id":"123"},bson.M{"name":"你好"})
	//return result
}

func Insert(table string,data interface{}){
	_, _ = Database.Collection(table).InsertOne(context.TODO(), data)
}