diff --git a/internal/cmd/client/discover.go b/internal/cmd/client/discover.go index 28054b2f..00f2820c 100644 --- a/internal/cmd/client/discover.go +++ b/internal/cmd/client/discover.go @@ -59,7 +59,7 @@ func discover(c *cli.Context) error { } discoverer, err := bootutil.DialString(h, c.String("discover"), - socket.WithLogger(logger), socket.WithRateLimiter(socket.NewPacketLimiter(500, 8))) + socket.WithLogger(logger), socket.WithRateLimiter(socket.NewPacketLimiter(1000, 8))) if err != nil { return err } diff --git a/internal/cmd/client/runtime.go b/internal/cmd/client/runtime.go index a468b0db..0a0b51da 100644 --- a/internal/cmd/client/runtime.go +++ b/internal/cmd/client/runtime.go @@ -12,6 +12,7 @@ import ( "github.com/lthibault/log" "github.com/urfave/cli/v2" "github.com/wetware/casm/pkg/boot" + "github.com/wetware/casm/pkg/boot/socket" bootutil "github.com/wetware/casm/pkg/boot/util" logutil "github.com/wetware/ww/internal/util/log" "github.com/wetware/ww/pkg/client" @@ -84,7 +85,8 @@ func dialer(c *cli.Context, h host.Host, lx fx.Lifecycle) (d client.Dialer, err return } - d.Boot, err = bootutil.DialString(h, c.String("discover")) + d.Boot, err = bootutil.DialString(h, c.String("discover"), + socket.WithRateLimiter(socket.NewPacketLimiter(1000, 8))) if err == nil { if b, ok := d.Boot.(io.Closer); ok { lx.Append(closer(b)) diff --git a/internal/runtime/network.go b/internal/runtime/network.go index 41cad0ae..6cf4edb9 100644 --- a/internal/runtime/network.go +++ b/internal/runtime/network.go @@ -26,6 +26,7 @@ import ( "go.uber.org/fx" "github.com/wetware/casm/pkg/boot" + "github.com/wetware/casm/pkg/boot/socket" bootutil "github.com/wetware/casm/pkg/boot/util" "github.com/wetware/casm/pkg/pex" protoutil "github.com/wetware/casm/pkg/util/proto" @@ -205,7 +206,8 @@ func bootstrap(config bootConfig) (b bootstrapper, err error) { if config.CLI.IsSet("addr") { b.Discovery, err = boot.NewStaticAddrStrings(config.CLI.StringSlice("addr")...) } else { - b.Discovery, err = bootutil.ListenString(config.Host(), config.CLI.String("discover")) + b.Discovery, err = bootutil.ListenString(config.Host(), config.CLI.String("discover"), + socket.WithRateLimiter(socket.NewPacketLimiter(1000, 8))) } if err == nil {