处理 ctrl+c 信号

main.go

@@ -5,8 +5,11 @@ import (
 	"context"
 	"flag"
 	"fmt"
+	"os"
+	"os/signal"
 	"runtime"
 	"strings"
+	"syscall"
 	"time"
 
 	"github.com/link1st/go-stress-testing/model"
@@ -21,7 +24,7 @@ func (a *array) String() string {
 	return fmt.Sprint(*a)
 }
 
-// Set set
+// Set arr set
 func (a *array) Set(s string) error {
 	*a = append(*a, s)
 
@@ -97,6 +100,15 @@ func main() {
 			fmt.Printf(" deadline %s", deadline)
 		}
 	}
+
+	// 处理 ctrl+c 信号
+	ctx, cancelFunc := context.WithCancel(ctx)
+	go func() {
+		c := make(chan os.Signal)
+		signal.Notify(c, syscall.SIGINT)
+		<-c
+		cancelFunc()
+	}()
 	server.Dispose(ctx, concurrency, totalNumber, request)
 	return
 }

server/golink/http_link.go

@@ -4,7 +4,6 @@ package golink
 import (
 	"compress/gzip"
 	"context"
-	"fmt"
 	"io"
 	"net/http"
 	"sync"
@@ -22,7 +21,7 @@ func HTTP(ctx context.Context, chanID uint64, ch chan<- *model.RequestResults, t
 	// fmt.Printf("启动协程 编号:%05d \n", chanID)
 	for i := uint64(0); i < totalNumber; i++ {
 		if ctx.Err() != nil {
-			fmt.Printf("ctx.Err err: %v \n", ctx.Err())
+			// fmt.Printf("ctx.Err err: %v \n", ctx.Err())
 			break
 		}