@eggjs/cluster 4.0.0-beta.20 → 4.0.0-beta.21

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (40) hide show
  1. package/dist/agent_worker.d.ts +1 -1
  2. package/dist/agent_worker.js +64 -51
  3. package/dist/app_worker.d.ts +1 -1
  4. package/dist/app_worker.js +166 -128
  5. package/dist/dirname.d.ts +1 -0
  6. package/dist/dirname.js +11 -0
  7. package/dist/error/ClusterAgentWorkerError.d.ts +10 -0
  8. package/dist/error/ClusterAgentWorkerError.js +19 -0
  9. package/dist/error/ClusterWorkerExceptionError.d.ts +7 -0
  10. package/dist/error/ClusterWorkerExceptionError.js +14 -0
  11. package/dist/error/index.d.ts +2 -0
  12. package/dist/error/index.js +3 -0
  13. package/dist/index.d.ts +5 -445
  14. package/dist/index.js +17 -709
  15. package/dist/master.d.ts +90 -0
  16. package/dist/master.js +553 -0
  17. package/dist/utils/messenger.d.ts +92 -0
  18. package/dist/utils/messenger.js +179 -0
  19. package/dist/utils/mode/base/agent.d.ts +38 -0
  20. package/dist/utils/mode/base/agent.js +65 -0
  21. package/dist/utils/mode/base/app.d.ts +48 -0
  22. package/dist/utils/mode/base/app.js +80 -0
  23. package/dist/utils/mode/impl/process/agent.d.ts +18 -0
  24. package/dist/utils/mode/impl/process/agent.js +103 -0
  25. package/dist/utils/mode/impl/process/app.d.ts +21 -0
  26. package/dist/utils/mode/impl/process/app.js +119 -0
  27. package/dist/utils/mode/impl/worker_threads/agent.d.ts +18 -0
  28. package/dist/utils/mode/impl/worker_threads/agent.js +84 -0
  29. package/dist/utils/mode/impl/worker_threads/app.d.ts +26 -0
  30. package/dist/utils/mode/impl/worker_threads/app.js +137 -0
  31. package/dist/utils/options.d.ts +80 -0
  32. package/dist/utils/options.js +81 -0
  33. package/dist/utils/terminate.d.ts +6 -0
  34. package/dist/utils/terminate.js +81 -0
  35. package/dist/utils/worker_manager.d.ts +25 -0
  36. package/dist/utils/worker_manager.js +74 -0
  37. package/package.json +6 -6
  38. package/dist/agent-griHEaCW.js +0 -246
  39. package/dist/app-5Was1vub.js +0 -315
  40. package/dist/terminate-w3g0oQgq.js +0 -71
