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

fullstack.org.ua

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

javascript

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

const PORT = 3000;
const SHUTDOWN_TIMEOUT = 10_000;

let isShuttingDown = false;
let shutdownPromise = null;

const sockets = new Set();
const activeRequests = new Set();

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

const server = http.createServer((req, res) => {
  let resolveRequest;

  const requestFinished = new Promise((resolve) => {
    resolveRequest = resolve;
  });

  activeRequests.add(requestFinished);

  let finished = false;
  const timer = setTimeout(() => {
    if (!res.writableEnded) {
      res.end('Відповідь сервера\n');
    }
  }, 2_000);

  const finishRequest = () => {
    if (finished) {
      return;
    }

    finished = true;
    clearTimeout(timer);
    activeRequests.delete(requestFinished);
    resolveRequest();
  };

  res.once('finish', finishRequest);
  res.once('close', finishRequest);

  if (isShuttingDown) {
    res.statusCode = 503;
    res.end('Сервер завершує роботу\n');
    return;
  }

  res.setHeader('Content-Type', 'text/plain; charset=utf-8');
  res.write('Запит прийнято. Обробка триває...\n');
});

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

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

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

      resolve();
    });

    // Закриваємо простоюючі keep-alive-з'єднання, якщо API доступне
    // у поточній версії Node.js.
    server.closeIdleConnections?.();
  });
}

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

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

    const forceShutdownTimer = setTimeout(() => {
      console.error('Час завершення вичерпано. Закриваємо з’єднання примусово.');

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

    // Таймер не повинен сам утримувати процес, якщо інших активних ресурсів уже немає.
    forceShutdownTimer.unref?.();

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

      // Додатково чекаємо завершення обробників запитів,
      // які ще залишилися зареєстрованими.
      await Promise.all(activeRequests);

      // Після HTTP-запитів можна закрити зовнішні ресурси.
      await externalResource.close();

      clearTimeout(forceShutdownTimer);
      console.log('Сервер коректно завершив роботу');
      process.exitCode = 0;
    } catch (error) {
      clearTimeout(forceShutdownTimer);
      console.error('Помилка під час завершення:', error);
      process.exitCode = 1;
    }
  })();

  return shutdownPromise;
}

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

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

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