⚙️ refactor: Signal handling in daemon

zutil/daemon/signal.go

@@ -2,41 +2,49 @@ package daemon
 
 import (
 	"sync"
-
-	"github.com/sohaha/zlsgo/zutil"
 )
 
 var (
 	singleSignal sync.Once
-	single       = make(chan bool)
-	singleNum    = zutil.NewInt32(0)
+	single            = make(chan bool)
+	singleNum    uint = 0
+	singleLock   sync.Mutex
 )
 
 func singelDo() {
 	singleSignal.Do(func() {
 		go func() {
 			kill := KillSignal()
+			singleLock.Lock()
 			for {
-				if int(singleNum.Sub(1)) < 0 {
-					singleNum.Add(1)
+				if singleNum == 0 {
 					break
 				}
+
+				singleNum--
 				single <- kill
 			}
 			close(single)
+			singleLock.Unlock()
 		}()
 	})
 }
 
 func SingleKillSignal() <-chan bool {
-	singleNum.Add(1)
+	singleLock.Lock()
+	defer singleLock.Unlock()
+
+	singleNum++
 	singelDo()
 
 	return single
 }
 
 func ReSingleKillSignal() {
-	if singleNum.Load() > 0 {
+	singleLock.Lock()
+	defer singleLock.Unlock()
+
+	if singleNum > 0 {
 		return
 	}
 

zutil/daemon/signal_test.go

@@ -35,15 +35,18 @@ func TestSignal(t *testing.T) {
 		tip = "kill"
 		t.Log(k, ok)
 	}
+}
 
+func TestSingleKillSignal(t *testing.T) {
+	tt := zls.NewTest(t)
 	go func() {
 		time.Sleep(time.Second * 1)
 		process, err := os.FindProcess(os.Getpid())
 		tt.NoError(err, true)
 		process.Signal(os.Interrupt)
 	}()
 
-	now = time.Now()
+	now := time.Now()
 	<-SingleKillSignal()
 	tt.EqualTrue(time.Since(now) > time.Second*1)