Revert "internal/ethapi, les: use slices package for sorting (#27492)"

This reverts commit 051b1a1fd9.
This commit is contained in:
devopsbo3 2023-11-10 12:27:53 -06:00 committed by GitHub
parent 339796446e
commit 39aaddb4c8
3 changed files with 56 additions and 22 deletions

View file

@ -17,6 +17,7 @@
package ethapi package ethapi
import ( import (
"bytes"
"context" "context"
"crypto/ecdsa" "crypto/ecdsa"
"encoding/json" "encoding/json"
@ -24,6 +25,7 @@ import (
"hash" "hash"
"math/big" "math/big"
"reflect" "reflect"
"sort"
"testing" "testing"
"time" "time"
@ -46,7 +48,6 @@ import (
"github.com/ethereum/go-ethereum/rpc" "github.com/ethereum/go-ethereum/rpc"
"github.com/stretchr/testify/require" "github.com/stretchr/testify/require"
"golang.org/x/crypto/sha3" "golang.org/x/crypto/sha3"
"golang.org/x/exp/slices"
) )
func TestTransaction_RoundTripRpcJSON(t *testing.T) { func TestTransaction_RoundTripRpcJSON(t *testing.T) {
@ -648,13 +649,19 @@ type Account struct {
addr common.Address addr common.Address
} }
func newAccounts(n int) (accounts []Account) { type Accounts []Account
func (a Accounts) Len() int { return len(a) }
func (a Accounts) Swap(i, j int) { a[i], a[j] = a[j], a[i] }
func (a Accounts) Less(i, j int) bool { return bytes.Compare(a[i].addr.Bytes(), a[j].addr.Bytes()) < 0 }
func newAccounts(n int) (accounts Accounts) {
for i := 0; i < n; i++ { for i := 0; i < n; i++ {
key, _ := crypto.GenerateKey() key, _ := crypto.GenerateKey()
addr := crypto.PubkeyToAddress(key.PublicKey) addr := crypto.PubkeyToAddress(key.PublicKey)
accounts = append(accounts, Account{key: key, addr: addr}) accounts = append(accounts, Account{key: key, addr: addr})
} }
slices.SortFunc(accounts, func(a, b Account) bool { return a.addr.Less(b.addr) }) sort.Sort(accounts)
return accounts return accounts
} }

View file

@ -17,12 +17,12 @@
package les package les
import ( import (
"sort"
"sync" "sync"
"sync/atomic" "sync/atomic"
"github.com/ethereum/go-ethereum/common/mclock" "github.com/ethereum/go-ethereum/common/mclock"
"github.com/ethereum/go-ethereum/common/prque" "github.com/ethereum/go-ethereum/common/prque"
"golang.org/x/exp/slices"
) )
// servingQueue allows running tasks in a limited number of threads and puts the // servingQueue allows running tasks in a limited number of threads and puts the
@ -180,19 +180,35 @@ func (sq *servingQueue) threadController() {
} }
} }
// peerTasks lists the tasks received from a given peer when selecting peers to freeze type (
type peerTasks struct { // peerTasks lists the tasks received from a given peer when selecting peers to freeze
peerTasks struct {
peer *clientPeer peer *clientPeer
list []*servingTask list []*servingTask
sumTime uint64 sumTime uint64
priority float64 priority float64
}
// peerList is a sortable list of peerTasks
peerList []*peerTasks
)
func (l peerList) Len() int {
return len(l)
}
func (l peerList) Less(i, j int) bool {
return l[i].priority < l[j].priority
}
func (l peerList) Swap(i, j int) {
l[i], l[j] = l[j], l[i]
} }
// freezePeers selects the peers with the worst priority queued tasks and freezes // freezePeers selects the peers with the worst priority queued tasks and freezes
// them until burstTime goes under burstDropLimit or all peers are frozen // them until burstTime goes under burstDropLimit or all peers are frozen
func (sq *servingQueue) freezePeers() { func (sq *servingQueue) freezePeers() {
peerMap := make(map[*clientPeer]*peerTasks) peerMap := make(map[*clientPeer]*peerTasks)
var peerList []*peerTasks var peerList peerList
if sq.best != nil { if sq.best != nil {
sq.queue.Push(sq.best, sq.best.priority) sq.queue.Push(sq.best, sq.best.priority)
} }
@ -215,9 +231,7 @@ func (sq *servingQueue) freezePeers() {
tasks.list = append(tasks.list, task) tasks.list = append(tasks.list, task)
tasks.sumTime += task.expTime tasks.sumTime += task.expTime
} }
slices.SortFunc(peerList, func(a, b *peerTasks) bool { sort.Sort(peerList)
return a.priority < b.priority
})
drop := true drop := true
for _, tasks := range peerList { for _, tasks := range peerList {
if drop { if drop {

View file

@ -17,10 +17,10 @@
package utils package utils
import ( import (
"sort"
"sync" "sync"
"github.com/ethereum/go-ethereum/p2p/enode" "github.com/ethereum/go-ethereum/p2p/enode"
"golang.org/x/exp/slices"
) )
const maxSelectionWeight = 1000000000 // maximum selection weight of each individual node/address group const maxSelectionWeight = 1000000000 // maximum selection weight of each individual node/address group
@ -340,9 +340,24 @@ func (l *Limiter) Stop() {
l.cond.Signal() l.cond.Signal()
} }
type dropListItem struct { type (
dropList []dropListItem
dropListItem struct {
nq *nodeQueue nq *nodeQueue
priority float64 priority float64
}
)
func (l dropList) Len() int {
return len(l)
}
func (l dropList) Less(i, j int) bool {
return l[i].priority < l[j].priority
}
func (l dropList) Swap(i, j int) {
l[i], l[j] = l[j], l[i]
} }
// dropRequests selects the nodes with the highest queued request cost to selection // dropRequests selects the nodes with the highest queued request cost to selection
@ -351,7 +366,7 @@ type dropListItem struct {
func (l *Limiter) dropRequests() { func (l *Limiter) dropRequests() {
var ( var (
sumValue float64 sumValue float64
list []dropListItem list dropList
) )
for _, nq := range l.nodes { for _, nq := range l.nodes {
sumValue += nq.value sumValue += nq.value
@ -369,9 +384,7 @@ func (l *Limiter) dropRequests() {
priority: w / float64(nq.sumCost), priority: w / float64(nq.sumCost),
}) })
} }
slices.SortFunc(list, func(a, b dropListItem) bool { sort.Sort(list)
return a.priority < b.priority
})
for _, item := range list { for _, item := range list {
for _, request := range item.nq.queue { for _, request := range item.nq.queue {
close(request.process) close(request.process)