configuration for request throttling

navidrome · Apr 20, 2020 · 716f4c5 · 716f4c5
1 parent ba2d4b6
commit 716f4c5
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/consts/consts.go b/consts/consts.go
@@ -26,6 +26,9 @@ const (
 
 	URLPathUI          = "/app"
 	URLPathSubsonicAPI = "/rest"
+
+	RequestThrottleBacklogLimit   = 100
+	RequestThrottleBacklogTimeout = time.Minute
 )
 
 // Cache options

diff --git a/server/server.go b/server/server.go
@@ -1,10 +1,12 @@
 package server
 
 import (
+	"math"
 	"net/http"
 	"os"
 	"path"
 	"path/filepath"
+	"runtime"
 	"time"
 
 	"github.com/deluan/navidrome/conf"
@@ -62,6 +64,10 @@ func (a *Server) initRoutes() {
 	r.Use(middleware.Heartbeat("/ping"))
 	r.Use(InjectLogger)
 
+	// configure request throttling
+	maxRequests := math.Max(2, float64(runtime.NumCPU()))
+	r.Use(middleware.ThrottleBacklog(int(maxRequests), consts.RequestThrottleBacklogLimit, consts.RequestThrottleBacklogTimeout))
+
 	indexHtml := path.Join(conf.Server.BaseURL, consts.URLPathUI, "index.html")
 	r.Get("/*", func(w http.ResponseWriter, r *http.Request) {
 		http.Redirect(w, r, indexHtml, 302)