Skip to content

Commit 7f2254e

Browse files
Thomas StrombergThomas Stromberg
authored andcommitted
simplify codebase
1 parent 24d6886 commit 7f2254e

File tree

6 files changed

+52
-269
lines changed

6 files changed

+52
-269
lines changed

cmd/server/main.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -91,7 +91,7 @@ func main() {
9191
go hub.Run(ctx)
9292

9393
// Create security components
94-
rateLimiter := security.NewRateLimiter(*rateLimit, time.Minute)
94+
rateLimiter := security.NewRateLimiter(*rateLimit)
9595
connLimiter := security.NewConnectionLimiter(*maxConnsPerIP, *maxConnsTotal)
9696

9797
mux := http.NewServeMux()

pkg/security/middleware.go

Lines changed: 0 additions & 134 deletions
This file was deleted.

pkg/security/middleware_test.go

Lines changed: 0 additions & 63 deletions
This file was deleted.

pkg/security/ratelimiter.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@ type bucket struct {
2525
}
2626

2727
// NewRateLimiter creates a new rate limiter.
28-
func NewRateLimiter(maxTokens int, _ time.Duration) *RateLimiter {
28+
func NewRateLimiter(maxTokens int) *RateLimiter {
2929
rl := &RateLimiter{
3030
buckets: make(map[string]*bucket),
3131
maxTokens: maxTokens,

pkg/security/security_test.go

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,11 +4,10 @@ import (
44
"net/http"
55
"net/http/httptest"
66
"testing"
7-
"time"
87
)
98

109
func TestRateLimiter(t *testing.T) {
11-
rl := NewRateLimiter(5, time.Second)
10+
rl := NewRateLimiter(5)
1211

1312
ip := "192.168.1.1"
1413

0 commit comments

Comments
 (0)