moony/blinksocks
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

core: refine _onClose() and terminate()

Browse Source
This commit is contained in:
Micooz 2017-10-26 14:18:39 +08:00
parent 8f46dc6ef4
commit 0b8ae8fe97
No known key found for this signature in database
GPG Key ID: 002FB5DD584D6CB1
1 changed files with 17 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
src/core/hub.js
View File

@ -36,14 +36,29 @@ export class Hub extends EventEmitter {
super(); super();
this._onConnection = this._onConnection.bind(this); this._onConnection = this._onConnection.bind(this);
this._onUdpAssociate = this._onUdpAssociate.bind(this); this._onUdpAssociate = this._onUdpAssociate.bind(this);
this._onClose = this._onClose.bind(this);
if (config !== undefined) { if (config !== undefined) {
Config.init(config); Config.init(config);
} }
} }
terminate() { terminate() {
this._onClose(); // relays
this._relays.forEach((relay) => relay.destroy());
MiddlewareManager.cleanup();
// balancer
if (__IS_CLIENT__) {
Balancer.destroy();
this._isFirstWorker && logger.info('[balancer] stopped');
}
// server
this._server.close();
this._isFirstWorker && logger.info('[hub] shutdown');
// udp server
if (__IS_SERVER__ && this._udpServer !== null) {
this._udpServer._handle && this._udpServer.close();
Object.keys(this._udpRelays).forEach((key) => this._udpRelays[key].destroy());
}
this.emit('close');
} }
async run() { async run() {
@ -239,24 +254,4 @@ export class Hub extends EventEmitter {
this._relays.push(relay); this._relays.push(relay);
} }
_onClose() {
// relays
this._relays.forEach((relay) => relay.destroy());
MiddlewareManager.cleanup();
// balancer
if (__IS_CLIENT__) {
Balancer.destroy();
this._isFirstWorker && logger.info('[balancer] stopped');
}
// server
this._server.close();
this._isFirstWorker && logger.info('[hub] shutdown');
// udp server
if (__IS_SERVER__ && this._udpServer !== null) {
this._udpServer._handle && this._udpServer.close();
Object.keys(this._udpRelays).forEach((key) => this._udpRelays[key].destroy());
}
this.emit('close');
}
} }