nodejs实现黑名单中间件设计

Node.js是一种在服务器端运行JavaScript的开源、跨平台、事件驱动的轻量级JavaScript运行时环境。黑名单中间件是Node.js中一个常见的功能，通过此中间件可以实现对请求的IP、URL、UA等信息进行过滤和限制。这里提供一种实现黑名单中间件的设计攻略。

1. 确定需求

首先，需要确定中间件的具体要求，例如：

• 支持对IP、URL、UA的黑名单过滤
• 支持自定义黑名单规则
• 支持回调函数处理被禁止请求的逻辑

2. 定义中间件函数

定义一个函数来作为中间件，输入参数是请求对象(request)、响应对象(response)和下一个中间件函数(next)。该函数需要实现对请求信息进行过滤的逻辑。

function blacklist(req, res, next) {
  // 黑名单规则判断逻辑
  next();
}

3. 实现黑名单规则

在黑名单中间件函数中实现黑名单规则的判断逻辑。可以使用正则表达式等方式来对IP、URL、UA等进行匹配判断。

function blacklist(req, res, next) {
  const ip = req.ip;
  const url = req.url;
  const ua = req.headers['user-agent'];
  const blackList = ['192.168.0.1', '127.0.0.1', '/admin'];
  if (blackList.some(item => ip.includes(item) || url.includes(item) || ua.includes(item))) {
    res.status(403).send('您的请求被禁止访问');
  } else {
    next();
  }
}

4. 使用中间件

将中间件函数应用到合适的路由上。例如，使用Express框架来实现：

const express = require('express');
const app = express();

app.use(blacklist);

app.get('/', (req, res) => {
  res.send('Hello World!');
});

app.listen(3000, () => {
  console.log('Server started on port 3000');
});

5. 扩展中间件功能

按需扩展中间件功能，例如添加自定义黑名单规则和使用回调函数处理被禁止请求的逻辑。

function blacklist(ruleList) {
  return function(req, res, next) {
    const ip = req.ip;
    const url = req.url;
    const ua = req.headers['user-agent'];
    const blackList = ruleList.concat(['192.168.0.1', '127.0.0.1', '/admin']);
    if (blackList.some(item => ip.includes(item) || url.includes(item) || ua.includes(item))) {
      // 处理被禁止请求的逻辑
      return callback(req, res, '您的请求被禁止访问');
    } else {
      next();
    }
  }
}

const ruleList = ['/api'];
app.use(blacklist(ruleList));

// 自定义回调函数
function callback(req, res, message) {
  res.status(403).send(message);
}

示例

以下是两个示例说明：

示例1: IP黑名单

function blacklist(req, res, next) {
  const ip = req.ip;
  const blackList = ['192.168.0.1', '127.0.0.1'];
  if (blackList.includes(ip)) {
    return res.status(403).send('您的请求被禁止访问');
  } else {
    next();
  }
}

示例2: UA黑名单

function blacklist(req, res, next) {
  const ua = req.headers['user-agent'];
  const blackList = ['Chrome'];
  if (blackList.some(item => ua.includes(item))) {
    return res.status(403).send('您的请求被禁止访问');
  } else {
    next();
  }
}