基于Node.js和MongoDB实现文档过期通知机制

npm install node-cron

const mongoose = require('mongoose');

const documentSchema = new mongoose.Schema({
  userId: { type: mongoose.Schema.Types.ObjectId, ref: 'User', required: true },
  title: { type: String, required: true },
  content: { type: String },
  expirationTime: { type: Date, required: true }, // 文档过期时间
  expired: { type: Boolean, default: false },     // 标记文档是否已过期
  createdAt: { type: Date, default: Date.now }
});

module.exports = mongoose.model('Document', documentSchema);

const cron = require('node-cron');
const Document = require('./path/to/your/models/Document'); // 替换为您的模型路径
// 假设您有一个用户模型用于获取用户邮箱
const User = require('./path/to/your/models/User'); // 替换为您的用户模型路径

/**
 * 检查并处理过期文档的异步函数
 */
const checkExpiredDocuments = async () => {
  try {
    const currentDate = new Date(); // 获取当前时间

    // 查找所有过期时间早于当前时间且尚未被标记为过期的文档
    const expiredDocuments = await Document.find({
      expirationTime: { $lt: currentDate },
      expired: false
    }).lean(); // 使用.lean()获取纯粹的JS对象，提高性能

    if (expiredDocuments.length === 0) {
      console.log('No new expired documents found.');
      return;
    }

    // 批量更新这些文档，将expired字段设置为true
    await Document.updateMany(
      { _id: { $in: expiredDocuments.map(doc => doc._id) } }, // 过滤条件
      { $set: { expired: true } } // 更新操作
    );

    console.log(`Found and marked ${expiredDocuments.length} documents as expired.`);

    // 遍历已过期的文档，并触发通知
    for (const doc of expiredDocuments) {
      // 可以在这里获取用户邮箱并发送通知
      // const user = await User.findById(doc.userId);
      // if (user && user.email) {
      //   await sendExpirationNotification(user.email, doc.title);
      //   console.log(`Sent expiration notification to ${user.email} for document: ${doc.title}`);
      // }
      // 这是一个占位符，实际的通知逻辑将在下一节讨论
      console.log(`Placeholder: Would notify user for document ID: ${doc._id}, Title: ${doc.title}`);
    }

  } catch (error) {
    console.error('Error checking or processing expired documents:', error);
  }
};

/**
 * 示例邮件通知函数 (需要您自行实现)
 * @param {string} email - 用户邮箱
 * @param {string} documentTitle - 过期文档标题
 */
async function sendExpirationNotification(email, documentTitle) {
  // 实际的邮件发送逻辑，可以使用Nodemailer等库
  // 例如：
  // const nodemailer = require('nodemailer');
  // let transporter = nodemailer.createTransport({...});
  // await transporter.sendMail({
  //   from: '"Your App" <noreply@yourapp.com>',
  //   to: email,
  //   subject: '您的文档已过期！',
  //   html: `<p>尊敬的用户，您的文档 "${documentTitle}" 已于 ${new Date().toLocaleDateString()} 过期。</p>`
  // });
  console.log(`Sending email to ${email} about expired document: ${documentTitle}`);
}

// Schedule the execution of the function to check for expired documents every hour
// cron.schedule('0 * * * *', checkExpiredDocuments);
// 上面的计划表示每小时的第0分钟执行一次。
// 您可以根据需求调整频率，例如：
// '*/5 * * * *'  - 每5分钟执行一次
// '0 0 * * *'    - 每天午夜执行一次

// 在您的应用启动文件 (例如 app.js 或 server.js) 中调用此调度器
const startExpirationChecker = () => {
  // 立即执行一次，确保启动时检查
  checkExpiredDocuments();
  // 每小时执行一次
  cron.schedule('0 * * * *', checkExpiredDocuments, {
    scheduled: true,
    timezone: "Asia/Shanghai" // 根据您的服务器时区设置
  });
  console.log('Expired document checker scheduled to run every hour.');
};

// 导出或直接调用此函数
module.exports = startExpirationChecker;
// 在您的主应用文件中 (e.g., app.js):
// const startExpirationChecker = require('./path/to/your/cron-job-file');
// startExpirationChecker();