brylix 0.2.9

GraphQL framework for AWS Lambda with SeaORM and multi-tenant support
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

//! Timestamp helpers for working with datetime fields.
//!
//! Provides utilities for getting the current UTC time and a trait
//! for models that have timestamp fields (created_at, updated_at).
//!
//! # Usage
//!
//! ```rust
//! use brylix::helpers::time::utc_now;
//!
//! let now = utc_now();
//! println!("Current UTC time: {}", now);
//! ```

use chrono::{DateTime, Utc};

/// Get the current UTC timestamp.
///
/// # Example
///
/// ```rust
/// use brylix::helpers::time::utc_now;
///
/// let now = utc_now();
/// ```
pub fn utc_now() -> DateTime<Utc> {
    Utc::now()
}

/// Trait for models with timestamp fields (created_at, updated_at).
///
/// Implement this on your SeaORM ActiveModel to standardize
/// timestamp management across entities.
///
/// # Example
///
/// ```rust,ignore
/// use brylix::helpers::time::{Timestamped, utc_now};
/// use sea_orm::Set;
///
/// impl Timestamped for users::ActiveModel {
///     fn set_created_at(&mut self) {
///         self.created_at = Set(utc_now());
///     }
///
///     fn set_updated_at(&mut self) {
///         self.updated_at = Set(utc_now());
///     }
/// }
///
/// // Usage:
/// let mut model = users::ActiveModel { .. };
/// model.set_timestamps(); // Sets both created_at and updated_at
/// ```
pub trait Timestamped {
    /// Set the created_at field to the current UTC time.
    fn set_created_at(&mut self);

    /// Set the updated_at field to the current UTC time.
    fn set_updated_at(&mut self);

    /// Set both created_at and updated_at to the current UTC time.
    fn set_timestamps(&mut self) {
        self.set_created_at();
        self.set_updated_at();
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_utc_now() {
        let now = utc_now();
        // Verify it returns a reasonable timestamp (after 2024)
        assert!(now.timestamp() > 1_700_000_000);
    }

    struct MockModel {
        created: bool,
        updated: bool,
    }

    impl Timestamped for MockModel {
        fn set_created_at(&mut self) {
            self.created = true;
        }

        fn set_updated_at(&mut self) {
            self.updated = true;
        }
    }

    #[test]
    fn test_set_timestamps() {
        let mut model = MockModel {
            created: false,
            updated: false,
        };
        model.set_timestamps();
        assert!(model.created);
        assert!(model.updated);
    }

    #[test]
    fn test_set_created_at_only() {
        let mut model = MockModel {
            created: false,
            updated: false,
        };
        model.set_created_at();
        assert!(model.created);
        assert!(!model.updated);
    }

    #[test]
    fn test_set_updated_at_only() {
        let mut model = MockModel {
            created: false,
            updated: false,
        };
        model.set_updated_at();
        assert!(!model.created);
        assert!(model.updated);
    }
}