From 59c396870ac2d81ec81113802d54277fe070d91b Mon Sep 17 00:00:00 2001
From: Huang Youliang <butterbright0504@gmail.com>
Date: Thu, 12 Sep 2024 13:21:43 +0800
Subject: [PATCH] Add benchmarks for topological queries (#530)

---
 pkg/test/query/metric.go              | 62 +++++++++++++++++++++++++++
 test/stress/trace/trace_suite_test.go |  4 ++
 2 files changed, 66 insertions(+)

diff --git a/pkg/test/query/metric.go b/pkg/test/query/metric.go
index a890a1189..0cfdfe746 100644
--- a/pkg/test/query/metric.go
+++ b/pkg/test/query/metric.go
@@ -25,6 +25,7 @@ import (
 	"sync"
 	"time"
 
+	"github.com/apache/skywalking-cli/pkg/graphql/dependency"
 	"github.com/apache/skywalking-cli/pkg/graphql/metrics"
 	"github.com/apache/skywalking-cli/pkg/graphql/utils"
 	"github.com/urfave/cli/v2"
@@ -191,3 +192,64 @@ func sortMetrics(name, svc string, limit int, order api.Order, fs *flag.FlagSet)
 	}
 	return elapsed, nil
 }
+
+// Topology verifies the topology.
+func Topology(basePath string, timeout time.Duration, groupNum int, fs *flag.FlagSet) {
+	basePath = path.Join(basePath, "topology")
+	err := os.MkdirAll(basePath, 0o755)
+	if err != nil {
+		panic(err)
+	}
+	stopCh := make(chan struct{})
+	go func() {
+		time.Sleep(timeout)
+		close(stopCh)
+	}()
+	header := make([]string, groupNum)
+	for i := 0; i < groupNum; i++ {
+		header[i] = fmt.Sprintf("topology-%d", i)
+	}
+	var wg sync.WaitGroup
+	wg.Add(1)
+	go func() {
+		defer wg.Done()
+		collect(basePath, func() ([]float64, error) {
+			data := make([]float64, groupNum)
+			var subWg sync.WaitGroup
+			for i := 0; i < groupNum; i++ {
+				subWg.Add(1)
+				go func() {
+					defer subWg.Done()
+					d, err := globalTopology(fs)
+					if err != nil {
+						fmt.Printf("query topology error: %v \n", err)
+					}
+					data[i] = d.Seconds()
+				}()
+			}
+			subWg.Wait()
+			return data, nil
+		}, 500*time.Millisecond, stopCh)
+		analyze(header, basePath)
+	}()
+	wg.Wait()
+}
+
+func globalTopology(fs *flag.FlagSet) (time.Duration, error) {
+	ctx := cli.NewContext(cli.NewApp(), fs, nil)
+	duration := api.Duration{
+		Start: time.Now().Add(-200 * time.Minute).Format(utils.StepFormats[api.StepMinute]),
+		End:   time.Now().Format(utils.StepFormats[api.StepMinute]),
+		Step:  api.StepMinute,
+	}
+	start := time.Now()
+	result, err := dependency.GlobalTopologyWithoutLayer(ctx, duration)
+	elapsed := time.Since(start)
+	if err != nil {
+		return 0, err
+	}
+	if len(result.Nodes) < 1 {
+		return 0, fmt.Errorf("no result")
+	}
+	return elapsed, nil
+}
diff --git a/test/stress/trace/trace_suite_test.go b/test/stress/trace/trace_suite_test.go
index 0aaee55b4..b26acbf69 100644
--- a/test/stress/trace/trace_suite_test.go
+++ b/test/stress/trace/trace_suite_test.go
@@ -76,4 +76,8 @@ var _ = Describe("Query", func() {
 	It("TopN", func() {
 		query.TopN(basePath, timeout, 1, fs)
 	})
+
+	It("Topology", func() {
+		query.Topology(basePath, timeout, 1, fs)
+	})
 })