Update admin socket response sorting

This commit is contained in:
Neil Alexander 2024-09-23 22:40:52 +01:00
parent e138fa679c
commit b8ab843a98
No known key found for this signature in database
GPG key ID: A02A2019A2BB0944
4 changed files with 30 additions and 20 deletions

View file

@ -3,7 +3,8 @@ package admin
import (
"encoding/hex"
"net"
"sort"
"slices"
"strings"
"time"
"github.com/yggdrasil-network/yggdrasil-go/src/address"
@ -61,17 +62,26 @@ func (a *AdminSocket) getPeersHandler(_ *GetPeersRequest, res *GetPeersResponse)
}
res.Peers = append(res.Peers, peer)
}
sort.Slice(res.Peers, func(i, j int) bool {
if res.Peers[i].Inbound == res.Peers[j].Inbound {
if res.Peers[i].PublicKey == res.Peers[j].PublicKey {
if res.Peers[i].Priority == res.Peers[j].Priority {
return res.Peers[i].Uptime > res.Peers[j].Uptime
}
return res.Peers[i].Priority < res.Peers[j].Priority
}
return res.Peers[i].PublicKey < res.Peers[j].PublicKey
slices.SortStableFunc(res.Peers, func(a, b PeerEntry) int {
if !a.Inbound && b.Inbound {
return -1
}
return !res.Peers[i].Inbound && res.Peers[j].Inbound
if a.Inbound && !b.Inbound {
return 1
}
if d := strings.Compare(a.PublicKey, b.PublicKey); d != 0 {
return d
}
if d := a.Priority - b.Priority; d != 0 {
return int(d)
}
if d := a.Cost - b.Cost; d != 0 {
return int(d)
}
if d := a.Uptime - b.Uptime; d != 0 {
return int(d)
}
return 0
})
return nil
}