ekko
71 lines
2.0 KiB
TypeScript
71 lines
2.0 KiB
TypeScript
import { logger } from './logger'
|
|
import { closeDb } from '../db'
|
|
import { stopPreviewRuntime } from '../controllers/update'
|
|
|
|
export function bindShutdown(server: any, groupChatServer?: any, chatRunServer?: any, agentBridgeManager?: any): void {
|
|
let isShuttingDown = false
|
|
|
|
const shutdown = async (signal: string) => {
|
|
if (isShuttingDown) return
|
|
isShuttingDown = true
|
|
|
|
// Force exit after 3s no matter what
|
|
setTimeout(() => process.exit(0), 3000)
|
|
|
|
logger.info('Shutting down (%s)...', signal)
|
|
console.log(`[shutdown] Received signal: ${signal}`)
|
|
|
|
try {
|
|
try {
|
|
await stopPreviewRuntime()
|
|
logger.info('Preview runtime stopped')
|
|
} catch (err) {
|
|
logger.warn(err, 'Failed to stop preview runtime (non-fatal)')
|
|
}
|
|
|
|
if (agentBridgeManager) {
|
|
try {
|
|
await agentBridgeManager.stop()
|
|
logger.info('Agent bridge stopped')
|
|
} catch (err) {
|
|
logger.warn(err, 'Failed to stop agent bridge (non-fatal)')
|
|
}
|
|
}
|
|
|
|
// Close ChatRunSocket first to abort all active runs and close EventSource connections
|
|
if (chatRunServer) {
|
|
chatRunServer.close()
|
|
logger.info('ChatRunSocket closed')
|
|
}
|
|
|
|
// Disconnect Socket.IO before HTTP server to prevent hanging
|
|
if (groupChatServer) {
|
|
groupChatServer.agentClients.disconnectAll()
|
|
groupChatServer.getIO().close()
|
|
logger.info('Socket.IO closed')
|
|
}
|
|
|
|
const servers = Array.isArray(server) ? server : [server].filter(Boolean)
|
|
if (servers.length) {
|
|
await Promise.all(servers.map((httpServer) => (
|
|
new Promise<void>((resolve) => {
|
|
httpServer.close(() => {
|
|
logger.info('HTTP server closed')
|
|
resolve()
|
|
})
|
|
})
|
|
)))
|
|
}
|
|
} catch (err) {
|
|
logger.error(err, 'Shutdown error')
|
|
}
|
|
|
|
closeDb()
|
|
process.exit(0)
|
|
}
|
|
|
|
process.once('SIGUSR2', shutdown)
|
|
process.on('SIGINT', shutdown)
|
|
process.on('SIGTERM', shutdown)
|
|
}
|