Add request limit and timeout to bfgd config

api/bfgapi/bfgapi.go

@@ -56,6 +56,8 @@ var (
 	DefaultPrometheusListen = "localhost:2112"
 	DefaultPrivateURL       = "ws://" + DefaultPrivateListen + "/" + RouteWebsocketPrivate
 	DefaultPublicURL        = "ws://" + DefaultPublicListen + "/" + RouteWebsocketPublic
+	DefaultRequestLimit     = 10000 // XXX this is a bandaid
+	DefaultRequestTimeout   = 9     // XXX PNOOMA
 )
 
 type AccessPublicKey struct {

cmd/bfgd/bfgd.go

@@ -87,6 +87,18 @@ var (
 			Help:         "address and port bfgd pprof listens on (open <address>/debug/pprof to see available profiles)",
 			Print:        config.PrintAll,
 		},
+		"BFG_REQUEST_LIMIT": config.Config{
+			Value:        &cfg.RequestLimit,
+			DefaultValue: bfgapi.DefaultRequestLimit,
+			Help:         "maximum request queue depth",
+			Print:        config.PrintAll,
+		},
+		"BFG_REQUEST_TIMEOUT": config.Config{
+			Value:        &cfg.RequestTimeout,
+			DefaultValue: bfgapi.DefaultRequestTimeout,
+			Help:         "request timeout in seconds",
+			Print:        config.PrintAll,
+		},
 	}
 )
 

e2e/e2e_ext_test.go

@@ -292,6 +292,8 @@ func createBfgServerWithAuth(ctx context.Context, t *testing.T, pgUri string, el
 		EXBTCAddress:         electrumxAddr,
 		BTCStartHeight:       btcStartHeight,
 		PublicKeyAuth:        auth,
+		RequestLimit:         bfgapi.DefaultRequestLimit,
+		RequestTimeout:       bfgapi.DefaultRequestTimeout,
 	})
 	if err != nil {
 		t.Fatal(err)

service/bfg/bfg.go

@@ -65,6 +65,8 @@ func NewDefaultConfig() *Config {
 		EXBTCAddress:         "localhost:18001",
 		PrivateListenAddress: ":8080",
 		PublicListenAddress:  ":8383",
+		RequestLimit:         bfgapi.DefaultRequestLimit,
+		RequestTimeout:       bfgapi.DefaultRequestTimeout,
 	}
 }
 
@@ -91,6 +93,8 @@ type Config struct {
 	PrometheusListenAddress string
 	PprofListenAddress      string
 	PublicKeyAuth           bool
+	RequestLimit            int
+	RequestTimeout          int // in seconds
 }
 
 type Server struct {
@@ -100,9 +104,8 @@ type Server struct {
 	cfg *Config
 
 	// requests
-	requestLimit   int       // Request limiter queue depth
 	requestLimiter chan bool // Maximum in progress websocket commands
-	requestTimeout time.Duration
+	// requestTimeout time.Duration
 
 	btcHeight uint64
 
@@ -187,21 +190,25 @@ func NewServer(cfg *Config) (*Server, error) {
 	if cfg == nil {
 		cfg = NewDefaultConfig()
 	}
-	defaultRequestTimeout := 9 * time.Second // XXX
-	requestLimit := 1000                     // XXX
+	if cfg.RequestLimit <= 0 {
+		return nil, fmt.Errorf("invalid request limit: %v", cfg.RequestLimit)
+	}
+	minRequestTimeout := 3
+	if cfg.RequestTimeout <= minRequestTimeout {
+		return nil, fmt.Errorf("invalid request timeout (minimum %v): %v",
+			minRequestTimeout, cfg.RequestTimeout)
+	}
 	s := &Server{
 		cfg:                   cfg,
-		requestLimiter:        make(chan bool, requestLimit),
-		requestLimit:          requestLimit,
-		requestTimeout:        defaultRequestTimeout,
+		requestLimiter:        make(chan bool, cfg.RequestLimit),
 		btcHeight:             cfg.BTCStartHeight,
 		server:                http.NewServeMux(),
 		publicServer:          http.NewServeMux(),
 		metrics:               newMetrics(),
 		sessions:              make(map[string]*bfgWs),
 		checkForInvalidBlocks: make(chan struct{}),
 	}
-	for range requestLimit {
+	for range cfg.RequestLimit {
 		s.requestLimiter <- true
 	}
 
@@ -251,7 +258,8 @@ func (s *Server) handleRequest(parentCtx context.Context, bws *bfgWs, wsid strin
 	log.Tracef("handleRequest: %v", bws.addr)
 	defer log.Tracef("handleRequest exit: %v", bws.addr)
 
-	ctx, cancel := context.WithTimeout(parentCtx, s.requestTimeout)
+	ctx, cancel := context.WithTimeout(parentCtx,
+		time.Duration(s.cfg.RequestTimeout)*time.Second)
 	defer cancel()
 
 	select {