Node.js 创建一个简单的服务实现反向代理

创建一个Node.js项目

打开文件管理器，在你想要的索引位置上新建一个文件夹，文件夹的名称即是项目名

用Visual Studio Code打开这个文件夹，并在终端执行

npm init

没有特定要求一直回车即可，也可参考其他Node.js项目教程

安装所需依赖

安装http-proxy-middleware

1	npm install http-proxy-middleware

导入代码

新建一个index.js，为主要执行程序
新建一个api文件夹，并在api文件下新建module.js。用于不走代理时执行自定义代码

index.js

const http = require('http');
const { createProxyMiddleware } = require('http-proxy-middleware');
const printStr = require('./api/module');
const fs = require('fs');

const newProxy = createProxyMiddleware({
    target: 'https://www.baidu.com',
    changeOrigin: true,
    pathRewrite: {
        '^/proxy': '', // 将路径中的 '/proxy' 替换为空字符串
        '^/': '', // 将路径中的 '/' 替换为空字符串
    },
});

const server = http.createServer((req, res) => {
    const clientIp = req.socket.remoteAddress;
    // 检查请求路径，如果是以 '/noproxy' 开头，则不进行代理
    if (req.url.startsWith('/noproxy')) {
        //判断是否为POST请求
        if (req.method === 'POST') {
            let body = '';
            // 方法: on() 方法用于监听事件，第一个参数是事件名称，这里是 'data'，表示接收到请求体数据时触发。
            // chunk 参数: 在 'data' 事件中，回调函数被调用多次，每次传递一个数据块 chunk。这些数据块通常是Buffer对象，表示来自客户端的一部分数据。
            req.on('data', (chunk) => {
                // chunk.toString(): 将每个数据块 chunk 转换为字符串形式。因为HTTP请求体可以是任何类型的数据，例如表单数据或JSON数据，因此需要将每个数据块转换为字符串进行处理。
                body += chunk.toString(); // 这行代码将每个接收到的数据块字符串拼接到 body 变量中。这样，当所有数据块都接收完毕时，body 将包含完整的请求体数据。
            });
            // 这里是 'end'，表示请求数据接收完毕时触发。
            req.on('end', () => {
                // 解析 POST 请求参数
                const postData = JSON.parse(body);
                const { domain } = postData;
                // 判断 domain 参数是否存在
                if (domain) {
                    const jsonData = printStr(domain);
                    // 设置HTTP状态码为200（成功）并将内容类型设置为 application/json。
                    res.writeHead(200, { 'Content-Type': 'application/json' });
                    // 将 jsonData 对象作为JSON字符串发送到响应体中。
                    res.end(JSON.stringify(jsonData));
                } else {
                    const jsonData = {
                        success: true,
                        message: "参数不完整",
                        code: 200,
                        timestamp: Math.round(new Date())
                    };
                    res.writeHead(200, { 'Content-Type': 'application/json' });
                    res.end(JSON.stringify(jsonData));
                }
            });
        } else {
            const jsonData = {
                success: true,
                message: "GET方式请求",
                code: 200,
                timestamp: Math.round(new Date())
            };
            res.writeHead(200, { 'Content-Type': 'application/json' });
            res.end(JSON.stringify(jsonData));
        }
    // 检查请求路径，如果是以 '/proxy' 开头，则进行代理
    } else if (req.url.startsWith('/proxy')) {
        // 执行newProxy代理
        newProxy(req, res);
    } else {
        sendPage('./page/404.html', res);
    }
});

function getClientIp(req) {
    const xForwardedFor = req.headers['x-forwarded-for'];
    if (xForwardedFor) {
        const ipList = xForwardedFor.split(',');
        return ipList[ipList.length - 1].trim();
    }
    return req.connection.remoteAddress;
}

function sendPage(pagePath, res) {
    // 读取HTML文件
    fs.readFile(pagePath, (err, data) => {
        if (err) {
            res.writeHead(404);
            res.end("404 Not Found");
        } else {
            res.writeHead(200, { 'Content-Type': 'text/html; charset=utf-8' });
            res.end(data);
        }
    });
}

const port = process.env.PORT || 3000;
server.listen(port, () => {
    console.log(`Proxy server running on port ${port}`);
});

module.js

function printStr(domain) {
    return {
        success: true,
        message: domain,
        code: 200,
        timestamp: Math.round(new Date())
    };
}

module.exports = printStr;