mirror of
https://github.com/ipfs/kubo.git
synced 2025-08-06 19:44:01 +08:00
feat: Provider.WorkerCount
and stats reprovide
(#10779)
* adjust ipfs stats provide * update boxo dep * bump boxo * fixing tests * docs/chore: mark stat reprovide as experimental * docs: Provider.Strategy explicitly document it is not used - without this legacy users will have it in their config and be very confused --------- Co-authored-by: Marcin Rataj <lidel@lidel.org>
This commit is contained in:
@ -184,6 +184,7 @@ func TestCommands(t *testing.T) {
|
||||
"/stats/bw",
|
||||
"/stats/dht",
|
||||
"/stats/provide",
|
||||
"/stats/reprovide",
|
||||
"/stats/repo",
|
||||
"/swarm",
|
||||
"/swarm/addrs",
|
||||
|
@ -27,11 +27,12 @@ for your IPFS node.`,
|
||||
},
|
||||
|
||||
Subcommands: map[string]*cmds.Command{
|
||||
"bw": statBwCmd,
|
||||
"repo": repoStatCmd,
|
||||
"bitswap": bitswapStatCmd,
|
||||
"dht": statDhtCmd,
|
||||
"provide": statProvideCmd,
|
||||
"bw": statBwCmd,
|
||||
"repo": repoStatCmd,
|
||||
"bitswap": bitswapStatCmd,
|
||||
"dht": statDhtCmd,
|
||||
"provide": statProvideCmd,
|
||||
"reprovide": statReprovideCmd,
|
||||
},
|
||||
}
|
||||
|
||||
|
@ -4,28 +4,20 @@ 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{
|
||||
Status: cmds.Deprecated,
|
||||
Helptext: cmds.HelpText{
|
||||
Tagline: "Returns statistics about the node's (re)provider system.",
|
||||
Tagline: "Deprecated command, use 'ipfs stats reprovide' instead.",
|
||||
ShortDescription: `
|
||||
Returns statistics about the content the node is advertising.
|
||||
|
||||
This interface is not stable and may change from release to release.
|
||||
'ipfs stats provide' is deprecated because provide and reprovide operations
|
||||
are now distinct. This command may be replaced by provide only stats in the
|
||||
future.
|
||||
`,
|
||||
},
|
||||
Arguments: []cmds.Argument{},
|
||||
@ -57,8 +49,8 @@ This interface is not stable and may change from release to release.
|
||||
wtr := tabwriter.NewWriter(w, 1, 2, 1, ' ', 0)
|
||||
defer wtr.Flush()
|
||||
|
||||
fmt.Fprintf(wtr, "TotalReprovides:\t%s\n", humanNumber(s.TotalReprovides))
|
||||
fmt.Fprintf(wtr, "AvgReprovideDuration:\t%s\n", humanDuration(s.AvgReprovideDuration))
|
||||
fmt.Fprintf(wtr, "TotalProvides:\t%s\n", humanNumber(s.TotalReprovides))
|
||||
fmt.Fprintf(wtr, "AvgProvideDuration:\t%s\n", humanDuration(s.AvgReprovideDuration))
|
||||
fmt.Fprintf(wtr, "LastReprovideDuration:\t%s\n", humanDuration(s.LastReprovideDuration))
|
||||
if !s.LastRun.IsZero() {
|
||||
fmt.Fprintf(wtr, "LastRun:\t%s\n", humanTime(s.LastRun))
|
||||
@ -71,30 +63,3 @@ This interface is not stable and may change from release to release.
|
||||
},
|
||||
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)
|
||||
}
|
||||
|
104
core/commands/stat_reprovide.go
Normal file
104
core/commands/stat_reprovide.go
Normal file
@ -0,0 +1,104 @@
|
||||
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 statReprovideCmd = &cmds.Command{
|
||||
Status: cmds.Experimental,
|
||||
Helptext: cmds.HelpText{
|
||||
Tagline: "Returns statistics about the node's reprovider system.",
|
||||
ShortDescription: `
|
||||
Returns statistics about the content the node is reproviding every
|
||||
Reprovider.Interval according to Reprovider.Strategy:
|
||||
https://github.com/ipfs/kubo/blob/master/docs/config.md#reprovider
|
||||
|
||||
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, "TotalReprovides:\t%s\n", humanNumber(s.TotalReprovides))
|
||||
fmt.Fprintf(wtr, "AvgReprovideDuration:\t%s\n", humanDuration(s.AvgReprovideDuration))
|
||||
fmt.Fprintf(wtr, "LastReprovideDuration:\t%s\n", humanDuration(s.LastReprovideDuration))
|
||||
if !s.LastRun.IsZero() {
|
||||
fmt.Fprintf(wtr, "LastReprovide:\t%s\n", humanTime(s.LastRun))
|
||||
if s.fullRT {
|
||||
fmt.Fprintf(wtr, "NextReprovide:\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)
|
||||
}
|
@ -359,6 +359,7 @@ func Online(bcfg *BuildCfg, cfg *config.Config, userResourceOverrides rcmgr.Part
|
||||
cfg.Reprovider.Strategy.WithDefault(config.DefaultReproviderStrategy),
|
||||
cfg.Reprovider.Interval.WithDefault(config.DefaultReproviderInterval),
|
||||
cfg.Routing.AcceleratedDHTClient.WithDefault(config.DefaultAcceleratedDHTClient),
|
||||
int(cfg.Provider.WorkerCount.WithDefault(config.DefaultProviderWorkerCount)),
|
||||
),
|
||||
)
|
||||
}
|
||||
|
@ -21,12 +21,13 @@ import (
|
||||
// and in 'ipfs stats provide' report.
|
||||
const sampledBatchSize = 1000
|
||||
|
||||
func ProviderSys(reprovideInterval time.Duration, acceleratedDHTClient bool) fx.Option {
|
||||
func ProviderSys(reprovideInterval time.Duration, acceleratedDHTClient bool, provideWorkerCount int) fx.Option {
|
||||
return fx.Provide(func(lc fx.Lifecycle, cr irouting.ProvideManyRouter, keyProvider provider.KeyChanFunc, repo repo.Repo, bs blockstore.Blockstore) (provider.System, error) {
|
||||
opts := []provider.Option{
|
||||
provider.Online(cr),
|
||||
provider.ReproviderInterval(reprovideInterval),
|
||||
provider.KeyProvider(keyProvider),
|
||||
provider.ProvideWorkerCount(provideWorkerCount),
|
||||
}
|
||||
if !acceleratedDHTClient && reprovideInterval > 0 {
|
||||
// The estimation kinda suck if you are running with accelerated DHT client,
|
||||
@ -131,7 +132,7 @@ https://github.com/ipfs/kubo/blob/master/docs/config.md#routingaccelerateddhtcli
|
||||
// ONLINE/OFFLINE
|
||||
|
||||
// OnlineProviders groups units managing provider routing records online
|
||||
func OnlineProviders(useStrategicProviding bool, reprovideStrategy string, reprovideInterval time.Duration, acceleratedDHTClient bool) fx.Option {
|
||||
func OnlineProviders(useStrategicProviding bool, reprovideStrategy string, reprovideInterval time.Duration, acceleratedDHTClient bool, provideWorkerCount int) fx.Option {
|
||||
if useStrategicProviding {
|
||||
return OfflineProviders()
|
||||
}
|
||||
@ -146,7 +147,7 @@ func OnlineProviders(useStrategicProviding bool, reprovideStrategy string, repro
|
||||
|
||||
return fx.Options(
|
||||
keyProvider,
|
||||
ProviderSys(reprovideInterval, acceleratedDHTClient),
|
||||
ProviderSys(reprovideInterval, acceleratedDHTClient, provideWorkerCount),
|
||||
)
|
||||
}
|
||||
|
||||
@ -169,7 +170,6 @@ func mfsProvider(mfsRoot *mfs.Root, fetcher fetcher.Factory) provider.KeyChanFun
|
||||
kcf := provider.NewDAGProvider(rootNode.Cid(), fetcher)
|
||||
return kcf(ctx)
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
func mfsRootProvider(mfsRoot *mfs.Root) provider.KeyChanFunc {
|
||||
|
Reference in New Issue
Block a user