Skip to content

Commit 34921d3

Browse files
committed
1 parent 526712a commit 34921d3

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

cmd/main.go

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@ import (
2121
"github.com/labstack/echo/v4"
2222
"github.com/labstack/echo/v4/middleware"
2323
"github.com/lmittmann/tint"
24+
"golang.org/x/time/rate"
2425
)
2526

2627
type config struct {
@@ -83,6 +84,9 @@ func main() {
8384

8485
e.Use(middleware.Logger())
8586
e.Use(middleware.Recover())
87+
e.Use(middleware.RateLimiter(
88+
middleware.NewRateLimiterMemoryStore(rate.Limit(20)),
89+
))
8690
e.Use(middleware.TimeoutWithConfig(middleware.TimeoutConfig{
8791
Timeout: 5 * time.Second,
8892
ErrorMessage: "Request timed out.",

0 commit comments

Comments
 (0)