Коректне завершення роботи · fullstack.org.ua

fullstack.org.ua

Коректне завершення роботи · fullstack.org.ua

javascript

const http = require('node:http');

const PORT = 3000;
const SHUTDOWN_TIMEOUT = 10_000;

const sockets = new Set();
let isShuttingDown = false;
let forceShutdownTimer;

const server = http.createServer(async (req, res) => {
  if (req.url === '/slow') {
    // Імітація довгого запиту
    await new Promise((resolve) => setTimeout(resolve, 3000));
  }

  res.writeHead(200, { 'Content-Type': 'text/plain; charset=utf-8' });
  res.end('Відповідь сервера\n');
});

server.on('connection', (socket) => {
  sockets.add(socket);

  socket.on('close', () => {
    sockets.delete(socket);
  });
});

server.listen(PORT, () => {
  console.log(`Сервер запущено на порту ${PORT}`);
});

async function closeExternalResources() {
  // Тут можуть бути закриття бази даних, черги або інших клієнтів
  await new Promise((resolve) => setTimeout(resolve, 100));
  console.log('Зовнішні ресурси закрито');
}

function closeServer() {
  return new Promise((resolve, reject) => {
    server.close((error) => {
      if (error) {
        reject(error);
        return;
      }

      resolve();
    });
  });
}

async function shutdown(signal) {
  if (isShuttingDown) {
    console.log('Завершення вже виконується');
    return;
  }

  isShuttingDown = true;
  console.log(`Отримано ${signal}. Починається завершення роботи`);

  forceShutdownTimer = setTimeout(() => {
    console.error('Тайм-аут завершення. Примусово закриваємо з’єднання');

    for (const socket of sockets) {
      socket.destroy();
    }

    // Це аварійний шлях, тому незавершені операції можуть бути перервані
    process.exit(1);
  }, SHUTDOWN_TIMEOUT);

  try {
    await closeServer();
    console.log('Нові з’єднання більше не приймаються');

    await closeExternalResources();

    clearTimeout(forceShutdownTimer);
    console.log('Завершення роботи виконано успішно');

    process.exitCode = 0;
  } catch (error) {
    clearTimeout(forceShutdownTimer);

    console.error('Помилка під час завершення:', error);
    process.exitCode = 1;
  }
}

process.on('SIGINT', () => {
  void shutdown('SIGINT');
});

process.on('SIGTERM', () => {
  void shutdown('SIGTERM');
});

javascript

async function shutdown(signal) {
  if (isShuttingDown) {
    return;
  }

  isShuttingDown = true;

  try {
    await stopAcceptingRequests();
    await waitForActiveRequests();
    await closeDatabase();
    await closeMessageQueue();

    process.exitCode = 0;
  } catch (error) {
    console.error(error);
    process.exitCode = 1;
  }
}