elephantry 5.2.0

Object model manager for PostgreSQL
Documentation 
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
140
141
142
143
144
145
146
147
148
149

#[derive(Debug)]
pub struct Event<T: elephantry::FromSql + elephantry::ToSql> {
    #[cfg(feature = "uuid")]
    pub uuid: Option<uuid::Uuid>,
    #[cfg(not(feature = "uuid"))]
    pub uuid: Option<String>,
    pub name: String,
    pub visitor_id: Option<i32>,
    #[cfg(feature = "json")]
    pub properties: serde_json::Value,
    #[cfg(not(feature = "json"))]
    pub properties: String,
    #[cfg(feature = "json")]
    pub browser: serde_json::Value,
    #[cfg(not(feature = "json"))]
    pub browser: String,
    pub generic: Option<T>,
}

impl<T: elephantry::FromSql + elephantry::ToSql> elephantry::Entity for Event<T> {
    fn from(tuple: &elephantry::Tuple) -> Self {
        Self {
            uuid: tuple.get("uuid"),
            name: tuple.get("name"),
            visitor_id: tuple.get("visitor_id"),
            properties: tuple.get("properties"),
            browser: tuple.get("browser"),
            generic: tuple.get("generic"),
        }
    }

    fn get(&self, field: &str) -> Option<&dyn elephantry::ToSql> {
        match field {
            "uuid" => self.uuid.as_ref().map(|x| x as _),
            "name" => Some(&self.name),
            "visitor_id" => self.visitor_id.as_ref().map(|x| x as _),
            "properties" => Some(&self.properties),
            "browser" => Some(&self.browser),
            "generic" => self.generic.as_ref().map(|x| x as _),
            _ => None,
        }
    }
}

pub struct EventModel {
    #[allow(dead_code)]
    connection: elephantry::Connection,
}

impl elephantry::Model for EventModel {
    type Entity = Event<String>;
    type Structure = EventStructure;

    fn new(connection: &elephantry::Connection) -> Self {
        Self {
            connection: connection.clone(),
        }
    }
}

impl EventModel {
    pub fn count_uniq_visitor(&self) -> elephantry::Result<u32> {
        self.connection
            .execute("select count(distinct visitor_id) as count from event")
            .map(|x| x.get(0).get("count"))
    }
}

#[derive(Debug)]
pub struct EventExtra<T: elephantry::Entity + elephantry::ToSql> {
    #[cfg(feature = "uuid")]
    pub uuid: Option<uuid::Uuid>,
    #[cfg(not(feature = "uuid"))]
    pub uuid: Option<String>,
    pub name: String,
    pub visitor_id: Option<i32>,
    #[cfg(feature = "json")]
    pub properties: serde_json::Value,
    #[cfg(not(feature = "json"))]
    pub properties: String,
    #[cfg(feature = "json")]
    pub browser: serde_json::Value,
    pub generic: Option<T>,
    #[cfg(not(feature = "json"))]
    pub browser: String,
    pub os: Option<String>,
}

impl<T: elephantry::Entity + elephantry::ToSql> elephantry::Entity for EventExtra<T> {
    fn from(tuple: &elephantry::Tuple) -> Self {
        let event = <Event<String> as elephantry::Entity>::from(tuple);

        Self {
            uuid: event.uuid,
            name: event.name,
            visitor_id: event.visitor_id,
            properties: event.properties,
            browser: event.browser,
            generic: None,
            os: tuple.get("os"),
        }
    }

    fn get(&self, field: &str) -> Option<&dyn elephantry::ToSql> {
        match field {
            "uuid" => self.uuid.as_ref().map(|x| x as _),
            "name" => Some(&self.name),
            "visitor_id" => self.visitor_id.as_ref().map(|x| x as _),
            "properties" => Some(&self.properties),
            "browser" => Some(&self.browser),
            "generic" => self.generic.as_ref().map(|x| x as _),
            "os" => self.os.as_ref().map(|x| x as _),
            _ => None,
        }
    }
}

pub struct EventExtraModel;

impl elephantry::Model for EventExtraModel {
    type Entity = EventExtra<String>;
    type Structure = EventStructure;

    fn new(_: &elephantry::Connection) -> Self {
        Self
    }

    fn create_projection() -> elephantry::Projection {
        Self::default_projection().add_field("os", "%:browser:% ->> 'os'")
    }
}

pub struct EventStructure;

impl elephantry::Structure for EventStructure {
    fn primary_key() -> &'static [&'static str] {
        &["uuid"]
    }
}

impl elephantry::Projectable for EventStructure {
    fn relation() -> &'static str {
        "public.event"
    }

    fn columns() -> &'static [&'static str] {
        &["uuid", "name", "visitor_id", "properties", "browser"]
    }
}