bee_ledger/types/
treasury_diff.rs

1// Copyright 2020-2021 IOTA Stiftung
2// SPDX-License-Identifier: Apache-2.0
3
4use crate::types::error::Error;
5
6use bee_common::packable::{Packable, Read, Write};
7use bee_message::payload::milestone::MilestoneId;
8
9/// Wraps together the identifiers of the milestones that created and consumed treasury outputs.
10#[derive(Clone, Debug, Eq, PartialEq)]
11pub struct TreasuryDiff {
12    created: MilestoneId,
13    consumed: MilestoneId,
14}
15
16impl TreasuryDiff {
17    /// Creates a new `TreasuryDiff`.
18    pub fn new(created: MilestoneId, consumed: MilestoneId) -> Self {
19        Self { created, consumed }
20    }
21
22    /// Returns the id of the milestone that created the treasury output associated to the `TreasuryDiff`.
23    pub fn created(&self) -> &MilestoneId {
24        &self.created
25    }
26
27    /// Returns the id of the milestone that consumed the treasury input associated to the `TreasuryDiff`.
28    pub fn consumed(&self) -> &MilestoneId {
29        &self.consumed
30    }
31}
32
33impl Packable for TreasuryDiff {
34    type Error = Error;
35
36    fn packed_len(&self) -> usize {
37        self.created.packed_len() + self.consumed.packed_len()
38    }
39
40    fn pack<W: Write>(&self, writer: &mut W) -> Result<(), Self::Error> {
41        self.created.pack(writer)?;
42        self.consumed.pack(writer)?;
43
44        Ok(())
45    }
46
47    fn unpack_inner<R: Read + ?Sized, const CHECK: bool>(reader: &mut R) -> Result<Self, Self::Error> {
48        let created = MilestoneId::unpack_inner::<R, CHECK>(reader)?;
49        let consumed = MilestoneId::unpack_inner::<R, CHECK>(reader)?;
50
51        Ok(Self { created, consumed })
52    }
53}
bee_ledger/types/treasury_diff.rs

bee_ledger/types/
treasury_diff.rs
