使用mongo go驱动程序更新mongo db collection以创建具有唯一值的新字段,而不会影响现有数据
我对mongo和mongo go是新手。需要使用mongo go驱动程序将新字段“uri”添加到包含现有数据的集合中。新字段需要填充唯一值,以便可以在其上创建唯一索引。如果有一种方法可以基于_id字段填充新字段,那么集合也会使用_id 我正在尝试下面的代码,不知道如何填充唯一值 //步骤1:更新所有文档以添加具有唯一值的新字段使用mongo go驱动程序更新mongo db collection以创建具有唯一值的新字段,而不会影响现有数据,go,bson,mongo-go,Go,Bson,Mongo Go,我对mongo和mongo go是新手。需要使用mongo go驱动程序将新字段“uri”添加到包含现有数据的集合中。新字段需要填充唯一值,以便可以在其上创建唯一索引。如果有一种方法可以基于_id字段填充新字段,那么集合也会使用_id 我正在尝试下面的代码,不知道如何填充唯一值 //步骤1:更新所有文档以添加具有唯一值的新字段 _, err := myColl.UpdateMany( ctx, bson.D{},// select all docs in collect
_, err := myColl.UpdateMany(
ctx,
bson.D{},// select all docs in collection
bson.D{
{"$set", bson.D{{"uri", GenerateRandomUniqueString()}}},
},
)
if err != nil {
return err
}
//然后,下一步是在此字段上创建索引:
key := bson.D{{"uri", 1}}
opt := options.Index().SetName("uri-index").SetUnique(true)
model := mongo.IndexModel{Keys: key, Options: opt}
_, err = myColl.Indexes().CreateOne(ctx, model)
if err != nil {
return err
}
一旦建立索引,旧记录将标记为只读,但我们不能删除这些记录。新数据将具有唯一的“uri”字符串值。
非常感谢您的帮助
在创建唯一索引时,使用上述代码失败,因为相同的值用于回填
我也试过这个:
func BackFillUri(db *mongo.Database) error {
myColl := db.Collection("myColl")
ctx := context.Background()
cursor, err := myColl.Find(ctx, bson.M{})
if err != nil {
return err
}
defer cursor.Close(ctx)
for cursor.Next(ctx) {
var ds bson.M
if err = cursor.Decode(&ds); err != nil {
return err
}
_, err1 := myColl.UpdateOne(
ctx,
bson.D{"_id": ds.ObjectId},
bson.D{
{"$set", bson.D{{"uri", rand.Float64()}}},
},
)
if err1 != nil {
return err1
}
}
return nil
}
但是我遇到了很多错误,不确定上面的逻辑是否正确我最终使用了下面的代码,希望它能帮助像我这样的新手:-)
那你需要什么帮助呢?使用唯一值的逻辑?@icza感谢您的回答,问题是,它只调用GeneratorDomainUniqueString()一次,因此尝试为myColl中的所有文档添加相同的字符串,导致创建索引失败。回填数据的全部目的是允许创建唯一的索引。我正在寻找一个选项,在这里我可以为每个文档更新不同的唯一值,以便创建唯一索引。我尝试了其他调用updateOne,但我不确定如何使用游标或向其传递文档并对其进行迭代。我可以用shell或java来实现,但mongogo对我来说太新了。
const charset = "abcdefghijklmnopqrstuvwxyz" +
"ABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789"
func RandomUniqueString(length int) string {
return StringWithCharset(length, charset)
}
var seededRand *rand.Rand = rand.New(
rand.NewSource(time.Now().UnixNano()))
func StringWithCharset(length int, charset string) string {
b := make([]byte, length)
for i := range b {
b[i] = charset[seededRand.Intn(len(charset))]
}
return string(b)
}
// Adds index on uri
func AddUriIndex(db *mongo.Database) error {
mycoll := db.Collection("mycoll")
ctx := context.Background()
//backfill code starts
type attribute struct {
Key string `bson:"key"`
Value interface{} `bson:"value"`
}
type item struct {
ID primitive.ObjectID `bson:"_id"`
ResourceAttributes []attribute `bson:"resourceAttributes,omitempty"`
}
cursor, err := mycoll.Find(ctx, primitive.M{})
if err != nil {
return err
}
defer cursor.Close(ctx)
for cursor.Next(ctx) {
var result item
if err = cursor.Decode(&result); err != nil {
return err
}
//fmt.Println("Found() result:", result)
filter := primitive.M{"_id": result.ID}
update := primitive.M{"$set": primitive.M{"uri": RandomUniqueString(32)}}
if _, err := mycoll.UpdateOne(ctx, filter, update); err != nil {
return err
}
}
//add uri-index starts
key := bson.D{{"uri", 1}}
opt := options.Index().
SetName("uri-index").
SetUnique(true)
model := mongo.IndexModel{Keys: key, Options: opt}
_, err = mycoll.Indexes().CreateOne(ctx, model)
if err != nil {
return err
}
return nil
}