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
// This file is part of Gear.

// Copyright (C) 2022-2024 Gear Technologies Inc.
// SPDX-License-Identifier: GPL-3.0-or-later WITH Classpath-exception-2.0

// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.

// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.

// You should have received a copy of the GNU General Public License
// along with this program. If not, see <https://www.gnu.org/licenses/>.

//! Module for limiter implementation.
//!
//! Limiter provides API for continually descending value.
//! Could be used to prevent (limit) some value from overflow.

use crate::storage::ValueStorage;
use core::marker::PhantomData;

/// Represents logic of limiting value decreasing.
pub trait Limiter {
    /// Limiter stored type.
    type Value;

    /// Decreases stored value.
    ///
    /// Should be safe from overflow.
    fn decrease(value: Self::Value);

    /// Returns stored value, if present,
    /// or default/nullable value.
    fn get() -> Self::Value;

    /// Puts given value into stored, to start from
    /// new one for future decreasing.
    fn put(value: Self::Value);
}

/// `Limiter` implementation based on `ValueStorage`.
///
/// Generic parameter `T` represents inner storing type.
pub struct LimiterImpl<T, VS: ValueStorage<Value = T>>(PhantomData<VS>);

/// Crate local macro for repeating `Limiter` implementation
/// over `ValueStorage` with Rust's numeric types.
macro_rules! impl_limiter {
    ($($t: ty), +) => { $(
        impl<VS: ValueStorage<Value = $t>> Limiter for LimiterImpl<$t, VS> {
            type Value = VS::Value;

            fn decrease(value: Self::Value) {
                VS::mutate(|opt_val| {
                    if let Some(val) = opt_val {
                        *val = val.saturating_sub(value);
                    }
                });
            }

            fn get() -> Self::Value {
                VS::get().unwrap_or(0)
            }

            fn put(value: Self::Value) {
                VS::put(value);
            }
        }
    ) + };
}

// Implementation for unsigned integers.
impl_limiter!(u8, u16, u32, u64, u128);

// Implementation for signed integers.
impl_limiter!(i8, i16, i32, i64, i128);