make it configurable

application-research · Dec 13, 2022 · 905d6af · 905d6af
1 parent 30cea4c
commit 905d6af
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/config/estuary.go b/config/estuary.go
@@ -86,6 +86,7 @@ func NewEstuary(appVersion string) *Estuary {
 		DisableFilecoinStorage: false,
 		DisableSwaggerEndpoint: false,
 		DisableAutoRetrieve:    true,
+		RateLimit:              rate.Limit(20),
 
 		Deal: Deal{
 			IsDisabled:            false,

diff --git a/main.go b/main.go
@@ -6,6 +6,7 @@ import (
 	"errors"
 	"flag"
 	"fmt"
+	"golang.org/x/time/rate"
 	"os"
 	"path/filepath"
 	"strings"
@@ -223,6 +224,9 @@ func overrideSetOptions(flags []cli.Flag, cctx *cli.Context, cfg *config.Estuary
 			}
 			cfg.Deal.MaxVerifiedPrice = abi.TokenAmount(maxVerifiedPrice)
 
+		case "rate-limit":
+			cfg.RateLimit = rate.Limit(cctx.Float64("rate-limit"))
+
 		default:
 		}
 	}