mirror of
https://github.com/luxfi/netrunner.git
synced 2026-07-27 00:04:23 +00:00
Zach Kelling
4e2a4e432e
- Update github.com/luxfi/keys v1.0.5 → v1.0.6 - Update github.com/luxfi/genesis v1.5.18 → v1.5.19 - Remove local replace directives for keys and genesis - Use published package versions for reproducible builds
75 lines
1.7 KiB
Go
75 lines
1.7 KiB
Go
// Copyright (C) 2021-2025, Lux Industries Inc. All rights reserved.
|
|
// SPDX-License-Identifier: BSD-3-Clause
|
|
|
|
package ping
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/luxfi/log"
|
|
"github.com/luxfi/log/level"
|
|
"github.com/luxfi/netrunner/client"
|
|
"github.com/luxfi/netrunner/utils/constants"
|
|
"github.com/luxfi/netrunner/ux"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
var (
|
|
logLevel string
|
|
endpoint string
|
|
dialTimeout time.Duration
|
|
requestTimeout time.Duration
|
|
)
|
|
|
|
func NewCommand() *cobra.Command {
|
|
cmd := &cobra.Command{
|
|
Use: "ping [options]",
|
|
Short: "Ping the server.",
|
|
RunE: pingFunc,
|
|
}
|
|
|
|
cmd.PersistentFlags().StringVar(&logLevel, "log-level", level.Info.String(), "log level")
|
|
cmd.PersistentFlags().StringVar(&endpoint, "endpoint", "0.0.0.0:9000", "server endpoint")
|
|
cmd.PersistentFlags().DurationVar(&dialTimeout, "dial-timeout", 10*time.Second, "server dial timeout")
|
|
cmd.PersistentFlags().DurationVar(&requestTimeout, "request-timeout", 10*time.Second, "client request timeout")
|
|
|
|
return cmd
|
|
}
|
|
|
|
func pingFunc(*cobra.Command, []string) error {
|
|
lvl, err := log.ToLevel(logLevel)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
lcfg := log.Config{
|
|
DisplayLevel: lvl,
|
|
LogLevel: level.Off,
|
|
}
|
|
logFactory := log.NewFactoryWithConfig(lcfg)
|
|
logger, err := logFactory.Make(constants.LogNameControl)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
cli, err := client.New(client.Config{
|
|
Endpoint: endpoint,
|
|
DialTimeout: dialTimeout,
|
|
}, logger)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer cli.Close()
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), requestTimeout)
|
|
resp, err := cli.Ping(ctx)
|
|
cancel()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
logString := "ping response: " + log.Green.Wrap("%s")
|
|
ux.Print(logger, logString, resp)
|
|
return nil
|
|
}
|