From 8cf209d9975197a777fe646e56405ca6fd4880ea Mon Sep 17 00:00:00 2001
From: Ansuman Sahoo <anshumansahoo500@gmail.com>
Date: Fri, 15 Aug 2025 04:03:29 +0530
Subject: [PATCH] shutdown server if no client gets connected

Signed-off-by: Ansuman Sahoo <anshumansahoo500@gmail.com>
---
 pkg/driver/external/server/server.go | 38 +++++++++++++++++++++++-----
 1 file changed, 31 insertions(+), 7 deletions(-)

diff --git a/pkg/driver/external/server/server.go b/pkg/driver/external/server/server.go
index b0c4140a4e5..95d9f774582 100644
--- a/pkg/driver/external/server/server.go
+++ b/pkg/driver/external/server/server.go
@@ -64,7 +64,7 @@ func Serve(driver driver.Driver) {
 
 	kaProps := keepalive.ServerParameters{
 		Time:    10 * time.Second,
-		Timeout: 20 * time.Second,
+		Timeout: 30 * time.Second,
 	}
 
 	kaPolicy := keepalive.EnforcementPolicy{
@@ -85,17 +85,41 @@ func Serve(driver driver.Driver) {
 	sigs := make(chan os.Signal, 1)
 	signal.Notify(sigs, syscall.SIGINT, syscall.SIGTERM)
 
+	shutdownCh := make(chan struct{})
+
 	go func() {
 		<-sigs
 		logger.Info("Received shutdown signal, stopping server...")
-		server.GracefulStop()
+		close(shutdownCh)
 	}()
 
-	logger.Infof("Starting external driver server for %s", driver.Info().DriverName)
-	logger.Infof("Server starting on Unix socket: %s", socketPath)
-	if err := server.Serve(listener); err != nil {
-		logger.Fatalf("Failed to serve: %v", err)
-	}
+	go func() {
+		timer := time.NewTimer(60 * time.Second)
+		defer timer.Stop()
+
+		select {
+		case <-timer.C:
+			logger.Info("No client connected within 60 seconds, shutting down server...")
+			close(shutdownCh)
+		case <-shutdownCh:
+			return
+		}
+	}()
+
+	go func() {
+		logger.Infof("Starting external driver server for %s", driver.Info().DriverName)
+		logger.Infof("Server starting on Unix socket: %s", socketPath)
+		if err := server.Serve(listener); err != nil {
+			if errors.Is(err, grpc.ErrServerStopped) {
+				logger.Errorf("Server stopped: %v", err)
+			} else {
+				logger.Errorf("Failed to serve: %v", err)
+			}
+		}
+	}()
+
+	<-shutdownCh
+	server.GracefulStop()
 }
 
 func Start(extDriver *registry.ExternalDriver, instName string) error {