removed redundant mutex declaration

Author: iambenkay

middleware/rate_limiter.go

@@ -21,7 +21,7 @@ type (
 		// SourceFunc uses echo.Context to extract the identifier for a visitor
 		SourceFunc func(context echo.Context) string
 		// Store defines a store for the rate limiter
-		Store      TokenStore
+		Store TokenStore
 	}
 )
 
@@ -83,6 +83,10 @@ func (store *InMemoryStore) ShouldAllow(identifier string) bool {
 	store.mutex.Lock()
 	defer store.mutex.Unlock()
 
+	if store.visitors == nil {
+		store.visitors = make(map[string]*rate.Limiter)
+	}
+
 	limiter, exists := store.visitors[identifier]
 	if !exists {
 		limiter = rate.NewLimiter(store.rate, store.burst)

middleware/rate_limiter_test.go

@@ -3,18 +3,14 @@ package middleware
 import (
 	"github.com/labstack/echo/v4"
 	"github.com/stretchr/testify/assert"
-	"golang.org/x/time/rate"
 	"net/http"
 	"net/http/httptest"
-	"sync"
 	"testing"
 	"time"
 )
 
 func TestRateLimiter(t *testing.T) {
 	var inMemoryStore = new(InMemoryStore)
-	inMemoryStore.visitors = map[string]*rate.Limiter{}
-	inMemoryStore.mutex = sync.Mutex{}
 	inMemoryStore.rate = 1
 	inMemoryStore.burst = 3
 
@@ -57,8 +53,6 @@ func TestRateLimiter(t *testing.T) {
 
 func TestRateLimiterWithConfig(t *testing.T) {
 	var inMemoryStore = new(InMemoryStore)
-	inMemoryStore.visitors = map[string]*rate.Limiter{}
-	inMemoryStore.mutex = sync.Mutex{}
 	inMemoryStore.rate = 1
 	inMemoryStore.burst = 3
 
@@ -104,8 +98,6 @@ func TestRateLimiterWithConfig(t *testing.T) {
 
 func TestInMemoryStore_ShouldAllow(t *testing.T) {
 	var inMemoryStore = new(InMemoryStore)
-	inMemoryStore.visitors = map[string]*rate.Limiter{}
-	inMemoryStore.mutex = sync.Mutex{}
 	inMemoryStore.rate = 1
 	inMemoryStore.burst = 3