diff --git a/balancer.go b/balancer.go index 7a1cc54..473deef 100644 --- a/balancer.go +++ b/balancer.go @@ -13,18 +13,26 @@ type balancePenalty struct { func BestNodeByPenalties(nodes []*Node) (*Node, error) { penalties := make([]*balancePenalty, len(nodes)) - var playerPenalty, cpuPenalty int + var playerPenalty, cpuPenalty, deficitFramePenalty, nullFramePenalty int for i, node := range nodes { - if node.stats == nil { - penalties[i] = &balancePenalty{node, 1} - continue + playerPenalty = 0 + cpuPenalty = 0 + deficitFramePenalty = 0 + nullFramePenalty = 0 + + if node.stats != nil { + playerPenalty = node.stats.ActivePlayers + cpuPenalty = int(math.Pow(1.05, 100*node.stats.Cpu.SystemLoad)*10 - 10) + + if node.stats.Frames != nil && node.stats.Frames.Deficit != -1 { + deficitFramePenalty = int(math.Pow(1.03, 500*float64(node.stats.Frames.Deficit/3000))*600 - 600) + nullFramePenalty = int(math.Pow(1.03, 500*float64(node.stats.Frames.Nulled/3000))*300 - 300) + nullFramePenalty *= 2 + } } - playerPenalty = node.stats.ActivePlayers - cpuPenalty = int(math.Pow(1.05, 100*node.stats.Cpu.SystemLoad)*10 - 10) - - penalties[i] = &balancePenalty{node, playerPenalty + cpuPenalty} + penalties[i] = &balancePenalty{node, playerPenalty + cpuPenalty + deficitFramePenalty + nullFramePenalty} } sort.SliceStable(penalties, func(i, j int) bool { diff --git a/node.go b/node.go index 9a1ea1e..f296f72 100644 --- a/node.go +++ b/node.go @@ -42,6 +42,7 @@ type RemoteStats struct { Uptime int64 `json:"uptime"` Memory *MemoryStats `json:"memory"` Cpu *CpuStats `json:"cpu"` + Frames *FrameStats `json:"frameStats"` } type MemoryStats struct { @@ -57,6 +58,12 @@ type CpuStats struct { LavalinkLoad float64 `json:"lavalinkLoad"` } +type FrameStats struct { + Sent int `json:"sent"` + Nulled int `json:"nulled"` + Deficit int `json:"deficit"` +} + func (node *Node) open() error { header := http.Header{}