69 lines
1.8 KiB
JavaScript
69 lines
1.8 KiB
JavaScript
/**
|
|
* Graceful shutdown handler - registers cleanup functions and handles shutdown signals.
|
|
*/
|
|
|
|
import { SHUTDOWN_TIMEOUT_MS } from './constants.js'
|
|
|
|
const cleanupFunctions = []
|
|
let isShuttingDown = false
|
|
|
|
export function clearCleanup() {
|
|
cleanupFunctions.length = 0
|
|
isShuttingDown = false
|
|
}
|
|
|
|
export function registerCleanup(fn) {
|
|
if (typeof fn !== 'function') throw new TypeError('Cleanup function must be a function')
|
|
cleanupFunctions.push(fn)
|
|
}
|
|
|
|
async function executeCleanup() {
|
|
if (isShuttingDown) return
|
|
isShuttingDown = true
|
|
for (let i = cleanupFunctions.length - 1; i >= 0; i--) {
|
|
try {
|
|
await cleanupFunctions[i]()
|
|
}
|
|
catch (error) {
|
|
console.error(`[shutdown] Cleanup function ${i} failed:`, error?.message || String(error))
|
|
}
|
|
}
|
|
}
|
|
|
|
export async function graceful(error) {
|
|
if (error) {
|
|
console.error('[shutdown] Shutting down due to error:', error?.message || String(error))
|
|
if (error.stack) console.error('[shutdown] Stack trace:', error.stack)
|
|
}
|
|
else {
|
|
console.log('[shutdown] Initiating graceful shutdown')
|
|
}
|
|
const timeout = setTimeout(() => {
|
|
console.error('[shutdown] Shutdown timeout exceeded, forcing exit')
|
|
process.exit(1)
|
|
}, SHUTDOWN_TIMEOUT_MS)
|
|
try {
|
|
await executeCleanup()
|
|
clearTimeout(timeout)
|
|
console.log('[shutdown] Cleanup complete')
|
|
process.exit(error ? 1 : 0)
|
|
}
|
|
catch (err) {
|
|
clearTimeout(timeout)
|
|
console.error('[shutdown] Error during cleanup:', err?.message || String(err))
|
|
process.exit(1)
|
|
}
|
|
}
|
|
|
|
export function initShutdownHandlers() {
|
|
for (const signal of ['SIGTERM', 'SIGINT']) {
|
|
process.on(signal, () => {
|
|
console.log(`[shutdown] Received ${signal}`)
|
|
graceful().catch((err) => {
|
|
console.error('[shutdown] Error in graceful shutdown:', err)
|
|
process.exit(1)
|
|
})
|
|
})
|
|
}
|
|
}
|