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

// Copyright 2023 ReductStore
// This Source Code Form is subject to the terms of the Mozilla Public
//    License, v. 2.0. If a copy of the MPL was not distributed with this
//    file, You can obtain one at https://mozilla.org/MPL/2.0/.
use crate::msg::entry_api::EntryInfo;
use serde::{Deserialize, Serialize};
use std::fmt::Display;
use std::str::FromStr;

/// Quota type
///
/// NONE: No quota
/// FIFO: When quota_size is reached, the oldest records are deleted
#[derive(Serialize, Deserialize, Default, Clone, Debug, PartialEq)]
pub enum QuotaType {
    #[default]
    NONE = 0,
    FIFO = 1,
}

impl From<i32> for QuotaType {
    fn from(value: i32) -> Self {
        match value {
            0 => QuotaType::NONE,
            1 => QuotaType::FIFO,
            _ => QuotaType::NONE,
        }
    }
}

impl FromStr for QuotaType {
    type Err = ();

    fn from_str(s: &str) -> Result<Self, Self::Err> {
        match s {
            "NONE" => Ok(QuotaType::NONE),
            "FIFO" => Ok(QuotaType::FIFO),
            _ => Err(()),
        }
    }
}

impl Display for QuotaType {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        match self {
            QuotaType::NONE => write!(f, "NONE"),
            QuotaType::FIFO => write!(f, "FIFO"),
        }
    }
}

/// Bucket settings
#[derive(Serialize, Deserialize, Default, Clone, Debug, PartialEq)]
pub struct BucketSettings {
    /// Quota type see QuotaType
    pub quota_type: Option<QuotaType>,
    /// Quota size in bytes
    pub quota_size: Option<u64>,
    /// Max size of a block in bytes to start a new one
    pub max_block_size: Option<u64>,
    /// Max records in a block to start a new block one
    pub max_block_records: Option<u64>,
}

/// Bucket information
#[derive(Serialize, Deserialize, Default, Clone, Debug, PartialEq)]
pub struct BucketInfo {
    /// Unique bucket name
    pub name: String,
    /// Number of records in bucket
    pub entry_count: u64,
    /// Total size of bucket in bytes
    pub size: u64,
    /// Oldest record in bucket
    pub oldest_record: u64,
    /// Latest record in bucket
    pub latest_record: u64,
    /// Provisioned
    pub is_provisioned: bool,
}

/// Full bucket information
#[derive(Serialize, Deserialize, Default, Clone, Debug, PartialEq)]
pub struct FullBucketInfo {
    /// Bucket information
    pub info: BucketInfo,
    /// Bucket settings
    pub settings: BucketSettings,
    /// Entries in bucket
    pub entries: Vec<EntryInfo>,
}

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

    #[rstest]
    fn test_enum_as_string() {
        let settings = BucketSettings {
            quota_type: Some(QuotaType::FIFO),
            quota_size: Some(100),
            max_block_size: Some(100),
            max_block_records: Some(100),
        };
        let serialized = serde_json::to_string(&settings).unwrap();

        assert_eq!(
            serialized,
            r#"{"quota_type":"FIFO","quota_size":100,"max_block_size":100,"max_block_records":100}"#
        );
    }
}