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

use std::{
    fmt,
    iter::Sum,
    ops::{Add, AddAssign, Sub},
};

/// A newtype for the JSON size of an event.
/// Used to emit the `component_received_event_bytes_total` and
/// `component_sent_event_bytes_total` metrics.
#[derive(Clone, Copy, Default, Debug, PartialEq, Eq, PartialOrd, Ord)]
pub struct JsonSize(usize);

impl fmt::Display for JsonSize {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        write!(f, "{}", self.0)
    }
}

impl Sub for JsonSize {
    type Output = JsonSize;

    #[inline]
    fn sub(mut self, rhs: Self) -> Self::Output {
        self.0 -= rhs.0;
        self
    }
}

impl Add for JsonSize {
    type Output = JsonSize;

    #[inline]
    fn add(mut self, rhs: Self) -> Self::Output {
        self.0 += rhs.0;
        self
    }
}

impl AddAssign for JsonSize {
    #[inline]
    fn add_assign(&mut self, rhs: Self) {
        self.0 += rhs.0;
    }
}

impl Sum for JsonSize {
    #[inline]
    fn sum<I: Iterator<Item = Self>>(iter: I) -> Self {
        let mut accum = 0;
        for val in iter {
            accum += val.get();
        }

        JsonSize::new(accum)
    }
}

impl From<usize> for JsonSize {
    #[inline]
    fn from(value: usize) -> Self {
        Self(value)
    }
}

impl JsonSize {
    /// Create a new instance with the specified size.
    #[must_use]
    #[inline]
    pub const fn new(size: usize) -> Self {
        Self(size)
    }

    /// Create a new instance with size 0.
    #[must_use]
    #[inline]
    pub const fn zero() -> Self {
        Self(0)
    }

    /// Returns the contained size.
    #[must_use]
    #[inline]
    pub fn get(&self) -> usize {
        self.0
    }
}

#[derive(Clone, Copy, Debug, PartialEq, Eq, PartialOrd, Ord)]
#[allow(clippy::module_name_repetitions)]
pub struct NonZeroJsonSize(JsonSize);

impl NonZeroJsonSize {
    #[must_use]
    #[inline]
    pub fn new(size: JsonSize) -> Option<Self> {
        (size.0 > 0).then_some(NonZeroJsonSize(size))
    }
}

impl From<NonZeroJsonSize> for JsonSize {
    #[inline]
    fn from(value: NonZeroJsonSize) -> Self {
        value.0
    }
}