Breaking News: Grepper is joining You.com. Read the official announcement!

nodejs graceful shutdown

Restu Wahyu Saputra answered on August 22, 2024 Popularity 2/10 Helpfulness 1/10

answer nodejs graceful shutdown

related nodejs graceful shutdown

nodejs graceful shutdown

Comment

import consola from 'consola'
import http, { IncomingMessage, OutgoingMessage, Server } from 'node:http'
import { Socket } from 'node:net'
import { Express } from 'express'

import { Environment } from '~/configs/config.env'
import { MultiThread } from '~/helpers/helper.multiThread'

export class GracefulShutdown {
	private multiThread: MultiThread
	private connection: Map<any, any>
	private env: string
	private port: number
	private singleThread: boolean
	private gracefulShutdownTimer: number
	private signals: string[]

	constructor() {
		this.multiThread = new MultiThread()
		this.connection = new Map()
		this.env = Environment.NODE_ENV
		this.port = Environment.PORT
		this.singleThread = Environment.SINGLE_THREAD
		this.gracefulShutdownTimer = this.env === 'development' ? 3000 : 5000
		this.signals = ['SIGTERM', 'SIGINT', 'SIGQUIT', 'SIGHUP', 'SIGABRT', 'SIGALRM', 'SIGUSR1', 'SIGUSR2']
	}

	private close(server: Server, signal: string) {
		const socket = this.connection.get('connection')
		if (!socket?.closed) socket?.destroy()

		server.close(async (err) => {
			await this.sleep(this.gracefulShutdownTimer)
			this.connection.clear()

			if (!err) {
				console.info(`Server Closed Without Error: ${err}`)
				console.info('Gracefull shutdown successfully')

				process.removeAllListeners()
				process.kill(process.pid, signal)
				process.exit(0)
			} else {
				console.info(`Server Closed With Error: ${err}`)
				console.info('Gracefull shutdown successfully')

				process.removeAllListeners()
				process.kill(process.pid, signal)
				process.exit(1)
			}
		})
	}

	private async sleep(timer: number): Promise<void> {
		await new Promise((resolve) => setTimeout(resolve, timer))
	}

	listen(app: Express): void {
		const server: Server = http.createServer(app)

		server.on('connection', (socket: Socket): void => {
			console.info('Server Connection Established')

			if (socket) this.connection.set('connection', socket)
			else socket.destroy()
		})

		server.on('request', (req: IncomingMessage, _res: OutgoingMessage): void => {
			const socketRequest: Socket = req.socket
			const socketConnection: Socket = this.connection.get('connection')

			if (!(socketRequest instanceof Socket) || !(socketConnection instanceof Socket)) {
				this.connection.clear()
				this.connection.set('connection', socketRequest)
			}
		})

		this.signals.forEach((event: string) => {
			if (this.env === 'development') {
				process.on(event, async (signal: string): Promise<void> => {
					console.info(`Server ${this.env} received signal: ${signal}`)

					if (this.singleThread) {
						this.multiThread.nodecluster.disconnect()
					}

					await this.close(server, signal)
				})
			} else {
				process.on(event, async (signal: string): Promise<void> => {
					console.info(`Server ${this.env} received signal: ${signal}`)

					if (this.singleThread) {
						this.multiThread.nodecluster.disconnect()
					}

					await this.close(server, signal)
				})
			}
		})

		this.multiThread.cluster(this.singleThread, (): void => {
			const serverInfo: string = `Server is running on port ${this.port}`
			server.listen(this.port, () => consola.info(serverInfo))
		})
	}
}

Popularity 2/10 Helpfulness 1/10 Language typescript

Source: Grepper

Tags: graceful-shutdown typescript

Link to this answer
Share Copy Link

Contributed on Oct 31 2024

Restu Wahyu Saputra

0 Answers Avg Quality 2/10

Closely Related Answers

nodejs graceful shutdown

Comment

const http = require('http')
const { Socket } = require('net')
const { env } = require('../configs/env')

class GracefulShutdown {
  static __env = env.NODE_ENV
  static __port = env.PORT
  static __signals = ['SIGTERM', 'SIGINT', 'SIGQUIT', 'SIGHUP', 'SIGABRT', 'SIGALRM', 'SIGUSR1']
  static __gracefulShutdownTimer = env.NODE_ENV === 'local' ? 3000 : 5000
  static __connection = new Map()

  static __timeout(server, signal) {
    setTimeout(() => {
      const socket = GracefulShutdown.__connection.get('connection')
      if (!socket?.closed) socket?.destroy()

      server.close((err) => {
        GracefulShutdown.__connection.clear()

        if (!err) {
          console.info(`Server Closed Without Error: ${err}`)
          process.kill(process.pid, signal)
          process.exit(0)
        } else {
          console.info(`Server Closed With Error: ${err}`)
          process.kill(process.pid, signal)
          process.exit(1)
        }
      })
    }, GracefulShutdown.__gracefulShutdownTimer)
  }

  static listen(handler) {
    const server = http.createServer(handler)

    server.on('connection', (socket) => {
      console.info('Server Connection Established')

      if (socket) GracefulShutdown.__connection.set('connection', socket)
      else socket.destroy()
    })

    server.on('request', (req, _res) => {
      const socketRequest = req.socket
      const socketConnection = GracefulShutdown.__connection.get('connection')

      if ((!socketRequest || !socketConnection) instanceof Socket) {
        GracefulShutdown.__connection.clear()
        GracefulShutdown.__connection.set('connection', socketRequest)
      }
    })

    GracefulShutdown.__signals.forEach((event) => {
      if (GracefulShutdown.__env === 'local') {
        process.on(event, (signal) => {
          console.info(`Server Received Signal ${signal}`)
          GracefulShutdown.__timeout(server, signal)
        })
      } else {
        process.on(event, (signal) => {
          console.info(`Server Received Signal ${signal}`)
          GracefulShutdown.__timeout(server, signal)
        })
      }
    })

    const serverInfo = `Server is running on port ${GracefulShutdown.__port}`
    server.listen(GracefulShutdown.__port, () => console.log(serverInfo))
  }
}

exports.GracefulShutdown = GracefulShutdown

Popularity 4/10 Helpfulness 2/10 Language javascript

Source: Grepper

Tags: graceful-shutdown javascript

Link to this answer
Share Copy Link

Contributed on Aug 22 2024

Restu Wahyu Saputra

0 Answers Avg Quality 2/10

nodejs graceful shutdown

Contents

More Related Answers

nodejs graceful shutdown

Closely Related Answers

nodejs graceful shutdown

Grepper

Documentation

Social

Legal

Contact

Oops, You will need to install Grepper and log-in to perform this action.