首页 > 解决方案 > 如何使用 koa 路由器复制和转发请求

问题描述

出于多种原因,我有一台服务器必须将请求转发到另一台服务器。响应应该是最终服务器的响应。我还需要在请求中添加一个额外的标头,但在返回之前再次从响应中删除此标头。因此,重定向不会削减它。

我目前正在根据需要手动复制标题和正文,但我想知道是否有一种简单的通用方法可以做到这一点?

标签: node.jskoakoa-router

解决方案


代理将为此工作。假设 @koa/router 或类似的东西和 http-proxy 模块(也有可能工作的 Koa 包装器模块:

const proxy = httpProxy.createProxyServer({
  target: 'https://some-other-server.com',
  // other options, see https://www.npmjs.com/package/http-proxy
})
proxy.on('proxyReq', (proxyReq, req, res, options) => {
  proxyReq.setHeader('x-foo', 'bar')
})
proxy.on('proxyRes', (proxyRes, req, res) => {
  proxyRes.removeHeader('x-foo')
})

router.get('/foo', async (ctx) => {
  // ctx.req and ctx.res are the Node req and res, not Koa objects
  proxy.web(ctx.req, ctx.res, {
    // other options, see docs
  })
})

如果您碰巧使用以下方式启动 Koa 服务器,您也可以将代理从路由中http.createServer移除app.listen

// where app = new Koa()
const handler = app.callback()
http.createServer((req, res) => {
  if (req.url === '/foo') {
    return proxy.web(req, res, options)
  }

  return handler(req, res)
})

推荐阅读