gitql_ast/types/
variant.rs

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
use std::any::Any;

use super::base::DataType;

#[derive(Clone)]
pub struct VariantType {
    pub variants: Vec<Box<dyn DataType>>,
}

impl VariantType {
    pub fn new(variants: Vec<Box<dyn DataType>>) -> Self {
        VariantType { variants }
    }
}

impl DataType for VariantType {
    fn literal(&self) -> String {
        let mut str = String::new();
        let last_position = self.variants.len() - 1;
        str += "[";
        for (pos, data_type) in self.variants.iter().enumerate() {
            str += &data_type.literal();
            if pos != last_position {
                str += " | ";
            }
        }
        str += "]";
        str
    }

    fn equals(&self, other: &Box<dyn DataType>) -> bool {
        if other.is_any() {
            return true;
        }

        for variant in self.variants.iter() {
            if variant.equals(other) {
                return true;
            }
        }

        false
    }

    fn as_any(&self) -> &dyn Any {
        self
    }
}