リバースプロキシの実装

2020-11-24-advent-calendar-2020-server-reverse-proxy-implementation.ts

async function apiReverseProxy(
  apiBasePath: string,
  clientReq: any,
  clientRes: any
) {
  return new Promise((resolve) => {
    const serverReq = request({
      // VMなどで実行されるならここはos.hostname()など環境に合わせること。幸いNode.jsなのでos.type()などを用いて柔軟に対応できる。
      host: "localhost",
      // stub serverのport番号をしていする。package.jsonのconfigで設定するのもよい
      // process.env.npm_package_config_stub_server_portのような形で取得可能
      port: 8000,
      method: clientReq.method,
      path: clientReq.url.slice(apiBasePath.length),
      headers: clientReq.headers,
    })
      .on("error", () => {
        clientRes.writeHead(502);
        resolve();
      })
      .on("timeout", () => {
        clientRes.writeHead(504).end();
        resolve();
      })
      .on("response", (serverRes) => {
        clientRes.writeHead(serverRes.statusCode, serverRes.headers);
        serverRes.pipe(clientRes);
        resolve();
      });
    clientReq.pipe(serverReq);
  });
}

// routeの方からの呼び出しはこう
async function route(pathname: string, req: any, res: any) {
  const apiBasePath = "/api";
  if (pathname.startsWith(apiBasePath)) {
    await apiReverseProxy(apiBasePath, req, res);
    return;
  }
  // 省略