如何在JS中根据条件更新或移除MongoDB聚合查询数组内的节点？

原始聚合查询 #

[
    { "$match": {"PaymentStatus" : "Submitted"} },
    { "$sort": { "Date": -1 } },
    { "$skip": "$skip" },
    { "$limit": "$limit" },
    {
        "$lookup": {
            "from": "bookings",
            "localField": "_id",
            "foreignField": "",
            "as": "bookings"
        }
    },
    {
        "$lookup": {
            "from": "users",
            "localField": "bookings.user_id",
            "foreignField": "_id",
            "as": "User"
        }
    },
    {
        "$project": {
            "_id": 0,
            "Number": "$Number",
            "Status": "$bookings.Status",
            "User": "$bookings.AssignedTo",
            "VendorName": "$booking.username",
            "SystemID": "$booking._id"
        }
    }
]

需求 #

• 当传入skip和limit参数时，替换聚合查询中对应$skip、$limit节点的占位符值
• 未传入参数或参数不合法时，移除这两个节点

当前实现代码 #

移除节点代码 #

var myArray = query.filter(function( obj ) {
    return obj.$skip !== '$skip'&& obj.$limit !== '$limit'
});

更新节点代码 #

var a = 100 ;
var b = 300;
const newArr = query.map(obj => {
  if (obj.$skip=== '$skip') {
    return {...obj, $skip: a};
  }
  if (obj.$limit=== '$limit') {
    return {...obj, $limit: b};
  }
  return obj;
});
console.log(newArr)

当前实现的问题 #

1. 移除节点逻辑错误：你的过滤条件是obj.$skip !== '$skip' && obj.$limit !== '$limit'，但每个聚合阶段对象只会包含一个操作符（比如$skip节点里没有$limit属性），这会导致判断逻辑混乱。虽然最终结果碰巧符合需求，但逻辑本身存在漏洞，容易引发其他问题。
2. 缺乏参数校验：如果只传入了其中一个参数，另一个节点会保留占位符'$skip'或'$limit'，MongoDB执行时会报错，因为$skip/$limit要求传入数字类型。
3. 原聚合查询的潜在问题：lookup的foreignField是空字符串，会导致关联失败；$project中的$booking应该是$bookings（因为lookup的as是bookings，是数组，可能需要先$unwind或者取数组第一个元素）。

优化方案 #

方案一：动态构建聚合管道（推荐） #

把固定阶段和动态阶段分离，根据参数合法性动态插入$skip/$limit，逻辑更清晰：

// 处理并校验用户传入的参数
const skip = parseInt(req.query.skip, 10);
const limit = parseInt(req.query.limit, 10);

// 固定聚合阶段模板（去掉占位符的$skip/$limit）
const basePipeline = [
    { "$match": {"PaymentStatus" : "Submitted"} },
    { "$sort": { "Date": -1 } },
    {
        "$lookup": {
            "from": "bookings",
            "localField": "_id",
            "foreignField": "payment_id", // 修正为空的关联字段，替换为实际字段名
            "as": "bookings"
        }
    },
    { "$unwind": "$bookings" }, // 可选：将bookings数组拆分为单个文档，方便后续投影
    {
        "$lookup": {
            "from": "users",
            "localField": "bookings.user_id",
            "foreignField": "_id",
            "as": "user"
        }
    },
    { "$unwind": "$user" }, // 可选：拆分user数组
    {
        "$project": {
            "_id": 0,
            "Number": "$Number",
            "Status": "$bookings.Status",
            "User": "$bookings.AssignedTo",
            "VendorName": "$user.username", // 修正投影字段，对应lookup结果
            "SystemID": "$bookings._id" // 修正投影字段
        }
    }
];

// 动态插入$skip和$limit阶段
const pipeline = [...basePipeline];
let insertPos = 2;
if (!isNaN(skip) && skip >= 0) {
    pipeline.splice(insertPos, 0, { $skip: skip });
    insertPos++; // 插入skip后，limit要放在skip后面
}
if (!isNaN(limit) && limit > 0) {
    pipeline.splice(insertPos, 0, { $limit: limit });
}

// 最终使用pipeline执行聚合查询
// db.collection.aggregate(pipeline);

方案二：修正原替换/过滤逻辑 #

如果坚持用模板替换的方式，修正后的代码如下：

修正后的移除节点代码

// 精准过滤掉占位符节点
const filteredPipeline = query.filter(stage => {
    if (stage.$skip !== undefined) {
        return stage.$skip !== '$skip';
    }
    if (stage.$limit !== undefined) {
        return stage.$limit !== '$limit';
    }
    return true;
});

修正后的更新+过滤逻辑

const skip = parseInt(req.query.skip, 10);
const limit = parseInt(req.query.limit, 10);

let processedPipeline = [...query];

// 替换合法参数的占位符，否则移除节点
if (!isNaN(skip) && skip >= 0) {
    processedPipeline = processedPipeline.map(stage => 
        stage.$skip === '$skip' ? { $skip: skip } : stage
    );
} else {
    processedPipeline = processedPipeline.filter(stage => stage.$skip !== '$skip');
}

if (!isNaN(limit) && limit > 0) {
    processedPipeline = processedPipeline.map(stage => 
        stage.$limit === '$limit' ? { $limit: limit } : stage
    );
} else {
    processedPipeline = processedPipeline.filter(stage => stage.$limit !== '$limit');
}

内容的提问来源于stack exchange，提问作者its me