fix: reduce graceful shutdown timeout for improved responsiveness

- Updated the GRACEFUL_SHUTDOWN_TIMEOUT constant from 3 seconds to 1 second.
- Enhanced the shutdown process in the client to allow for quicker cleanup and response to interrupts.

Author: pratikbin

constants/main.go

@@ -51,7 +51,7 @@ const (
 
 	MAX_TUNNELS_PER_IP            = 5
 	TUNNEL_RECONNECT_TIMEOUT      = 3
-	GRACEFUL_SHUTDOWN_TIMEOUT     = 3
+	GRACEFUL_SHUTDOWN_TIMEOUT     = 1
 	TUNNEL_CREATE_TIMEOUT         = 3
 	REQ_BODY_READ_CHUNK_TIMEOUT   = 3
 	DEST_REQUEST_TIMEOUT          = 30

internal/client/main.go

@@ -392,7 +392,20 @@ func Run(config ConfigOptions) {
 	logger.Log(constants.YELLOW, "Gracefully shutting down client...")
 	disconnectMsg := protocol.TunnelMessage{MsgType: protocol.CLIENT_DISCONNECT}
 	_ = mmarClient.SendMessage(disconnectMsg)
+
+	// Cancel context to stop goroutines
 	cancel()
+
+	// Give a short time for cleanup, but don't block unnecessarily
 	gracefulShutdownTimer := time.NewTimer(constants.GRACEFUL_SHUTDOWN_TIMEOUT * time.Second)
-	<-gracefulShutdownTimer.C
+	defer gracefulShutdownTimer.Stop()
+
+	// Wait for either the timer or a second interrupt (force quit)
+	select {
+	case <-gracefulShutdownTimer.C:
+		// Normal graceful shutdown completed
+	case <-sigInt:
+		// Force quit on second interrupt
+		logger.Log(constants.RED, "Force quitting...")
+	}
 }