使用存储在数组中的 _id,从 Golang 查询 MongoDB。

huangapple go评论89阅读模式
英文:

Querying mongodb from golang using the _id stored in an array

问题

这是我的问题。我有一个数组,其中存储了mongodbs对象的_ids。使用mgo和bson包,如何以一次查询的方式检索它们?

如果数组是这样的:ids := ["543d171c5b2c12420dd016", "543d171c5b2dd016"]

我们如何进行查询?我尝试了下面的代码,但我知道它是错误的。

query := bson.M{"_id": bson.M{"$in": ids}}
c.Find(query).All()

谢谢。

英文:

So here is my question. I have an array which are stored the _ids of mongodbs objects. Whats the right way to retrieve them all in one query using the mgo and bson package?

So if the array is like that: ids:=["543d171c5b2c12420dd016","543d171c5b2dd016"]

How we make the query ? I tried that but I know its wrong.

query := bson.M{"_id": bson.M{"$in": ids}}
c.Find(query).All()

Thanks in advance

答案1

得分: 17

如果文档是以字符串ID的形式存储的,那么代码看起来是正确的。

这些ID看起来像是十六进制编码的对象ID。如果这些对象标识符是对象ID,那么你需要将十六进制字符串转换为对象ID:

oids := make([]bson.ObjectId, len(ids))
for i := range ids {
  oids[i] = bson.ObjectIdHex(ids[i])
}
query := bson.M{"_id": bson.M{"$in": oids}}
英文:

If the documents are stored with string ids, then the code looks correct.

The ids look like hex encoded object ids. If the object identifiers are object ids, then you need to the convert the hex strings to object ids:

oids := make([]bson.ObjectId, len(ids))
for i := range ids {
  oids[i] = bson.ObjectIdHex(ids[i])
}
query := bson.M{"_id": bson.M{"$in": oids}}

答案2

得分: 2

以下是翻译好的内容:

go.mongodb.org/mongo-driver 的 MongoDB 语法已更新,这应该可以使用官方驱动程序。

oids := make([]primitive.ObjectID, len(ids))
for i := range ids {
    objID, err := primitive.ObjectIDFromHex(ids[i])
    if err == nil {
        oids = append(oids, objID)
    }
}
英文:

MongoDB syntax for go.mongodb.org/mongo-driver has been updated, this should work using the official driver.

oids := make([]primitive.ObjectID, len(ids))
for i := range ids {
	objID, err := primitive.ObjectIDFromHex(ids[i])
	if err == nil {
		oids = append(oids, objID)
	}
}

答案3

得分: 1

这是将其转换回可在整个应用程序中使用的结构体的代码部分:

type MongoUser struct {
	ID        *primitive.ObjectID `json:"id" bson:"_id"`
	FirstName string              `json:"first_name" bson:"firstName"`
	LastName  string              `json:"last_name" bson:"lastName"`
	Email     string              `json:"email" bson:"email"`
}

这是一个辅助方法,它将您的 ID 切片转换为对象 ID 类型:

func formatObjectIdMultiple(hex []string) ([]primitive.ObjectID, error) {
	var list []primitive.ObjectID

	oids := make([]primitive.ObjectID, len(hex))
	for _, i := range hex {
		objectId, err := primitive.ObjectIDFromHex(i)
		if err != nil {
			return nil, err
		}
		oids = append(oids, objectId)
	}
	return list, nil
}

这是用于数据库的方法。重要的是要使用 bson.M,因为某种原因 bson.D 在这里不起作用。还要记得关闭游标,defer 函数将在 GetMultipleUser 函数结束时关闭它。

func (mongo *Mongo) GetMultipleUser(ids []string) ([]*MongoUser, error) {

	objectIDs, err := formatObjectIdMultiple(ids)
	if err != nil {
		return nil, err
	}

	query := bson.M{"_id": bson.M{"$in": objectIDs}}

	coll := mongo.Con.Database("dbName").Collection("users")
	cursor, err := coll.Find(context.Background(), query)
	if err != nil {
		return nil, err
	}
	defer func() {
		cursor.Close(context.Background())
	}()
	var output []*MongoUser
	for cursor.Next(context.Background()) {
		var temp *MongoUser
		cursor.Decode(&temp)
		output = append(output, temp)
	}

	return output, nil
}
英文:

This is to convert back to a struct that can be used through out the app

type MongoUser struct {
	ID        *primitive.ObjectID `json:"id" bson:"_id"`
	FirstName string              `json:"first_name" bson:"firstName"`
	LastName  string              `json:"last_name" bson:"lastName"`
	Email     string              `json:"email" bson:"email"`
}

This is a helper method that takes your slice of ids and turns it into the object id type.

  func formatObjectIdMultiple(hex []string) ([]primitive.ObjectID, error) {
	var list []primitive.ObjectID

	oids := make([]primitive.ObjectID, len(hex))
	for _, i := range hex {
		objectId, err := primitive.ObjectIDFromHex(i)
		if err != nil {
			return nil, err
		}
		oids = append(oids, objectId)
	}
	return list, nil
}

Here is my method for the db. Its important you use bson.M for some reason bson.D does not work with this. Also dont forget to close your cursor the defer function will close it at the end of your GetMultipleUser function.

    func (mongo *Mongo) GetMultipleUser(ids []string) ([]*MongoUser, error) {

	objectIDs, err := formatObjectIdMultiple(ids)
	if err != nil {
		return nil, err
	}

	query := bson.M{"_id": bson.M{"$in": objectIDs}}

	coll := mongo.Con.Database("dbName").Collection("users")
	cursor, err := coll.Find(context.Background(), query)
	if err != nil {
		return nil, err
	}
	defer func() {
		cursor.Close(context.Background())
	}()
	var output []*MongoUser
	for cursor.Next(context.Background()) {
		var temp *MongoUser
		cursor.Decode(&temp)
		output = append(output, temp)
	}

	return output, nil
}



</details>



huangapple
  • 本文由 发表于 2014年10月17日 23:22:39
  • 转载请务必保留本文链接:https://go.coder-hub.com/26428281.html
匿名

发表评论

匿名网友

:?: :razz: :sad: :evil: :!: :smile: :oops: :grin: :eek: :shock: :???: :cool: :lol: :mad: :twisted: :roll: :wink: :idea: :arrow: :neutral: :cry: :mrgreen:

确定