Set complexity limit

darleet · May 28, 2024 · 8e8d0b7 · 8e8d0b7
1 parent c19d817
commit 8e8d0b7
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/cmd/server/main.go b/cmd/server/main.go
@@ -3,6 +3,7 @@ package main
 import (
 	"context"
 	"github.com/99designs/gqlgen/graphql/handler"
+	"github.com/99designs/gqlgen/graphql/handler/extension"
 	"github.com/99designs/gqlgen/graphql/handler/transport"
 	"github.com/darleet/blog-graphql/internal/middleware/auth"
 	"github.com/darleet/blog-graphql/internal/middleware/loader"
@@ -58,6 +59,8 @@ func main() {
 
 	srv.AddTransport(&transport.Websocket{})
 
+	srv.Use(extension.FixedComplexityLimit(300))
+
 	authMW := auth.NewMiddleware()
 	logMW := logging.NewMiddleware(log)