fatedier hace 3 años
padre
commit
f1a51eba18
Se han modificado 3 ficheros con 9 adiciones y 9 borrados
  1. 2 2
      client/control.go
  2. 6 6
      client/service.go
  3. 1 1
      cmd/frpc/sub/root.go

+ 2 - 2
client/control.go

@@ -185,11 +185,11 @@ func (ctl *Control) Close() error {
 	return ctl.GracefulClose(0)
 }
 
-func (ctl *Control) GracefulClose(t time.Duration) error {
+func (ctl *Control) GracefulClose(d time.Duration) error {
 	ctl.pm.Close()
 	ctl.vm.Close()
 
-	time.Sleep(t)
+	time.Sleep(d)
 
 	ctl.conn.Close()
 	if ctl.session != nil {

+ 6 - 6
client/service.go

@@ -311,14 +311,14 @@ func (svr *Service) ReloadConf(pxyCfgs map[string]config.ProxyConf, visitorCfgs
 	return svr.ctl.ReloadConf(pxyCfgs, visitorCfgs)
 }
 
-func (svr *Service) Close(t time.Duration) {
+func (svr *Service) Close() {
+	svr.GracefulClose(time.Duration(0))
+}
+
+func (svr *Service) GracefulClose(d time.Duration) {
 	atomic.StoreUint32(&svr.exit, 1)
 	if svr.ctl != nil {
-		if t > 0 {
-			svr.ctl.GracefulClose(t)
-		} else {
-			svr.ctl.Close()
-		}
+		svr.ctl.GracefulClose(d)
 	}
 	svr.cancel()
 }

+ 1 - 1
cmd/frpc/sub/root.go

@@ -124,7 +124,7 @@ func handleSignal(svr *client.Service) {
 	ch := make(chan os.Signal)
 	signal.Notify(ch, syscall.SIGINT, syscall.SIGTERM)
 	<-ch
-	svr.Close(time.Millisecond * 250)
+	svr.GracefulClose(500 * time.Millisecond)
 	close(kcpDoneCh)
 }