diff options
| author | Viktor Kuzmin <kvaster@gmail.com> | 2023-08-06 23:23:54 +0300 |
|---|---|---|
| committer | Chris Lu <chrislusf@users.noreply.github.com> | 2023-08-07 10:22:05 -0700 |
| commit | 251eb9b4b8dc52ca55f3903b64f4f7027343fd5b (patch) | |
| tree | eb92e65f3dcea999d094bf5e446d57d0d407ec30 /pkg/driver/server.go | |
| parent | d84db4bbebec5d8c0d0e20dbf9deffdd24ba1151 (diff) | |
| download | seaweedfs-csi-driver-251eb9b4b8dc52ca55f3903b64f4f7027343fd5b.tar.xz seaweedfs-csi-driver-251eb9b4b8dc52ca55f3903b64f4f7027343fd5b.zip | |
Graceful stop with mounts cleanup
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) - } |