mirror of
https://github.com/ipfs/kubo.git
synced 2025-05-17 15:06:47 +08:00
b35555978d
* Add lastRun, NextRun, move reprovide cmd to routing. * acceleratedDHT logic * changelog * depend on latest boxo --------- Co-authored-by: guillaumemichel <guillaume@michel.id> Co-authored-by: Guillaume Michel <guillaumemichel@users.noreply.github.com>
101 lines
2.5 KiB
Go
101 lines
2.5 KiB
Go
package commands
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"text/tabwriter"
|
|
"time"
|
|
|
|
humanize "github.com/dustin/go-humanize"
|
|
"github.com/ipfs/boxo/provider"
|
|
cmds "github.com/ipfs/go-ipfs-cmds"
|
|
"github.com/ipfs/kubo/core/commands/cmdenv"
|
|
"github.com/libp2p/go-libp2p-kad-dht/fullrt"
|
|
"golang.org/x/exp/constraints"
|
|
)
|
|
|
|
type reprovideStats struct {
|
|
provider.ReproviderStats
|
|
fullRT bool
|
|
}
|
|
|
|
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
|
|
}
|
|
|
|
stats, err := nd.Provider.Stat()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
_, fullRT := nd.DHTClient.(*fullrt.FullRT)
|
|
|
|
if err := res.Emit(reprovideStats{stats, fullRT}); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
},
|
|
Encoders: cmds.EncoderMap{
|
|
cmds.Text: cmds.MakeTypedEncoder(func(req *cmds.Request, w io.Writer, s reprovideStats) 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))
|
|
if !s.LastRun.IsZero() {
|
|
fmt.Fprintf(wtr, "LastRun:\t%s\n", humanTime(s.LastRun))
|
|
if s.fullRT {
|
|
fmt.Fprintf(wtr, "NextRun:\t%s\n", humanTime(s.LastRun.Add(s.ReprovideInterval)))
|
|
}
|
|
}
|
|
return nil
|
|
}),
|
|
},
|
|
Type: reprovideStats{},
|
|
}
|
|
|
|
func humanDuration(val time.Duration) string {
|
|
return val.Truncate(time.Microsecond).String()
|
|
}
|
|
|
|
func humanTime(val time.Time) string {
|
|
return val.Format("2006-01-02 15:04:05")
|
|
}
|
|
|
|
func humanNumber[T constraints.Float | constraints.Integer](n T) 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)
|
|
}
|