Update cors.md to allow CORS for websockets

99designs · Aug 5, 2019 · ab228f1 · ab228f1
1 parent 373359d
commit ab228f1
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/docs/content/recipes/cors.md b/docs/content/recipes/cors.md
@@ -34,9 +34,18 @@ func main() {
 		Debug:            true,
 	}).Handler)
 
+	upgrader := websocket.Upgrader{
+		CheckOrigin: func(r *http.Request) bool {
+			// Check against your desired domains here
+			return true
+		},
+		ReadBufferSize:  1024,
+		WriteBufferSize: 1024,
+	}
+
 	router.Handle("/", handler.Playground("Starwars", "/query"))
 	router.Handle("/query",
-		handler.GraphQL(starwars.NewExecutableSchema(starwars.NewResolver())),
+		handler.GraphQL(starwars.NewExecutableSchema(starwars.NewResolver()), handler.WebsocketUpgrader(upgrader)),
 	)
 
 	err := http.ListenAndServe(":8080", router)