feat:add redisIoAdapter (#105)

packages/api/package.json

@@ -29,6 +29,7 @@
     "@nestjs/schedule": "^3.0.1",
     "@nestjs/swagger": "^6.3.0",
     "@nestjs/websockets": "^9.4.3",
+    "@socket.io/redis-adapter": "^8.2.1",
     "bull": "^4.10.4",
     "config": "^3.3.9",
     "cookie-parser": "^1.4.6",

packages/api/src/chats/adapters/redis-io.adapter.ts

@@ -0,0 +1,30 @@
+import { ConfigService } from '@nestjs/config';
+import { IoAdapter } from '@nestjs/platform-socket.io';
+import { createAdapter } from '@socket.io/redis-adapter';
+import { createClient } from 'redis';
+import { ServerOptions } from 'socket.io';
+
+export class RedisIoAdapter extends IoAdapter {
+  constructor(app, private readonly configService: ConfigService) {
+    super(app);
+  }
+  private adapterConstructor: ReturnType<typeof createAdapter>;
+
+  async connectToRedis(): Promise<void> {
+    const pubClient = createClient({
+      url: `redis://${this.configService.get('REDIS_CONNECTION_URL')}`,
+      password: this.configService.get('REDIS_HOST_PASSWORD'),
+    });
+    const subClient = pubClient.duplicate();
+
+    await Promise.all([pubClient.connect(), subClient.connect()]);
+
+    this.adapterConstructor = createAdapter(pubClient, subClient);
+  }
+
+  createIOServer(port: number, options?: ServerOptions): any {
+    const server = super.createIOServer(port, options);
+    server.adapter(this.adapterConstructor);
+    return server;
+  }
+}

packages/api/src/chats/chat-socket.gateway.ts

@@ -18,6 +18,7 @@ import { Server, Socket } from 'socket.io';
 @WebSocketGateway({
   cors: {
     origin: '*',
+    credentials: true,
   },
 })
 export class ChatSocketGateway {

packages/api/src/main.ts

@@ -6,6 +6,7 @@ import { DocumentBuilder, SwaggerModule } from '@nestjs/swagger';
 import * as cookieParser from 'cookie-parser';
 
 import { AppModule } from './app.module';
+import { RedisIoAdapter } from './chats/adapters/redis-io.adapter';
 
 async function bootstrap() {
   const app = await NestFactory.create(AppModule);
@@ -31,6 +32,10 @@ async function bootstrap() {
   const document = SwaggerModule.createDocument(app, config);
   SwaggerModule.setup('docs', app, document);
 
+  const redisIoAdapter = new RedisIoAdapter(app, configService);
+  await redisIoAdapter.connectToRedis();
+  app.useWebSocketAdapter(redisIoAdapter);
+
   await app.listen(configService.get('API_PORT'));
 }
 

packages/web-ui/app/state/socket.ts

@@ -2,7 +2,9 @@ import { atom } from 'recoil';
 import io, { Socket } from 'socket.io-client';
 
 // @ts-ignore
-const socket = io(process.env.NEXT_PUBLIC_SOCKET_URL);
+const socket = io(process.env.NEXT_PUBLIC_SOCKET_URL, {
+  withCredentials: true,
+});
 
 export const socketState = atom<Socket>({
   key: 'Socket',