Added stringent type check to rr

ws-os · Aug 30, 2016 · fb497c4 · fb497c4
1 parent 902d12a
commit fb497c4
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/distributed_rate_limiter.go b/distributed_rate_limiter.go
@@ -7,13 +7,6 @@ import (
 	"time"
 )
 
-// TODO:
-/*
-
-2. Add rate check to init so that we have a load indication
-
-*/
-
 var DRLManager drl.DRL
 
 func SetupDRL() {

diff --git a/round_robin.go b/round_robin.go
@@ -7,6 +7,13 @@ type RoundRobin struct {
 }
 
 func (r *RoundRobin) SetMax(rp interface{}) {
+	switch rp.(type) {
+	case *[]string:
+		r.max = len(*rp.(*[]string)) - 1
+	case []string:
+		r.max = len(rp.([]string)) - 1
+	}
+
 	r.max = len(*rp.(*[]string)) - 1
 
 	if r.max < 0 {