anna/async-graphql
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #79 from vkill/async-graphql-tide-request-ext

Browse Source 
async-graphql-tide: support Extension Traits
This commit is contained in:
Sunli 2020-05-12 16:36:15 +08:00 committed by GitHub
commit 3b3608fee7
3 changed files with 106 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
async-graphql-tide/Cargo.toml
View File

@ -15,6 +15,7 @@ categories = ["network-programming", "asynchronous"]
[dependencies]
async-graphql = { path = "..", version = "1.11.3" }
tide = "0.8"
async-trait = "0.1.30"
[dev-dependencies]
async-std = "1.5.0"

122
async-graphql-tide/src/lib.rs
View File

@ -8,6 +8,7 @@ use async_graphql::http::GQLResponse;
use async_graphql::{
IntoQueryBuilder, IntoQueryBuilderOpts, ObjectType, QueryBuilder, Schema, SubscriptionType,
};
use async_trait::async_trait;
use tide::{http::headers, Request, Response, Status, StatusCode};
/// GraphQL request handler
@ -53,7 +54,7 @@ where
Mutation: ObjectType + Send + Sync + 'static,
Subscription: SubscriptionType + Send + Sync + 'static,
TideState: Send + Sync + 'static,
F: Fn(QueryBuilder) -> QueryBuilder,
F: Fn(QueryBuilder) -> QueryBuilder + Send,
{
graphql_opts(req, schema, query_builder_configuration, Default::default()).await
}
@ -70,24 +71,107 @@ where
Mutation: ObjectType + Send + Sync + 'static,
Subscription: SubscriptionType + Send + Sync + 'static,
TideState: Send + Sync + 'static,
F: Fn(QueryBuilder) -> QueryBuilder,
F: Fn(QueryBuilder) -> QueryBuilder + Send,
{
let content_type = req
.header(&headers::CONTENT_TYPE)
.and_then(|values| values.first().map(|value| value.to_string()));
let mut query_builder = (content_type, req)
.into_query_builder_opts(&opts)
req.graphql_opts(schema, query_builder_configuration, opts)
.await
.status(StatusCode::BadRequest)?;
query_builder = query_builder_configuration(query_builder);
let query_response = query_builder.execute(&schema).await;
let gql_response = GQLResponse(query_response);
let resp = Response::new(StatusCode::Ok).body_json(&gql_response)?;
Ok(resp)
}
/// Tide request extension
///
#[async_trait]
pub trait RequestExt<State>: Sized {
/// ```no_run
/// use async_graphql::*;
/// use async_std::task;
/// use tide::Request;
/// use async_graphql_tide::RequestExt as _;
///
/// struct QueryRoot;
/// #[Object]
/// impl QueryRoot {
/// #[field(desc = "Returns the sum of a and b")]
/// async fn add(&self, a: i32, b: i32) -> i32 {
/// a + b
/// }
/// }
///
/// fn main() -> std::result::Result<(), Box<dyn std::error::Error + Send + Sync>> {
/// task::block_on(async {
/// let mut app = tide::new();
/// app.at("/").post(|req: Request<()>| async move {
/// let schema = Schema::build(QueryRoot, EmptyMutation, EmptySubscription).finish();
/// req.graphql(schema, |query_builder| query_builder).await
/// });
/// app.listen("0.0.0.0:8000").await?;
///
/// Ok(())
/// })
/// }
/// ```
async fn graphql<Query, Mutation, Subscription, F>(
self,
schema: Schema<Query, Mutation, Subscription>,
query_builder_configuration: F,
) -> tide::Result<Response>
where
Query: ObjectType + Send + Sync + 'static,
Mutation: ObjectType + Send + Sync + 'static,
Subscription: SubscriptionType + Send + Sync + 'static,
State: Send + Sync + 'static,
F: Fn(QueryBuilder) -> QueryBuilder + Send,
{
self.graphql_opts(schema, query_builder_configuration, Default::default())
.await
}
/// Similar to graphql, but you can set the options `IntoQueryBuilderOpts`.
async fn graphql_opts<Query, Mutation, Subscription, F>(
self,
schema: Schema<Query, Mutation, Subscription>,
query_builder_configuration: F,
opts: IntoQueryBuilderOpts,
) -> tide::Result<Response>
where
Query: ObjectType + Send + Sync + 'static,
Mutation: ObjectType + Send + Sync + 'static,
Subscription: SubscriptionType + Send + Sync + 'static,
State: Send + Sync + 'static,
F: Fn(QueryBuilder) -> QueryBuilder + Send;
}
#[async_trait]
impl<State> RequestExt<State> for Request<State> {
async fn graphql_opts<Query, Mutation, Subscription, F>(
self,
schema: Schema<Query, Mutation, Subscription>,
query_builder_configuration: F,
opts: IntoQueryBuilderOpts,
) -> tide::Result<Response>
where
Query: ObjectType + Send + Sync + 'static,
Mutation: ObjectType + Send + Sync + 'static,
Subscription: SubscriptionType + Send + Sync + 'static,
State: Send + Sync + 'static,
F: Fn(QueryBuilder) -> QueryBuilder + Send,
{
let content_type = self
.header(&headers::CONTENT_TYPE)
.and_then(|values| values.first().map(|value| value.to_string()));
let mut query_builder = (content_type, self)
.into_query_builder_opts(&opts)
.await
.status(StatusCode::BadRequest)?;
query_builder = query_builder_configuration(query_builder);
let query_response = query_builder.execute(&schema).await;
let gql_response = GQLResponse(query_response);
let resp = Response::new(StatusCode::Ok).body_json(&gql_response)?;
Ok(resp)
}
}

3
async-graphql-tide/tests/graphql.rs
View File

@ -6,6 +6,7 @@ use std::time::Duration;
use tide::Request;
use async_graphql::*;
use async_graphql_tide::RequestExt as _;
type Result<T> = std::result::Result<T, Box<dyn std::error::Error + Send + Sync>>;
@ -27,7 +28,7 @@ fn quickstart() -> Result<()> {
let mut app = tide::new();
app.at("/").post(|req: Request<()>| async move {
let schema = Schema::build(QueryRoot, EmptyMutation, EmptySubscription).finish();
async_graphql_tide::graphql(req, schema, |query_builder| query_builder).await
req.graphql(schema, |query_builder| query_builder).await
});
app.listen(&listen_addr).await?;