mirror of
https://github.com/yusing/godoxy.git
synced 2026-03-19 07:43:54 +01:00
27 lines
642 B
Go
27 lines
642 B
Go
package loadbalancer
|
|
|
|
import (
|
|
"net/http"
|
|
"sync/atomic"
|
|
|
|
"github.com/yusing/godoxy/internal/types"
|
|
)
|
|
|
|
type roundRobin struct {
|
|
index atomic.Uint32
|
|
}
|
|
|
|
var _ impl = (*roundRobin)(nil)
|
|
|
|
func (*LoadBalancer) newRoundRobin() impl { return &roundRobin{} }
|
|
func (lb *roundRobin) OnAddServer(srv types.LoadBalancerServer) {}
|
|
func (lb *roundRobin) OnRemoveServer(srv types.LoadBalancerServer) {}
|
|
|
|
func (lb *roundRobin) ChooseServer(srvs types.LoadBalancerServers, r *http.Request) types.LoadBalancerServer {
|
|
if len(srvs) == 0 {
|
|
return nil
|
|
}
|
|
index := (lb.index.Add(1) - 1) % uint32(len(srvs))
|
|
return srvs[index]
|
|
}
|