x

Author: chenquan

zrpc/internal/rpcpubserver.go

@@ -54,15 +54,16 @@ type keepAliveServer struct {
 func (s keepAliveServer) Start(fn RegisterFn) error {
 	errCh := make(chan error)
 	stopCh := make(chan struct{})
+	defer close(stopCh)
 	ctx, cancel := context.WithTimeout(context.Background(), time.Minute)
 	defer cancel()
 
 	go func() {
 		defer close(errCh)
 		select {
-		case <-ctx.Done():
 		case errCh <- s.Server.Start(fn):
 		case <-stopCh:
+			// prevent goroutine leak
 		}
 	}()
 
@@ -76,7 +77,6 @@ l:
 			if health.IsReady() {
 				err := s.registerEtcd()
 				if err != nil {
-					close(stopCh)
 					return err
 				}
 				// break for loop