72 lines
1.8 KiB
TypeScript
72 lines
1.8 KiB
TypeScript
|
// MessageDispatcher.ts
|
|||
|
|
|
|||
|
|
/** 消息处理器类型 */
|
|||
|
|
export type MessageHandler = (data: any) => void
|
|||
|
|
|
|||
|
|
export class MessageDispatcher {
|
|||
|
|
private static _instance: MessageDispatcher
|
|||
|
|
private _handlers: Map<number, MessageHandler[]> = new Map()
|
|||
|
|
|
|||
|
|
/** 单例模式 */
|
|||
|
|
public static get Instance(): MessageDispatcher {
|
|||
|
|
if (!this._instance) {
|
|||
|
|
this._instance = new MessageDispatcher()
|
|||
|
|
}
|
|||
|
|
return this._instance
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 注册消息处理器（允许多个 Handler）
|
|||
|
|
* @param msgId 消息ID
|
|||
|
|
* @param handler 回调函数
|
|||
|
|
*/
|
|||
|
|
public On(msgId: number, handler: MessageHandler): void {
|
|||
|
|
if (!this._handlers.has(msgId)) {
|
|||
|
|
this._handlers.set(msgId, [])
|
|||
|
|
}
|
|||
|
|
this._handlers.get(msgId)?.push(handler) // 添加到队列
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 移除指定的消息处理器（通过函数引用精准匹配）
|
|||
|
|
* @param msgId 消息ID
|
|||
|
|
* @param handler 要移除的回调函数
|
|||
|
|
*/
|
|||
|
|
public Off(msgId: number, handler: MessageHandler): void {
|
|||
|
|
const handlers = this._handlers.get(msgId)
|
|||
|
|
if (!handlers) return
|
|||
|
|
|
|||
|
|
// 过滤掉指定的 handler
|
|||
|
|
const newHandlers = handlers.filter((h) => h !== handler)
|
|||
|
|
this._handlers.set(msgId, newHandlers)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 消息派发器中handler数量
|
|||
|
|
* @param msgId 消息ID
|
|||
|
|
* @returns 返回消息派发器中handler数量
|
|||
|
|
*/
|
|||
|
|
public Size(msgId: number): number {
|
|||
|
|
if (msgId === 0) {
|
|||
|
|
return this._handlers.size
|
|||
|
|
}
|
|||
|
|
const handlers = this._handlers.get(msgId)
|
|||
|
|
if (handlers) {
|
|||
|
|
return handlers.length
|
|||
|
|
}
|
|||
|
|
return 0
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 派发消息（按注册顺序执行所有 Handler）
|
|||
|
|
* @param msgId 消息ID
|
|||
|
|
* @param msg 消息数据
|
|||
|
|
*/
|
|||
|
|
public Dispatch(msgId: number, msg: any): void {
|
|||
|
|
const handlers = this._handlers.get(msgId)
|
|||
|
|
if (handlers) {
|
|||
|
|
handlers.forEach((handler) => handler(msg)) // 顺序执行
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|