|
|
@@ -9,29 +9,37 @@
|
|
|
- By now the node app should exit on its own.
|
|
|
*/
|
|
|
|
|
|
-const logger = require('@overleaf/logger')
|
|
|
-const OError = require('@overleaf/o-error')
|
|
|
-const Settings = require('@overleaf/settings')
|
|
|
-const Metrics = require('@overleaf/metrics')
|
|
|
-const sleep = require('util').promisify(setTimeout)
|
|
|
+import logger from '@overleaf/logger'
|
|
|
+
|
|
|
+import OError from '@overleaf/o-error'
|
|
|
+import Settings from '@overleaf/settings'
|
|
|
+import Metrics from '@overleaf/metrics'
|
|
|
+import { setTimeout as sleep } from 'node:timers/promises'
|
|
|
+
|
|
|
const optionalCleanupHandlersBeforeStoppingTraffic = []
|
|
|
const requiredCleanupHandlersBeforeDrainingConnections = []
|
|
|
const optionalCleanupHandlersAfterDrainingConnections = []
|
|
|
const connectionDrainer = []
|
|
|
|
|
|
-function addConnectionDrainer(label, handler) {
|
|
|
+export function addConnectionDrainer(label, handler) {
|
|
|
connectionDrainer.push({ label, handler })
|
|
|
}
|
|
|
|
|
|
-function addOptionalCleanupHandlerBeforeStoppingTraffic(label, handler) {
|
|
|
+export function addOptionalCleanupHandlerBeforeStoppingTraffic(label, handler) {
|
|
|
optionalCleanupHandlersBeforeStoppingTraffic.push({ label, handler })
|
|
|
}
|
|
|
|
|
|
-function addRequiredCleanupHandlerBeforeDrainingConnections(label, handler) {
|
|
|
+export function addRequiredCleanupHandlerBeforeDrainingConnections(
|
|
|
+ label,
|
|
|
+ handler
|
|
|
+) {
|
|
|
requiredCleanupHandlersBeforeDrainingConnections.push({ label, handler })
|
|
|
}
|
|
|
|
|
|
-function addOptionalCleanupHandlerAfterDrainingConnections(label, handler) {
|
|
|
+export function addOptionalCleanupHandlerAfterDrainingConnections(
|
|
|
+ label,
|
|
|
+ handler
|
|
|
+) {
|
|
|
optionalCleanupHandlersAfterDrainingConnections.push({ label, handler })
|
|
|
}
|
|
|
|
|
|
@@ -55,7 +63,7 @@ async function runHandlers(stage, handlers, logOnly) {
|
|
|
* @param {import('net').Server} [server]
|
|
|
* @param {number|string} [signal]
|
|
|
*/
|
|
|
-async function gracefulShutdown(server, signal) {
|
|
|
+export async function gracefulShutdown(server, signal) {
|
|
|
logger.warn({ signal }, 'graceful shutdown: started shutdown sequence')
|
|
|
Settings.shuttingDown = true
|
|
|
|
|
|
@@ -110,7 +118,7 @@ async function gracefulShutdown(server, signal) {
|
|
|
logger.info({}, 'graceful shutdown: ready to exit')
|
|
|
}
|
|
|
|
|
|
-function triggerGracefulShutdown(server, signal) {
|
|
|
+export function triggerGracefulShutdown(server, signal) {
|
|
|
gracefulShutdown(server, signal).catch(err => {
|
|
|
logger.err(
|
|
|
{ err },
|
|
|
@@ -119,7 +127,7 @@ function triggerGracefulShutdown(server, signal) {
|
|
|
})
|
|
|
}
|
|
|
|
|
|
-class BackgroundTaskTracker {
|
|
|
+export class BackgroundTaskTracker {
|
|
|
constructor(label) {
|
|
|
// Do not leak any handles, just record the number of pending jobs.
|
|
|
// In case we miss the cleanup of one job, the worst thing that can happen
|
|
|
@@ -141,7 +149,7 @@ class BackgroundTaskTracker {
|
|
|
}
|
|
|
}
|
|
|
|
|
|
-module.exports = {
|
|
|
+export default {
|
|
|
BackgroundTaskTracker,
|
|
|
addConnectionDrainer,
|
|
|
addOptionalCleanupHandlerBeforeStoppingTraffic,
|