mirror of
https://github.com/ipfs/kubo.git
synced 2026-02-21 10:27:46 +08:00
90 lines
2.2 KiB
Go
90 lines
2.2 KiB
Go
package commands
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"text/tabwriter"
|
|
"time"
|
|
|
|
humanize "github.com/dustin/go-humanize"
|
|
cmds "github.com/ipfs/go-ipfs-cmds"
|
|
"github.com/ipfs/kubo/core/commands/cmdenv"
|
|
|
|
"github.com/ipfs/boxo/provider/batched"
|
|
)
|
|
|
|
var statProvideCmd = &cmds.Command{
|
|
Helptext: cmds.HelpText{
|
|
Tagline: "Returns statistics about the node's (re)provider system.",
|
|
ShortDescription: `
|
|
Returns statistics about the content the node is advertising.
|
|
|
|
This interface is not stable and may change from release to release.
|
|
`,
|
|
},
|
|
Arguments: []cmds.Argument{},
|
|
Options: []cmds.Option{},
|
|
Run: func(req *cmds.Request, res cmds.ResponseEmitter, env cmds.Environment) error {
|
|
nd, err := cmdenv.GetNode(env)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if !nd.IsOnline {
|
|
return ErrNotOnline
|
|
}
|
|
|
|
sys, ok := nd.Provider.(*batched.BatchProvidingSystem)
|
|
if !ok {
|
|
return fmt.Errorf("can only return stats if Experimental.AcceleratedDHTClient is enabled")
|
|
}
|
|
|
|
stats, err := sys.Stat(req.Context)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if err := res.Emit(stats); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
},
|
|
Encoders: cmds.EncoderMap{
|
|
cmds.Text: cmds.MakeTypedEncoder(func(req *cmds.Request, w io.Writer, s *batched.BatchedProviderStats) error {
|
|
wtr := tabwriter.NewWriter(w, 1, 2, 1, ' ', 0)
|
|
defer wtr.Flush()
|
|
|
|
fmt.Fprintf(wtr, "TotalProvides:\t%s\n", humanNumber(s.TotalProvides))
|
|
fmt.Fprintf(wtr, "AvgProvideDuration:\t%s\n", humanDuration(s.AvgProvideDuration))
|
|
fmt.Fprintf(wtr, "LastReprovideDuration:\t%s\n", humanDuration(s.LastReprovideDuration))
|
|
fmt.Fprintf(wtr, "LastReprovideBatchSize:\t%s\n", humanNumber(s.LastReprovideBatchSize))
|
|
return nil
|
|
}),
|
|
},
|
|
Type: batched.BatchedProviderStats{},
|
|
}
|
|
|
|
func humanDuration(val time.Duration) string {
|
|
return val.Truncate(time.Microsecond).String()
|
|
}
|
|
|
|
func humanNumber(n int) string {
|
|
nf := float64(n)
|
|
str := humanSI(nf, 0)
|
|
fullStr := humanFull(nf, 0)
|
|
if str != fullStr {
|
|
return fmt.Sprintf("%s\t(%s)", str, fullStr)
|
|
}
|
|
return str
|
|
}
|
|
|
|
func humanSI(val float64, decimals int) string {
|
|
v, unit := humanize.ComputeSI(val)
|
|
return fmt.Sprintf("%s%s", humanFull(v, decimals), unit)
|
|
}
|
|
|
|
func humanFull(val float64, decimals int) string {
|
|
return humanize.CommafWithDigits(val, decimals)
|
|
}
|