nestjs编写一个网关

以下是nestjs编写一个网关的示例代码：

import { Injectable } from '@nestjs/common';
import { HttpAdapterHost, NestFactory } from '@nestjs/core';
import { FastifyAdapter, NestFastifyApplication } from '@nestjs/platform-fastify';
import { FastifyRequest, FastifyReply } from 'fastify';
import * as httpProxy from 'http-proxy';

@Injectable()
export class GatewayService {
  private proxy: httpProxy;

  constructor(private readonly httpAdapterHost: HttpAdapterHost) {
    const adapter = this.httpAdapterHost.httpAdapter as FastifyAdapter;
    const app = this.httpAdapterHost.applicationRef as NestFastifyApplication;
    this.proxy = httpProxy.createProxyServer({
      target: 'http://localhost:3000', // 设置代理目标地址
      ws: true, // 是否代理websocket请求
    });
    adapter.getHttpServer().on('upgrade', (req, socket, head) => {
      this.proxy.ws(req, socket, head);
    });
    app.get('*', (req: FastifyRequest, res: FastifyReply) => {
      this.proxy.web(req.raw, res.raw);
    });
  }
}

async function bootstrap() {
  const app = await NestFactory.create<NestFastifyApplication>(AppModule, new FastifyAdapter());
  const httpAdapterHost = app.get(HttpAdapterHost);
  await app.listen(8080);
}
bootstrap();

在上面的示例中，我们使用了http-proxy模块来实现网关的代理功能。在构造函数中，我们创建了一个httpProxy实例，并设置了代理的目标地址和是否代理websocket请求。然后我们监听了http和websocket请求，将请求转发给代理实例处理。最后，在NestJS的应用程序中，我们使用FastifyAdapter来设置http服务器和请求处理程序。

在上方的代码中，我们创建了一个名为GatewayService的Injectable类。 在构造函数中，我们获取了HttpAdapterHost的实例。这使我们能够在FastifyAdapter上注册HTTP路由和WebSocket路由。 然后，我们使用http-proxy库创建了代理实例。 在代理实例中，我们将目标地址设置为http://localhost:3000。 这意味着所有进入网关的请求都将被转发到该地址。 我们还将ws选项设置为true，这意味着代理将代理WebSockets请求。

在FastifyAdapter上，我们使用app.get('*')来匹配所有入站请求。 在匹配的请求中，我们调用代理实例的web方法来转发请求。 对于WebSocket请求，我们使用adapter.getHttpServer()来获取Fastify服务器实例，并使用它来处理升级请求。 在处理程序中，我们使用代理实例的ws方法来代理WebSocket请求。

最后，我们将GatewayService注入到AppModule中，并使用FastifyAdapter创建NestJS应用程序实例。 我们还使用httpAdapterHost来获取FastifyAdapter实例。 最后，我们在8080端口上启动应用程序。

这是一个简单的NestJS网关的示例代码。 它使用http-proxy库来实现代理功能，并使用FastifyAdapter来处理HTTP和WebSocket请求。