Move code to pkg
This commit is contained in:
parent
bd4c822670
commit
f1b085fa36
465 changed files with 656 additions and 680 deletions
|
@ -1,44 +0,0 @@
|
|||
package tcp
|
||||
|
||||
import (
|
||||
"net"
|
||||
"sync"
|
||||
|
||||
"github.com/containous/traefik/log"
|
||||
)
|
||||
|
||||
// RRLoadBalancer is a naive RoundRobin load balancer for TCP services
|
||||
type RRLoadBalancer struct {
|
||||
servers []Handler
|
||||
lock sync.RWMutex
|
||||
current int
|
||||
}
|
||||
|
||||
// NewRRLoadBalancer creates a new RRLoadBalancer
|
||||
func NewRRLoadBalancer() *RRLoadBalancer {
|
||||
return &RRLoadBalancer{}
|
||||
}
|
||||
|
||||
// ServeTCP forwards the connection to the right service
|
||||
func (r *RRLoadBalancer) ServeTCP(conn net.Conn) {
|
||||
r.next().ServeTCP(conn)
|
||||
}
|
||||
|
||||
// AddServer appends a server to the existing list
|
||||
func (r *RRLoadBalancer) AddServer(server Handler) {
|
||||
r.servers = append(r.servers, server)
|
||||
}
|
||||
|
||||
func (r *RRLoadBalancer) next() Handler {
|
||||
r.lock.Lock()
|
||||
defer r.lock.Unlock()
|
||||
|
||||
// FIXME handle weight
|
||||
if r.current >= len(r.servers) {
|
||||
r.current = 0
|
||||
log.Debugf("Load balancer: going back to the first available server")
|
||||
}
|
||||
handler := r.servers[r.current]
|
||||
r.current++
|
||||
return handler
|
||||
}
|
Loading…
Add table
Add a link
Reference in a new issue