proxy.js 1.8 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758
  1. var express = require('express');
  2. var router = express.Router();
  3. const fetch = require('node-fetch');
  4. const config = require('../config/app');
  5. // 代理请求的中间件
  6. router.use('/', async (req, res) => {
  7. console.log(`proxy req.url = ${req.url}`);
  8. const targetUrl = `${config.resHost}${req.url}`; // 替换为目标服务器的 URL
  9. const method = req.method;
  10. const headers = {
  11. ...req.headers,
  12. // 可能需要根据目标服务器的要求调整或添加请求头
  13. };
  14. // 如果请求包含请求体(例如 POST 请求),则读取它
  15. let body;
  16. if (req.method === 'POST' || req.method === 'PUT') {
  17. body = await getRawBody(req); // 使用 getRawBody 函数读取请求体
  18. }
  19. try {
  20. const response = await fetch(targetUrl, {
  21. method,
  22. headers,
  23. body,
  24. });
  25. // 在这里添加 CORS 响应头
  26. res.setHeader('Access-Control-Allow-Origin', '*');
  27. res.setHeader('Access-Control-Allow-Methods', 'GET, POST, PUT, DELETE, OPTIONS');
  28. res.setHeader('Access-Control-Allow-Headers', 'Content-Type, Authorization, X-Requested-With');
  29. // 将目标服务器的响应转发给客户端
  30. const responseData = await response.buffer(); // 获取响应的 Buffer
  31. res.status(response.status).send(responseData);
  32. } catch (error) {
  33. // 处理错误,并将错误信息返回给客户端
  34. console.error('Proxy error:', error);
  35. res.status(500).send('Internal Server Error');
  36. }
  37. });
  38. // 辅助函数:读取请求体的原始数据
  39. async function getRawBody(req) {
  40. return new Promise((resolve, reject) => {
  41. let chunks = [];
  42. req.on('data', (chunk) => {
  43. chunks.push(chunk);
  44. });
  45. req.on('end', () => {
  46. resolve(Buffer.concat(chunks));
  47. });
  48. req.on('error', reject);
  49. });
  50. }
  51. module.exports = router;