chore: refactor (#39)

* feat: support connection stats * chore: refactor
kevwan · May 6, 2023 · c713ba7 · c713ba7
1 parent 6a555eb
commit c713ba7
Showing 1 changed file with 12 additions and 15 deletions.
diff --git a/stat.go b/stat.go
@@ -31,21 +31,18 @@ func NewStater(staters ...Stater) Stater {
 		c := make(chan os.Signal, 1)
 		signal.Notify(c, os.Interrupt, syscall.SIGTERM)
 
-		for {
-			select {
-			case sig := <-c:
-				signal.Stop(c)
-				stat.Stop()
-
-				p, err := os.FindProcess(syscall.Getpid())
-				if err != nil {
-					fmt.Println(err)
-					os.Exit(0)
-				}
-
-				if err := p.Signal(sig); err != nil {
-					fmt.Println(err)
-				}
+		for sig := range c {
+			signal.Stop(c)
+			stat.Stop()
+
+			p, err := os.FindProcess(syscall.Getpid())
+			if err != nil {
+				fmt.Println(err)
+				os.Exit(0)
+			}
+
+			if err := p.Signal(sig); err != nil {
+				fmt.Println(err)
 			}
 		}
 	}()