From 8ab9bcc1fe4d947ff612976a0447adebf3de5d6d Mon Sep 17 00:00:00 2001
From: yiGmMk <marvelousme@163.com>
Date: Fri, 10 Nov 2023 11:28:52 +0800
Subject: [PATCH] =?UTF-8?q?refactor:=20=E7=94=A8=E6=88=B7=E4=BB=BB?=
 =?UTF-8?q?=E5=8A=A1,go=20trace?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 base/runtime/trace/trace_test.go | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/base/runtime/trace/trace_test.go b/base/runtime/trace/trace_test.go
index bc1f641..099e72c 100644
--- a/base/runtime/trace/trace_test.go
+++ b/base/runtime/trace/trace_test.go
@@ -1,6 +1,7 @@
 package trace
 
 import (
+	"context"
 	"fmt"
 	"log"
 	"os"
@@ -29,6 +30,10 @@ func TestTrace(t *testing.T) {
 	}
 	defer trace.Stop()
 
+	// 用户任务
+	ctx, task := trace.NewTask(context.Background(), "trace 用户任务")
+	defer task.End()
+
 	// 下面就是你的监控的程序
 	// 我简单写了一个文件读写
 	var wg sync.WaitGroup
@@ -37,6 +42,9 @@ func TestTrace(t *testing.T) {
 	// 一个协程用来读文件
 	go func() {
 		defer wg.Done()
+		r := trace.StartRegion(ctx, "reading file")
+		defer r.End()
+
 		content, err := os.ReadFile(`mxc.txt`)
 		fmt.Println(content, err)
 	}()
@@ -44,6 +52,9 @@ func TestTrace(t *testing.T) {
 	// 写文件协程
 	go func() {
 		defer wg.Done()
+		r := trace.StartRegion(ctx, "reading file")
+		defer r.End()
+
 		fmt.Println("hello trace")
 	}()
 	wg.Wait()