MongoDB批量更新实战：使用随机数生成唯一字段

2018-11-01

引言

在 MongoDB 数据库维护和数据迁移过程中，经常需要对现有数据进行批量更新。本文将介绍如何使用 MongoDB 的 shell 脚本进行批量数据更新，并以一个实际案例——将 name 字段为 “Guest” 的文档更新为 “Guest” + 4位随机数——来演示完整的操作流程。

需求分析

场景描述

假设我们有一个用户集合（robots），其中部分用户的 name 字段值为 “Guest”。为了区分这些用户，我们需要将其更新为 “Guest” + 4位随机数的形式，例如 “Guest1234”。

数据示例

// 更新前的数据
{ "_id" : ObjectId("..."), "name" : "Guest" }
{ "_id" : ObjectId("..."), "name" : "Guest" }
{ "_id" : ObjectId("..."), "name" : "Alice" }

// 更新后的数据
{ "_id" : ObjectId("..."), "name" : "Guest8321" }
{ "_id" : ObjectId("..."), "name" : "Guest4756" }
{ "_id" : ObjectId("..."), "name" : "Alice" }

解决方案

基础实现

db.robots.find({name:"Guest"}).forEach(
   function(item){
       db.robots.update(
           {"_id": item._id},
           {$set: {name: "Guest" + Math.floor(Math.random() * 10000 + 10000) % 10000}}
       )
   }
)

代码解析

1. 查询条件

1	db.robots.find({name: "Guest"})

只查询 name 字段为 “Guest” 的文档
避免对不需要更新的数据执行操作

2. 遍历游标

1	.forEach(function(item) { ... })

遍历查询结果中的每个文档
item 代表当前处理的文档对象

3. 更新操作

db.robots.update(
    {"_id": item._id},  // 查询条件：使用 _id 精确匹配
    {$set: {name: newName}}  // 更新操作：设置新的 name 值
)

4. 随机数生成

1	Math.floor(Math.random() * 10000 + 10000) % 10000

分解步骤：

Math.random()：生成 0-1 之间的随机小数
* 10000：扩大到 0-10000 范围
+ 10000：偏移到 10000-20000 范围
% 10000：取模得到 0-9999
Math.floor()：取整数部分

优化版本

// 确保生成 4 位数字（不足补零）
function generateRandomSuffix() {
    return Math.floor(Math.random() * 10000).toString().padStart(4, '0');
}

// 批量更新
db.robots.find({name: "Guest"}).forEach(function(item) {
    let newName = "Guest" + generateRandomSuffix();

    db.robots.update(
        {_id: item._id},
        {$set: {name: newName}}
    );

    print("Updated: " + item._id + " -> " + newName);
});

进阶应用

1. 确保随机数唯一

// 使用 Set 跟踪已使用的数字
let usedNumbers = new Set();

// 生成唯一随机数
function generateUniqueSuffix() {
    let num;
    do {
        num = Math.floor(Math.random() * 10000);
    } while (usedNumbers.has(num));

    usedNumbers.add(num);
    return num.toString().padStart(4, '0');
}

// 更新并确保唯一
let count = db.robots.countDocuments({name: "Guest"});
if (count > 10000) {
    print("警告：文档数量超过可能的随机数范围");
}

db.robots.find({name: "Guest"}).forEach(function(item) {
    db.robots.update(
        {_id: item._id},
        {$set: {name: "Guest" + generateUniqueSuffix()}}
    );
});

2. 使用聚合管道（MongoDB 4.2+）

db.robots.updateMany(
    {name: "Guest"},
    [
        {
            $set: {
                name: {
                    $concat: [
                        "Guest",
                        {
                            $toString: {
                                $floor: {
                                    $multiply: [{$rand: {}}, 10000]
                                }
                            }
                        }
                    ]
                }
            }
        }
    ]
);

3. 批量更新脚本（Node.js）

const { MongoClient } = require('mongodb');

async function updateGuestNames() {
    const client = new MongoClient('mongodb://localhost:27017');

    try {
        await client.connect();
        const db = client.db('mydatabase');
        const collection = db.collection('robots');

        // 查询所有 Guest 用户
        const guests = await collection.find({name: "Guest"}).toArray();

        console.log(`Found ${guests.length} Guest users`);

        // 批量更新
        const updates = guests.map(guest => ({
            updateOne: {
                filter: {_id: guest._id},
                update: {
                    $set: {
                        name: "Guest" + Math.floor(Math.random() * 10000)
                            .toString().padStart(4, '0')
                    }
                }
            }
        }));

        if (updates.length > 0) {
            const result = await collection.bulkWrite(updates);
            console.log(`Updated ${result.modifiedCount} documents`);
        }

    } finally {
        await client.close();
    }
}

updateGuestNames().catch(console.error);

性能优化

1. 使用批量操作

// 收集批量更新操作
let bulkOps = [];

db.robots.find({name: "Guest"}).forEach(function(item) {
    bulkOps.push({
        updateOne: {
            filter: {_id: item._id},
            update: {$set: {name: "Guest" + Math.floor(Math.random() * 10000)}}
        }
    });

    // 每 1000 条执行一次
    if (bulkOps.length === 1000) {
        db.robots.bulkWrite(bulkOps);
        bulkOps = [];
    }
});

// 处理剩余的
if (bulkOps.length > 0) {
    db.robots.bulkWrite(bulkOps);
}

2. 添加索引

// 更新前添加索引
db.robots.createIndex({name: 1});

// 执行更新
// ...

// 更新后（可选）删除索引
db.robots.dropIndex("name_1");

注意事项

备份数据：执行批量更新前务必备份
测试环境：先在测试环境验证脚本
分批处理：大数据量时分批处理避免锁表
监控性能：关注更新操作的性能影响

总结

MongoDB 的 forEach 结合 update 提供了一种灵活的数据更新方式。对于简单的批量更新，使用 MongoDB Shell 脚本即可；对于复杂的业务逻辑，建议使用 Node.js 等编程语言配合 MongoDB 驱动程序实现。

后端开发数据库

引言