基于NestJS 实现 API 网关服务，支持身份验证、负载均衡

typescript
import { Module, MiddlewareConsumer, NestModule } from '@nestjs/common';
import { createProxyMiddleware } from 'http-proxy-middleware';

@Module({})
export class ApiGatewayModule implements NestModule {
  public configure(consumer: MiddlewareConsumer): void {
    consumer
      .apply(
        createProxyMiddleware('/service1', { target: 'http://localhost:3001', changeOrigin: true }),
        createProxyMiddleware('/service2', { target: 'http://localhost:3002', changeOrigin: true }),
      )
      .forRoutes('*');
  }
}

在上面的代码中，我们为 /service1 和 /service2 配置了两个代理规则。每个服务的请求都会被转发到相应的微服务。

三、添加身份验证和权限

API 网关通常也负责用户的身份验证和权限校验。你可以利用 NestJS 的守卫（Guard）和拦截器（Interceptor）来实现这些功能。

例如，创建一个简单的守卫:

typescript
import { Injectable, CanActivate, ExecutionContext } from '@nestjs/common';

@Injectable()
export class AuthGuard implements CanActivate {
  canActivate(context: ExecutionContext): boolean {
    const request = context.switchToHttp().getRequest();
    // 逻辑来检查用户是否认证
    return true; // 或者 false，如果用户未认证
  }
}

将守卫应用于路由:

typescript
consumer
  .apply(AuthGuard)
  .forRoutes('*');

四、启动网关

现在，启动你的网关，让它开始工作:

bash
npm run start

API 网关应该可以处理请求并将它们路由到适当的后端服务了，同时还可以处理身份验证等事务。

总结

NestJS 提供了强大的抽象和工具来构建复杂的网关逻辑，比如使用拦截器来处理响应，或者利用管道来转换请求。继续探索 NestJS 的文档，你将发现更多高级特性来提升你的网关。

前言为什么选择 NestJS 作为 API 网关?实现方式一、创建 API 网关模块二、配置路由和代理三、添加身份验证和权限四、启动网关总结