如何在NestJS中启用类似Node.js的粘性会话？集群是否支持？

核心问题回答 #

NestJS完全支持在集群模式下启用粘性会话，但框架本身并未内置该功能，需要结合第三方粘性会话库（如sticky-session）或自定义逻辑来实现，原理和你提供的Node.js原生示例一致。

实现方法一：使用sticky-session库直接集成 #

这是最贴近你给出的原生示例的实现方式，步骤如下：

1. 安装依赖

npm install sticky-session

2. 修改main.ts文件

import { NestFactory } from '@nestjs/core';
import { AppModule } from './app.module';
import * as sticky from 'sticky-session';
import * as http from 'http';

async function bootstrap() {
  const app = await NestFactory.create(AppModule);
  
  // 此处添加你的应用配置（如CORS、全局管道等）
  const httpServer = app.getHttpServer();

  // 启动粘性会话集群
  if (!sticky.listen(httpServer, 3000, {
    loadBalancingMethod: 'least-connection' // 和你示例一致的负载均衡策略
  })) {
    // 主进程逻辑
    httpServer.once('listening', () => {
      console.log(`Master process ${process.pid} listening on port 3000`);
    });
  } else {
    // 工作进程逻辑
    console.log(`Worker process ${process.pid} started`);
  }
}

bootstrap();

实现方法二：结合@nestjs/cluster模块 #

如果你已经在使用NestJS官方的集群模块，可以将其与粘性会话逻辑结合：

1. 安装依赖

npm install @nestjs/cluster sticky-session

2. 修改main.ts文件

import { NestFactory } from '@nestjs/core';
import { AppModule } from './app.module';
import { ClusterService } from '@nestjs/cluster';
import { setupMaster } from 'sticky-session';

async function bootstrap() {
  const app = await NestFactory.create(AppModule);
  
  // 初始化NestJS集群服务
  const clusterService = app.get(ClusterService);
  clusterService.setup({
    workers: require('os').cpus().length, // 按CPU核心数生成工作进程
    mode: 'fork',
  });

  const httpServer = app.getHttpServer();

  // 仅在主进程中配置粘性会话
  if (clusterService.isMaster) {
    setupMaster(httpServer, {
      loadBalancingMethod: 'least-connection',
    });
  }

  await app.listen(3000);
}

bootstrap();

关键注意事项 #

• 粘性会话依赖客户端请求的标识信息（如IP、Cookie中的Session ID），需确保客户端请求能稳定传递这些信息
• 若使用Fastify作为NestJS的HTTP适配器，只需将app.getHttpServer()替换为Fastify的服务器实例，sticky-session同样兼容
• 生产环境中如果前端有额外的负载均衡器（如Nginx），需确保该负载均衡器也配置了粘性会话策略，避免请求被错误路由

内容的提问来源于stack exchange，提问作者Rick