Implemented priority + weight balancing #204

config/gobetween.toml

@@ -81,8 +81,9 @@ bind = "0.0.0.0:3000"
   [servers.sample.discovery]
   kind = "static"
   static_list = [
-      "localhost:8000",
-      "localhost:8001"
+      "localhost:8000 weight=40 priority=1",
+      "localhost:8001 weight=60 priority=1",
+      "localhost:8002 weight=100 priority=0"
   ]
 
 # ---------- udp example ----------- #

src/balance/weight.go

@@ -8,6 +8,7 @@ package balance
 
 import (
 	"errors"
+	"fmt"
 	"math/rand"
 
 	"github.com/yyyar/gobetween/core"
@@ -19,26 +20,47 @@ import (
 type WeightBalancer struct{}
 
 /**
- * Elect backend based on weight strategy
+ * Elect backend based on weight with priority strategy. See https://tools.ietf.org/html/rfc2782, Priority and Weight sections
  */
 func (b *WeightBalancer) Elect(context core.Context, backends []*core.Backend) (*core.Backend, error) {
 
 	if len(backends) == 0 {
 		return nil, errors.New("Can't elect backend, Backends empty")
 	}
 
+	lp := backends[0].Priority
+	lpBackends := make([]*core.Backend, 0)
 	totalWeight := 0
+
 	for _, backend := range backends {
+
+		if backend.Priority > lp {
+			continue
+		}
+
+		if backend.Priority < 0 {
+			return nil, fmt.Errorf("Invalid backend priority, shold not be less than 0: %v", backend.Priority)
+		}
+
 		if backend.Weight <= 0 {
-			return nil, errors.New("Invalid backend weight 0")
+			return nil, fmt.Errorf("Invalid backend weight, should not be less or equal to 0: %v", backend.Weight)
+		}
+
+		// got new lower priority, reset
+		if backend.Priority < lp {
+			lp = backend.Priority
+			lpBackends = make([]*core.Backend, 0)
+			totalWeight = 0
 		}
+
+		lpBackends = append(lpBackends, backend)
 		totalWeight += backend.Weight
 	}
 
 	r := rand.Intn(totalWeight)
 	pos := 0
 
-	for _, backend := range backends {
+	for _, backend := range lpBackends {
 		pos += backend.Weight
 		if r >= pos {
 			continue