forked from cerc-io/plugeth
be65b47645
The Go authors updated golang/x/ext to change the function signature of the slices sort method. It's an entire shitshow now because x/ext is not tagged, so everyone's codebase just picked a new version that some other dep depends on, causing our code to fail building. This PR updates the dep on our code too and does all the refactorings to follow upstream...
134 lines
3.4 KiB
Go
134 lines
3.4 KiB
Go
// Copyright 2019 The go-ethereum Authors
|
|
// This file is part of go-ethereum.
|
|
//
|
|
// go-ethereum is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
//
|
|
// go-ethereum is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU General Public License
|
|
// along with go-ethereum. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
package main
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"fmt"
|
|
"os"
|
|
"time"
|
|
|
|
"github.com/ethereum/go-ethereum/common"
|
|
"github.com/ethereum/go-ethereum/p2p/enode"
|
|
"golang.org/x/exp/slices"
|
|
)
|
|
|
|
const jsonIndent = " "
|
|
|
|
// nodeSet is the nodes.json file format. It holds a set of node records
|
|
// as a JSON object.
|
|
type nodeSet map[enode.ID]nodeJSON
|
|
|
|
type nodeJSON struct {
|
|
Seq uint64 `json:"seq"`
|
|
N *enode.Node `json:"record"`
|
|
|
|
// The score tracks how many liveness checks were performed. It is incremented by one
|
|
// every time the node passes a check, and halved every time it doesn't.
|
|
Score int `json:"score,omitempty"`
|
|
// These two track the time of last successful contact.
|
|
FirstResponse time.Time `json:"firstResponse,omitempty"`
|
|
LastResponse time.Time `json:"lastResponse,omitempty"`
|
|
// This one tracks the time of our last attempt to contact the node.
|
|
LastCheck time.Time `json:"lastCheck,omitempty"`
|
|
}
|
|
|
|
func loadNodesJSON(file string) nodeSet {
|
|
var nodes nodeSet
|
|
if err := common.LoadJSON(file, &nodes); err != nil {
|
|
exit(err)
|
|
}
|
|
return nodes
|
|
}
|
|
|
|
func writeNodesJSON(file string, nodes nodeSet) {
|
|
nodesJSON, err := json.MarshalIndent(nodes, "", jsonIndent)
|
|
if err != nil {
|
|
exit(err)
|
|
}
|
|
if file == "-" {
|
|
os.Stdout.Write(nodesJSON)
|
|
return
|
|
}
|
|
if err := os.WriteFile(file, nodesJSON, 0644); err != nil {
|
|
exit(err)
|
|
}
|
|
}
|
|
|
|
// nodes returns the node records contained in the set.
|
|
func (ns nodeSet) nodes() []*enode.Node {
|
|
result := make([]*enode.Node, 0, len(ns))
|
|
for _, n := range ns {
|
|
result = append(result, n.N)
|
|
}
|
|
// Sort by ID.
|
|
slices.SortFunc(result, func(a, b *enode.Node) int {
|
|
return bytes.Compare(a.ID().Bytes(), b.ID().Bytes())
|
|
})
|
|
return result
|
|
}
|
|
|
|
// add ensures the given nodes are present in the set.
|
|
func (ns nodeSet) add(nodes ...*enode.Node) {
|
|
for _, n := range nodes {
|
|
v := ns[n.ID()]
|
|
v.N = n
|
|
v.Seq = n.Seq()
|
|
ns[n.ID()] = v
|
|
}
|
|
}
|
|
|
|
// topN returns the top n nodes by score as a new set.
|
|
func (ns nodeSet) topN(n int) nodeSet {
|
|
if n >= len(ns) {
|
|
return ns
|
|
}
|
|
|
|
byscore := make([]nodeJSON, 0, len(ns))
|
|
for _, v := range ns {
|
|
byscore = append(byscore, v)
|
|
}
|
|
slices.SortFunc(byscore, func(a, b nodeJSON) int {
|
|
if a.Score > b.Score {
|
|
return -1
|
|
}
|
|
if a.Score < b.Score {
|
|
return 1
|
|
}
|
|
return 0
|
|
})
|
|
result := make(nodeSet, n)
|
|
for _, v := range byscore[:n] {
|
|
result[v.N.ID()] = v
|
|
}
|
|
return result
|
|
}
|
|
|
|
// verify performs integrity checks on the node set.
|
|
func (ns nodeSet) verify() error {
|
|
for id, n := range ns {
|
|
if n.N.ID() != id {
|
|
return fmt.Errorf("invalid node %v: ID does not match ID %v in record", id, n.N.ID())
|
|
}
|
|
if n.N.Seq() != n.Seq {
|
|
return fmt.Errorf("invalid node %v: 'seq' does not match seq %d from record", id, n.N.Seq())
|
|
}
|
|
}
|
|
return nil
|
|
}
|