nash_protocol/protocol/multi_request/
response.rs

1use 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}