refactor(discord-gateway): optimize redis usage and audio stream handling

- refactor(command-handler): replace per-call Redis connection creation with a persistent publisher connection to reduce overhead
- refactor(recorder): switch from manual audio stream subscription to direct event listeners on the existing stream
- feat(recorder): implement exponential backoff for voice connection retries
- chore(config): update default DECODER_COOLDOWN_MS to 30000ms
This commit is contained in:
MythEclipse
2026-06-08 18:09:17 +07:00
parent d036fbf568
commit 2643a3c278
3 changed files with 32 additions and 36 deletions
@@ -53,11 +53,13 @@ const MEDIA_STATUS_KEY = "media:status";
export class CommandHandler {
private redisSub: Redis;
private redisPub: Redis;
private client: Client | null = null;
private voiceController: VoiceController | null = null;
constructor() {
this.redisSub = new Redis(config.REDIS_URL);
this.redisPub = new Redis(config.REDIS_URL);
this.redisSub.on("error", (err) => {
logger.error({ error: err }, "Redis subscriber connection error");
@@ -99,7 +101,7 @@ export class CommandHandler {
}
async close(): Promise<void> {
await this.redisSub.quit();
await Promise.allSettled([this.redisSub.quit(), this.redisPub.quit()]);
}
// ---- Command dispatch ----
@@ -395,19 +397,14 @@ export class CommandHandler {
}
/**
* Fire-and-forget SET on a separate Redis connection so we never block the
* subscriber loop.
* Fire-and-forget SET using the persistent Redis publisher connection.
*/
private setKey(key: string, value: string): void {
const redis = new Redis(config.REDIS_URL);
redis
this.redisPub
.set(key, value)
.catch((err: unknown) => {
const msg = err instanceof Error ? err.message : String(err);
logger.warn({ key, error: msg }, "Failed to update Redis status key");
})
.finally(() => {
void redis.quit();
});
}
}
@@ -15,7 +15,6 @@ import type { Client, VoiceChannel } from "discord.js-selfbot-v13";
import { config } from "../../shared/config/config.js";
import type { PcmBroadcaster } from "../message-capture/types.js";
import { PacketFilter } from "./packetFilter.js";
import { subscribeToAudioStream } from "./recorder/audioStream.js";
import { OpusDecoder } from "./recorder/decoder.js";
import {
collectUserMetadata,
@@ -92,9 +91,9 @@ export async function startRecording(
config.VOICE_CONNECTION_TIMEOUT_MS,
),
{
retries: 0,
minTimeout: 0,
maxTimeout: 0,
retries: 3,
minTimeout: 1000,
maxTimeout: 5000,
},
);
logger.info("Connected to voice channel. Recording started");
@@ -237,29 +236,29 @@ export async function startRecording(
logger.error({ userId, error: msg }, "File write error");
});
// Feed Opus packets one-by-one
subscribeToAudioStream(receiver, userId, {
onPacket: (chunk) => {
if (chunk.length < 8) return;
segmentManager.rotateIfNeeded(oggPacketStream);
if (!broadcaster.broadcastPcmToWeb) return;
decoder.rotateIfNeeded();
decoder.write(chunk);
},
onEnd: () => {
segmentManager.close(oggPacketStream);
decoder.destroy();
broadcaster.updateActiveUser?.(userId, {
username: userMetadata.username,
avatar: userMetadata.avatarUrl,
speaking: false,
});
},
onError: (error) => {
segmentManager.close(oggPacketStream);
decoder.destroy();
logger.error({ userId, error: error.message }, "Audio stream error");
},
// Attach event handlers directly to the existing audioStream (no double subscription)
audioStream.on("data", (chunk: Buffer) => {
if (chunk.length < 8) return;
segmentManager.rotateIfNeeded(oggPacketStream);
if (!broadcaster.broadcastPcmToWeb) return;
decoder.rotateIfNeeded();
decoder.write(chunk);
});
audioStream.on("end", () => {
segmentManager.close(oggPacketStream);
decoder.destroy();
broadcaster.updateActiveUser?.(userId, {
username: userMetadata.username,
avatar: userMetadata.avatarUrl,
speaking: false,
});
});
audioStream.on("error", (error: Error) => {
segmentManager.close(oggPacketStream);
decoder.destroy();
logger.error({ userId, error: error.message }, "Audio stream error");
});
packetFilterForOgg.on("error", (err) => {
@@ -21,7 +21,7 @@ const configSchema = z
RECORDINGS_DIR: z.string().default("./recordings"),
RECORDING_SEGMENT_MS: z.coerce.number().positive().default(5000),
DECODER_ROTATE_MS: z.coerce.number().positive().default(5000),
DECODER_COOLDOWN_MS: z.coerce.number().positive().default(0),
DECODER_COOLDOWN_MS: z.coerce.number().positive().default(30000),
WEBSERVER_PORT: z.coerce.number().positive().default(3000),
VOICE_CONNECTION_TIMEOUT_MS: z.coerce.number().positive().default(15000),
RECONNECT_TIMEOUT_MS: z.coerce.number().positive().default(5000),