From 251eb9b4b8dc52ca55f3903b64f4f7027343fd5b Mon Sep 17 00:00:00 2001 From: Viktor Kuzmin Date: Sun, 6 Aug 2023 23:23:54 +0300 Subject: Graceful stop with mounts cleanup --- pkg/driver/server.go | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) (limited to 'pkg/driver/server.go') diff --git a/pkg/driver/server.go b/pkg/driver/server.go index 99580ae..b998533 100644 --- a/pkg/driver/server.go +++ b/pkg/driver/server.go @@ -34,10 +34,12 @@ type nonBlockingGRPCServer struct { } func (s *nonBlockingGRPCServer) Start(endpoint string, ids csi.IdentityServer, cs csi.ControllerServer, ns csi.NodeServer) { - s.wg.Add(1) - go s.serve(endpoint, ids, cs, ns) + go func() { + defer s.wg.Done() + s.serve(endpoint, ids, cs, ns) + }() return } @@ -92,5 +94,4 @@ func (s *nonBlockingGRPCServer) serve(endpoint string, ids csi.IdentityServer, c glog.Infof("Listening for connections on address: %#v", listener.Addr()) server.Serve(listener) - } -- cgit v1.2.3