add stop pxy.StopCheck

main.go

@@ -68,6 +68,7 @@ func main() {
 
 	// enable checkers
 	pxy.Check()
+	defer pxy.StopCheck()
 
 	// run proxy servers
 	for _, listen := range config.Listens {

rule/group.go

@@ -33,6 +33,7 @@ type FwdrGroup struct {
 	index    uint32
 	priority uint32
 	next     func(addr string) *Forwarder
+	stopChan chan bool
 }
 
 // NewFwdrGroup returns a new forward group.
@@ -235,6 +236,11 @@ func (p *FwdrGroup) check(fwdr *Forwarder, checker Checker) {
 	intval := time.Duration(p.config.CheckInterval) * time.Second
 
 	for {
+		select {
+		case <-p.stopChan:
+			log.F("[check] %s: stop checking", p.name)
+			return
+		default:
 			time.Sleep(intval * time.Duration(wait))
 
 			// check all forwarders at least one time
@@ -272,6 +278,7 @@ func (p *FwdrGroup) check(fwdr *Forwarder, checker Checker) {
 			fwdr.Enable()
 		}
 	}
+}
 
 func (p *FwdrGroup) setLatency(fwdr *Forwarder, elapsed time.Duration) {
 	newLatency := int64(elapsed)
@@ -316,3 +323,8 @@ func (p *FwdrGroup) scheduleDH(dstAddr string) *Forwarder {
 	fnv1a.Write([]byte(dstAddr))
 	return p.avail[fnv1a.Sum32()%uint32(len(p.avail))]
 }
+
+// Stop Check
+func (p *FwdrGroup) StopCheck() {
+	p.stopChan <- true
+}

rule/proxy.go

@@ -154,3 +154,12 @@ func (p *Proxy) Check() {
 		fwdrGroup.Check()
 	}
 }
+
+// Stop All checkers.
+func (p *Proxy) StopCheck() {
+	p.main.StopCheck()
+
+	for _, fwdrGroup := range p.all {
+		fwdrGroup.StopCheck()
+	}
+}