1
  2
  3
  4
  5
  6
  7
  8
  9
 10
 11
 12
 13
 14
 15
 16
 17
 18
 19
 20
 21
 22
 23
 24
 25
 26
 27
 28
 29
 30
 31
 32
 33
 34
 35
 36
 37
 38
 39
 40
 41
 42
 43
 44
 45
 46
 47
 48
 49
 50
 51
 52
 53
 54
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139

use crate::{
    client_type_error, FromCell, GraphError, GraphResult, GraphString, Node, Relation, ResultSet,
    Scalar,
};

impl FromCell for Scalar {
    fn from_cell(result_set: &ResultSet, row_idx: usize, column_idx: usize) -> GraphResult<Self> {
        let scalar = result_set.get_scalar(row_idx, column_idx)?;
        Ok(scalar.clone())
    }
}

impl FromCell for () {
    fn from_cell(result_set: &ResultSet, row_idx: usize, column_idx: usize) -> GraphResult<Self> {
        let scalar = result_set.get_scalar(row_idx, column_idx)?;
        match scalar {
            Scalar::Nil => Ok(()),
            any => client_type_error!("failed to construct value: expected nil, found {:?}", any),
        }
    }
}

impl<T: FromCell> FromCell for Option<T> {
    fn from_cell(result_set: &ResultSet, row_idx: usize, column_idx: usize) -> GraphResult<Self> {
        let scalar = result_set.get_scalar(row_idx, column_idx)?;
        match scalar {
            Scalar::Nil => Ok(None),
            _ => T::from_cell(result_set, row_idx, column_idx).map(Some),
        }
    }
}

impl FromCell for bool {
    fn from_cell(result_set: &ResultSet, row_idx: usize, column_idx: usize) -> GraphResult<Self> {
        let scalar = result_set.get_scalar(row_idx, column_idx)?;
        match scalar {
            Scalar::Boolean(boolean) => Ok(*boolean),
            any => client_type_error!(
                "failed to construct value: expected boolean, found {:?}",
                any
            ),
        }
    }
}

// The following code and macros produce the requisite type "magic" to allow
// code in an actor to extract strongly-typed data from a result set in
// tuples (or vecs of tuples)

macro_rules! impl_from_scalar_for_integer {
    ($t:ty) => {
        impl FromCell for $t {
            fn from_cell(
                result_set: &ResultSet,
                row_idx: usize,
                column_idx: usize,
            ) -> GraphResult<Self> {
                let scalar = result_set.get_scalar(row_idx, column_idx)?;
                match scalar {
                    Scalar::Integer(int) => Ok(*int as $t),
                    any => client_type_error!(
                        "failed to construct value: expected integer, found {:?}",
                        any
                    ),
                }
            }
        }
    };
}

impl_from_scalar_for_integer!(u8);
impl_from_scalar_for_integer!(u16);
impl_from_scalar_for_integer!(u32);
impl_from_scalar_for_integer!(u64);
impl_from_scalar_for_integer!(usize);

impl_from_scalar_for_integer!(i8);
impl_from_scalar_for_integer!(i16);
impl_from_scalar_for_integer!(i32);
impl_from_scalar_for_integer!(i64);
impl_from_scalar_for_integer!(isize);

macro_rules! impl_from_scalar_for_float {
    ($t:ty) => {
        impl FromCell for $t {
            fn from_cell(
                result_set: &ResultSet,
                row_idx: usize,
                column_idx: usize,
            ) -> GraphResult<Self> {
                let scalar = result_set.get_scalar(row_idx, column_idx)?;
                match scalar {
                    Scalar::Double(double) => Ok(*double as $t),
                    any => client_type_error!(
                        "failed to construct value: expected double, found {:?}",
                        any
                    ),
                }
            }
        }
    };
}

impl_from_scalar_for_float!(f32);
impl_from_scalar_for_float!(f64);

impl FromCell for GraphString {
    fn from_cell(result_set: &ResultSet, row_idx: usize, column_idx: usize) -> GraphResult<Self> {
        let scalar = result_set.get_scalar(row_idx, column_idx)?;
        match scalar {
            Scalar::String(data) => Ok(data.clone()),
            any => client_type_error!(
                "failed to construct value: expected string, found {:?}",
                any
            ),
        }
    }
}

impl FromCell for String {
    fn from_cell(result_set: &ResultSet, row_idx: usize, column_idx: usize) -> GraphResult<Self> {
        let redis_string = GraphString::from_cell(result_set, row_idx, column_idx)?;
        String::from_utf8(redis_string.into()).map_err(|_| GraphError::InvalidUtf8)
    }
}

impl FromCell for Node {
    fn from_cell(result_set: &ResultSet, row_idx: usize, column_idx: usize) -> GraphResult<Self> {
        let node = result_set.get_node(row_idx, column_idx)?;
        Ok(node.clone())
    }
}

impl FromCell for Relation {
    fn from_cell(result_set: &ResultSet, row_idx: usize, column_idx: usize) -> GraphResult<Self> {
        let relation = result_set.get_relation(row_idx, column_idx)?;
        Ok(relation.clone())
    }
}