fix: kcp protocol cause delay release resource (#2621)

Co-authored-by: blizard863 <760076784@qq.com>
This commit is contained in:
Blizard
2021-10-19 14:57:26 +08:00
committed by GitHub
parent 998e678a7f
commit 1d26ea440b
3 changed files with 15 additions and 5 deletions

View File

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