CrescentLeaf/LingChair
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

feat(unstable): 断开连接时储存事件并重发

Browse Source
This commit is contained in:
CrescentLeaf
2025-11-30 01:45:42 +08:00
parent 4b5f0bcdd6
commit d57b023769
5 changed files with 52 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
server/api/ApiManager.ts
View File

@@ -7,6 +7,7 @@ import EventCallbackFunction from "../typedef/EventCallbackFunction.ts"
import BaseApi from "./BaseApi.ts"
import DataWrongError from "./DataWrongError.ts"
import chalk from "chalk"
import EventStorer from "./EventStorer.ts";
function stringifyNotIncludeArrayBuffer(value: any) {
return JSON.stringify(value, (_k, v) => {
@@ -46,7 +47,7 @@ export default class ApiManager {
}
static clients: { [key: string]: { [key: string]: SocketIo.Socket<SocketIo.DefaultEventsMap, SocketIo.DefaultEventsMap, SocketIo.DefaultEventsMap, any> } } = {}
static checkUserIsOnline(userId: string) {
return this.getUserClientSockets(userId) != null
return this.getUserClientSockets(userId) != null && Object.keys(this.getUserClientSockets(userId)).length > 0
}
/**
* 獲取用戶所有的客戶端表 (格式遵循 設備ID_當前Session)
@@ -79,13 +80,15 @@ export default class ApiManager {
else {
console.log(chalk.green('[断]') + ` ${ip} disconnected`)
delete this.clients[clientInfo.userId][deviceId + '_' + sessionId]
if (Object.keys(this.clients[clientInfo.userId]).length == 0)
EventStorer.getInstanceForUser(clientInfo.userId).clearEvents()
}
})
console.log(chalk.yellow('[连]') + ` ${ip} connected`)
socket.on("The_White_Silk", async (name: string, args: { [key: string]: unknown }, callback_: (ret: ApiCallbackMessage) => void) => {
function callback(ret: ApiCallbackMessage) {
console.log(chalk.blue('[发]') + ` ${ip} <- ${ret.code == 200 ? chalk.green(ret.msg) : chalk.red(ret.msg)} [${ret.code}]${ret.data ? (' <extras: ' + stringifyNotIncludeArrayBuffer(ret.data) + '>') : ''}`)
console.log(chalk.blue('[回]') + ` ${ip} <- ${ret.code == 200 ? chalk.green(ret.msg) : chalk.red(ret.msg)} [${ret.code}]${ret.data ? (' <extras: ' + stringifyNotIncludeArrayBuffer(ret.data) + '>') : ''}`)
return callback_(ret)
}
async function checkIsPromiseAndAwait(value: Promise<unknown> | unknown) {

21
server/api/BaseApi.ts
View File

@@ -5,6 +5,8 @@ import User from "../data/User.ts"
import Token from "./Token.ts"
import TokenManager from './TokenManager.ts'
import * as SocketIo from "socket.io"
import EventStorer from "./EventStorer.ts"
import chalk from "chalk"
export default abstract class BaseApi {
abstract getName(): string
@@ -31,18 +33,23 @@ export default abstract class BaseApi {
if (!name.startsWith(this.getName() + ".")) throw Error("注冊的事件應該與接口集合命名空間相匹配: " + name)
ApiManager.addEventListener(name, func)
}
emitToClient(client: SocketIo.Socket, name: ClientEvent, args: { [key: string]: unknown }) {
client.emit("The_White_Silk", name, args)
emitToClient(client: SocketIo.Socket, name: ClientEvent, args: { [key: string]: unknown }, user: User | string, deviceSession: string) {
console.log(chalk.magenta('[发]') + ` ${client.handshake.address} <- ${chalk.yellow(name)} ${' <extras: ' + JSON.stringify(args) + '>'}`)
client.timeout(5000).emit("The_White_Silk", name, args, (err: Error) => {
if (err) EventStorer.getInstanceForUser(user).addEvent(name, args, deviceSession)
})
}
boardcastToUsers(users: string[], name: ClientEvent, args: { [key: string]: unknown }) {
for (const user of users) {
if (ApiManager.checkUserIsOnline(user)) {
const sockets = ApiManager.getUserClientSockets(user)
for (const socket of Object.keys(sockets))
this.emitToClient(sockets[socket], name, args)
} else {
// TODO: EventStore
}
for (const deviceSession of Object.keys(sockets))
if (sockets[deviceSession].connected)
this.emitToClient(sockets[deviceSession], name, args, user, deviceSession)
else
EventStorer.getInstanceForUser(user).addEvent(name, args, deviceSession)
} else
EventStorer.getInstanceForUser(user).clearEvents()
}
}
}

7
server/api/EventBean.ts
View File

@@ -1,4 +1,7 @@
import { ClientEvent } from "./ApiDeclare.ts"
export default class EventBean {
declare event_name: string
declare data: unknown
declare event_name: ClientEvent
declare data: { [key: string]: unknown }
declare device_session: string
}

34
server/api/EventStorer.ts
View File

@@ -2,8 +2,9 @@ import { DatabaseSync } from "node:sqlite"
import path from 'node:path'
import config from "../config.ts"
import User from "../data/User.ts";
import EventBean from "./EventBean.ts";
import User from "../data/User.ts"
import EventBean from "./EventBean.ts"
import { ClientEvent } from "./ApiDeclare.ts"
export default class EventStorer {
static database: DatabaseSync = this.init()
@@ -13,41 +14,44 @@ export default class EventStorer {
return db
}
static getInstanceForUser(user: User) {
static getInstanceForUser(user: User | string) {
return new EventStorer(user)
}
declare user: User
constructor(user: User) {
this.user = user
declare user_id: string
constructor(user: User | string) {
this.user_id = user instanceof User ? user.bean.id : user
EventStorer.database.exec(`
CREATE TABLE IF NOT EXISTS ${this.getTableName()} (
/* 序号 */ count INTEGER PRIMARY KEY AUTOINCREMENT,
/* 事件 */ event_name TEXT NOT NULL,
/* 数据 */ data TEXT NOT NULL,
/* 会话 */ device_session TEXT
);
`)
}
protected getTableName() {
return `events_${this.user.bean.id}`
return `events_${this.user_id.replaceAll('-', '_')}`
}
addEvent(eventName: string, data: unknown) {
addEvent(eventName: ClientEvent, data: object, device_session: string) {
EventStorer.database.prepare(`INSERT INTO ${this.getTableName()} (
event_name,
data
) VALUES (?, ?);`).run(
data,
device_session
) VALUES (?, ?, ?);`).run(
eventName,
JSON.stringify(data)
JSON.stringify(data),
device_session || null
)
}
getEvents() {
return EventStorer.database.prepare(`SELECT * FROM ${this.getTableName()};`).all().map((v: any) => ({
getEvents(device_session?: string) {
return EventStorer.database.prepare(`SELECT * FROM ${this.getTableName()}${device_session ? ` WHERE device_session = ?` : ''};`).all(...(device_session ? [ device_session ] : [])).map((v: any) => ({
...v,
data: JSON.parse(v.data)
})) as unknown as EventBean[]
}
clearEvents() {
EventStorer.database.prepare(`DELETE FROM ${this.getTableName()};`).run()
clearEvents(device_session?: string) {
EventStorer.database.prepare(`DELETE FROM ${this.getTableName()}${device_session ? ` WHERE device_session = ?` : ''};`).run(...(device_session ? [ device_session ] : []))
}
}

11
server/api/UserApi.ts
View File

@@ -6,6 +6,7 @@ import ChatPrivate from "../data/ChatPrivate.ts"
import Chat from "../data/Chat.ts"
import chalk from "chalk"
import ApiManager from "./ApiManager.ts"
import EventStorer from "./EventStorer.ts";
export default class UserApi extends BaseApi {
override getName(): string {
@@ -37,10 +38,16 @@ export default class UserApi extends BaseApi {
clientInfo.userId = access_token.author
console.log(chalk.green('[验]') + ` ${access_token.author} authed on Client ${deviceId} (ip = ${ip})`)
const deviceSession = deviceId + '_' + sessionId
if (ApiManager.clients[clientInfo.userId] == null) ApiManager.clients[clientInfo.userId] = {
[deviceId + '_' + sessionId]: socket
[deviceSession]: socket
}
else ApiManager.clients[clientInfo.userId][deviceId + '_' + sessionId] = socket
else ApiManager.clients[clientInfo.userId][deviceSession] = socket
// 事件恢复
console.log(EventStorer.getInstanceForUser(access_token.author).getEvents(deviceSession))
for (const event of EventStorer.getInstanceForUser(access_token.author).getEvents(deviceSession))
this.emitToClient(socket, event.event_name, event.data, access_token.author, deviceSession)
return {
msg: "成功",