acc50a5d36
* Update dependencies and force-update qs This is mainly an attempt to get rid of as many resolutions as possible since it seems they are unnecessary except for qs (according to yarn/npm audit). For qs use 6.9.7 since Express is using 6.9.6 and that matches the most closely. Also add overrides since this is npm's version of yarn's resolutions and we need it for the shrinkwrap to generate with the right dependencies. Decided to keep pinning @types/node as well although I am not sure it is necessary. Express is pulling in v20 types. Since this is development-only we only need it in resolutions. * Run formatter Some rules seem to have changed with the dependency updates. * Replace deprecated bodyParser.json() usage * Audit npm shrinkwrap as well * Skip installing dependencies in audit It seems the tools only require the lock files. * Fix tests when using ipv6 * Add missing openssl dependency to flake
73 lines
1.8 KiB
TypeScript
73 lines
1.8 KiB
TypeScript
import { logger } from "@coder/logger"
|
|
import { promises as fs } from "fs"
|
|
|
|
/**
|
|
* Provides a heartbeat using a local file to indicate activity.
|
|
*/
|
|
export class Heart {
|
|
private heartbeatTimer?: NodeJS.Timeout
|
|
private heartbeatInterval = 60000
|
|
public lastHeartbeat = 0
|
|
|
|
public constructor(
|
|
private readonly heartbeatPath: string,
|
|
private readonly isActive: () => Promise<boolean>,
|
|
) {
|
|
this.beat = this.beat.bind(this)
|
|
this.alive = this.alive.bind(this)
|
|
}
|
|
|
|
public alive(): boolean {
|
|
const now = Date.now()
|
|
return now - this.lastHeartbeat < this.heartbeatInterval
|
|
}
|
|
/**
|
|
* Write to the heartbeat file if we haven't already done so within the
|
|
* timeout and start or reset a timer that keeps running as long as there is
|
|
* activity. Failures are logged as warnings.
|
|
*/
|
|
public async beat(): Promise<void> {
|
|
if (this.alive()) {
|
|
return
|
|
}
|
|
|
|
logger.trace("heartbeat")
|
|
this.lastHeartbeat = Date.now()
|
|
if (typeof this.heartbeatTimer !== "undefined") {
|
|
clearTimeout(this.heartbeatTimer)
|
|
}
|
|
this.heartbeatTimer = setTimeout(() => heartbeatTimer(this.isActive, this.beat), this.heartbeatInterval)
|
|
try {
|
|
return await fs.writeFile(this.heartbeatPath, "")
|
|
} catch (error: any) {
|
|
logger.warn(error.message)
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Call to clear any heartbeatTimer for shutdown.
|
|
*/
|
|
public dispose(): void {
|
|
if (typeof this.heartbeatTimer !== "undefined") {
|
|
clearTimeout(this.heartbeatTimer)
|
|
}
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Helper function for the heartbeatTimer.
|
|
*
|
|
* If heartbeat is active, call beat. Otherwise do nothing.
|
|
*
|
|
* Extracted to make it easier to test.
|
|
*/
|
|
export async function heartbeatTimer(isActive: Heart["isActive"], beat: Heart["beat"]) {
|
|
try {
|
|
if (await isActive()) {
|
|
beat()
|
|
}
|
|
} catch (error: unknown) {
|
|
logger.warn((error as Error).message)
|
|
}
|
|
}
|