Update examples

async-graphql · Sep 24, 2022 · 5d29914 · 5d29914
1 parent b1b2e9a
commit 5d29914
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 16 deletions.
diff --git a/actix-web/token-from-header/src/main.rs b/actix-web/token-from-header/src/main.rs
@@ -1,7 +1,7 @@
 use actix_web::{
     guard, http::header::HeaderMap, web, App, HttpRequest, HttpResponse, HttpServer, Result,
 };
-use async_graphql::{http::GraphiQLSource, Data, EmptyMutation, Schema};
+use async_graphql::{http::GraphiQLSource, EmptyMutation, Schema};
 use async_graphql_actix_web::{GraphQLRequest, GraphQLResponse, GraphQLSubscription};
 use token::{on_connection_init, QueryRoot, SubscriptionRoot, Token, TokenSchema};
 
@@ -39,13 +39,7 @@ async fn index_ws(
     req: HttpRequest,
     payload: web::Payload,
 ) -> Result<HttpResponse> {
-    let mut data = Data::default();
-    if let Some(token) = get_token_from_headers(req.headers()) {
-        data.insert(token);
-    }
-
     GraphQLSubscription::new(Schema::clone(&*schema))
-        .with_data(data)
         .on_connection_init(on_connection_init)
         .start(&req, payload)
 }

diff --git a/axum/token-from-header/src/main.rs b/axum/token-from-header/src/main.rs
@@ -1,6 +1,6 @@
 use async_graphql::{
     http::{playground_source, GraphQLPlaygroundConfig, ALL_WEBSOCKET_PROTOCOLS},
-    Data, EmptyMutation, Schema,
+    EmptyMutation, Schema,
 };
 use async_graphql_axum::{GraphQLProtocol, GraphQLRequest, GraphQLResponse, GraphQLWebSocket};
 use axum::{
@@ -40,18 +40,11 @@ async fn graphql_ws_handler(
     Extension(schema): Extension<TokenSchema>,
     protocol: GraphQLProtocol,
     websocket: WebSocketUpgrade,
-    headers: HeaderMap,
 ) -> Response {
-    let mut data = Data::default();
-    if let Some(token) = get_token_from_headers(&headers) {
-        data.insert(token);
-    }
-
     websocket
         .protocols(ALL_WEBSOCKET_PROTOCOLS)
         .on_upgrade(move |stream| {
             GraphQLWebSocket::new(stream, schema.clone(), protocol)
-                .with_data(data)
                 .on_connection_init(on_connection_init)
                 .serve()
         })

diff --git a/poem/token-from-header/src/main.rs b/poem/token-from-header/src/main.rs
@@ -44,7 +44,6 @@ async fn index(
 #[handler]
 async fn ws(
     schema: Data<&TokenSchema>,
-    headers: &HeaderMap,
     protocol: GraphQLProtocol,
     websocket: WebSocket,
 ) -> impl IntoResponse {