新增测试代码

Signed-off-by: allan716 <[email protected]>

TestCode/test_context.go

@@ -0,0 +1,67 @@
+package TestCode
+
+import (
+	"errors"
+	"fmt"
+	"golang.org/x/net/context"
+	"time"
+)
+
+func baseHardWork(job interface{}) error {
+	index := job.(int)
+	println("Index:", index, "Start")
+	time.Sleep(time.Second * 1)
+	println("Index:", index, "End")
+	return nil
+}
+
+func DoThings(ctx context.Context) error {
+	const total = 10
+	for i := 0; i < total; i++ {
+		// 创建一个 chan 用于任务的中断和超时
+		done := make(chan interface{}, 1)
+		// 接收内部任务的 panic
+		panicChan := make(chan interface{}, 1)
+
+		go func() {
+			if p := recover(); p != nil {
+				panicChan <- p
+			}
+			// 匹配对应的 Eps 去处理
+			done <- baseHardWork(i)
+		}()
+
+		select {
+		case errInterface := <-done:
+			if errInterface != nil {
+				println("Index:", i, "Error:", errInterface)
+			}
+			break
+		case p := <-panicChan:
+			// 遇到内部的 panic，向外抛出
+			panic(p)
+		case <-ctx.Done():
+			{
+				err := errors.New(fmt.Sprintf("cancel at index: %d", i))
+				return err
+			}
+		}
+	}
+
+	return nil
+}
+
+func MainProcess() {
+	ctx, cancel := context.WithTimeout(context.Background(), time.Second*5)
+	defer cancel()
+
+	go func() {
+		err := DoThings(ctx)
+		if err != nil {
+			println("Error:", err.Error())
+		}
+	}()
+
+	time.Sleep(5 * time.Second)
+	cancel()
+}

TestCode/test_context_test.go

@@ -0,0 +1,8 @@
+package TestCode
+
+import "testing"
+
+func TestMainProcess(t *testing.T) {
+
+	MainProcess()
+}