songtianlun/happy-server
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: fix broken rpc

Browse Source
This commit is contained in:
Steve Korshakov 2025-09-01 14:51:24 -07:00
parent c40801b329
commit 80a3a64d84
2 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
sources/app/api/socket.ts
View File

@ -33,6 +33,7 @@ export function startSocket(app: Fastify, eventRouter: EventRouter) {
serveClient: false // Don't serve the client files serveClient: false // Don't serve the client files
}); });
let rpcListeners = new Map<string, Map<string, Socket>>();
io.on("connection", async (socket) => { io.on("connection", async (socket) => {
log({ module: 'websocket' }, `New connection attempt from socket: ${socket.id}`); log({ module: 'websocket' }, `New connection attempt from socket: ${socket.id}`);
const token = socket.handshake.auth.token as string; const token = socket.handshake.auth.token as string;
@ -645,7 +646,12 @@ export function startSocket(app: Fastify, eventRouter: EventRouter) {
}); });
// Handlers // Handlers
rpcHandler(userId, socket, eventRouter); let userRpcListeners = rpcListeners.get(userId);
if (!userRpcListeners) {
userRpcListeners = new Map<string, Socket>();
rpcListeners.set(userId, userRpcListeners);
}
rpcHandler(userId, socket, eventRouter, userRpcListeners);
usageHandler(userId, socket, eventRouter); usageHandler(userId, socket, eventRouter);
pingHandler(socket); pingHandler(socket);

4
sources/app/api/socket/rpcHandler.ts
View File

@ -2,8 +2,8 @@ import { EventRouter } from "@/modules/eventRouter";
import { log } from "@/utils/log"; import { log } from "@/utils/log";
import { Socket } from "socket.io"; import { Socket } from "socket.io";
export function rpcHandler(userId: string, socket: Socket, eventRouter: EventRouter) { export function rpcHandler(userId: string, socket: Socket, eventRouter: EventRouter, rpcListeners: Map<string, Socket>) {
const rpcListeners = new Map<string, Socket>();
// RPC register - Register this socket as a listener for an RPC method // RPC register - Register this socket as a listener for an RPC method
socket.on('rpc-register', async (data: any) => { socket.on('rpc-register', async (data: any) => {
try { try {