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

// 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 counter implementation.
//!
//! Counter provides API for step-by-step changing of the value.
//! Could be used to count amount of some parameter.

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

/// Represents logic of managing step-by-step changeable value.
pub trait Counter {
    /// Counter stored type.
    type Value;

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

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

    /// Increases stored value.
    ///
    /// Should be safe from overflow.
    fn increase();

    /// Resets stored value by setting default/starting value.
    fn reset();
}

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

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

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

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

            fn increase() {
                VS::mutate(|opt_val| {
                    if let Some(val) = opt_val {
                        *val = val.saturating_add(1);
                    } else {
                        *opt_val = Some(1)
                    }
                });
            }

            fn reset() {
                VS::put(0);
            }
        }
    ) + };
}

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

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