diff options
Diffstat (limited to 'pkg/driver/server.go')
| -rw-r--r-- | pkg/driver/server.go | 7 |
1 files changed, 4 insertions, 3 deletions
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) - } |