mirror of
https://gitlab.com/pulsechaincom/erigon-pulse.git
synced 2025-01-06 02:52:19 +00:00
93 lines
2.2 KiB
Go
93 lines
2.2 KiB
Go
package reports
|
|
|
|
import (
|
|
"context"
|
|
"github.com/ledgerwatch/erigon/cmd/utils"
|
|
"github.com/spf13/cobra"
|
|
"github.com/urfave/cli"
|
|
)
|
|
|
|
type CommandFlags struct {
|
|
DataDir string
|
|
Chain string
|
|
ClientsLimit uint
|
|
MaxPingTries uint
|
|
Estimate bool
|
|
}
|
|
|
|
type Command struct {
|
|
command cobra.Command
|
|
flags CommandFlags
|
|
}
|
|
|
|
func NewCommand() *Command {
|
|
command := cobra.Command{
|
|
Use: "report",
|
|
Short: "P2P network crawler database report",
|
|
}
|
|
|
|
instance := Command{
|
|
command: command,
|
|
}
|
|
instance.withDatadir()
|
|
instance.withChain()
|
|
instance.withClientsLimit()
|
|
instance.withMaxPingTries()
|
|
instance.withEstimate()
|
|
|
|
return &instance
|
|
}
|
|
|
|
func (command *Command) withDatadir() {
|
|
flag := utils.DataDirFlag
|
|
command.command.Flags().StringVar(&command.flags.DataDir, flag.Name, flag.Value.String(), flag.Usage)
|
|
must(command.command.MarkFlagDirname(utils.DataDirFlag.Name))
|
|
}
|
|
|
|
func (command *Command) withChain() {
|
|
flag := utils.ChainFlag
|
|
command.command.Flags().StringVar(&command.flags.Chain, flag.Name, flag.Value, flag.Usage)
|
|
}
|
|
|
|
func (command *Command) withClientsLimit() {
|
|
flag := cli.UintFlag{
|
|
Name: "clients-limit",
|
|
Usage: "A number of top clients to show",
|
|
Value: uint(10),
|
|
}
|
|
command.command.Flags().UintVar(&command.flags.ClientsLimit, flag.Name, flag.Value, flag.Usage)
|
|
}
|
|
|
|
func (command *Command) withMaxPingTries() {
|
|
flag := cli.UintFlag{
|
|
Name: "max-ping-tries",
|
|
Usage: "A number of PING failures for a node to be considered dead",
|
|
Value: 3,
|
|
}
|
|
command.command.Flags().UintVar(&command.flags.MaxPingTries, flag.Name, flag.Value, flag.Usage)
|
|
}
|
|
|
|
func (command *Command) withEstimate() {
|
|
flag := cli.BoolFlag{
|
|
Name: "estimate",
|
|
Usage: "Estimate totals including nodes that replied with 'too many peers'",
|
|
}
|
|
command.command.Flags().BoolVar(&command.flags.Estimate, flag.Name, false, flag.Usage)
|
|
}
|
|
|
|
func (command *Command) RawCommand() *cobra.Command {
|
|
return &command.command
|
|
}
|
|
|
|
func (command *Command) OnRun(runFunc func(ctx context.Context, flags CommandFlags) error) {
|
|
command.command.RunE = func(cmd *cobra.Command, args []string) error {
|
|
return runFunc(cmd.Context(), command.flags)
|
|
}
|
|
}
|
|
|
|
func must(err error) {
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|