增加FactoryManager

Author: xfali

factory/multisource.go

@@ -0,0 +1,84 @@
+// Copyright (C) 2019-2020, Xiongfa Li.
+// @author xiongfa.li
+// @version V1.0
+// Description: 
+
+package factory
+
+import (
+	"github.com/xfali/loadbalance"
+)
+
+type LoadBalanceType int
+
+const (
+	LBRoundRobbin       LoadBalanceType = loadbalance.LBRoundRobbin
+	LBRoundRobbinWeight LoadBalanceType = loadbalance.LBRoundRobbinWeight
+	LBRandom            LoadBalanceType = loadbalance.LBRandom
+	LBRandomWeight      LoadBalanceType = loadbalance.LBRandomWeight
+
+	DefaultGroup = "default"
+)
+
+type Manager interface {
+	Bind(action string, weight int, factory Factory)
+	Select(action string) Factory
+}
+
+type SingleSource struct {
+	fac Factory
+}
+
+func NewSingleSource(fac Factory) *SingleSource {
+	return &SingleSource{fac: fac}
+}
+
+func (lb *SingleSource) Bind(action string, weight int, factory Factory) {
+	lb.fac = factory
+}
+
+func (lb *SingleSource) Select(action string) Factory {
+	return lb.fac
+}
+
+type DefaultMultiSource struct {
+	lbType      int
+	actionMaps  map[string]loadbalance.LoadBalance
+	factoryMaps map[Factory]loadbalance.LoadBalance
+}
+
+func NewMultiSource(t LoadBalanceType) *DefaultMultiSource {
+	return &DefaultMultiSource{
+		actionMaps: map[string]loadbalance.LoadBalance{},
+		lbType:     int(t),
+	}
+}
+
+func (lb *DefaultMultiSource) Bind(action string, weight int, factory Factory) {
+	if action == "" {
+		action = DefaultGroup
+	}
+
+	if v, ok := lb.actionMaps[action]; ok {
+		v.Add(weight, factory)
+	} else {
+		if f, ok := lb.factoryMaps[factory]; ok {
+			lb.actionMaps[action] = f
+		} else {
+			newlb := loadbalance.Create(lb.lbType)
+			newlb.Add(weight, factory)
+			lb.actionMaps[action] = newlb
+			lb.factoryMaps[factory] = newlb
+		}
+	}
+}
+
+func (lb *DefaultMultiSource) Select(action string) Factory {
+	if v, ok := lb.actionMaps[action]; ok {
+		f := v.Select(nil)
+		if f != nil {
+			return f.(Factory)
+		}
+	}
+	return nil
+}

go.mod

@@ -6,4 +6,5 @@ require (
 	github.com/go-sql-driver/mysql v1.4.1 // indirect
 	github.com/lib/pq v1.3.0 // indirect
 	github.com/mattn/go-sqlite3 v1.10.0 // indirect
+	github.com/xfali/loadbalance v0.0.1 // indirect
 )

go.sum

@@ -4,3 +4,5 @@ github.com/lib/pq v1.3.0 h1:/qkRGz8zljWiDcFvgpwUpwIAPu3r07TDvs3Rws+o/pU=
 github.com/lib/pq v1.3.0/go.mod h1:5WUZQaWbwv1U+lTReE5YruASi9Al49XbQIvNi/34Woo=
 github.com/mattn/go-sqlite3 v1.10.0 h1:jbhqpg7tQe4SupckyijYiy0mJJ/pRyHvXf7JdWK860o=
 github.com/mattn/go-sqlite3 v1.10.0/go.mod h1:FPy6KqzDD04eiIsT53CuJW3U88zkxoIYsOqkbpncsNc=
+github.com/xfali/loadbalance v0.0.1 h1:UVOuuDipJ740KyTaBUxTeW6UEC+fukiQCOn/5YQgexA=
+github.com/xfali/loadbalance v0.0.1/go.mod h1:yrzHHRZMdt2wBpLnBDeU2zbjnRYeNvUWV6sq6+3KVG0=