diff options
author | Pierre de La Morinerie <kemenaran@gmail.com> | 2018-02-06 10:57:34 +0530 |
---|---|---|
committer | Chris <ccbrown112@gmail.com> | 2018-02-05 23:27:34 -0600 |
commit | 034dbc07e3068c482e654b6a1a8fcbe4b01c44f3 (patch) | |
tree | 338bac5ef5795901f1c04718c0cc9b3330bdaa52 /app/ratelimit.go | |
parent | 323d717a402c4407ca14c8b94b5eebf861ccf44a (diff) | |
download | chat-034dbc07e3068c482e654b6a1a8fcbe4b01c44f3.tar.gz chat-034dbc07e3068c482e654b6a1a8fcbe4b01c44f3.tar.bz2 chat-034dbc07e3068c482e654b6a1a8fcbe4b01c44f3.zip |
handle RateLimiter initialization errors (#8199)
Previously, an error occuring in NewRateLimiter would return a nil
reference – which would be de-referenced just after, making the server
crash.
Diffstat (limited to 'app/ratelimit.go')
-rw-r--r-- | app/ratelimit.go | 11 |
1 files changed, 5 insertions, 6 deletions
diff --git a/app/ratelimit.go b/app/ratelimit.go index 460088598..13508f36f 100644 --- a/app/ratelimit.go +++ b/app/ratelimit.go @@ -12,6 +12,7 @@ import ( l4g "github.com/alecthomas/log4go" "github.com/mattermost/mattermost-server/model" "github.com/mattermost/mattermost-server/utils" + "github.com/pkg/errors" throttled "gopkg.in/throttled/throttled.v2" "gopkg.in/throttled/throttled.v2/store/memstore" ) @@ -23,11 +24,10 @@ type RateLimiter struct { header string } -func NewRateLimiter(settings *model.RateLimitSettings) *RateLimiter { +func NewRateLimiter(settings *model.RateLimitSettings) (*RateLimiter, error) { store, err := memstore.New(*settings.MemoryStoreSize) if err != nil { - l4g.Critical(utils.T("api.server.start_server.rate_limiting_memory_store")) - return nil + return nil, errors.Wrap(err, utils.T("api.server.start_server.rate_limiting_memory_store")) } quota := throttled.RateQuota{ @@ -37,8 +37,7 @@ func NewRateLimiter(settings *model.RateLimitSettings) *RateLimiter { throttledRateLimiter, err := throttled.NewGCRARateLimiter(store, quota) if err != nil { - l4g.Critical(utils.T("api.server.start_server.rate_limiting_rate_limiter")) - return nil + return nil, errors.Wrap(err, utils.T("api.server.start_server.rate_limiting_rate_limiter")) } return &RateLimiter{ @@ -46,7 +45,7 @@ func NewRateLimiter(settings *model.RateLimitSettings) *RateLimiter { useAuth: *settings.VaryByUser, useIP: *settings.VaryByRemoteAddr, header: settings.VaryByHeader, - } + }, nil } func (rl *RateLimiter) GenerateKey(r *http.Request) string { |