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

简单的配置什么的,可以取官网看,连接地址 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)
}

相关推荐