@@ -0,0 +1,119 @@
1
+ import cluster, {} from 'node:cluster';
2
+ import { cfork } from 'cfork';
3
+ import { sendmessage } from 'sendmessage';
4
+ import { graceful as gracefulExit } from 'graceful-process';
5
+ import { BaseAppWorker, BaseAppUtils } from "../../base/app.js";
6
+ import { terminate } from "../../../terminate.js";
7
+ export class AppProcessWorker extends BaseAppWorker {
8
+ get id() {
9
+ return this.instance.id;
10
+ }
11
+ get workerId() {
12
+ return this.instance.process.pid;
13
+ }
14
+ get exitedAfterDisconnect() {
15
+ return this.instance.exitedAfterDisconnect;
16
+ }
17
+ get exitCode() {
18
+ return this.instance.process.exitCode;
19
+ }
20
+ send(message) {
21
+ sendmessage(this.instance, message);
22
+ }
23
+ clean() {
24
+ this.instance.removeAllListeners();
25
+ }
26
+ // static methods use on src/app_worker.ts
27
+ static get workerId() {
28
+ return process.pid;
29
+ }
30
+ static on(event, listener) {
31
+ process.on(event, listener);
32
+ }
33
+ static send(message) {
34
+ message.senderWorkerId = String(process.pid);
35
+ process.send(message);
36
+ }
37
+ static kill() {
38
+ process.exitCode = 1;
39
+ process.kill(process.pid);
40
+ }
41
+ static gracefulExit(options) {
42
+ gracefulExit(options);
43
+ }
44
+ }
45
+ export class AppProcessUtils extends BaseAppUtils {
46
+ fork() {
47
+ this.startTime = Date.now();
48
+ this.startSuccessCount = 0;
49
+ const args = [JSON.stringify(this.options)];
50
+ this.log('[master] start appWorker with args %j (process)', args);
51
+ cfork({
52
+ exec: this.getAppWorkerFile(),
53
+ args,
54
+ silent: false,
55
+ count: this.options.workers,
56
+ // don't refork in local env
57
+ refork: this.isProduction,
58
+ windowsHide: process.platform === 'win32',
59
+ });
60
+ let debugPort = process.debugPort;
61
+ cluster.on('fork', worker => {
62
+ const appWorker = new AppProcessWorker(worker);
63
+ this.emit('worker_forked', appWorker);
64
+ appWorker.disableRefork = true;
65
+ worker.on('message', msg => {
66
+ if (typeof msg === 'string') {
67
+ msg = {
68
+ action: msg,
69
+ data: msg,
70
+ };
71
+ }
72
+ msg.from = 'app';
73
+ this.messenger.send(msg);
74
+ });
75
+ this.log('[master] app_worker#%s:%s start, state: %s, current workers: %j', appWorker.id, appWorker.workerId, appWorker.state, Object.keys(cluster.workers));
76
+ // send debug message, due to `brk` scene, send here instead of app_worker.js
77
+ if (this.options.isDebug) {
78
+ debugPort++;
79
+ this.messenger.send({
80
+ to: 'parent',
81
+ from: 'app',
82
+ action: 'debug',
83
+ data: {
84
+ debugPort,
85
+ // keep compatibility, should use workerId instead
86
+ pid: appWorker.workerId,
87
+ workerId: appWorker.workerId,
88
+ },
89
+ });
90
+ }
91
+ });
92
+ cluster.on('disconnect', worker => {
93
+ const appWorker = new AppProcessWorker(worker);
94
+ this.log('[master] app_worker#%s:%s disconnect, suicide: %s, state: %s, current workers: %j', appWorker.id, appWorker.workerId, appWorker.exitedAfterDisconnect, appWorker.state, Object.keys(cluster.workers));
95
+ });
96
+ cluster.on('exit', (worker, code, signal) => {
97
+ const appWorker = new AppProcessWorker(worker);
98
+ this.messenger.send({
99
+ action: 'app-exit',
100
+ data: {
101
+ workerId: appWorker.workerId,
102
+ code,
103
+ signal,
104
+ },
105
+ to: 'master',
106
+ from: 'app',
107
+ });
108
+ });
109
+ return this;
110
+ }
111
+ async kill(timeout) {
112
+ await Promise.all(Object.keys(cluster.workers).map(id => {
113
+ const worker = cluster.workers[id];
114
+ Reflect.set(worker, 'disableRefork', true);
115
+ return terminate(worker.process, timeout);
116
+ }));
117
+ }
118
+ }
119
+ //# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiYXBwLmpzIiwic291cmNlUm9vdCI6IiIsInNvdXJjZXMiOlsiLi4vLi4vLi4vLi4vLi4vc3JjL3V0aWxzL21vZGUvaW1wbC9wcm9jZXNzL2FwcC50cyJdLCJuYW1lcyI6W10sIm1hcHBpbmdzIjoiQUFBQSxPQUFPLE9BQU8sRUFBRSxFQUF1QyxNQUFNLGNBQWMsQ0FBQztBQUU1RSxPQUFPLEVBQUUsS0FBSyxFQUFFLE1BQU0sT0FBTyxDQUFDO0FBQzlCLE9BQU8sRUFBRSxXQUFXLEVBQUUsTUFBTSxhQUFhLENBQUM7QUFDMUMsT0FBTyxFQUFFLFFBQVEsSUFBSSxZQUFZLEVBQXVDLE1BQU0sa0JBQWtCLENBQUM7QUFFakcsT0FBTyxFQUFFLGFBQWEsRUFBRSxZQUFZLEVBQUUsTUFBTSxtQkFBbUIsQ0FBQztBQUNoRSxPQUFPLEVBQUUsU0FBUyxFQUFFLE1BQU0sdUJBQXVCLENBQUM7QUFHbEQsTUFBTSxPQUFPLGdCQUFpQixTQUFRLGFBQW1DO0lBQ3ZFLElBQUksRUFBRTtRQUNKLE9BQU8sSUFBSSxDQUFDLFFBQVEsQ0FBQyxFQUFFLENBQUM7SUFDMUIsQ0FBQztJQUVELElBQUksUUFBUTtRQUNWLE9BQU8sSUFBSSxDQUFDLFFBQVEsQ0FBQyxPQUFPLENBQUMsR0FBSSxDQUFDO0lBQ3BDLENBQUM7SUFFRCxJQUFJLHFCQUFxQjtRQUN2QixPQUFPLElBQUksQ0FBQyxRQUFRLENBQUMscUJBQXFCLENBQUM7SUFDN0MsQ0FBQztJQUVELElBQUksUUFBUTtRQUNWLE9BQU8sSUFBSSxDQUFDLFFBQVEsQ0FBQyxPQUFPLENBQUMsUUFBUyxDQUFDO0lBQ3pDLENBQUM7SUFFRCxJQUFJLENBQUMsT0FBb0I7UUFDdkIsV0FBVyxDQUFDLElBQUksQ0FBQyxRQUFRLEVBQUUsT0FBTyxDQUFDLENBQUM7SUFDdEMsQ0FBQztJQUVELEtBQUs7UUFDSCxJQUFJLENBQUMsUUFBUSxDQUFDLGtCQUFrQixFQUFFLENBQUM7SUFDckMsQ0FBQztJQUVELDBDQUEwQztJQUUxQyxNQUFNLEtBQUssUUFBUTtRQUNqQixPQUFPLE9BQU8sQ0FBQyxHQUFHLENBQUM7SUFDckIsQ0FBQztJQUVELE1BQU0sQ0FBQyxFQUFFLENBQUMsS0FBYSxFQUFFLFFBQWtDO1FBQ3pELE9BQU8sQ0FBQyxFQUFFLENBQUMsS0FBSyxFQUFFLFFBQVEsQ0FBQyxDQUFDO0lBQzlCLENBQUM7SUFFRCxNQUFNLENBQUMsSUFBSSxDQUFDLE9BQW9CO1FBQzlCLE9BQU8sQ0FBQyxjQUFjLEdBQUcsTUFBTSxDQUFDLE9BQU8sQ0FBQyxHQUFHLENBQUMsQ0FBQztRQUM3QyxPQUFPLENBQUMsSUFBSyxDQUFDLE9BQU8sQ0FBQyxDQUFDO0lBQ3pCLENBQUM7SUFFRCxNQUFNLENBQUMsSUFBSTtRQUNULE9BQU8sQ0FBQyxRQUFRLEdBQUcsQ0FBQyxDQUFDO1FBQ3JCLE9BQU8sQ0FBQyxJQUFJLENBQUMsT0FBTyxDQUFDLEdBQUcsQ0FBQyxDQUFDO0lBQzVCLENBQUM7SUFFRCxNQUFNLENBQUMsWUFBWSxDQUFDLE9BQTRCO1FBQzlDLFlBQVksQ0FBQyxPQUFPLENBQUMsQ0FBQztJQUN4QixDQUFDO0NBQ0Y7QUFFRCxNQUFNLE9BQU8sZUFBZ0IsU0FBUSxZQUFZO0lBQy9DLElBQUk7UUFDRixJQUFJLENBQUMsU0FBUyxHQUFHLElBQUksQ0FBQyxHQUFHLEVBQUUsQ0FBQztRQUM1QixJQUFJLENBQUMsaUJBQWlCLEdBQUcsQ0FBQyxDQUFDO1FBRTNCLE1BQU0sSUFBSSxHQUFHLENBQUMsSUFBSSxDQUFDLFNBQVMsQ0FBQyxJQUFJLENBQUMsT0FBTyxDQUFDLENBQUMsQ0FBQztRQUM1QyxJQUFJLENBQUMsR0FBRyxDQUFDLGlEQUFpRCxFQUFFLElBQUksQ0FBQyxDQUFDO1FBQ2xFLEtBQUssQ0FBQztZQUNKLElBQUksRUFBRSxJQUFJLENBQUMsZ0JBQWdCLEVBQUU7WUFDN0IsSUFBSTtZQUNKLE1BQU0sRUFBRSxLQUFLO1lBQ2IsS0FBSyxFQUFFLElBQUksQ0FBQyxPQUFPLENBQUMsT0FBTztZQUMzQiw0QkFBNEI7WUFDNUIsTUFBTSxFQUFFLElBQUksQ0FBQyxZQUFZO1lBQ3pCLFdBQVcsRUFBRSxPQUFPLENBQUMsUUFBUSxLQUFLLE9BQU87U0FDMUMsQ0FBQyxDQUFDO1FBRUgsSUFBSSxTQUFTLEdBQUcsT0FBTyxDQUFDLFNBQVMsQ0FBQztRQUNsQyxPQUFPLENBQUMsRUFBRSxDQUFDLE1BQU0sRUFBRSxNQUFNLENBQUMsRUFBRTtZQUMxQixNQUFNLFNBQVMsR0FBRyxJQUFJLGdCQUFnQixDQUFDLE1BQU0sQ0FBQyxDQUFDO1lBQy9DLElBQUksQ0FBQyxJQUFJLENBQUMsZUFBZSxFQUFFLFNBQVMsQ0FBQyxDQUFDO1lBQ3RDLFNBQVMsQ0FBQyxhQUFhLEdBQUcsSUFBSSxDQUFDO1lBQy9CLE1BQU0sQ0FBQyxFQUFFLENBQUMsU0FBUyxFQUFFLEdBQUcsQ0FBQyxFQUFFO2dCQUN6QixJQUFJLE9BQU8sR0FBRyxLQUFLLFFBQVEsRUFBRSxDQUFDO29CQUM1QixHQUFHLEdBQUc7d0JBQ0osTUFBTSxFQUFFLEdBQUc7d0JBQ1gsSUFBSSxFQUFFLEdBQUc7cUJBQ1YsQ0FBQztnQkFDSixDQUFDO2dCQUNELEdBQUcsQ0FBQyxJQUFJLEdBQUcsS0FBSyxDQUFDO2dCQUNqQixJQUFJLENBQUMsU0FBUyxDQUFDLElBQUksQ0FBQyxHQUFHLENBQUMsQ0FBQztZQUMzQixDQUFDLENBQUMsQ0FBQztZQUNILElBQUksQ0FBQyxHQUFHLENBQ04saUVBQWlFLEVBQ2pFLFNBQVMsQ0FBQyxFQUFFLEVBQ1osU0FBUyxDQUFDLFFBQVEsRUFDbEIsU0FBUyxDQUFDLEtBQUssRUFDZixNQUFNLENBQUMsSUFBSSxDQUFDLE9BQU8sQ0FBQyxPQUFRLENBQUMsQ0FDOUIsQ0FBQztZQUVGLDZFQUE2RTtZQUM3RSxJQUFJLElBQUksQ0FBQyxPQUFPLENBQUMsT0FBTyxFQUFFLENBQUM7Z0JBQ3pCLFNBQVMsRUFBRSxDQUFDO2dCQUNaLElBQUksQ0FBQyxTQUFTLENBQUMsSUFBSSxDQUFDO29CQUNsQixFQUFFLEVBQUUsUUFBUTtvQkFDWixJQUFJLEVBQUUsS0FBSztvQkFDWCxNQUFNLEVBQUUsT0FBTztvQkFDZixJQUFJLEVBQUU7d0JBQ0osU0FBUzt3QkFDVCxrREFBa0Q7d0JBQ2xELEdBQUcsRUFBRSxTQUFTLENBQUMsUUFBUTt3QkFDdkIsUUFBUSxFQUFFLFNBQVMsQ0FBQyxRQUFRO3FCQUM3QjtpQkFDRixDQUFDLENBQUM7WUFDTCxDQUFDO1FBQ0gsQ0FBQyxDQUFDLENBQUM7UUFDSCxPQUFPLENBQUMsRUFBRSxDQUFDLFlBQVksRUFBRSxNQUFNLENBQUMsRUFBRTtZQUNoQyxNQUFNLFNBQVMsR0FBRyxJQUFJLGdCQUFnQixDQUFDLE1BQU0sQ0FBQyxDQUFDO1lBQy9DLElBQUksQ0FBQyxHQUFHLENBQ04sbUZBQW1GLEVBQ25GLFNBQVMsQ0FBQyxFQUFFLEVBQ1osU0FBUyxDQUFDLFFBQVEsRUFDbEIsU0FBUyxDQUFDLHFCQUFxQixFQUMvQixTQUFTLENBQUMsS0FBSyxFQUNmLE1BQU0sQ0FBQyxJQUFJLENBQUMsT0FBTyxDQUFDLE9BQVEsQ0FBQyxDQUM5QixDQUFDO1FBQ0osQ0FBQyxDQUFDLENBQUM7UUFDSCxPQUFPLENBQUMsRUFBRSxDQUFDLE1BQU0sRUFBRSxDQUFDLE1BQU0sRUFBRSxJQUFJLEVBQUUsTUFBTSxFQUFFLEVBQUU7WUFDMUMsTUFBTSxTQUFTLEdBQUcsSUFBSSxnQkFBZ0IsQ0FBQyxNQUFNLENBQUMsQ0FBQztZQUMvQyxJQUFJLENBQUMsU0FBUyxDQUFDLElBQUksQ0FBQztnQkFDbEIsTUFBTSxFQUFFLFVBQVU7Z0JBQ2xCLElBQUksRUFBRTtvQkFDSixRQUFRLEVBQUUsU0FBUyxDQUFDLFFBQVE7b0JBQzVCLElBQUk7b0JBQ0osTUFBTTtpQkFDUDtnQkFDRCxFQUFFLEVBQUUsUUFBUTtnQkFDWixJQUFJLEVBQUUsS0FBSzthQUNaLENBQUMsQ0FBQztRQUNMLENBQUMsQ0FBQyxDQUFDO1FBQ0gsT0FBTyxJQUFJLENBQUM7SUFDZCxDQUFDO0lBRUQsS0FBSyxDQUFDLElBQUksQ0FBQyxPQUFlO1FBQ3hCLE1BQU0sT0FBTyxDQUFDLEdBQUcsQ0FDZixNQUFNLENBQUMsSUFBSSxDQUFDLE9BQU8sQ0FBQyxPQUFRLENBQUMsQ0FBQyxHQUFHLENBQUMsRUFBRSxDQUFDLEVBQUU7WUFDckMsTUFBTSxNQUFNLEdBQUcsT0FBTyxDQUFDLE9BQVEsQ0FBQyxFQUFFLENBQUUsQ0FBQztZQUNyQyxPQUFPLENBQUMsR0FBRyxDQUFDLE1BQU0sRUFBRSxlQUFlLEVBQUUsSUFBSSxDQUFDLENBQUM7WUFDM0MsT0FBTyxTQUFTLENBQUMsTUFBTSxDQUFDLE9BQU8sRUFBRSxPQUFPLENBQUMsQ0FBQztRQUM1QyxDQUFDLENBQUMsQ0FDSCxDQUFDO0lBQ0osQ0FBQztDQUNGIn0=
@@ -0,0 +1,18 @@
1
+ import { type Worker } from 'node:worker_threads';
2
+ import { type Options as gracefulExitOptions } from 'graceful-process';
3
+ import { BaseAgentUtils, BaseAgentWorker } from '../../base/agent.ts';
4
+ import type { MessageBody } from '../../../messenger.ts';
5
+ export declare class AgentThreadWorker extends BaseAgentWorker<Worker> {
6
+ get workerId(): number;
7
+ send(message: MessageBody): void;
8
+ static send(message: MessageBody): void;
9
+ static kill(): void;
10
+ static gracefulExit(options: gracefulExitOptions): void;
11
+ }
12
+ export declare class AgentThreadUtils extends BaseAgentUtils {
13
+ #private;
14
+ instance: AgentThreadWorker;
15
+ fork(): void;
16
+ clean(): void;
17
+ kill(): Promise<void>;
18
+ }
@@ -0,0 +1,84 @@
1
+ import workerThreads, {} from 'node:worker_threads';
2
+ import {} from 'graceful-process';
3
+ import { BaseAgentUtils, BaseAgentWorker } from "../../base/agent.js";
4
+ import { ClusterAgentWorkerError } from "../../../../error/ClusterAgentWorkerError.js";
5
+ export class AgentThreadWorker extends BaseAgentWorker {
6
+ get workerId() {
7
+ return this.instance.threadId;
8
+ }
9
+ send(message) {
10
+ this.instance.postMessage(message);
11
+ }
12
+ static send(message) {
13
+ message.senderWorkerId = String(workerThreads.threadId);
14
+ workerThreads.parentPort.postMessage(message);
15
+ }
16
+ static kill() {
17
+ // in worker_threads, process.exit
18
+ // does not stop the whole program, just the single thread
19
+ process.exit(1);
20
+ }
21
+ static gracefulExit(options) {
22
+ const { beforeExit } = options;
23
+ process.on('exit', async (code) => {
24
+ if (typeof beforeExit === 'function') {
25
+ await beforeExit();
26
+ }
27
+ process.exit(code);
28
+ });
29
+ }
30
+ }
31
+ export class AgentThreadUtils extends BaseAgentUtils {
32
+ #worker;
33
+ #id = 0;
34
+ instance;
35
+ fork() {
36
+ this.startTime = Date.now();
37
+ // start agent worker
38
+ const argv = [JSON.stringify(this.options)];
39
+ const agentPath = this.getAgentWorkerFile();
40
+ const worker = (this.#worker = new workerThreads.Worker(agentPath, { argv }));
41
+ // wrap agent worker
42
+ const agentWorker = (this.instance = new AgentThreadWorker(worker));
43
+ this.emit('agent_forked', agentWorker);
44
+ agentWorker.status = 'starting';
45
+ agentWorker.id = ++this.#id;
46
+ this.log('[master] agent_worker#%s:%s start with worker_threads', agentWorker.id, agentWorker.workerId);
47
+ worker.on('message', msg => {
48
+ if (typeof msg === 'string') {
49
+ msg = {
50
+ action: msg,
51
+ data: msg,
52
+ };
53
+ }
54
+ msg.from = 'agent';
55
+ this.messenger.send(msg);
56
+ });
57
+ worker.on('error', err => {
58
+ this.logger.error(new ClusterAgentWorkerError(agentWorker.id, agentWorker.workerId, agentWorker.status, err));
59
+ });
60
+ // agent exit message
61
+ worker.once('exit', (code, signal) => {
62
+ this.messenger.send({
63
+ action: 'agent-exit',
64
+ data: {
65
+ code,
66
+ signal,
67
+ },
68
+ to: 'master',
69
+ from: 'agent',
70
+ });
71
+ });
72
+ }
73
+ clean() {
74
+ this.#worker.removeAllListeners();
75
+ }
76
+ async kill() {
77
+ if (this.#worker) {
78
+ this.log(`[master] kill agent worker#${this.#id} (worker_threads) by worker.terminate()`);
79
+ this.clean();
80
+ await this.#worker.terminate();
81
+ }
82
+ }
83
+ }
84
+ //# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiYWdlbnQuanMiLCJzb3VyY2VSb290IjoiIiwic291cmNlcyI6WyIuLi8uLi8uLi8uLi8uLi9zcmMvdXRpbHMvbW9kZS9pbXBsL3dvcmtlcl90aHJlYWRzL2FnZW50LnRzIl0sIm5hbWVzIjpbXSwibWFwcGluZ3MiOiJBQUFBLE9BQU8sYUFBYSxFQUFFLEVBQWUsTUFBTSxxQkFBcUIsQ0FBQztBQUVqRSxPQUFPLEVBQXVDLE1BQU0sa0JBQWtCLENBQUM7QUFFdkUsT0FBTyxFQUFFLGNBQWMsRUFBRSxlQUFlLEVBQUUsTUFBTSxxQkFBcUIsQ0FBQztBQUV0RSxPQUFPLEVBQUUsdUJBQXVCLEVBQUUsTUFBTSw4Q0FBOEMsQ0FBQztBQUV2RixNQUFNLE9BQU8saUJBQWtCLFNBQVEsZUFBdUI7SUFDNUQsSUFBSSxRQUFRO1FBQ1YsT0FBTyxJQUFJLENBQUMsUUFBUSxDQUFDLFFBQVEsQ0FBQztJQUNoQyxDQUFDO0lBRUQsSUFBSSxDQUFDLE9BQW9CO1FBQ3ZCLElBQUksQ0FBQyxRQUFRLENBQUMsV0FBVyxDQUFDLE9BQU8sQ0FBQyxDQUFDO0lBQ3JDLENBQUM7SUFFRCxNQUFNLENBQUMsSUFBSSxDQUFDLE9BQW9CO1FBQzlCLE9BQU8sQ0FBQyxjQUFjLEdBQUcsTUFBTSxDQUFDLGFBQWEsQ0FBQyxRQUFRLENBQUMsQ0FBQztRQUN4RCxhQUFhLENBQUMsVUFBVyxDQUFDLFdBQVcsQ0FBQyxPQUFPLENBQUMsQ0FBQztJQUNqRCxDQUFDO0lBRUQsTUFBTSxDQUFDLElBQUk7UUFDVCxrQ0FBa0M7UUFDbEMsMERBQTBEO1FBQzFELE9BQU8sQ0FBQyxJQUFJLENBQUMsQ0FBQyxDQUFDLENBQUM7SUFDbEIsQ0FBQztJQUVELE1BQU0sQ0FBQyxZQUFZLENBQUMsT0FBNEI7UUFDOUMsTUFBTSxFQUFFLFVBQVUsRUFBRSxHQUFHLE9BQU8sQ0FBQztRQUMvQixPQUFPLENBQUMsRUFBRSxDQUFDLE1BQU0sRUFBRSxLQUFLLEVBQUMsSUFBSSxFQUFDLEVBQUU7WUFDOUIsSUFBSSxPQUFPLFVBQVUsS0FBSyxVQUFVLEVBQUUsQ0FBQztnQkFDckMsTUFBTSxVQUFVLEVBQUUsQ0FBQztZQUNyQixDQUFDO1lBQ0QsT0FBTyxDQUFDLElBQUksQ0FBQyxJQUFJLENBQUMsQ0FBQztRQUNyQixDQUFDLENBQUMsQ0FBQztJQUNMLENBQUM7Q0FDRjtBQUVELE1BQU0sT0FBTyxnQkFBaUIsU0FBUSxjQUFjO0lBQ2xELE9BQU8sQ0FBUztJQUNoQixHQUFHLEdBQUcsQ0FBQyxDQUFDO0lBQ1IsUUFBUSxDQUFvQjtJQUU1QixJQUFJO1FBQ0YsSUFBSSxDQUFDLFNBQVMsR0FBRyxJQUFJLENBQUMsR0FBRyxFQUFFLENBQUM7UUFFNUIscUJBQXFCO1FBQ3JCLE1BQU0sSUFBSSxHQUFHLENBQUMsSUFBSSxDQUFDLFNBQVMsQ0FBQyxJQUFJLENBQUMsT0FBTyxDQUFDLENBQUMsQ0FBQztRQUM1QyxNQUFNLFNBQVMsR0FBRyxJQUFJLENBQUMsa0JBQWtCLEVBQUUsQ0FBQztRQUM1QyxNQUFNLE1BQU0sR0FBRyxDQUFDLElBQUksQ0FBQyxPQUFPLEdBQUcsSUFBSSxhQUFhLENBQUMsTUFBTSxDQUFDLFNBQVMsRUFBRSxFQUFFLElBQUksRUFBRSxDQUFDLENBQUMsQ0FBQztRQUU5RSxvQkFBb0I7UUFDcEIsTUFBTSxXQUFXLEdBQUcsQ0FBQyxJQUFJLENBQUMsUUFBUSxHQUFHLElBQUksaUJBQWlCLENBQUMsTUFBTSxDQUFDLENBQUMsQ0FBQztRQUNwRSxJQUFJLENBQUMsSUFBSSxDQUFDLGNBQWMsRUFBRSxXQUFXLENBQUMsQ0FBQztRQUN2QyxXQUFXLENBQUMsTUFBTSxHQUFHLFVBQVUsQ0FBQztRQUNoQyxXQUFXLENBQUMsRUFBRSxHQUFHLEVBQUUsSUFBSSxDQUFDLEdBQUcsQ0FBQztRQUM1QixJQUFJLENBQUMsR0FBRyxDQUFDLHVEQUF1RCxFQUFFLFdBQVcsQ0FBQyxFQUFFLEVBQUUsV0FBVyxDQUFDLFFBQVEsQ0FBQyxDQUFDO1FBRXhHLE1BQU0sQ0FBQyxFQUFFLENBQUMsU0FBUyxFQUFFLEdBQUcsQ0FBQyxFQUFFO1lBQ3pCLElBQUksT0FBTyxHQUFHLEtBQUssUUFBUSxFQUFFLENBQUM7Z0JBQzVCLEdBQUcsR0FBRztvQkFDSixNQUFNLEVBQUUsR0FBRztvQkFDWCxJQUFJLEVBQUUsR0FBRztpQkFDVixDQUFDO1lBQ0osQ0FBQztZQUNELEdBQUcsQ0FBQyxJQUFJLEdBQUcsT0FBTyxDQUFDO1lBQ25CLElBQUksQ0FBQyxTQUFTLENBQUMsSUFBSSxDQUFDLEdBQUcsQ0FBQyxDQUFDO1FBQzNCLENBQUMsQ0FBQyxDQUFDO1FBRUgsTUFBTSxDQUFDLEVBQUUsQ0FBQyxPQUFPLEVBQUUsR0FBRyxDQUFDLEVBQUU7WUFDdkIsSUFBSSxDQUFDLE1BQU0sQ0FBQyxLQUFLLENBQUMsSUFBSSx1QkFBdUIsQ0FBQyxXQUFXLENBQUMsRUFBRSxFQUFFLFdBQVcsQ0FBQyxRQUFRLEVBQUUsV0FBVyxDQUFDLE1BQU0sRUFBRSxHQUFHLENBQUMsQ0FBQyxDQUFDO1FBQ2hILENBQUMsQ0FBQyxDQUFDO1FBRUgscUJBQXFCO1FBQ3JCLE1BQU0sQ0FBQyxJQUFJLENBQUMsTUFBTSxFQUFFLENBQUMsSUFBWSxFQUFFLE1BQWMsRUFBRSxFQUFFO1lBQ25ELElBQUksQ0FBQyxTQUFTLENBQUMsSUFBSSxDQUFDO2dCQUNsQixNQUFNLEVBQUUsWUFBWTtnQkFDcEIsSUFBSSxFQUFFO29CQUNKLElBQUk7b0JBQ0osTUFBTTtpQkFDUDtnQkFDRCxFQUFFLEVBQUUsUUFBUTtnQkFDWixJQUFJLEVBQUUsT0FBTzthQUNkLENBQUMsQ0FBQztRQUNMLENBQUMsQ0FBQyxDQUFDO0lBQ0wsQ0FBQztJQUVELEtBQUs7UUFDSCxJQUFJLENBQUMsT0FBTyxDQUFDLGtCQUFrQixFQUFFLENBQUM7SUFDcEMsQ0FBQztJQUVELEtBQUssQ0FBQyxJQUFJO1FBQ1IsSUFBSSxJQUFJLENBQUMsT0FBTyxFQUFFLENBQUM7WUFDakIsSUFBSSxDQUFDLEdBQUcsQ0FBQyw4QkFBOEIsSUFBSSxDQUFDLEdBQUcseUNBQXlDLENBQUMsQ0FBQztZQUMxRixJQUFJLENBQUMsS0FBSyxFQUFFLENBQUM7WUFDYixNQUFNLElBQUksQ0FBQyxPQUFPLENBQUMsU0FBUyxFQUFFLENBQUM7UUFDakMsQ0FBQztJQUNILENBQUM7Q0FDRiJ9
@@ -0,0 +1,26 @@
1
+ import { Worker as ThreadWorker } from 'node:worker_threads';
2
+ import type { Options as gracefulExitOptions } from 'graceful-process';
3
+ import { BaseAppWorker, BaseAppUtils } from '../../base/app.ts';
4
+ import type { MessageBody } from '../../../messenger.ts';
5
+ export declare class AppThreadWorker extends BaseAppWorker<ThreadWorker> {
6
+ #private;
7
+ constructor(instance: ThreadWorker, id: number);
8
+ get id(): number;
9
+ get workerId(): number;
10
+ get state(): string;
11
+ set state(val: string);
12
+ get exitedAfterDisconnect(): boolean;
13
+ get exitCode(): number;
14
+ send(message: MessageBody): void;
15
+ clean(): void;
16
+ static get workerId(): number;
17
+ static on(event: string, listener: (...args: any[]) => void): void;
18
+ static send(message: MessageBody): void;
19
+ static kill(): void;
20
+ static gracefulExit(options: gracefulExitOptions): void;
21
+ }
22
+ export declare class AppThreadUtils extends BaseAppUtils {
23
+ #private;
24
+ fork(): this;
25
+ kill(): Promise<void>;
26
+ }
@@ -0,0 +1,137 @@
1
+ import { setTimeout as sleep } from 'node:timers/promises';
2
+ import { Worker as ThreadWorker, threadId, parentPort } from 'node:worker_threads';
3
+ import { BaseAppWorker, BaseAppUtils } from "../../base/app.js";
4
+ export class AppThreadWorker extends BaseAppWorker {
5
+ #state = 'none';
6
+ #id;
7
+ constructor(instance, id) {
8
+ super(instance);
9
+ this.#id = id;
10
+ }
11
+ get id() {
12
+ return this.#id;
13
+ }
14
+ get workerId() {
15
+ return this.instance.threadId;
16
+ }
17
+ get state() {
18
+ return this.#state;
19
+ }
20
+ set state(val) {
21
+ this.#state = val;
22
+ }
23
+ get exitedAfterDisconnect() {
24
+ return true;
25
+ }
26
+ get exitCode() {
27
+ return 0;
28
+ // return this.instance.exitCode;
29
+ }
30
+ send(message) {
31
+ this.instance.postMessage(message);
32
+ }
33
+ clean() {
34
+ this.instance.removeAllListeners();
35
+ }
36
+ // static methods use on src/app_worker.ts
37
+ static get workerId() {
38
+ return threadId;
39
+ }
40
+ static on(event, listener) {
41
+ parentPort.on(event, listener);
42
+ }
43
+ static send(message) {
44
+ message.senderWorkerId = String(threadId);
45
+ parentPort.postMessage(message);
46
+ }
47
+ static kill() {
48
+ process.exit(1);
49
+ }
50
+ static gracefulExit(options) {
51
+ process.on('exit', async (code) => {
52
+ if (typeof options.beforeExit === 'function') {
53
+ await options.beforeExit();
54
+ }
55
+ process.exit(code);
56
+ });
57
+ }
58
+ }
59
+ export class AppThreadUtils extends BaseAppUtils {
60
+ #workers = [];
61
+ #forkSingle(appPath, options, id) {
62
+ // start app worker
63
+ const worker = new ThreadWorker(appPath, options);
64
+ this.#workers.push(worker);
65
+ // wrap app worker
66
+ const appWorker = new AppThreadWorker(worker, id);
67
+ this.emit('worker_forked', appWorker);
68
+ appWorker.disableRefork = true;
69
+ worker.on('message', (msg) => {
70
+ if (typeof msg === 'string') {
71
+ msg = {
72
+ action: msg,
73
+ data: msg,
74
+ };
75
+ }
76
+ msg.from = 'app';
77
+ this.messenger.send(msg);
78
+ });
79
+ this.log('[master] app_worker#%s (tid:%s) start', appWorker.id, appWorker.workerId);
80
+ // send debug message, due to `brk` scene, send here instead of app_worker.js
81
+ let debugPort = process.debugPort;
82
+ if (this.options.isDebug) {
83
+ debugPort++;
84
+ this.messenger.send({
85
+ to: 'parent',
86
+ from: 'app',
87
+ action: 'debug',
88
+ data: {
89
+ debugPort,
90
+ pid: appWorker.workerId,
91
+ workerId: appWorker.workerId,
92
+ },
93
+ });
94
+ }
95
+ // handle worker exit
96
+ worker.on('exit', async (code) => {
97
+ appWorker.state = 'dead';
98
+ this.messenger.send({
99
+ action: 'app-exit',
100
+ data: {
101
+ workerId: appWorker.workerId,
102
+ code,
103
+ },
104
+ to: 'master',
105
+ from: 'app',
106
+ });
107
+ // refork app worker
108
+ await sleep(1000);
109
+ this.#forkSingle(appPath, options, id);
110
+ });
111
+ }
112
+ fork() {
113
+ this.startTime = Date.now();
114
+ this.startSuccessCount = 0;
115
+ const ports = this.options.ports ?? [];
116
+ if (!ports.length) {
117
+ ports.push(this.options.port);
118
+ }
119
+ this.options.workers = ports.length;
120
+ let i = 0;
121
+ do {
122
+ const options = Object.assign({}, this.options, { port: ports[i] });
123
+ const argv = [JSON.stringify(options)];
124
+ this.#forkSingle(this.getAppWorkerFile(), { argv }, ++i);
125
+ } while (i < ports.length);
126
+ return this;
127
+ }
128
+ async kill() {
129
+ for (const worker of this.#workers) {
130
+ const id = Reflect.get(worker, 'id');
131
+ this.log(`[master] kill app worker#${id} (worker_threads) by worker.terminate()`);
132
+ worker.removeAllListeners();
133
+ worker.terminate();
134
+ }
135
+ }
136
+ }
137
+ //# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiYXBwLmpzIiwic291cmNlUm9vdCI6IiIsInNvdXJjZXMiOlsiLi4vLi4vLi4vLi4vLi4vc3JjL3V0aWxzL21vZGUvaW1wbC93b3JrZXJfdGhyZWFkcy9hcHAudHMiXSwibmFtZXMiOltdLCJtYXBwaW5ncyI6IkFBQUEsT0FBTyxFQUFFLFVBQVUsSUFBSSxLQUFLLEVBQUUsTUFBTSxzQkFBc0IsQ0FBQztBQUMzRCxPQUFPLEVBQUUsTUFBTSxJQUFJLFlBQVksRUFBRSxRQUFRLEVBQUUsVUFBVSxFQUFzQixNQUFNLHFCQUFxQixDQUFDO0FBSXZHLE9BQU8sRUFBRSxhQUFhLEVBQUUsWUFBWSxFQUFFLE1BQU0sbUJBQW1CLENBQUM7QUFHaEUsTUFBTSxPQUFPLGVBQWdCLFNBQVEsYUFBMkI7SUFDOUQsTUFBTSxHQUFHLE1BQU0sQ0FBQztJQUNoQixHQUFHLENBQVM7SUFFWixZQUFZLFFBQXNCLEVBQUUsRUFBVTtRQUM1QyxLQUFLLENBQUMsUUFBUSxDQUFDLENBQUM7UUFDaEIsSUFBSSxDQUFDLEdBQUcsR0FBRyxFQUFFLENBQUM7SUFDaEIsQ0FBQztJQUVELElBQUksRUFBRTtRQUNKLE9BQU8sSUFBSSxDQUFDLEdBQUcsQ0FBQztJQUNsQixDQUFDO0lBRUQsSUFBSSxRQUFRO1FBQ1YsT0FBTyxJQUFJLENBQUMsUUFBUSxDQUFDLFFBQVEsQ0FBQztJQUNoQyxDQUFDO0lBRUQsSUFBSSxLQUFLO1FBQ1AsT0FBTyxJQUFJLENBQUMsTUFBTSxDQUFDO0lBQ3JCLENBQUM7SUFFRCxJQUFJLEtBQUssQ0FBQyxHQUFHO1FBQ1gsSUFBSSxDQUFDLE1BQU0sR0FBRyxHQUFHLENBQUM7SUFDcEIsQ0FBQztJQUVELElBQUkscUJBQXFCO1FBQ3ZCLE9BQU8sSUFBSSxDQUFDO0lBQ2QsQ0FBQztJQUVELElBQUksUUFBUTtRQUNWLE9BQU8sQ0FBQyxDQUFDO1FBQ1QsaUNBQWlDO0lBQ25DLENBQUM7SUFFRCxJQUFJLENBQUMsT0FBb0I7UUFDdkIsSUFBSSxDQUFDLFFBQVEsQ0FBQyxXQUFXLENBQUMsT0FBTyxDQUFDLENBQUM7SUFDckMsQ0FBQztJQUVELEtBQUs7UUFDSCxJQUFJLENBQUMsUUFBUSxDQUFDLGtCQUFrQixFQUFFLENBQUM7SUFDckMsQ0FBQztJQUVELDBDQUEwQztJQUUxQyxNQUFNLEtBQUssUUFBUTtRQUNqQixPQUFPLFFBQVEsQ0FBQztJQUNsQixDQUFDO0lBRUQsTUFBTSxDQUFDLEVBQUUsQ0FBQyxLQUFhLEVBQUUsUUFBa0M7UUFDekQsVUFBVyxDQUFDLEVBQUUsQ0FBQyxLQUFLLEVBQUUsUUFBUSxDQUFDLENBQUM7SUFDbEMsQ0FBQztJQUVELE1BQU0sQ0FBQyxJQUFJLENBQUMsT0FBb0I7UUFDOUIsT0FBTyxDQUFDLGNBQWMsR0FBRyxNQUFNLENBQUMsUUFBUSxDQUFDLENBQUM7UUFDMUMsVUFBVyxDQUFDLFdBQVcsQ0FBQyxPQUFPLENBQUMsQ0FBQztJQUNuQyxDQUFDO0lBRUQsTUFBTSxDQUFDLElBQUk7UUFDVCxPQUFPLENBQUMsSUFBSSxDQUFDLENBQUMsQ0FBQyxDQUFDO0lBQ2xCLENBQUM7SUFFRCxNQUFNLENBQUMsWUFBWSxDQUFDLE9BQTRCO1FBQzlDLE9BQU8sQ0FBQyxFQUFFLENBQUMsTUFBTSxFQUFFLEtBQUssRUFBQyxJQUFJLEVBQUMsRUFBRTtZQUM5QixJQUFJLE9BQU8sT0FBTyxDQUFDLFVBQVUsS0FBSyxVQUFVLEVBQUUsQ0FBQztnQkFDN0MsTUFBTSxPQUFPLENBQUMsVUFBVSxFQUFFLENBQUM7WUFDN0IsQ0FBQztZQUNELE9BQU8sQ0FBQyxJQUFJLENBQUMsSUFBSSxDQUFDLENBQUM7UUFDckIsQ0FBQyxDQUFDLENBQUM7SUFDTCxDQUFDO0NBQ0Y7QUFFRCxNQUFNLE9BQU8sY0FBZSxTQUFRLFlBQVk7SUFDOUMsUUFBUSxHQUFtQixFQUFFLENBQUM7SUFFOUIsV0FBVyxDQUFDLE9BQWUsRUFBRSxPQUFzQixFQUFFLEVBQVU7UUFDN0QsbUJBQW1CO1FBQ25CLE1BQU0sTUFBTSxHQUFHLElBQUksWUFBWSxDQUFDLE9BQU8sRUFBRSxPQUFPLENBQUMsQ0FBQztRQUNsRCxJQUFJLENBQUMsUUFBUSxDQUFDLElBQUksQ0FBQyxNQUFNLENBQUMsQ0FBQztRQUUzQixrQkFBa0I7UUFDbEIsTUFBTSxTQUFTLEdBQUcsSUFBSSxlQUFlLENBQUMsTUFBTSxFQUFFLEVBQUUsQ0FBQyxDQUFDO1FBQ2xELElBQUksQ0FBQyxJQUFJLENBQUMsZUFBZSxFQUFFLFNBQVMsQ0FBQyxDQUFDO1FBQ3RDLFNBQVMsQ0FBQyxhQUFhLEdBQUcsSUFBSSxDQUFDO1FBQy9CLE1BQU0sQ0FBQyxFQUFFLENBQUMsU0FBUyxFQUFFLENBQUMsR0FBZ0IsRUFBRSxFQUFFO1lBQ3hDLElBQUksT0FBTyxHQUFHLEtBQUssUUFBUSxFQUFFLENBQUM7Z0JBQzVCLEdBQUcsR0FBRztvQkFDSixNQUFNLEVBQUUsR0FBRztvQkFDWCxJQUFJLEVBQUUsR0FBRztpQkFDVixDQUFDO1lBQ0osQ0FBQztZQUNELEdBQUcsQ0FBQyxJQUFJLEdBQUcsS0FBSyxDQUFDO1lBQ2pCLElBQUksQ0FBQyxTQUFTLENBQUMsSUFBSSxDQUFDLEdBQUcsQ0FBQyxDQUFDO1FBQzNCLENBQUMsQ0FBQyxDQUFDO1FBQ0gsSUFBSSxDQUFDLEdBQUcsQ0FBQyx1Q0FBdUMsRUFBRSxTQUFTLENBQUMsRUFBRSxFQUFFLFNBQVMsQ0FBQyxRQUFRLENBQUMsQ0FBQztRQUVwRiw2RUFBNkU7UUFDN0UsSUFBSSxTQUFTLEdBQUcsT0FBTyxDQUFDLFNBQVMsQ0FBQztRQUNsQyxJQUFJLElBQUksQ0FBQyxPQUFPLENBQUMsT0FBTyxFQUFFLENBQUM7WUFDekIsU0FBUyxFQUFFLENBQUM7WUFDWixJQUFJLENBQUMsU0FBUyxDQUFDLElBQUksQ0FBQztnQkFDbEIsRUFBRSxFQUFFLFFBQVE7Z0JBQ1osSUFBSSxFQUFFLEtBQUs7Z0JBQ1gsTUFBTSxFQUFFLE9BQU87Z0JBQ2YsSUFBSSxFQUFFO29CQUNKLFNBQVM7b0JBQ1QsR0FBRyxFQUFFLFNBQVMsQ0FBQyxRQUFRO29CQUN2QixRQUFRLEVBQUUsU0FBUyxDQUFDLFFBQVE7aUJBQzdCO2FBQ0YsQ0FBQyxDQUFDO1FBQ0wsQ0FBQztRQUVELHFCQUFxQjtRQUNyQixNQUFNLENBQUMsRUFBRSxDQUFDLE1BQU0sRUFBRSxLQUFLLEVBQUMsSUFBSSxFQUFDLEVBQUU7WUFDN0IsU0FBUyxDQUFDLEtBQUssR0FBRyxNQUFNLENBQUM7WUFDekIsSUFBSSxDQUFDLFNBQVMsQ0FBQyxJQUFJLENBQUM7Z0JBQ2xCLE1BQU0sRUFBRSxVQUFVO2dCQUNsQixJQUFJLEVBQUU7b0JBQ0osUUFBUSxFQUFFLFNBQVMsQ0FBQyxRQUFRO29CQUM1QixJQUFJO2lCQUNMO2dCQUNELEVBQUUsRUFBRSxRQUFRO2dCQUNaLElBQUksRUFBRSxLQUFLO2FBQ1osQ0FBQyxDQUFDO1lBRUgsb0JBQW9CO1lBQ3BCLE1BQU0sS0FBSyxDQUFDLElBQUksQ0FBQyxDQUFDO1lBQ2xCLElBQUksQ0FBQyxXQUFXLENBQUMsT0FBTyxFQUFFLE9BQU8sRUFBRSxFQUFFLENBQUMsQ0FBQztRQUN6QyxDQUFDLENBQUMsQ0FBQztJQUNMLENBQUM7SUFFRCxJQUFJO1FBQ0YsSUFBSSxDQUFDLFNBQVMsR0FBRyxJQUFJLENBQUMsR0FBRyxFQUFFLENBQUM7UUFDNUIsSUFBSSxDQUFDLGlCQUFpQixHQUFHLENBQUMsQ0FBQztRQUUzQixNQUFNLEtBQUssR0FBRyxJQUFJLENBQUMsT0FBTyxDQUFDLEtBQUssSUFBSSxFQUFFLENBQUM7UUFDdkMsSUFBSSxDQUFDLEtBQUssQ0FBQyxNQUFNLEVBQUUsQ0FBQztZQUNsQixLQUFLLENBQUMsSUFBSSxDQUFDLElBQUksQ0FBQyxPQUFPLENBQUMsSUFBSyxDQUFDLENBQUM7UUFDakMsQ0FBQztRQUNELElBQUksQ0FBQyxPQUFPLENBQUMsT0FBTyxHQUFHLEtBQUssQ0FBQyxNQUFNLENBQUM7UUFDcEMsSUFBSSxDQUFDLEdBQUcsQ0FBQyxDQUFDO1FBQ1YsR0FBRyxDQUFDO1lBQ0YsTUFBTSxPQUFPLEdBQUcsTUFBTSxDQUFDLE1BQU0sQ0FBQyxFQUFFLEVBQUUsSUFBSSxDQUFDLE9BQU8sRUFBRSxFQUFFLElBQUksRUFBRSxLQUFLLENBQUMsQ0FBQyxDQUFDLEVBQUUsQ0FBQyxDQUFDO1lBQ3BFLE1BQU0sSUFBSSxHQUFHLENBQUMsSUFBSSxDQUFDLFNBQVMsQ0FBQyxPQUFPLENBQUMsQ0FBQyxDQUFDO1lBQ3ZDLElBQUksQ0FBQyxXQUFXLENBQUMsSUFBSSxDQUFDLGdCQUFnQixFQUFFLEVBQUUsRUFBRSxJQUFJLEVBQUUsRUFBRSxFQUFFLENBQUMsQ0FBQyxDQUFDO1FBQzNELENBQUMsUUFBUSxDQUFDLEdBQUcsS0FBSyxDQUFDLE1BQU0sRUFBRTtRQUUzQixPQUFPLElBQUksQ0FBQztJQUNkLENBQUM7SUFFRCxLQUFLLENBQUMsSUFBSTtRQUNSLEtBQUssTUFBTSxNQUFNLElBQUksSUFBSSxDQUFDLFFBQVEsRUFBRSxDQUFDO1lBQ25DLE1BQU0sRUFBRSxHQUFHLE9BQU8sQ0FBQyxHQUFHLENBQUMsTUFBTSxFQUFFLElBQUksQ0FBQyxDQUFDO1lBQ3JDLElBQUksQ0FBQyxHQUFHLENBQUMsNEJBQTRCLEVBQUUseUNBQXlDLENBQUMsQ0FBQztZQUNsRixNQUFNLENBQUMsa0JBQWtCLEVBQUUsQ0FBQztZQUM1QixNQUFNLENBQUMsU0FBUyxFQUFFLENBQUM7UUFDckIsQ0FBQztJQUNILENBQUM7Q0FDRiJ9
@@ -0,0 +1,80 @@
1
+ import { type SecureContextOptions } from 'node:tls';
2
+ export interface ClusterHTTPSSecureOptions {
3
+ key: SecureContextOptions['key'];
4
+ cert: SecureContextOptions['cert'];
5
+ ca?: SecureContextOptions['ca'];
6
+ passphrase?: SecureContextOptions['passphrase'];
7
+ }
8
+ export type ClusterStartMode = 'process' | 'worker_threads';
9
+ /** Cluster start options */
10
+ export interface ClusterOptions {
11
+ /**
12
+ * specify framework that can be absolute path or npm package
13
+ */
14
+ framework?: string;
15
+ /**
16
+ * @deprecated please use framework instead
17
+ */
18
+ customEgg?: string;
19
+ /** directory of application, default to `process.cwd()` */
20
+ baseDir?: string;
21
+ /**
22
+ * numbers of app workers, default to `os.cpus().length`
23
+ */
24
+ workers?: number | string;
25
+ /**
26
+ * listening port, default to `7001`(http) or `8443`(https)
27
+ */
28
+ port?: number | string | null;
29
+ /**
30
+ * listening a debug port on http protocol
31
+ */
32
+ debugPort?: number;
33
+ /**
34
+ * https options, { key, cert, ca }, full path
35
+ */
36
+ https?: ClusterHTTPSSecureOptions | boolean;
37
+ /**
38
+ * @deprecated please use `options.https.key` instead
39
+ */
40
+ key?: ClusterHTTPSSecureOptions['key'];
41
+ /**
42
+ * @deprecated please use `options.https.cert` instead
43
+ */
44
+ cert?: ClusterHTTPSSecureOptions['cert'];
45
+ /**
46
+ * will inject into worker/agent process
47
+ */
48
+ require?: string | string[];
49
+ /**
50
+ * will save master pid to this file
51
+ */
52
+ pidFile?: string;
53
+ /**
54
+ * custom env, default is `process.env.EGG_SERVER_ENV`
55
+ */
56
+ env?: string;
57
+ /**
58
+ * default is `'process'`, use `'worker_threads'` to start the app & agent worker by worker_threads
59
+ */
60
+ startMode?: ClusterStartMode;
61
+ /**
62
+ * startup port of each app worker, such as: `[7001, 7002, 7003]`, only effects when the startMode is `'worker_threads'`
63
+ */
64
+ ports?: number[];
65
+ /**
66
+ * sticky mode server
67
+ */
68
+ sticky?: boolean;
69
+ /** customized plugins, for unittest */
70
+ plugins?: object;
71
+ isDebug?: boolean;
72
+ }
73
+ export interface ParsedClusterOptions extends ClusterOptions {
74
+ port?: number;
75
+ baseDir: string;
76
+ workers: number;
77
+ framework: string;
78
+ startMode: ClusterStartMode;
79
+ }
80
+ export declare function parseOptions(options?: ClusterOptions): Promise<ParsedClusterOptions>;
@@ -0,0 +1,81 @@
1
+ import os from 'node:os';
2
+ import fs from 'node:fs';
3
+ import path from 'node:path';
4
+ import assert from 'node:assert';
5
+ import { debuglog } from 'node:util';
6
+ import {} from 'node:tls';
7
+ import { getFrameworkPath, importModule } from '@eggjs/utils';
8
+ const debug = debuglog('egg/cluster/utils/options');
9
+ export async function parseOptions(options) {
10
+ options = {
11
+ baseDir: process.cwd(),
12
+ port: options?.https ? 8443 : undefined,
13
+ startMode: 'process',
14
+ // ports: [],
15
+ env: process.env.EGG_SERVER_ENV,
16
+ ...options,
17
+ };
18
+ const pkgPath = path.join(options.baseDir, 'package.json');
19
+ assert(fs.existsSync(pkgPath), `${pkgPath} should exist`);
20
+ options.framework = getFrameworkPath({
21
+ baseDir: options.baseDir,
22
+ // compatible customEgg only when call startCluster directly without framework
23
+ framework: options.framework ?? options.customEgg,
24
+ });
25
+ debug('[parseOptions] %o', options);
26
+ const egg = await importModule(options.framework, {
27
+ paths: [options.baseDir],
28
+ });
29
+ assert(egg.Application, `should define Application in ${options.framework}`);
30
+ assert(egg.Agent, `should define Agent in ${options.framework}`);
31
+ if (options.https === true) {
32
+ // Keep compatible options.key, options.cert
33
+ console.warn('[@eggjs/cluster:deprecated] [master] Please use `https: { key, cert, ca }` instead of `https: true`');
34
+ options.https = {
35
+ key: options.key,
36
+ cert: options.cert,
37
+ };
38
+ }
39
+ // https
40
+ if (options.https) {
41
+ assert(options.https.key, 'options.https.key should exists');
42
+ if (typeof options.https.key === 'string') {
43
+ assert(fs.existsSync(options.https.key), 'options.https.key file should exists');
44
+ }
45
+ assert(options.https.cert, 'options.https.cert should exists');
46
+ if (typeof options.https.cert === 'string') {
47
+ assert(fs.existsSync(options.https.cert), 'options.https.cert file should exists');
48
+ }
49
+ if (typeof options.https.ca === 'string') {
50
+ assert(fs.existsSync(options.https.ca), 'options.https.ca file should exists');
51
+ }
52
+ }
53
+ if (options.port && typeof options.port === 'string') {
54
+ options.port = parseInt(options.port);
55
+ }
56
+ if (options.port === null) {
57
+ options.port = undefined;
58
+ }
59
+ if (options.workers && typeof options.workers === 'string') {
60
+ options.workers = parseInt(options.workers);
61
+ }
62
+ if (!options.workers) {
63
+ options.workers = os.cpus().length;
64
+ }
65
+ if (options.require) {
66
+ if (typeof options.require === 'string') {
67
+ options.require = [options.require];
68
+ }
69
+ }
70
+ // don't print deprecated message in production env.
71
+ // it will print to stderr.
72
+ if (process.env.NODE_ENV === 'production') {
73
+ process.env.NO_DEPRECATION = '*';
74
+ }
75
+ const isDebug = process.execArgv.some(argv => argv.includes('--debug') || argv.includes('--inspect'));
76
+ if (isDebug) {
77
+ options.isDebug = isDebug;
78
+ }
79
+ return options;
80
+ }
81
+ //# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoib3B0aW9ucy5qcyIsInNvdXJjZVJvb3QiOiIiLCJzb3VyY2VzIjpbIi4uLy4uL3NyYy91dGlscy9vcHRpb25zLnRzIl0sIm5hbWVzIjpbXSwibWFwcGluZ3MiOiJBQUFBLE9BQU8sRUFBRSxNQUFNLFNBQVMsQ0FBQztBQUN6QixPQUFPLEVBQUUsTUFBTSxTQUFTLENBQUM7QUFDekIsT0FBTyxJQUFJLE1BQU0sV0FBVyxDQUFDO0FBQzdCLE9BQU8sTUFBTSxNQUFNLGFBQWEsQ0FBQztBQUNqQyxPQUFPLEVBQUUsUUFBUSxFQUFFLE1BQU0sV0FBVyxDQUFDO0FBQ3JDLE9BQU8sRUFBNkIsTUFBTSxVQUFVLENBQUM7QUFFckQsT0FBTyxFQUFFLGdCQUFnQixFQUFFLFlBQVksRUFBRSxNQUFNLGNBQWMsQ0FBQztBQUU5RCxNQUFNLEtBQUssR0FBRyxRQUFRLENBQUMsMkJBQTJCLENBQUMsQ0FBQztBQW9GcEQsTUFBTSxDQUFDLEtBQUssVUFBVSxZQUFZLENBQUMsT0FBd0I7SUFDekQsT0FBTyxHQUFHO1FBQ1IsT0FBTyxFQUFFLE9BQU8sQ0FBQyxHQUFHLEVBQUU7UUFDdEIsSUFBSSxFQUFFLE9BQU8sRUFBRSxLQUFLLENBQUMsQ0FBQyxDQUFDLElBQUksQ0FBQyxDQUFDLENBQUMsU0FBUztRQUN2QyxTQUFTLEVBQUUsU0FBUztRQUNwQixhQUFhO1FBQ2IsR0FBRyxFQUFFLE9BQU8sQ0FBQyxHQUFHLENBQUMsY0FBYztRQUMvQixHQUFHLE9BQU87S0FDWCxDQUFDO0lBRUYsTUFBTSxPQUFPLEdBQUcsSUFBSSxDQUFDLElBQUksQ0FBQyxPQUFPLENBQUMsT0FBUSxFQUFFLGNBQWMsQ0FBQyxDQUFDO0lBQzVELE1BQU0sQ0FBQyxFQUFFLENBQUMsVUFBVSxDQUFDLE9BQU8sQ0FBQyxFQUFFLEdBQUcsT0FBTyxlQUFlLENBQUMsQ0FBQztJQUUxRCxPQUFPLENBQUMsU0FBUyxHQUFHLGdCQUFnQixDQUFDO1FBQ25DLE9BQU8sRUFBRSxPQUFPLENBQUMsT0FBUTtRQUN6Qiw4RUFBOEU7UUFDOUUsU0FBUyxFQUFFLE9BQU8sQ0FBQyxTQUFTLElBQUksT0FBTyxDQUFDLFNBQVM7S0FDbEQsQ0FBQyxDQUFDO0lBQ0gsS0FBSyxDQUFDLG1CQUFtQixFQUFFLE9BQU8sQ0FBQyxDQUFDO0lBRXBDLE1BQU0sR0FBRyxHQUFHLE1BQU0sWUFBWSxDQUFDLE9BQU8sQ0FBQyxTQUFTLEVBQUU7UUFDaEQsS0FBSyxFQUFFLENBQUMsT0FBTyxDQUFDLE9BQVEsQ0FBQztLQUMxQixDQUFDLENBQUM7SUFDSCxNQUFNLENBQUMsR0FBRyxDQUFDLFdBQVcsRUFBRSxnQ0FBZ0MsT0FBTyxDQUFDLFNBQVMsRUFBRSxDQUFDLENBQUM7SUFDN0UsTUFBTSxDQUFDLEdBQUcsQ0FBQyxLQUFLLEVBQUUsMEJBQTBCLE9BQU8sQ0FBQyxTQUFTLEVBQUUsQ0FBQyxDQUFDO0lBRWpFLElBQUksT0FBTyxDQUFDLEtBQUssS0FBSyxJQUFJLEVBQUUsQ0FBQztRQUMzQiw0Q0FBNEM7UUFDNUMsT0FBTyxDQUFDLElBQUksQ0FBQyxxR0FBcUcsQ0FBQyxDQUFDO1FBQ3BILE9BQU8sQ0FBQyxLQUFLLEdBQUc7WUFDZCxHQUFHLEVBQUUsT0FBTyxDQUFDLEdBQUc7WUFDaEIsSUFBSSxFQUFFLE9BQU8sQ0FBQyxJQUFJO1NBQ25CLENBQUM7SUFDSixDQUFDO0lBRUQsUUFBUTtJQUNSLElBQUksT0FBTyxDQUFDLEtBQUssRUFBRSxDQUFDO1FBQ2xCLE1BQU0sQ0FBQyxPQUFPLENBQUMsS0FBSyxDQUFDLEdBQUcsRUFBRSxpQ0FBaUMsQ0FBQyxDQUFDO1FBQzdELElBQUksT0FBTyxPQUFPLENBQUMsS0FBSyxDQUFDLEdBQUcsS0FBSyxRQUFRLEVBQUUsQ0FBQztZQUMxQyxNQUFNLENBQUMsRUFBRSxDQUFDLFVBQVUsQ0FBQyxPQUFPLENBQUMsS0FBSyxDQUFDLEdBQUcsQ0FBQyxFQUFFLHNDQUFzQyxDQUFDLENBQUM7UUFDbkYsQ0FBQztRQUNELE1BQU0sQ0FBQyxPQUFPLENBQUMsS0FBSyxDQUFDLElBQUksRUFBRSxrQ0FBa0MsQ0FBQyxDQUFDO1FBQy9ELElBQUksT0FBTyxPQUFPLENBQUMsS0FBSyxDQUFDLElBQUksS0FBSyxRQUFRLEVBQUUsQ0FBQztZQUMzQyxNQUFNLENBQUMsRUFBRSxDQUFDLFVBQVUsQ0FBQyxPQUFPLENBQUMsS0FBSyxDQUFDLElBQUksQ0FBQyxFQUFFLHVDQUF1QyxDQUFDLENBQUM7UUFDckYsQ0FBQztRQUNELElBQUksT0FBTyxPQUFPLENBQUMsS0FBSyxDQUFDLEVBQUUsS0FBSyxRQUFRLEVBQUUsQ0FBQztZQUN6QyxNQUFNLENBQUMsRUFBRSxDQUFDLFVBQVUsQ0FBQyxPQUFPLENBQUMsS0FBSyxDQUFDLEVBQUUsQ0FBQyxFQUFFLHFDQUFxQyxDQUFDLENBQUM7UUFDakYsQ0FBQztJQUNILENBQUM7SUFFRCxJQUFJLE9BQU8sQ0FBQyxJQUFJLElBQUksT0FBTyxPQUFPLENBQUMsSUFBSSxLQUFLLFFBQVEsRUFBRSxDQUFDO1FBQ3JELE9BQU8sQ0FBQyxJQUFJLEdBQUcsUUFBUSxDQUFDLE9BQU8sQ0FBQyxJQUFJLENBQUMsQ0FBQztJQUN4QyxDQUFDO0lBQ0QsSUFBSSxPQUFPLENBQUMsSUFBSSxLQUFLLElBQUksRUFBRSxDQUFDO1FBQzFCLE9BQU8sQ0FBQyxJQUFJLEdBQUcsU0FBUyxDQUFDO0lBQzNCLENBQUM7SUFFRCxJQUFJLE9BQU8sQ0FBQyxPQUFPLElBQUksT0FBTyxPQUFPLENBQUMsT0FBTyxLQUFLLFFBQVEsRUFBRSxDQUFDO1FBQzNELE9BQU8sQ0FBQyxPQUFPLEdBQUcsUUFBUSxDQUFDLE9BQU8sQ0FBQyxPQUFPLENBQUMsQ0FBQztJQUM5QyxDQUFDO0lBQ0QsSUFBSSxDQUFDLE9BQU8sQ0FBQyxPQUFPLEVBQUUsQ0FBQztRQUNyQixPQUFPLENBQUMsT0FBTyxHQUFHLEVBQUUsQ0FBQyxJQUFJLEVBQUUsQ0FBQyxNQUFNLENBQUM7SUFDckMsQ0FBQztJQUVELElBQUksT0FBTyxDQUFDLE9BQU8sRUFBRSxDQUFDO1FBQ3BCLElBQUksT0FBTyxPQUFPLENBQUMsT0FBTyxLQUFLLFFBQVEsRUFBRSxDQUFDO1lBQ3hDLE9BQU8sQ0FBQyxPQUFPLEdBQUcsQ0FBQyxPQUFPLENBQUMsT0FBTyxDQUFDLENBQUM7UUFDdEMsQ0FBQztJQUNILENBQUM7SUFFRCxvREFBb0Q7SUFDcEQsMkJBQTJCO0lBQzNCLElBQUksT0FBTyxDQUFDLEdBQUcsQ0FBQyxRQUFRLEtBQUssWUFBWSxFQUFFLENBQUM7UUFDMUMsT0FBTyxDQUFDLEdBQUcsQ0FBQyxjQUFjLEdBQUcsR0FBRyxDQUFDO0lBQ25DLENBQUM7SUFFRCxNQUFNLE9BQU8sR0FBRyxPQUFPLENBQUMsUUFBUSxDQUFDLElBQUksQ0FBQyxJQUFJLENBQUMsRUFBRSxDQUFDLElBQUksQ0FBQyxRQUFRLENBQUMsU0FBUyxDQUFDLElBQUksSUFBSSxDQUFDLFFBQVEsQ0FBQyxXQUFXLENBQUMsQ0FBQyxDQUFDO0lBQ3RHLElBQUksT0FBTyxFQUFFLENBQUM7UUFDWixPQUFPLENBQUMsT0FBTyxHQUFHLE9BQU8sQ0FBQztJQUM1QixDQUFDO0lBRUQsT0FBTyxPQUErQixDQUFDO0FBQ3pDLENBQUMifQ==
@@ -0,0 +1,6 @@
1
+ import { ChildProcess } from 'node:child_process';
2
+ interface SubProcess extends ChildProcess {
3
+ process?: ChildProcess;
4
+ }
5
+ export declare function terminate(subProcess: SubProcess, timeout: number): Promise<void>;
6
+ export {};