更改 MongoDB 集合以添加独特值的新字段，不影响现有数据

亲爱的编程学习爱好者，如果你点开了这篇文章，说明你对《更改 MongoDB 集合以添加独特值的新字段，不影响现有数据》很感兴趣。本篇文章就来给大家详细解析一下，主要介绍一下，希望所有认真读完的童鞋们，都有实质性的提高。

我是 mongo 和 mongo go 驱动程序的新手。需要使用 mongo go 驱动程序将新字段“uri”添加到我的现有数据集合中。新字段需要填充唯一值，以便可以在其上创建唯一索引。该集合也使用 _id，如果有一种方法我们可以基于 _id 字段填充新字段，该方法也可以工作。

我正在尝试下面的代码，不确定如何填充唯一值。

//step1：更新所有文档以添加具有唯一值的新字段

_, 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

}

但是我遇到了很多错误，不确定上述逻辑是否正确

解决方案

我最终使用了下面的代码，希望它能帮助像我这样的新手:-)

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
}

到这里，我们也就讲完了《更改 MongoDB 集合以添加独特值的新字段，不影响现有数据》的内容了。个人认为，基础知识的学习和巩固，是为了更好的将其运用到项目中，欢迎关注golang学习网公众号，带你了解更多关于的知识点！