perf: 优化监听进程
This commit is contained in:
@@ -1,5 +1,6 @@
|
||||
import { EventEmitter } from "eventemitter3";
|
||||
import { QueryRouterServer } from "../route.ts"
|
||||
import { QueryRouterServer, RouterContextT, RunMessage } from "../route.ts"
|
||||
import { merge } from 'es-toolkit'
|
||||
export class MockProcess {
|
||||
emitter?: EventEmitter
|
||||
process?: NodeJS.Process;
|
||||
@@ -37,13 +38,31 @@ export class MockProcess {
|
||||
this.process = undefined;
|
||||
}
|
||||
}
|
||||
export type ListenProcessParams = {
|
||||
message?: RunMessage,
|
||||
context?: any
|
||||
}
|
||||
export type ListenProcessResponse = {
|
||||
// 调用进程的功能
|
||||
success?: boolean
|
||||
data?: {
|
||||
// 调用router的结果
|
||||
code?: number
|
||||
data?: any
|
||||
message?: string
|
||||
[key: string]: any
|
||||
};
|
||||
error?: any
|
||||
timestamp?: string
|
||||
[key: string]: any
|
||||
}
|
||||
export type ListenProcessOptions = {
|
||||
app?: QueryRouterServer; // 传入的应用实例
|
||||
mockProcess?: MockProcess; // 可选的事件发射器
|
||||
params?: any; // 可选的参数
|
||||
params?: ListenProcessParams; // 可选的参数
|
||||
timeout?: number; // 可选的超时时间 (单位: 毫秒) 默认 10 分钟
|
||||
};
|
||||
export const listenProcess = async ({ app, mockProcess, params, timeout = 10 * 60 * 60 * 1000 }: ListenProcessOptions) => {
|
||||
export const listenProcess = async ({ app, mockProcess, params = {}, timeout = 10 * 60 * 60 * 1000 }: ListenProcessOptions) => {
|
||||
const process = mockProcess || new MockProcess();
|
||||
let isEnd = false;
|
||||
const timer = setTimeout(() => {
|
||||
@@ -57,11 +76,11 @@ export const listenProcess = async ({ app, mockProcess, params, timeout = 10 * 6
|
||||
// 监听来自主进程的消息
|
||||
const getParams = async (): Promise<any> => {
|
||||
return new Promise((resolve) => {
|
||||
process.on((msg) => {
|
||||
process.on((params) => {
|
||||
if (isEnd) return;
|
||||
isEnd = true;
|
||||
clearTimeout(timer);
|
||||
resolve(msg)
|
||||
resolve(params || {})
|
||||
})
|
||||
})
|
||||
}
|
||||
@@ -70,11 +89,11 @@ export const listenProcess = async ({ app, mockProcess, params, timeout = 10 * 6
|
||||
/**
|
||||
* 如果不提供path,默认是main
|
||||
*/
|
||||
const {
|
||||
payload = {},
|
||||
...rest
|
||||
} = await getParams()
|
||||
const msg = { ...params, ...rest, payload: { ...params?.payload, ...payload } }
|
||||
const _params = await getParams()
|
||||
const mergeParams = merge(params, _params)
|
||||
|
||||
const msg = mergeParams?.message || {};
|
||||
const ctx: RouterContextT = mergeParams?.context || {}
|
||||
/**
|
||||
* 如果没有提供path和id,默认取第一个路由, 而且路由path不是router的
|
||||
*/
|
||||
@@ -83,7 +102,7 @@ export const listenProcess = async ({ app, mockProcess, params, timeout = 10 * 6
|
||||
msg.id = route?.id
|
||||
}
|
||||
// 执行主要逻辑
|
||||
const result = await app.run(msg)
|
||||
const result = await app.run(msg, ctx);
|
||||
// 发送结果回主进程
|
||||
const response = {
|
||||
success: true,
|
||||
@@ -95,6 +114,7 @@ export const listenProcess = async ({ app, mockProcess, params, timeout = 10 * 6
|
||||
process.exit?.(0)
|
||||
})
|
||||
} catch (error) {
|
||||
console.error('Error in listenProcess:', error);
|
||||
process.send?.({
|
||||
success: false,
|
||||
error: error.message
|
||||
|
||||
Reference in New Issue
Block a user