increased discovery rate limiter (#46)

wetware · Jul 25, 2022 · 5f83dfc · 5f83dfc
1 parent 6b6c05e
commit 5f83dfc
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git a/internal/cmd/client/discover.go b/internal/cmd/client/discover.go
@@ -60,7 +60,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
@@ -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,7 @@ 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
@@ -28,6 +28,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"
@@ -211,7 +212,7 @@ 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 {