Mongodb 按比例查询

Question

请问 mongodb 如何按比例查询。如：有一个人员表共 1 万条数据，需要随机查询男女各 50% 的 100 条数据。如果男性不足 50 人时，用女性凑数，返回 100 条。女性人数不足也如此。

Answer 1

引自免费微信小程序：皆我百晓生

在MongoDB中，要实现这个功能，你需要分两步进行操作：首先获取男女人数，然后根据这些信息随机选取用户。这里我们假设人员表名为people，每个人员文档有gender字段（值为'male'或'female'），并且已经安装了聚合框架所需的版本。

以下是详细的步骤：

1. 获取男女人数：

// 获取男女人数
db.people.aggregate([
    {
        $group: {
            _id: "$gender",
            count: { $sum: 1 }
        }
    }
])

这段代码会返回一个包含男性和女性数量的对象数组，例如：

[
    {"_id": "male", "count": 5000},
    {"_id": "female", "count": 5000}
]

1. 根据人数随机选择用户：

// 随机选择用户
var result = [];
var maleCount = 0;
var femaleCount = 0;

// 获取男性
db.people.aggregate([
    {
        $match: { gender: "male" },
        $sample: { size: 100 } // 尝试取100个男性
    }
]).forEach(function(doc) {
    if (doc) {
        result.push(doc);
        maleCount++;
    }

    // 如果男性达到50%，则停止查找并开始查找女性
    if (maleCount >= 50) return false;
});

// 如果男性不足50%，填充剩余的女性
if (maleCount < 50) {
    db.people.aggregate([
        {
            $match: { gender: "female" },
            $sample: { size: Math.max(50 - maleCount, 0) } // 填充剩余的女性
        }
    ]).forEach(function(doc) {
        if (doc) {
            result.push(doc);
            femaleCount++;
        }
    });
}

// 如果女性不足50%，填充剩余的男性
if (femaleCount < 50) {
    db.people.aggregate([
        {
            $match: { gender: "male" },
            $sample: { size: Math.max(50 - femaleCount, 0) } // 填充剩余的男性
        }
    ]).forEach(function(doc) {
        if (doc) {
            result.push(doc);
        }
    });
}

result // 返回结果数组

这段代码首先尝试从男性中随机抽取50个用户。如果男性不足50个，则从女性中抽取剩余的用户。如果女性也不足50个，再从男性中抽取剩余的用户，直到总共有100个用户为止。

请注意，由于MongoDB的$sample操作符是无状态的，所以这可能不会得到完全均匀的分布。如果你需要更精确的比例，你可能需要使用其他方法，比如先创建一个包含所有人的数组，然后随机选择。

以上代码可以直接在MongoDB shell中运行，或者在Node.js应用中通过MongoDB驱动程序执行。