sea_orm/
value.rs

1use crate::sea_query::{Nullable, ValueType};
2use crate::{ActiveValue, Value};
3
4/// Default value for T
5pub trait DefaultActiveValue {
6    /// `Default::default()` if implemented, dummy value otherwise
7    fn default_value(&self) -> Self;
8}
9
10/// Default value for Option<T>
11pub trait DefaultActiveValueNone {
12    /// Always `None`
13    fn default_value(&self) -> Self;
14}
15
16/// Default value for types that's not nullable
17pub trait DefaultActiveValueNotSet {
18    /// The owned value type
19    type Value;
20
21    /// Always `NotSet`
22    fn default_value(&self) -> Self::Value;
23}
24
25impl<V> DefaultActiveValue for ActiveValue<V>
26where
27    V: Into<Value> + ValueType + Nullable,
28{
29    fn default_value(&self) -> Self {
30        match V::try_from(V::null().dummy_value()) {
31            Ok(v) => ActiveValue::Set(v),
32            Err(_) => ActiveValue::NotSet,
33        }
34    }
35}
36
37impl<V> DefaultActiveValueNone for ActiveValue<Option<V>>
38where
39    V: Into<Value> + Nullable,
40{
41    fn default_value(&self) -> Self {
42        ActiveValue::Set(None)
43    }
44}
45
46impl<V> DefaultActiveValueNotSet for &ActiveValue<V>
47where
48    V: Into<Value>,
49{
50    type Value = ActiveValue<V>;
51
52    fn default_value(&self) -> Self::Value {
53        ActiveValue::NotSet
54    }
55}
56
57#[cfg(test)]
58mod test {
59    use super::*;
60    use crate::prelude::TimeDateTime;
61
62    #[test]
63    fn test_default_value() {
64        let v = (&ActiveValue::<i32>::NotSet).default_value();
65        assert_eq!(v, ActiveValue::Set(0));
66
67        let v = (&ActiveValue::<Option<i32>>::NotSet).default_value();
68        assert_eq!(v, ActiveValue::Set(None));
69
70        let v = (&ActiveValue::<String>::NotSet).default_value();
71        assert_eq!(v, ActiveValue::Set("".to_owned()));
72
73        let v = (&ActiveValue::<Option<String>>::NotSet).default_value();
74        assert_eq!(v, ActiveValue::Set(None));
75
76        let v = (&ActiveValue::<TimeDateTime>::NotSet).default_value();
77        assert!(matches!(v, ActiveValue::Set(_)));
78    }
79}
sea_orm/value.rs

sea_orm/
value.rs
