Add subscription support for tide with tide-websockets.
This commit is contained in:
parent
b54421ad6c
commit
244aa37738
2
examples
2
examples
|
@ -1 +1 @@
|
||||||
Subproject commit 9034777541c542d1f53d5b8adcc988b8e53a6ed8
|
Subproject commit b36b5c44543b7323cb199ed229ea247e83b85d18
|
|
@ -13,8 +13,12 @@ categories = ["network-programming", "asynchronous"]
|
||||||
|
|
||||||
[dependencies]
|
[dependencies]
|
||||||
async-graphql = { path = "../..", version = "=2.5.1" }
|
async-graphql = { path = "../..", version = "=2.5.1" }
|
||||||
|
async-trait = "0.1.42"
|
||||||
|
futures-util = "0.3.12"
|
||||||
|
serde_json = "1.0.62"
|
||||||
|
|
||||||
tide = { version = "0.16.0", default-features = false, features = ["h1-server"] }
|
tide = { version = "0.16.0", default-features = false, features = ["h1-server"] }
|
||||||
|
tide-websockets = "0.2.0"
|
||||||
|
|
||||||
[dev-dependencies]
|
[dev-dependencies]
|
||||||
# Surf lacks multipart support
|
# Surf lacks multipart support
|
||||||
|
|
|
@ -11,6 +11,8 @@
|
||||||
#![allow(clippy::needless_doctest_main)]
|
#![allow(clippy::needless_doctest_main)]
|
||||||
#![forbid(unsafe_code)]
|
#![forbid(unsafe_code)]
|
||||||
|
|
||||||
|
mod subscription;
|
||||||
|
|
||||||
use async_graphql::http::MultipartOptions;
|
use async_graphql::http::MultipartOptions;
|
||||||
use async_graphql::{ObjectType, ParseRequestError, Schema, SubscriptionType};
|
use async_graphql::{ObjectType, ParseRequestError, Schema, SubscriptionType};
|
||||||
use tide::utils::async_trait;
|
use tide::utils::async_trait;
|
||||||
|
@ -22,6 +24,8 @@ use tide::{
|
||||||
Body, Request, Response, StatusCode,
|
Body, Request, Response, StatusCode,
|
||||||
};
|
};
|
||||||
|
|
||||||
|
pub use subscription::Subscription;
|
||||||
|
|
||||||
/// Create a new GraphQL endpoint with the schema.
|
/// Create a new GraphQL endpoint with the schema.
|
||||||
///
|
///
|
||||||
/// Default multipart options are used and batch operations are supported.
|
/// Default multipart options are used and batch operations are supported.
|
||||||
|
|
99
integrations/tide/src/subscription.rs
Normal file
99
integrations/tide/src/subscription.rs
Normal file
|
@ -0,0 +1,99 @@
|
||||||
|
use std::future::Future;
|
||||||
|
use std::pin::Pin;
|
||||||
|
use std::str::FromStr;
|
||||||
|
|
||||||
|
use async_graphql::http::{WebSocket as AGWebSocket, WebSocketProtocols};
|
||||||
|
use async_graphql::{Data, ObjectType, Result, Schema, SubscriptionType};
|
||||||
|
use futures_util::{future, StreamExt};
|
||||||
|
use tide::{Endpoint, Request, Response};
|
||||||
|
use tide_websockets::Message;
|
||||||
|
|
||||||
|
/// GraphQL subscription endpoint.
|
||||||
|
pub struct Subscription<S> {
|
||||||
|
inner: Pin<Box<dyn Endpoint<S>>>,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[async_trait::async_trait]
|
||||||
|
impl<S> Endpoint<S> for Subscription<S>
|
||||||
|
where
|
||||||
|
S: Send + Sync + Clone + 'static,
|
||||||
|
{
|
||||||
|
async fn call(&self, req: Request<S>) -> tide::Result<Response> {
|
||||||
|
self.inner.call(req).await
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<S> Subscription<S>
|
||||||
|
where
|
||||||
|
S: Send + Sync + Clone + 'static,
|
||||||
|
{
|
||||||
|
/// Create a graphql subscription endpoint.
|
||||||
|
pub fn new<Query, Mutation, Subscription>(schema: Schema<Query, Mutation, Subscription>) -> Self
|
||||||
|
where
|
||||||
|
Query: ObjectType + 'static,
|
||||||
|
Mutation: ObjectType + 'static,
|
||||||
|
Subscription: SubscriptionType + 'static,
|
||||||
|
{
|
||||||
|
Self::new_with_initializer(schema, |_| {
|
||||||
|
futures_util::future::ready(Ok(Default::default()))
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Create a graphql subscription endpoint.
|
||||||
|
///
|
||||||
|
/// Specifies that a function converts the init payload to data.
|
||||||
|
pub fn new_with_initializer<Query, Mutation, Subscription, F, R>(
|
||||||
|
schema: Schema<Query, Mutation, Subscription>,
|
||||||
|
initializer: F,
|
||||||
|
) -> Self
|
||||||
|
where
|
||||||
|
Query: ObjectType + 'static,
|
||||||
|
Mutation: ObjectType + 'static,
|
||||||
|
Subscription: SubscriptionType + 'static,
|
||||||
|
F: FnOnce(serde_json::Value) -> R + Unpin + Send + Sync + Clone + 'static,
|
||||||
|
R: Future<Output = Result<Data>> + Send + 'static,
|
||||||
|
{
|
||||||
|
let endpoint = tide_websockets::WebSocket::<S, _>::new(move |request, connection| {
|
||||||
|
let schema = schema.clone();
|
||||||
|
let initializer = initializer.clone();
|
||||||
|
async move {
|
||||||
|
let protocol = match request
|
||||||
|
.header("sec-websocket-protocol")
|
||||||
|
.map(|value| value.as_str())
|
||||||
|
.and_then(|protocols| {
|
||||||
|
protocols
|
||||||
|
.split(',')
|
||||||
|
.find_map(|p| WebSocketProtocols::from_str(p.trim()).ok())
|
||||||
|
}) {
|
||||||
|
Some(protocol) => protocol,
|
||||||
|
None => {
|
||||||
|
// default to the prior standard
|
||||||
|
WebSocketProtocols::SubscriptionsTransportWS
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
let sink = connection.clone();
|
||||||
|
let mut stream = AGWebSocket::with_data(
|
||||||
|
schema.clone(),
|
||||||
|
connection
|
||||||
|
.take_while(|msg| future::ready(msg.is_ok()))
|
||||||
|
.map(Result::unwrap)
|
||||||
|
.map(Message::into_data),
|
||||||
|
initializer,
|
||||||
|
protocol,
|
||||||
|
);
|
||||||
|
while let Some(data) = stream.next().await {
|
||||||
|
if sink.send_string(data).await.is_err() {
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
})
|
||||||
|
.with_protocols(&["graphql-transport-ws", "graphql-ws"]);
|
||||||
|
Self {
|
||||||
|
inner: Box::pin(endpoint),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in New Issue
Block a user