forks/go-ethereum
1
0
Fork 1
mirror of https://github.com/ethereum/go-ethereum.git synced 2026-08-13 07:23:46 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

feat: remove flags for hive test

Browse source
This commit is contained in:
fearlessfe 2024-03-17 22:33:02 +08:00 committed by Chen Kai
parent afeaa102b4
commit 0b6714dea0
3 changed files with 38 additions and 54 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

31
cmd/shisui/main.go
View file

@ -44,20 +44,12 @@ var (
utils.PortalDataCapacityFlag, utils.PortalDataCapacityFlag,
utils.PortalLogLevelFlag, utils.PortalLogLevelFlag,
} }
hiveTestFlags = []cli.Flag{
utils.HiveBootNodeFlag,
utils.HiveClientPrivateKeyFlag,
utils.HiveLogLevelFlag,
}
) )
func init() { func init() {
app.Action = shisui app.Action = shisui
app.Flags = flags.Merge(portalProtocolFlags, historyRpcFlags) app.Flags = flags.Merge(portalProtocolFlags, historyRpcFlags)
flags.AutoEnvVars(app.Flags, "SHISUI") flags.AutoEnvVars(app.Flags, "SHISUI")
app.Flags = flags.Merge(app.Flags, hiveTestFlags)
flags.AutoEnvVars(hiveTestFlags, "HIVE")
} }
func main() { func main() {
@ -140,16 +132,15 @@ func getPortalHistoryConfig(ctx *cli.Context) (*PortalHistoryConfig, error) {
config.Protocol.NodeIP = netIp config.Protocol.NodeIP = netIp
} }
if ctx.IsSet(utils.HiveLogLevelFlag.Name) { if ctx.IsSet(utils.PortalBootNodesFlag.Name) {
config.LogLevel = ctx.Int(utils.HiveLogLevelFlag.Name) for _, node := range ctx.StringSlice(utils.PortalBootNodesFlag.Name) {
} bootNode := new(enode.Node)
if ctx.IsSet(utils.HiveBootNodeFlag.Name) { err = bootNode.UnmarshalText([]byte(node))
bootNode := new(enode.Node) if err != nil {
err = bootNode.UnmarshalText([]byte(ctx.String(utils.HiveBootNodeFlag.Name))) return config, err
if err != nil { }
return config, err config.Protocol.BootstrapNodes = append(config.Protocol.BootstrapNodes, bootNode)
} }
config.Protocol.BootstrapNodes = append(config.Protocol.BootstrapNodes, bootNode)
} }
return config, nil return config, nil
} }
@ -157,9 +148,9 @@ func getPortalHistoryConfig(ctx *cli.Context) (*PortalHistoryConfig, error) {
func setPrivateKey(ctx *cli.Context, config *PortalHistoryConfig) error { func setPrivateKey(ctx *cli.Context, config *PortalHistoryConfig) error {
var privateKey *ecdsa.PrivateKey var privateKey *ecdsa.PrivateKey
var err error var err error
if ctx.IsSet(utils.HiveClientPrivateKeyFlag.Name) { if ctx.IsSet(utils.PortalPrivateKeyFlag.Name) {
keyStr := ctx.String(utils.HiveClientPrivateKeyFlag.Name) keyStr := ctx.String(utils.PortalPrivateKeyFlag.Name)
keyBytes, err := hexutil.Decode("0x" + keyStr) keyBytes, err := hexutil.Decode(keyStr)
if err != nil { if err != nil {
return err return err
} }

22
cmd/utils/flags.go
View file

@ -994,22 +994,16 @@ Please note that --` + MetricsHTTPFlag.Name + ` must be set to start the server.
Category: flags.PortalNetworkCategory, Category: flags.PortalNetworkCategory,
} }
HiveBootNodeFlag = &cli.StringFlag{ PortalPrivateKeyFlag = &cli.StringFlag{
Name: "private.key",
Usage: "private key of p2p node, hex format",
Category: flags.PortalNetworkCategory,
}
PortalBootNodesFlag = &cli.StringSliceFlag{
Name: "bootnode", Name: "bootnode",
Usage: "bootnode of p2p network with ENR format for portal hive test", Usage: "bootnode of p2p network with ENR format for portal hive test",
Category: flags.PortalNetworkHiveCategory, Category: flags.PortalNetworkCategory,
}
HiveClientPrivateKeyFlag = &cli.StringFlag{
Name: "client.private.key",
Usage: "private key of current p2p node for portal hive test",
Category: flags.PortalNetworkHiveCategory,
}
HiveLogLevelFlag = &cli.IntFlag{
Name: "loglevel",
Usage: "loglevel for portal hive test",
Category: flags.PortalNetworkHiveCategory,
} }
) )

39
internal/flags/categories.go
View file

@ -19,26 +19,25 @@ package flags
import "github.com/urfave/cli/v2" import "github.com/urfave/cli/v2"
const ( const (
EthCategory = "ETHEREUM" EthCategory = "ETHEREUM"
BeaconCategory = "BEACON CHAIN" BeaconCategory = "BEACON CHAIN"
DevCategory = "DEVELOPER CHAIN" DevCategory = "DEVELOPER CHAIN"
StateCategory = "STATE HISTORY MANAGEMENT" StateCategory = "STATE HISTORY MANAGEMENT"
TxPoolCategory = "TRANSACTION POOL (EVM)" TxPoolCategory = "TRANSACTION POOL (EVM)"
BlobPoolCategory = "TRANSACTION POOL (BLOB)" BlobPoolCategory = "TRANSACTION POOL (BLOB)"
PerfCategory = "PERFORMANCE TUNING" PerfCategory = "PERFORMANCE TUNING"
AccountCategory = "ACCOUNT" AccountCategory = "ACCOUNT"
APICategory = "API AND CONSOLE" APICategory = "API AND CONSOLE"
NetworkingCategory = "NETWORKING" NetworkingCategory = "NETWORKING"
MinerCategory = "MINER" MinerCategory = "MINER"
GasPriceCategory = "GAS PRICE ORACLE" GasPriceCategory = "GAS PRICE ORACLE"
VMCategory = "VIRTUAL MACHINE" VMCategory = "VIRTUAL MACHINE"
LoggingCategory = "LOGGING AND DEBUGGING" LoggingCategory = "LOGGING AND DEBUGGING"
MetricsCategory = "METRICS AND STATS" MetricsCategory = "METRICS AND STATS"
MiscCategory = "MISC" MiscCategory = "MISC"
TestingCategory = "TESTING" TestingCategory = "TESTING"
DeprecatedCategory = "ALIASED (deprecated)" DeprecatedCategory = "ALIASED (deprecated)"
PortalNetworkCategory = "PORTAL NETWORK" PortalNetworkCategory = "PORTAL NETWORK"
PortalNetworkHiveCategory = "PORTAL NETWORK FOR HIVE TEST"
) )
func init() { func init() {