[incubator-eventmesh] branch master updated: go sdk add loadbalance, add weight round robin selector

[mi...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

mikexue pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-eventmesh.git


The following commit(s) were added to refs/heads/master by this push:
     new 026e35e7 go sdk add loadbalance, add weight round robin selector
     new 6f196be5 Merge pull request #1751 from horoc/add-go-sdk-loadbalance-weight-round-robin
026e35e7 is described below

commit 026e35e7d8ed2fcdd8739ae4a92f00c0ecbe79fe
Author: horoc <ho...@gmail.com>
AuthorDate: Sat Oct 22 17:16:49 2022 +0800

    go sdk add loadbalance, add weight round robin selector
---
 .../http/selector/weight_round_robin_selector.go   | 84 ++++++++++++++++++++++
 1 file changed, 84 insertions(+)

diff --git a/eventmesh-sdk-go/http/selector/weight_round_robin_selector.go b/eventmesh-sdk-go/http/selector/weight_round_robin_selector.go
new file mode 100644
index 00000000..b9a09a13
--- /dev/null
+++ b/eventmesh-sdk-go/http/selector/weight_round_robin_selector.go
@@ -0,0 +1,84 @@
+// Licensed to the Apache Software Foundation (ASF) under one or more
+// contributor license agreements.  See the NOTICE file distributed with
+// this work for additional information regarding copyright ownership.
+// The ASF licenses this file to You under the Apache License, Version 2.0
+// (the "License"); you may not use this file except in compliance with
+// the License.  You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package selector
+
+import (
+	"errors"
+	"github.com/apache/incubator-eventmesh/eventmesh-sdk-go/http/conf"
+	"go.uber.org/atomic"
+)
+
+const WeightRoundRobinSelectorType = "weight_round_robin"
+
+func init() {
+	registerSelectorBuilder(WeightRoundRobinSelectorType, buildWeightRoundRobinLoadSelector)
+}
+
+func buildWeightRoundRobinLoadSelector(config *conf.EventMeshHttpClientConfig) (LoadBalanceSelector, error) {
+	meshNodes, err := parseWeightedMeshNodeFromConfig(config)
+	if err != nil {
+		return nil, err
+	}
+
+	return NewWeightRoundRobinLoadSelector(meshNodes)
+}
+
+type WeightRoundRobinMeshNode struct {
+	MeshNode
+	currentWeight *atomic.Int64
+}
+
+type WeightRoundRobinLoadSelector struct {
+	clusterGroup []WeightRoundRobinMeshNode
+	totalWeight  int64
+}
+
+func NewWeightRoundRobinLoadSelector(clusterGroup []MeshNode) (*WeightRoundRobinLoadSelector, error) {
+	if len(clusterGroup) == 0 {
+		return nil, errors.New("fail to create weight round robin load balancer selector, cluster group is empty")
+	}
+	var totalWeight int64 = 0
+	weightRoundRobinClusterGroup := make([]WeightRoundRobinMeshNode, 0, len(clusterGroup))
+	for _, node := range clusterGroup {
+		totalWeight += node.Weight
+		weightRoundRobinClusterGroup = append(weightRoundRobinClusterGroup, WeightRoundRobinMeshNode{
+			MeshNode:      node,
+			currentWeight: atomic.NewInt64(0),
+		})
+	}
+
+	return &WeightRoundRobinLoadSelector{
+		clusterGroup: weightRoundRobinClusterGroup,
+		totalWeight:  totalWeight,
+	}, nil
+}
+
+func (s *WeightRoundRobinLoadSelector) Select() MeshNode {
+	var targetWeight *WeightRoundRobinMeshNode = nil
+	for i := range s.clusterGroup {
+		node := &s.clusterGroup[i]
+		node.currentWeight.Add(node.Weight)
+		if targetWeight == nil || targetWeight.currentWeight.Load() < node.currentWeight.Load() {
+			targetWeight = node
+		}
+	}
+	targetWeight.currentWeight.Sub(s.totalWeight)
+	return targetWeight.MeshNode
+}
+
+func (s *WeightRoundRobinLoadSelector) GetType() string {
+	return WeightRoundRobinSelectorType
+}


---------------------------------------------------------------------
To unsubscribe, e-mail: commits-unsubscribe@eventmesh.apache.org
For additional commands, e-mail: commits-help@eventmesh.apache.org