QQ/NapCatQQ
1
0
Fork 0
mirror of https://github.com/NapNeko/NapCatQQ.git synced 2024-09-28 12:07:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore: 初步完成network

Browse Source
This commit is contained in:
手瓜一十雪 2024-08-11 00:11:51 +08:00
parent a2ad39f78d
commit 73a8fcd35b
1 changed files with 32 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
src/onebot/network/passive-websocket.ts
View File

@ -29,8 +29,7 @@ export class OB11PassiveWebSocketAdapter implements IOB11NetworkAdapter {
}
}
wsClient.on('message', (message) => {
// TODO: extract action name and payload from the message, then call the corresponding action.
// TODO: consider using a utility function
this.handleMessage(message);
});
wsClient.once('close', () => {
this.wsClientsMutex.runExclusive(async () => {
@ -51,15 +50,22 @@ export class OB11PassiveWebSocketAdapter implements IOB11NetworkAdapter {
}
registerHeartBeat() {
//WS正向心跳
setInterval(() => {
this.wsClientsMutex.runExclusive(async () => {
this.wsClients.forEach((wsClient) => {
if (wsClient.readyState === WebSocket.OPEN) {
wsClient.ping();
}
});
});
}, this.heartbeatInterval);
}
onEvent<T extends OB11EmitEventContent>(event: T) {
this.wsClientsMutex.runExclusive(async () => {
this.wsClients.forEach((wsClient) => {
// wsClient.send(JSON.stringify(event));
// TODO: wrap the event, and send the wrapped to the client.
// TODO: consider using a utility function
const wrappedEvent = this.wrapEvent(event);
wsClient.send(JSON.stringify(wrappedEvent));
});
});
}
@ -76,4 +82,24 @@ export class OB11PassiveWebSocketAdapter implements IOB11NetworkAdapter {
this.hasBeenClosed = true;
this.wsServer.close();
}
private handleMessage(message: any) {
try {
const parsedMessage = JSON.parse(message);
const action = this.actionMap.get(parsedMessage.actionName);
if (action) {
action.handle(parsedMessage.payload);
}
} catch (e) {
console.error('Failed to handle message:', e);
}
}
private wrapEvent<T extends OB11EmitEventContent>(event: T) {
return {
type: 'event',
data: event
};
}
}