express之 创建定时任务

要使用node-cron来实现定时任务，你需要先安装node-cron包。以下是如何修改代码以使用node-cron来实现定时清理旧用户数据的步骤：

1.安装node-cron

在你的项目目录下，运行以下命令来安装node-cron：

npm install node-cron

2.创建定时任务模块

创建一个新的文件tasks.js，用于定义定时任务。

// tasks.js
const cron = require('node-cron');
const UserService = require('./services/UserService');

// 定时任务：每小时清理数据库中的旧用户数据
cron.schedule('0 0 * * *', () => { // 每小时的第0分钟执行
  console.log('Running cleanup task...');
  UserService.deleteOldUsers().then(() => {
    console.log('Cleanup task completed.');
  }).catch(error => {
    console.error('Error during cleanup task:', error);
  });
});

// 服务层中添加删除旧用户的函数
async function deleteOldUsers() {
  // 假设用户模型有一个createdAt字段
  return User.deleteMany({ createdAt: { $lt: new Date(Date.now() - 30 * 24 * 60 * 60 * 1000) } });
}

module.exports = {
  deleteOldUsers
};

3.修改服务层以支持定时任务

在UserService.js中添加deleteOldUsers函数。

// services/UserService.js
const User = require('../models/User');

// ...其他函数...

async function deleteOldUsers() {
  return User.deleteMany({
    createdAt: { $lt: new Date(Date.now() - 30 * 24 * 60 * 60 * 1000) } // 假设用户模型有一个createdAt字段
  });
}

module.exports = {
  // ...其他函数...
  deleteOldUsers
};

4.修改模型层以支持创建时间

在User.js模型中添加createdAt字段。

// models/User.js
const mongoose = require('mongoose');

const userSchema = new mongoose.Schema({
  name: { type: String, required: true },
  email: { type: String, required: true, unique: true },
  createdAt: { type: Date, default: Date.now } // 添加创建时间字段
});

module.exports = mongoose.model('User', userSchema);

5.在主服务器文件中引入定时任务

在index.js中引入tasks.js。

// index.js
const express = require('express');
const mongoose = require('mongoose');
const bodyParser = require('body-parser');
const UserController = require('./controllers/UserController');
const tasks = require('./tasks'); // 引入定时任务

const app = express();
const PORT = process.env.PORT || 3000;

// 连接MongoDB
mongoose.connect('mongodb://localhost/mycrudapp', { useNewUrlParser: true, useUnifiedTopology: true });

// 中间件
app.use(bodyParser.json());

// 路由
app.post('/users', UserController.createUser);
app.get('/users', UserController.getAllUsers);
app.get('/users/:id', UserController.getUserById);
app.put('/users/:id', UserController.updateUser);
app.delete('/users/:id', UserController.deleteUser');

// 启动定时任务
// 注意：这里不需要显式调用tasks，因为node-cron会在指定的时间自动执行任务

app.listen(PORT, () => {
  console.log(`Server is running on port ${PORT}`);
});

使用node-cron，你可以定义更复杂的时间规则，例如每天、每周或每月执行任务。node-cron的语法是基于cron的，其中0 0 * * *表示每小时的第0分钟执行任务。

这样，你就使用node-cron成功地在你的Express应用中实现了定时任务功能。