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

add cbor response support for actix integration

Browse Source
This commit is contained in:
Follpvosten 2022-01-15 21:00:15 +01:00
parent c6bb4c026c
commit 9f8720c5df
2 changed files with 16 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
integrations/actix-web/Cargo.toml
View File

@ -21,6 +21,7 @@ actix-web-actors = "4.0.0-beta.9"
async-channel = "1.6.1"
futures-util = { version = "0.3.0", default-features = false }
serde_json = "1.0.64"
serde_cbor = "0.11"
serde_urlencoded = "0.7.0"
futures-channel = "0.3.13"
thiserror = "1.0.30"

19
integrations/actix-web/src/request.rs
View File

@ -156,17 +156,28 @@ impl From<async_graphql::BatchResponse> for GraphQLResponse {
impl Responder for GraphQLResponse {
type Body = BoxBody;
fn respond_to(self, _req: &HttpRequest) -> HttpResponse {
fn respond_to(self, req: &HttpRequest) -> HttpResponse {
let mut res = HttpResponse::build(StatusCode::OK);
res.content_type("application/json");
if self.0.is_ok() {
if let Some(cache_control) = self.0.cache_control().value() {
res.append_header(("cache-control", cache_control));
res.append_header((http::header::CACHE_CONTROL, cache_control));
}
}
for (name, value) in self.0.http_headers() {
res.append_header((name, value));
}
res.body(serde_json::to_string(&self.0).unwrap())
let accept = req
.headers()
.get(http::header::ACCEPT)
.and_then(|val| val.to_str().ok());
// TODO: Error handling
// Neither of these branches should potentially panic.
if accept == Some("application/cbor") {
res.content_type("application/cbor");
res.body(serde_cbor::to_vec(&self.0).unwrap())
} else {
res.content_type("application/json");
res.body(serde_json::to_vec(&self.0).unwrap())
}
}
}