milksad
/
rust-bitcoin-unsafe-fast
14
0
Fork 0
Code Issues Pull Requests Activity

Mark functions `const` 

Mark `checked_` functions const.
Replace `map()` and `?` operators, which are not allowed in const
context, with match statements.
This commit is contained in:
Jamil Lambert, PhD 2024-11-15 09:02:15 +00:00
parent 3ba3079edf
commit 68aac98753
No known key found for this signature in database
GPG Key ID: 54DC29234AB5D2C0
1 changed files with 37 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

44
units/src/fee_rate.rs
View File

@ -79,12 +79,24 @@ impl FeeRate {
/// Checked multiplication. /// Checked multiplication.
/// ///
/// Computes `self * rhs` returning [`None`] if overflow occurred. /// Computes `self * rhs` returning [`None`] if overflow occurred.
pub fn checked_mul(self, rhs: u64) -> Option<Self> { self.0.checked_mul(rhs).map(Self) } pub const fn checked_mul(self, rhs: u64) -> Option<Self> {
// No `map()` in const context.
match self.0.checked_mul(rhs) {
Some(res) => Some(Self(res)),
None => None,
}
}
/// Checked division. /// Checked division.
/// ///
/// Computes `self / rhs` returning [`None`] if `rhs == 0`. /// Computes `self / rhs` returning [`None`] if `rhs == 0`.
pub fn checked_div(self, rhs: u64) -> Option<Self> { self.0.checked_div(rhs).map(Self) } pub const fn checked_div(self, rhs: u64) -> Option<Self> {
// No `map()` in const context.
match self.0.checked_div(rhs) {
Some(res) => Some(Self(res)),
None => None,
}
}
/// Checked weight multiplication. /// Checked weight multiplication.
/// ///
@ -94,20 +106,38 @@ impl FeeRate {
/// rounded down. /// rounded down.
/// ///
/// [`None`] is returned if an overflow occurred. /// [`None`] is returned if an overflow occurred.
pub fn checked_mul_by_weight(self, rhs: Weight) -> Option<Amount> { pub const fn checked_mul_by_weight(self, rhs: Weight) -> Option<Amount> {
let sats = self.0.checked_mul(rhs.to_wu())?.checked_add(999)? / 1000; // No `?` operator in const context.
Some(Amount::from_sat(sats)) match self.0.checked_mul(rhs.to_wu()) {
Some(mul_res) => match mul_res.checked_add(999) {
Some(add_res) => Some(Amount::from_sat(add_res / 1000)),
None => None,
},
None => None,
}
} }
/// Checked addition. /// Checked addition.
/// ///
/// Computes `self + rhs` returning [`None`] if overflow occured. /// Computes `self + rhs` returning [`None`] if overflow occured.
pub fn checked_add(self, rhs: u64) -> Option<Self> { self.0.checked_add(rhs).map(Self) } pub const fn checked_add(self, rhs: u64) -> Option<Self> {
// No `map()` in const context.
match self.0.checked_add(rhs) {
Some(res) => Some(Self(res)),
None => None,
}
}
/// Checked subtraction. /// Checked subtraction.
/// ///
/// Computes `self - rhs` returning [`None`] if overflow occured. /// Computes `self - rhs` returning [`None`] if overflow occured.
pub fn checked_sub(self, rhs: u64) -> Option<Self> { self.0.checked_sub(rhs).map(Self) } pub const fn checked_sub(self, rhs: u64) -> Option<Self> {
// No `map()` in const context.
match self.0.checked_sub(rhs) {
Some(res) => Some(Self(res)),
None => None,
}
}
/// Calculates the fee by multiplying this fee rate by weight, in weight units, returning [`None`] /// Calculates the fee by multiplying this fee rate by weight, in weight units, returning [`None`]
/// if an overflow occurred. /// if an overflow occurred.