update Daemon.gracefulShutdownInterval

2025-10-17 09:10:21 +01:00 · 2025-09-18 18:49:50 -04:00 · 2025-09-18 18:49:50 -04:00 · fd4635e8f4
commit fd4635e8f4
parent 50fa5eee2f
1 changed files with 10 additions and 4 deletions
--- a/daemon/HaikuDaemon.cpp
+++ b/daemon/HaikuDaemon.cpp
@ -62,14 +62,20 @@ void MainWindow::MessageReceived (BMessage * msg)
 	switch (msg->what)
 	{
 		case M_GRACEFUL_SHUTDOWN:
-			i2p::context.SetAcceptsTunnels (false);
-			m_GracefulShutdownTimer = std::make_unique<BMessageRunner>(m_Messenger, 
-				BMessage (C_GRACEFUL_SHUTDOWN_UPDATE), GRACEFUL_SHUTDOWN_UPDATE_INTERVAL);
+			if (!m_GracefulShutdownTimer)
+			{
+				i2p::context.SetAcceptsTunnels (false);
+				Daemon.gracefulShutdownInterval = GRACEFUL_SHUTDOWN_UPDATE_COUNT;
+				m_GracefulShutdownTimer = std::make_unique<BMessageRunner>(m_Messenger, 
+					BMessage (C_GRACEFUL_SHUTDOWN_UPDATE), GRACEFUL_SHUTDOWN_UPDATE_INTERVAL);
+			}	
 		break;
 		case C_GRACEFUL_SHUTDOWN_UPDATE:
-			if (i2p::tunnel::tunnels.CountTransitTunnels () <= 0)
+			if (Daemon.gracefulShutdownInterval > 0) Daemon.gracefulShutdownInterval--;
+			if (!Daemon.gracefulShutdownInterval || i2p::tunnel::tunnels.CountTransitTunnels () <= 0)
 			{
 				m_GracefulShutdownTimer = nullptr;
+				Daemon.gracefulShutdownInterval = 0;
 				m_Messenger.SendMessage (B_QUIT_REQUESTED);
 			}	
 		break;