82 lines
2.7 KiB
TypeScript
82 lines
2.7 KiB
TypeScript
import { IoAdapter } from '@nestjs/platform-socket.io';
|
|
import { ServerOptions } from 'socket.io';
|
|
import { createAdapter } from '@socket.io/redis-adapter';
|
|
import Redis from 'ioredis';
|
|
import { ConfigService } from '@nestjs/config';
|
|
import { INestApplicationContext, Logger } from '@nestjs/common';
|
|
|
|
export class RedisIoAdapter extends IoAdapter {
|
|
private adapterConstructor: ReturnType<typeof createAdapter>;
|
|
private readonly logger = new Logger(RedisIoAdapter.name);
|
|
|
|
constructor(
|
|
private app: INestApplicationContext,
|
|
private configService: ConfigService,
|
|
) {
|
|
super(app);
|
|
}
|
|
|
|
async connectToRedis(): Promise<void> {
|
|
const host = this.configService.get<string>('REDIS_HOST');
|
|
const port = this.configService.get<number>('REDIS_PORT');
|
|
const password = this.configService.get<string>('REDIS_PASSWORD');
|
|
|
|
// Only set up Redis adapter if host is configured
|
|
if (!host) {
|
|
this.logger.log('Redis adapter disabled (REDIS_HOST not set)');
|
|
return;
|
|
}
|
|
|
|
this.logger.log(`Connecting Redis adapter to ${host}:${port || 6379}`);
|
|
|
|
try {
|
|
const pubClient = new Redis({
|
|
host,
|
|
port: port || 6379,
|
|
password: password,
|
|
retryStrategy(times) {
|
|
// Retry connecting but don't crash if Redis is temporarily down during startup
|
|
return Math.min(times * 50, 2000);
|
|
},
|
|
});
|
|
|
|
const subClient = pubClient.duplicate();
|
|
|
|
// Wait for connection to ensure it's valid
|
|
await new Promise<void>((resolve, reject) => {
|
|
pubClient.once('connect', () => {
|
|
this.logger.log('Redis Pub client connected');
|
|
resolve();
|
|
});
|
|
pubClient.once('error', (err) => {
|
|
this.logger.error('Redis Pub client error', err);
|
|
reject(err);
|
|
});
|
|
});
|
|
|
|
// Handle subsequent errors gracefully
|
|
pubClient.on('error', (err) => {
|
|
this.logger.error('Redis Pub client error', err);
|
|
});
|
|
subClient.on('error', (err) => {
|
|
this.logger.error('Redis Sub client error', err);
|
|
});
|
|
|
|
this.adapterConstructor = createAdapter(pubClient, subClient);
|
|
this.logger.log('Redis adapter initialized successfully');
|
|
} catch (error) {
|
|
this.logger.error('Failed to initialize Redis adapter', error);
|
|
// We don't throw here to allow the app to start without Redis if connection fails,
|
|
// though functionality will be degraded if multiple instances are running.
|
|
}
|
|
}
|
|
|
|
createIOServer(port: number, options?: ServerOptions): any {
|
|
const server = super.createIOServer(port, options);
|
|
if (this.adapterConstructor) {
|
|
server.adapter(this.adapterConstructor);
|
|
}
|
|
return server;
|
|
}
|
|
}
|