Azure CosmosDB $sample聚合配合Azure Functions未返回预期随机结果

基础问题说明 #

• 关于不调用toArray()返回异常结构：aggregate方法执行后返回的是AggregationCursor游标实例，并非直接的查询结果，直接序列化返回只会暴露游标本身的属性，也就是你看到的_events相关字段，属于正常表现，必须调用toArray()、forEach()或next()等方法遍历游标才能拿到实际文档数据。

$sample不返回随机结果的排查思路 #

1. 先在Mongo Shell中执行相同的聚合语句验证：

db.moments.aggregate([{$sample:{size:3}}])

多次执行看返回结果是否随机，以此判断是MongoDB侧问题还是代码侧问题。
2. 若Shell执行也返回固定结果：

• 确认集合内文档总数量大于3，若集合只有3条文档自然每次返回相同结果
• 确认MongoDB版本 >= 3.2，低于该版本不支持$sample聚合算子
• 若为小集合（文档数<100）可以替换为随机跳过的实现方案，避免$sample的适配问题

3. 若Shell执行结果正常，排查代码侧问题：

• 检查MongoDB Node.js驱动版本，升级到最新稳定版，旧版本驱动存在$sample实现异常的已知bug
• 调整MongoClient初始化逻辑，将连接逻辑放到函数外部全局复用，不要每次函数触发都重新建立连接
• 执行聚合时显式传入游标配置参数，避免驱动默认配置导致的游标复用问题
• 修正原代码的空值判断逻辑：toArray永远返回数组，无数据时为长度为0的空数组，原if(!obj)的判断永远不会触发。

修复后的参考代码 #

// 全局初始化MongoClient，复用连接，避免每次函数调用重连
const { MongoClient } = require('mongodb');
const client = new MongoClient(process.env.MONGODB_CONN_URI);
// 全局缓存连接Promise
const clientConnPromise = client.connect();

module.exports = async function (context, req) {
    // 等待连接完成，无需每次执行connect
    await clientConnPromise;
    const database = client.db("crud");
    const collection = database.collection("moments");
    // 聚合时显式指定游标参数，兼容不同版本驱动
    let obj = await collection.aggregate(
        [{$sample:{ size: 3 }}], 
        { cursor: { batchSize: 10 } }
    ).toArray();
    if (obj.length === 0) {
        return context.res = {
            status: 400,
            body: "not found"
        };
    }
    return context.res = {
        status: 200,
        body: obj
    };
}

备选兼容方案（适合小集合） #

如果$sample始终无法正常工作，可以使用随机跳过的实现方式，兼容性更强：

// 全局连接逻辑同上
module.exports = async function (context, req) {
    await clientConnPromise;
    const database = client.db("crud");
    const collection = database.collection("moments");
    // 先查询集合总条数
    const totalCount = await collection.countDocuments();
    const randomRes = [];
    // 生成3个随机偏移量取数据
    for(let i = 0; i < 3; i++) {
        const randomOffset = Math.floor(Math.random() * totalCount);
        const doc = await collection.find().skip(randomOffset).limit(1).next();
        randomRes.push(doc);
    }
    return context.res = {
        status: 200,
        body: randomRes
    };
}

内容的提问来源于stack exchange，提问作者Junzhong Loo