From 14230d7b5257da18c8e3756fe36a41764b808647 Mon Sep 17 00:00:00 2001 From: InfiniteTF Date: Tue, 23 Feb 2021 03:03:02 +0100 Subject: [PATCH] Enable keepalive for websocket connection (#1134) --- pkg/api/server.go | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/pkg/api/server.go b/pkg/api/server.go index 0fc201cc6..331e203f3 100644 --- a/pkg/api/server.go +++ b/pkg/api/server.go @@ -14,6 +14,7 @@ import ( "runtime/debug" "strconv" "strings" + "time" "github.com/99designs/gqlgen/handler" "github.com/go-chi/chi" @@ -135,12 +136,14 @@ func Start() { }, }) maxUploadSize := handler.UploadMaxSize(config.GetMaxUploadSize()) + websocketKeepAliveDuration := handler.WebsocketKeepAliveDuration(10 * time.Second) txnManager := manager.GetInstance().TxnManager resolver := &Resolver{ txnManager: txnManager, } - gqlHandler := handler.GraphQL(models.NewExecutableSchema(models.Config{Resolvers: resolver}), recoverFunc, websocketUpgrader, maxUploadSize) + + gqlHandler := handler.GraphQL(models.NewExecutableSchema(models.Config{Resolvers: resolver}), recoverFunc, websocketUpgrader, websocketKeepAliveDuration, maxUploadSize) r.Handle("/graphql", gqlHandler) r.Handle("/playground", handler.Playground("GraphQL playground", "/graphql"))