通过 context 通知 go 超时退出

Author: hwholiday

context/go_ctx.go

@@ -0,0 +1,50 @@
+package main
+
+import (
+	"context"
+	"fmt"
+	"time"
+)
+
+func main() {
+	ctx, cancel := context.WithTimeout(context.Background(), time.Second*4)
+	defer cancel()
+	go fmt.Println(doFuncSuccess(ctx))
+	go fmt.Println(doFuncFail(ctx))
+	time.Sleep(time.Second * 6)
+	fmt.Println("doFuncFail")
+	go fmt.Println(doFuncFail(ctx))
+	time.Sleep(time.Second * 2)
+}
+
+func doFuncSuccess(ctx context.Context) string {
+	select {
+	case <-ctx.Done():
+		return ctx.Err().Error()
+	default:
+	}
+	time.Sleep(time.Second * 3)
+	select {
+	case <-ctx.Done():
+		return ctx.Err().Error()
+	default:
+	}
+	return "success"
+}
+
+func doFuncFail(ctx context.Context) string {
+	select {
+	case <-ctx.Done():
+		return ctx.Err().Error()
+	default:
+	}
+
+	time.Sleep(time.Second * 5)
+
+	select {
+	case <-ctx.Done():
+		return ctx.Err().Error()
+	default:
+	}
+	return "fail"
+}

context/main.go

@@ -76,6 +76,10 @@ func ContextWithDeadline() {
 	req = req.WithContext(ctx)
 	client := &http.Client{}
 	res, err := client.Do(req)
+	if err != nil {
+		return
+	}
+	defer res.Body.Close()
 
 }