Initial commit (code only without large binaries)

EdgeNode/internal/utils/trackers/label.go

@@ -0,0 +1,32 @@
+// Copyright 2021 Liuxiangchao iwind.liu@gmail.com. All rights reserved.
+
+package trackers
+
+import "time"
+
+type tracker struct {
+	label     string
+	startTime time.Time
+}
+
+func Begin(label string) *tracker {
+	return &tracker{label: label, startTime: time.Now()}
+}
+
+func Run(label string, f func()) {
+	var tr = Begin(label)
+	f()
+	tr.End()
+}
+
+func (this *tracker) End() {
+	SharedManager.Add(this.label, time.Since(this.startTime).Seconds()*1000)
+}
+
+func (this *tracker) Begin(subLabel string) *tracker {
+	return Begin(this.label + ":" + subLabel)
+}
+
+func (this *tracker) Add(duration time.Duration) {
+	this.startTime = this.startTime.Add(-duration)
+}

EdgeNode/internal/utils/trackers/manager.go

@@ -0,0 +1,47 @@
+// Copyright 2021 Liuxiangchao iwind.liu@gmail.com. All rights reserved.
+
+package trackers
+
+import (
+	"sync"
+)
+
+var SharedManager = NewManager()
+
+type Manager struct {
+	m      map[string][]float64 // label => time costs ms
+	locker sync.Mutex
+}
+
+func NewManager() *Manager {
+	return &Manager{m: map[string][]float64{}}
+}
+
+func (this *Manager) Add(label string, costMs float64) {
+	this.locker.Lock()
+	costs, ok := this.m[label]
+	if ok {
+		costs = append(costs, costMs)
+		if len(costs) > 5 { // 只取最近的N条
+			costs = costs[1:]
+		}
+		this.m[label] = costs
+	} else {
+		this.m[label] = []float64{costMs}
+	}
+	this.locker.Unlock()
+}
+
+func (this *Manager) Labels() map[string]float64 {
+	var result = map[string]float64{}
+	this.locker.Lock()
+	for label, costs := range this.m {
+		var sum float64
+		for _, cost := range costs {
+			sum += cost
+		}
+		result[label] = sum / float64(len(costs))
+	}
+	this.locker.Unlock()
+	return result
+}

EdgeNode/internal/utils/trackers/manager_test.go

@@ -0,0 +1,56 @@
+// Copyright 2021 Liuxiangchao iwind.liu@gmail.com. All rights reserved.
+
+package trackers_test
+
+import (
+	"github.com/TeaOSLab/EdgeNode/internal/utils/trackers"
+	"github.com/iwind/TeaGo/logs"
+	"testing"
+	"time"
+)
+
+func TestNewManager(t *testing.T) {
+	{
+		var tr = trackers.Begin("a")
+		tr.End()
+	}
+	{
+		var tr = trackers.Begin("a")
+		time.Sleep(1 * time.Millisecond)
+		tr.End()
+	}
+	{
+		var tr = trackers.Begin("a")
+		time.Sleep(2 * time.Millisecond)
+		tr.End()
+	}
+	{
+		var tr = trackers.Begin("a")
+		time.Sleep(3 * time.Millisecond)
+		tr.End()
+	}
+	{
+		var tr = trackers.Begin("a")
+		time.Sleep(4 * time.Millisecond)
+		tr.End()
+	}
+	{
+		var tr = trackers.Begin("a")
+		time.Sleep(5 * time.Millisecond)
+		tr.End()
+	}
+	{
+		var tr = trackers.Begin("b")
+		tr.End()
+	}
+
+	logs.PrintAsJSON(trackers.SharedManager.Labels(), t)
+}
+
+func TestTrackers_Add(t *testing.T) {
+	var tr = trackers.Begin("a")
+	time.Sleep(50 * time.Millisecond)
+	tr.Add(-10 * time.Millisecond)
+	tr.End()
+	t.Log(trackers.SharedManager.Labels())
+}