From 8075e19d5b86674a7f622b8f20d72487537d7f94 Mon Sep 17 00:00:00 2001
From: GoEdgeLab <goedgecloud@gmail.com>
Date: Thu, 18 Apr 2024 10:02:09 +0800
Subject: [PATCH] =?UTF-8?q?=E6=94=B9=E8=BF=9B=E7=A9=BA=E9=97=B2=E6=97=B6?=
 =?UTF-8?q?=E9=97=B4=E7=AE=97=E6=B3=95?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 internal/utils/idles/run.go      | 82 +++++++++++++++++++++++---------
 internal/utils/idles/run_test.go | 15 ++++--
 2 files changed, 69 insertions(+), 28 deletions(-)

diff --git a/internal/utils/idles/run.go b/internal/utils/idles/run.go
index 8048e01..69cd383 100644
--- a/internal/utils/idles/run.go
+++ b/internal/utils/idles/run.go
@@ -9,6 +9,7 @@ import (
 	fsutils "github.com/TeaOSLab/EdgeNode/internal/utils/fs"
 	"github.com/iwind/TeaGo/Tea"
 	"github.com/shirou/gopsutil/v3/load"
+	"math"
 	"os"
 	"sort"
 	"time"
@@ -18,7 +19,7 @@ const maxSamples = 7
 const cacheFile = "idles.cache"
 
 var hourlyLoadMap = map[int]*HourlyLoad{}
-var minLoadHour = -1
+var sharedMinLoadHours []int
 
 type HourlyLoad struct {
 	Hour   int       `json:"hour"`
@@ -35,7 +36,10 @@ func init() {
 	{
 		data, err := os.ReadFile(Tea.Root + "/data/" + cacheFile)
 		if err == nil {
-			_ = json.Unmarshal(data, &hourlyLoadMap)
+			err = json.Unmarshal(data, &hourlyLoadMap)
+			if err == nil {
+				calculateMinLoadHours()
+			}
 		}
 	}
 
@@ -70,36 +74,37 @@ func CheckHourlyLoad(hour int) {
 	for _, v := range hourlyLoad.Values {
 		sum += v
 	}
-	hourlyLoad.Avg = sum / float64(len(hourlyLoad.Values))
+	hourlyLoad.Avg = math.Ceil(sum/float64(len(hourlyLoad.Values))*10) / 10 // fix precision
 
-	// calculate min load hour
-	var allLoads = []*HourlyLoad{}
-	for _, v := range hourlyLoadMap {
-		allLoads = append(allLoads, v)
-	}
-
-	sort.Slice(allLoads, func(i, j int) bool {
-		return allLoads[i].Avg < allLoads[j].Avg
-	})
-
-	minLoadHour = allLoads[0].Hour
-
-	// write to cache
-	hourlyLoadMapJSON, err := json.Marshal(hourlyLoadMap)
-	if err == nil {
-		_ = os.WriteFile(Tea.Root+"/data/"+cacheFile, hourlyLoadMapJSON, 0666)
-	}
+	calculateMinLoadHours()
 }
 
 func Run(f func()) {
 	defer f()
 
-	if minLoadHour < 0 {
+	var minLoadHours = sharedMinLoadHours // copy
+
+	if len(minLoadHours) == 0 {
 		fsutils.WaitLoad(15, 8, time.Hour)
 		return
 	}
 
 	var hour = time.Now().Hour()
+	var minLoadHour = -1
+	for _, v := range minLoadHours {
+		if v == hour {
+			minLoadHour = v
+			break
+		}
+		if v > hour {
+			minLoadHour = v
+			break
+		}
+	}
+	if minLoadHour < 0 {
+		minLoadHour = minLoadHours[0]
+	}
+
 	if minLoadHour == hour {
 		fsutils.WaitLoad(15, 10, time.Minute)
 		return
@@ -119,8 +124,39 @@ func RunTicker(ticker *time.Ticker, f func()) {
 	}
 }
 
-func TestMinLoadHour() int {
-	return minLoadHour
+func calculateMinLoadHours() {
+	var allLoads = []*HourlyLoad{}
+	for _, v := range hourlyLoadMap {
+		allLoads = append(allLoads, v)
+	}
+
+	sort.Slice(allLoads, func(i, j int) bool {
+		return allLoads[i].Avg < allLoads[j].Avg
+	})
+
+	var minAvgLoad = allLoads[0].Avg
+	var newMinLoadHours []int
+	for _, v := range allLoads {
+		if v.Avg == minAvgLoad {
+			newMinLoadHours = append(newMinLoadHours, v.Hour)
+		}
+	}
+	sort.Ints(newMinLoadHours)
+	sharedMinLoadHours = newMinLoadHours
+
+	// write to cache
+	hourlyLoadMapJSON, err := json.Marshal(hourlyLoadMap)
+	if err == nil {
+		_ = os.WriteFile(Tea.Root+"/data/"+cacheFile, hourlyLoadMapJSON, 0666)
+	}
+}
+
+func TestMinLoadHours() []int {
+	return sharedMinLoadHours
+}
+
+func TestSetMinLoadHours(minLoadHours []int) {
+	sharedMinLoadHours = minLoadHours
 }
 
 func TestHourlyLoadMap() map[int]*HourlyLoad {
diff --git a/internal/utils/idles/run_test.go b/internal/utils/idles/run_test.go
index c68a6c0..8aeae47 100644
--- a/internal/utils/idles/run_test.go
+++ b/internal/utils/idles/run_test.go
@@ -5,7 +5,8 @@ package idles_test
 import (
 	"github.com/TeaOSLab/EdgeNode/internal/utils/idles"
 	"github.com/TeaOSLab/EdgeNode/internal/utils/testutils"
-	"github.com/iwind/TeaGo/logs"
+	_ "github.com/iwind/TeaGo/bootstrap"
+	"github.com/iwind/TeaGo/types"
 	timeutil "github.com/iwind/TeaGo/utils/time"
 	"testing"
 	"time"
@@ -13,17 +14,21 @@ import (
 
 func TestCheckHourlyLoad(t *testing.T) {
 	for i := 0; i < 10; i++ {
+		idles.CheckHourlyLoad(5)
 		idles.CheckHourlyLoad(1)
-		idles.CheckHourlyLoad(2)
 		idles.CheckHourlyLoad(3)
+		idles.CheckHourlyLoad(2)
+		idles.CheckHourlyLoad(4)
 	}
 
-	t.Log(idles.TestMinLoadHour())
-	logs.PrintAsJSON(idles.TestHourlyLoadMap(), t)
+	t.Log(idles.TestMinLoadHours())
+	for h, v := range idles.TestHourlyLoadMap() {
+		t.Log(types.String(h)+":", v.Avg)
+	}
 }
 
 func TestRun(t *testing.T) {
-	//idles.CheckHourlyLoad(time.Now().Hour())
+	idles.TestSetMinLoadHours([]int{0, time.Now().Hour()})
 	idles.Run(func() {
 		t.Log("run once")
 	})