在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) }