GuanM/NapCatQQ
1
0
Fork 0
mirror of https://github.com/NapNeko/NapCatQQ.git synced 2024-11-21 09:36:35 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

refactor: NTEvent

Browse Source
This commit is contained in:
手瓜一十雪 2024-05-31 20:38:31 +08:00
parent 7f2d57ef62
commit 6248991b01
1 changed files with 16 additions and 42 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

58
src/common/utils/EventTask.ts
View File

@ -2,13 +2,8 @@ import { NodeIKernelMsgListener } from "@/core";
import { NodeIQQNTWrapperSession } from "@/core/wrapper";
import { randomUUID } from "crypto";
export enum NTEventMode {
Once = 1,
Twice = 2
}
interface Internal_MapKey {
mode: NTEventMode,
timeout: number,
createtime: number,
func: Function
@ -78,50 +73,30 @@ export class NTEventWrapper {
this.EventTask.delete(uuid);
return;
}
if (task.mode == NTEventMode.Once) {
this.EventTask.delete(uuid);
}
task.func(...args);
})
}
async CallOnceEvent<EventType extends (...args: any[]) => any, ListenerType extends (...args: any[]) => void>(EventName = '', ListenerName = '', timeout: number = 3000, ...args: Parameters<EventType>) {
return new Promise<ArrayLike<Parameters<ListenerType>>>((resolve, reject) => {
const id = randomUUID();
let complete = false;
let retData: ArrayLike<Parameters<ListenerType>> | undefined = undefined;
let databack = () => {
if (!complete) {
this.EventTask.delete(id);
reject(new Error('NTEvent EventName:' + EventName + ' ListenerName:' + ListenerName + ' timeout'));
} else {
resolve(retData as ArrayLike<Parameters<ListenerType>>);
}
}
let Timeouter = setTimeout(databack, timeout);
this.EventTask.set(id, {
mode: NTEventMode.Once,
timeout: timeout,
createtime: Date.now(),
func: (...args: any[]) => {
clearTimeout(Timeouter);
complete = true;
retData = args as ArrayLike<Parameters<ListenerType>>;
databack();
}
});
async CallNoListenerEvent<EventType extends (...args: any[]) => Promise<any>,>(EventName = '', timeout: number = 3000, ...args: Parameters<EventType>) {
return new Promise<ReturnType<EventType>>((resolve, reject) => {
let EventFunc = this.CreatEventFunction<EventType>(EventName);
EventFunc!(...args);
});
let complete = false;
let Timeouter = setTimeout(() => {
if (!complete) {
reject(new Error('NTEvent EventName:' + EventName + ' timeout'));
}
async CallTwiceEvent<EventType extends (...args: any[]) => any, ListenerType extends (...args: any[]) => void>(EventName = '', ListenerName = '', timeout: number = 3000, ...args: Parameters<EventType>) {
}, timeout);
let retData = await EventFunc!(...args);
complete = true;
resolve(retData);
}
}
async CallNormalEvent<EventType extends (...args: any[]) => Promise<any>, ListenerType extends (...args: any[]) => void>(EventName = '', ListenerName = '', waitTimes = 1, timeout: number = 3000, ...args: Parameters<EventType>) {
return new Promise<ArrayLike<Parameters<ListenerType>>>((resolve, reject) => {
const id = randomUUID();
let complete = 0;
let retData: ArrayLike<Parameters<ListenerType>> | undefined = undefined;
let databack = () => {
if (complete < 2) {
this.EventTask.delete(id);
if (complete < waitTimes) {
reject(new Error('NTEvent EventName:' + EventName + ' ListenerName:' + ListenerName + ' timeout'));
} else {
resolve(retData as ArrayLike<Parameters<ListenerType>>);
@ -130,20 +105,19 @@ export class NTEventWrapper {
let Timeouter = setTimeout(databack, timeout);
this.EventTask.set(id, {
mode: NTEventMode.Once,
timeout: timeout,
createtime: Date.now(),
func: (...args: any[]) => {
complete++;
retData = args as ArrayLike<Parameters<ListenerType>>;
if (complete == 2) {
if (complete == waitTimes) {
clearTimeout(Timeouter);
databack();
}
}
});
let EventFunc = this.CreatEventFunction<EventType>(EventName);
EventFunc!(...args);
await EventFunc!(...args);
});
}