MongoDB如何从数组中获取指定数量元素实现分页功能

你当前使用的$size是查询条件操作符，作用是筛选tasks数组总长度等于_limit的用户文档，无法实现截取数组指定数量元素的需求。

MongoDB提供了$slice投影操作符，专门用于返回数组的指定子集，完全满足分页截取tasks的需求。

1. 简单获取前N个任务 #

直接在find的投影参数中配置$slice即可：

async getUserTasks( _id, _limit ) {
    try {
        // findOne更适合按唯一_id查询，无需处理数组返回结果
        const user = await UserModel.findOne( 
            { _id: _id },
            // 投影配置：仅返回tasks数组的前_limit个元素
            { tasks: { $slice: _limit } }
        );
        return user?.tasks || [];
    } catch ( e ) {
        console.log( e );
    }
}

2. 分页实现（支持偏移量跳页） #

如果需要按页码分页，传入偏移量即可，$slice支持[偏移量, 截取数量]的传参格式：

/**
 * @param {Number} page 当前页码，默认第1页
 * @param {Number} pageSize 每页条数，默认10条
 */
async getUserTasksByPage( _id, page = 1, pageSize = 10 ) {
    try {
        const skipCount = (page - 1) * pageSize;
        const user = await UserModel.findOne( 
            { _id: _id },
            { tasks: { $slice: [skipCount, pageSize] } }
        );
        return user?.tasks || [];
    } catch ( e ) {
        console.log( e );
    }
}

3. 携带总任务数的分页方案（适合前端渲染分页器） #

如果需要同时返回tasks总条数计算总页数，可以使用聚合管道同时完成数组长度统计和截取：

const mongoose = require('mongoose');

async getUserTasksWithTotal( _id, page = 1, pageSize = 10 ) {
    try {
        const skipCount = (page - 1) * pageSize;
        const result = await UserModel.aggregate([
            // 匹配指定用户，注意字符串_id要转ObjectId类型
            { $match: { _id: mongoose.Types.ObjectId(_id) } },
            {
                $project: {
                    totalTasks: { $size: "$tasks" }, // 统计tasks总长度
                    tasks: { $slice: ["$tasks", skipCount, pageSize] } // 截取当前页数据
                }
            }
        ]);
        return result[0] || { totalTasks: 0, tasks: [] };
    } catch ( e ) {
        console.log( e );
    }
}

补充说明 #

• $slice传负数时会从数组末尾开始截取，比如$slice: -5会返回最后添加的5个任务，适合最新记录查询场景
• 如果tasks数组元素量极大（单个数组超过1000条），建议单独拆分为tasks集合存储，查询性能会远高于内嵌数组截取

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