From 8b9fc6e69f1668bfd56959c5d25da3560a2616ad Mon Sep 17 00:00:00 2001
From: baixl <baixl>
Date: Wed, 5 Jul 2023 18:35:07 +0800
Subject: [PATCH] Immediately stop check

---
 rule/group.go | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/rule/group.go b/rule/group.go
index e1a8fc0..2272116 100644
--- a/rule/group.go
+++ b/rule/group.go
@@ -33,7 +33,7 @@ type FwdrGroup struct {
 	index    uint32
 	priority uint32
 	next     func(addr string) *Forwarder
-	stopChan chan bool
+	stopChan chan struct{}
 }
 
 // NewFwdrGroup returns a new forward group.
@@ -66,7 +66,7 @@ func NewFwdrGroup(rulePath string, s []string, c *Strategy) *FwdrGroup {
 
 // newFwdrGroup returns a new FwdrGroup.
 func newFwdrGroup(name string, fwdrs []*Forwarder, c *Strategy) *FwdrGroup {
-	p := &FwdrGroup{name: name, fwdrs: fwdrs, config: c}
+	p := &FwdrGroup{name: name, fwdrs: fwdrs, config: c, stopChan: make(chan struct{})}
 	sort.Sort(p.fwdrs)
 
 	p.init()
@@ -240,7 +240,7 @@ func (p *FwdrGroup) check(fwdr *Forwarder, checker Checker) {
 		case <-p.stopChan:
 			log.F("[check] %s: stop checking", p.name)
 			return
-		default:
+		case <-time.After(intval * time.Duration(wait)):
 			time.Sleep(intval * time.Duration(wait))
 
 			// check all forwarders at least one time
@@ -326,5 +326,5 @@ func (p *FwdrGroup) scheduleDH(dstAddr string) *Forwarder {
 
 // Stop Check
 func (p *FwdrGroup) StopCheck() {
-	p.stopChan <- true
+	close(p.stopChan)
 }