-
-
Notifications
You must be signed in to change notification settings - Fork 33
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
aff481c
commit 7ff8174
Showing
2 changed files
with
102 additions
and
73 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,93 @@ | ||
| import Bluebird from 'bluebird' | ||
| import {clone} from 'lodash-es' | ||
| import ms from 'ms' | ||
| import {clearTimeout} from 'node:timers' | ||
| import pTimeout from 'p-timeout' | ||
| import prettyBytes from 'pretty-bytes' | ||
| import {Socket} from 'socket.io-client' | ||
| import {Cluster} from './cluster.js' | ||
| import {logger} from './logger.js' | ||
|
|
||
| export class Keepalive { | ||
| public timer?: NodeJS.Timeout | ||
| private socket?: Socket | ||
| private keepAliveError = 0 | ||
|
|
||
| constructor( | ||
| private readonly interval: number, | ||
| private readonly cluster: Cluster, | ||
| ) {} | ||
|
|
||
| public start(socket: Socket): void { | ||
| this.socket = socket | ||
| this.schedule() | ||
| } | ||
|
|
||
| public stop(): void { | ||
| if (this.timer) { | ||
| clearTimeout(this.timer) | ||
| } | ||
| } | ||
|
|
||
| private schedule(): void { | ||
| if (this.timer) { | ||
| clearTimeout(this.timer) | ||
| } | ||
| this.timer = setTimeout(() => { | ||
| logger.trace('start keep alive') | ||
| void this.emitKeepAlive() | ||
| }, this.interval) | ||
| } | ||
|
|
||
| private async emitKeepAlive(): Promise<void> { | ||
| try { | ||
| const status = await pTimeout(this.keepAlive(), { | ||
| milliseconds: ms('10s'), | ||
| }) | ||
| if (!status) { | ||
| logger.fatal('kicked by server') | ||
| this.cluster.exit(1) | ||
| } | ||
| this.keepAliveError = 0 | ||
| } catch (e) { | ||
| this.keepAliveError++ | ||
| logger.error(e, 'keep alive error') | ||
| if (this.keepAliveError >= 1) { | ||
| await Bluebird.try(async () => { | ||
| await this.cluster.disable() | ||
| await this.cluster.connect() | ||
| await this.cluster.enable() | ||
| }) | ||
| .timeout(ms('10m'), 'restart timeout') | ||
| .catch((e) => { | ||
| logger.error(e, 'restart failed') | ||
| this.cluster.exit(1) | ||
| }) | ||
| } | ||
| } finally { | ||
| void this.schedule() | ||
| } | ||
| } | ||
|
|
||
| private async keepAlive(): Promise<boolean> { | ||
| if (!this.cluster.isEnabled) { | ||
| throw new Error('节点未启用') | ||
| } | ||
| if (!this.socket) { | ||
| throw new Error('未连接到服务器') | ||
| } | ||
|
|
||
| const counters = clone(this.cluster.counters) | ||
| const [err, date] = (await this.socket.emitWithAck('keep-alive', { | ||
| time: new Date(), | ||
| ...counters, | ||
| })) as [object, unknown] | ||
|
|
||
| if (err) throw new Error('keep alive error', {cause: err}) | ||
| const bytes = prettyBytes(counters.bytes, {binary: true}) | ||
| logger.info(`keep alive success, serve ${counters.hits} files, ${bytes}`) | ||
| this.cluster.counters.hits -= counters.hits | ||
| this.cluster.counters.bytes -= counters.bytes | ||
| return !!date | ||
| } | ||
| } |