Merge pull request #28 from tstromberg/main

simplify codebase

Author: tstromberg

cmd/server/main.go

@@ -91,7 +91,7 @@ func main() {
 	go hub.Run(ctx)
 
 	// Create security components
-	rateLimiter := security.NewRateLimiter(*rateLimit, time.Minute)
+	rateLimiter := security.NewRateLimiter(*rateLimit)
 	connLimiter := security.NewConnectionLimiter(*maxConnsPerIP, *maxConnsTotal)
 
 	mux := http.NewServeMux()

pkg/security/middleware.go

@@ -25,7 +25,7 @@ type bucket struct {
 }
 
 // NewRateLimiter creates a new rate limiter.
-func NewRateLimiter(maxTokens int, _ time.Duration) *RateLimiter {
+func NewRateLimiter(maxTokens int) *RateLimiter {
 	rl := &RateLimiter{
 		buckets:    make(map[string]*bucket),
 		maxTokens:  maxTokens,

pkg/security/middleware_test.go

@@ -4,11 +4,10 @@ import (
 	"net/http"
 	"net/http/httptest"
 	"testing"
-	"time"
 )
 
 func TestRateLimiter(t *testing.T) {
-	rl := NewRateLimiter(5, time.Second)
+	rl := NewRateLimiter(5)
 
 	ip := "192.168.1.1"