阶段性提交

pkg/serverconfigs/scheduling/scheduling_random.go

@@ -0,0 +1,78 @@
+package scheduling
+
+import (
+	"github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs/shared"
+	"github.com/iwind/TeaGo/maps"
+	"math"
+	"math/rand"
+	"time"
+)
+
+// 随机调度算法
+type RandomScheduling struct {
+	Scheduling
+
+	array []CandidateInterface
+	count uint // 实际总的服务器数
+}
+
+// 启动
+func (this *RandomScheduling) Start() {
+	sumWeight := uint(0)
+	for _, c := range this.Candidates {
+		weight := c.CandidateWeight()
+		if weight == 0 {
+			weight = 1
+		} else if weight > 10000 {
+			weight = 10000
+		}
+		sumWeight += weight
+	}
+
+	if sumWeight == 0 {
+		return
+	}
+
+	for _, c := range this.Candidates {
+		weight := c.CandidateWeight()
+		if weight == 0 {
+			weight = 1
+		} else if weight > 10000 {
+			weight = 10000
+		}
+		count := uint(0)
+		if sumWeight <= 1000 {
+			count = weight
+		} else {
+			count = uint(math.Round(float64(weight*10000) / float64(sumWeight))) // 1% 产生 100个数据，最多支持10000个服务器
+		}
+		for i := uint(0); i < count; i++ {
+			this.array = append(this.array, c)
+		}
+		this.count += count
+	}
+
+	rand.Seed(time.Now().UnixNano())
+}
+
+// 获取下一个候选对象
+func (this *RandomScheduling) Next(call *shared.RequestCall) CandidateInterface {
+	if this.count == 0 {
+		return nil
+	}
+	if this.count == 1 {
+		return this.array[0]
+	}
+	index := rand.Int() % int(this.count)
+	return this.array[index]
+}
+
+// 获取简要信息
+func (this *RandomScheduling) Summary() maps.Map {
+	return maps.Map{
+		"code":        "random",
+		"name":        "Random随机算法",
+		"description": "根据权重设置随机分配后端服务器",
+		"networks":    []string{"http", "tcp"},
+	}
+}