26 lines
902 B
TypeScript
26 lines
902 B
TypeScript
import { IoAdapter } from '@nestjs/platform-socket.io';
|
|
import { ServerOptions } from 'socket.io';
|
|
import { createAdapter } from '@socket.io/redis-adapter';
|
|
import Redis, { RedisOptions } from 'ioredis';
|
|
import { createRetryStrategy } from '../../common/helpers';
|
|
|
|
export class WsRedisIoAdapter extends IoAdapter {
|
|
private adapterConstructor: ReturnType<typeof createAdapter>;
|
|
|
|
async connectToRedis(): Promise<void> {
|
|
const options: RedisOptions = {
|
|
retryStrategy: createRetryStrategy(),
|
|
};
|
|
|
|
const pubClient = new Redis(process.env.REDIS_URL, options);
|
|
const subClient = new Redis(process.env.REDIS_URL, options);
|
|
|
|
this.adapterConstructor = createAdapter(pubClient, subClient);
|
|
}
|
|
|
|
createIOServer(port: number, options?: ServerOptions): any {
|
|
const server = super.createIOServer(port, options);
|
|
server.adapter(this.adapterConstructor);
|
|
return server;
|
|
}
|
|
}
|