nash_protocol/protocol/multi_request/
response.rs1use std::marker::PhantomData;
2use std::fmt;
3use serde::de::{self, MapAccess, Visitor};
4use serde_json::{Value, Map};
5use serde::Deserialize;
6use crate::protocol::multi_request::MultiResponse;
7use crate::errors::Result;
8
9pub struct ResponseData<T> {
10 pub responses: Vec<Result<T>>
11}
12
13impl<T> From<Vec<T>> for ResponseData<T> {
14 fn from(from: Vec<T>) -> Self {
15 Self { responses: from.into_iter().map(|x| Ok(x)).collect() }
16 }
17}
18
19impl<T, U: From<T>> From<ResponseData<T>> for MultiResponse<U> {
20 fn from(response: ResponseData<T>) -> Self {
21 let responses = response
22 .responses
23 .into_iter()
24 .map(|r|
25 r.map(|x|
26 x.into()
27 )
28 ).collect();
29 Self { responses }
30 }
31}
32
33impl<T> From<Vec<Result<T>>> for ResponseData<T> {
34 fn from(responses: Vec<Result<T>>) -> Self {
35 Self { responses }
36 }
37}
38
39pub struct ResponseVisitor<T, U> {
40 pub field: String,
41 response_data: PhantomData<T>,
42 graphql_response_data: PhantomData<U>
43}
44
45impl<T, U> ResponseVisitor<T, U> {
46 pub fn new(field: String) -> Self {
47 Self {
48 field,
49 response_data: PhantomData,
50 graphql_response_data: PhantomData
51 }
52 }
53}
54
55impl<'de, U: for <'a> Deserialize<'a>, T: From<Vec<U>>> Visitor<'de> for ResponseVisitor<T, U> {
56 type Value = T;
57
58 fn expecting(&self, formatter: &mut fmt::Formatter) -> fmt::Result {
59 formatter.write_str("Expected multiple responses")
60 }
61
62 fn visit_map<A>(self, mut map: A) -> std::result::Result<Self::Value, A::Error>
63 where A: MapAccess<'de> {
64 let mut responses = Vec::new();
65 while let Some((_, value)) = map.next_entry::<String, Value>()? {
66 let mut object = Map::new();
67 object.insert(self.field.clone(), value);
68 let response = serde_json::from_value::<U>(Value::Object(object)).map_err(|e| {
69 de::Error::custom(format!("Couldn't parse: {:#?}", e))
70 })?;
71 responses.push(response);
72 }
73 Ok(T::from(responses))
74 }
75}