Node.js + Express + mongodb 的博客项目之前端ejs模块化及博客分类管理、分页功能实现（六）

/*
    自定义分页渲染模块
        需要传递一个对象进来，该对象的属性包括：
        pagination = {
        // 每页显示的条数
        limit: 10,
        // 需要操作的数据库模型
        model: userModel,
        // 需要控制分页的url
        url: "/admin/user",
        // 渲染的模板页面
        ejs: "admin/user",
        res: res,
        req: req,
        // 查询的条件
        where: {},
        // 联合查询的条件
        populate: []
        // 名称
        docs
    }
*/

const pagination = (object) => {
    /*
        实现分页
            limit(Number): 限制获取的数据条数
            skip(Numer): 跳过数据的条数
        每页显示2条
    */
    // 当前页数,使用get获取前端传递的当前页数
    let page = object.req.query.page || 1;
    // 每页显示数据条数默认为10
    let limit = object.limit || 10;
    // 总页数
    let pages = 0;
    // 跨集合查询的条件
    let populate = object.populate || [];
    // 查询该文档的数据条数
    object.model.countDocuments(object.where).then((count) => {
        // 根据总条数计算总页数
        pages = Math.ceil(count / limit);
        // 限制当前页数，避免溢出
        // page不能超过pages
        page = Math.min(page, pages);
        // page不能小于1
        page = Math.max(page, 1);
        // 跳过数据的条数
        let skip = (page - 1) * limit;
        // 分页查询出数据
        object.model.find(object.where).populate(populate).skip(skip).limit(limit).then((docs) => {
            object.res.render(object.ejs, {
                userInfo: object.req.userInfo,
                docs: docs,
                page: page,
                pages: pages,
                limit: limit,
                url: object.url,
                count: count
            });
        });
    });
};

// 暴露给外部使用
module.exports = pagination;

// 用户管理首页
router.get("/user", (req, res, next) => {
    // 调用自定义的分页渲染方法
    pagination({
        // 每页显示的条数
        limit: 10,
        // 需要操作的数据库模型
        model: userModel,
        // 需要控制分页的url
        url: "/admin/user",
        // 渲染的模板页面
        ejs: "admin/user",
        // 查询的条件
        where: {},
        res: res,
        req: req
    });
});

// 引入相关模块
const mongoose = require("mongoose");

/*
    文章分类的数据结构
    {
        文章标题，字符串类型
    }
*/
module.exports = new mongoose.Schema({
    name: String
});

// 引入mongoose模块
const mongoose = require("mongoose");
// 引入文章分类的schema
const categoriesSchema = require("../schemas/categories");

// 创建文章分类模型
module.exports = mongoose.model("category", categoriesSchema);

// 分类的保存
router.post("/category/add", (req, res, next) => {
    // 获取分类名称，默认为""
    let name = req.body.name || "";

    // 如果名称为空
    if (name === "") {
        // 渲染一个错误提示
        res.render("admin/error", {
            userInfo: req.userInfo,
            url: null,
            message: "分类名称不能为空！"
        });
        return;
    }
    // 从数据库中查询该名称是否已存在
    categoryModel.findOne({name: name}, (err, docs) => {
        // 如果数库库中已存在该名称
        if (docs) {
            // 渲染一个错误提示
            res.render("admin/error", {
                userInfo: req.userInfo,
                url: null,
                message: "该分类名称已存在！"
            });
            return;
        } else {
            // 不存在则新建一个数据
            categoryModel.create({
                name: name
            }, (err) => {
                if (!err) {
                    // 渲染一个错误提示
                    res.render("admin/success", {
                        userInfo: req.userInfo,
                        message: "添加成功！",
                        // 跳转到该路由
                        url: "/admin/category"
                    });
                    return;
                }
            });
        }
    });
});

// 分类的修改界面
router.get("/category/edit", (req, res, next) => {
    // 获取用户提交过来的id
    let id = req.query.id || "";
    // 根据id从数据库中查询相关数据
    categoryModel.findOne({_id: id}, (err, category) => {
        if (category) {
            // 如何数据存在则渲染修改界面
            res.render("admin/category/edit", {
                userInfo: req.userInfo,
                category: category
            });
        } else {
            // 若不存在渲染错误提示面板
            res.render("admin/error", {
                userInfo: req.userInfo,
                url: null,
                message: "该分类不存在！"
            });
        }
    });
});

// 分类修改的保存
router.post("/category/edit", (req, res, next) => {
    // 获取修改后的id及名称
    let id = req.query.id;
    let name = req.body.name;

    // 根据id从数据库中查询相关数据
    categoryModel.findById(id, (err, category) => {
        if (category) {
            // 若数据存在
            // 简单验证---如果数据没修改
            if (name === category.name) {
                res.render("admin/success", {
                    url: "/admin/category",
                    userInfo: req.userInfo,
                    message: "修改成功！"
                });
                return;
            }
            // 查询用户修改的分类是否与数据库中的冲突
            categoryModel.findOne({
                _id: {$ne: id},
                name: name
            }, (err, docs) => {
                if (docs) {
                    // 数据冲突
                    res.render("admin/error", {
                        userInfo: req.userInfo,
                        url: null,
                        message: "该分类已存在！"
                    });
                    return;
                } else {
                    // 更新数据
                    categoryModel.update({_id: id}, {$set: {name: name}}, (err) => {
                        if (!err) {
                            // 不出错
                            res.render("admin/success", {
                                userInfo: req.userInfo,
                                url: "/admin/category",
                                message: "修改成功！"
                            });
                            return;
                        } else {
                            // 出错
                            res.render("admin/error", {
                                userInfo: req.userInfo,
                                url: null,
                                message: "修改失败！"
                            });
                            return;
                        }
                    });
                }
            });
        } else {
            // 若不存在
            res.render("admin/error", {
                userInfo: req.userInfo,
                url: null,
                message: "该分类不存在！"
            });
            return;
        }
    });
});

// 分类的删除
router.get("/category/delete", (req, res, next) => {
    // 获取需要删除的分类id
    let id = req.query.id || "";
    // 从数据库中删除数据
    categoryModel.remove({_id: id}, (err) => {
        if (!err) {
            // 删除成功
            res.render("admin/success", {
                url: "/admin/category",
                userInfo: req.userInfo,
                message: "删除成功！"
            });
        } else {
            // 删除失败
            res.render("admin/error", {
                url: null,
                userInfo: req.userInfo,
                message: "删除失败！"
            });
        }
    });
});