123456#![recursion_limit = "1024"]78#![cfg_attr(not(feature = "std"), no_std)]910#[cfg(feature = "std")]11pub use std::*;1213pub use serde::{Serialize, Deserialize};1415#[cfg(feature = "runtime-benchmarks")]16mod benchmarking;1718#[cfg(test)]19mod tests;2021pub use frame_support::{22 construct_runtime, decl_module, decl_storage,23 ensure,24 traits::{25 Currency, ExistenceRequirement, Get, Imbalance, KeyOwnerProofSystem, OnUnbalanced,26 Randomness, IsSubType, WithdrawReasons,27 },28 weights::{29 constants::{BlockExecutionWeight, ExtrinsicBaseWeight, RocksDbWeight, WEIGHT_PER_SECOND},30 DispatchInfo, GetDispatchInfo, IdentityFee, Pays, PostDispatchInfo, Weight,31 WeightToFeePolynomial, DispatchClass,32 },33 StorageValue,34 transactional,35};363738pub use frame_support::dispatch::DispatchResult;3940use sp_runtime::{41 Perbill,42 traits::{Zero}43};44use sp_std::convert::TryInto;4546use frame_system::{self as system};474849pub type BalanceOf<T> =50 <<T as Config>::Currency as Currency<<T as frame_system::Config>::AccountId>>::Balance;5152pub const YEAR: u32 = 5_259_600;53pub const TOTAL_YEARS_UNTIL_FLAT: u32 = 9;54pub const START_INFLATION_PERCENT: u32 = 10;55pub const END_INFLATION_PERCENT: u32 = 4;5657pub trait Config: system::Config {58 type Currency: Currency<Self::AccountId>;59 type TreasuryAccountId: Get<Self::AccountId>;60 type InflationBlockInterval: Get<Self::BlockNumber>;61}6263decl_storage! {64 trait Store for Module<T: Config> as Inflation {65 66 pub StartingYearTotalIssuance get(fn starting_year_total_issuance): BalanceOf<T>;6768 69 pub BlockInflation get(fn block_inflation): BalanceOf<T>;70 }71}7273decl_module! {74 pub struct Module<T: Config> for enum Call 75 where 76 origin: T::Origin,77 {78 const InflationBlockInterval: T::BlockNumber = T::InflationBlockInterval::get();7980 fn on_initialize(now: T::BlockNumber) -> Weight 81 {82 let mut consumed_weight = 0;83 let mut add_weight = |reads, writes, weight| {84 consumed_weight += T::DbWeight::get().reads_writes(reads, writes);85 consumed_weight += weight;86 };8788 let block_interval: u32 = T::InflationBlockInterval::get().try_into().unwrap_or(0);8990 91 if (now % T::BlockNumber::from(YEAR)).is_zero() || <BlockInflation<T>>::get().is_zero() {92 let current_year: u32 = (now / T::BlockNumber::from(YEAR)).try_into().unwrap_or(0);9394 let one_percent = Perbill::from_percent(1);9596 if current_year <= TOTAL_YEARS_UNTIL_FLAT {97 let amount: BalanceOf<T> = Perbill::from_rational_approximation(98 block_interval * (START_INFLATION_PERCENT * TOTAL_YEARS_UNTIL_FLAT - current_year * (START_INFLATION_PERCENT - END_INFLATION_PERCENT)), 99 YEAR * TOTAL_YEARS_UNTIL_FLAT100 ) * ( one_percent * T::Currency::total_issuance() );101 <BlockInflation<T>>::put(amount);102 }103 else {104 let amount: BalanceOf<T> = Perbill::from_rational_approximation(105 block_interval * END_INFLATION_PERCENT, 106 YEAR107 ) * (one_percent * T::Currency::total_issuance());108 <BlockInflation<T>>::put(amount);109 }110 <StartingYearTotalIssuance<T>>::set(T::Currency::total_issuance());111112 113 T::Currency::deposit_creating(&T::TreasuryAccountId::get(), <BlockInflation<T>>::get());114115 add_weight(7, 6, 28_300_000);116 }117118 119 else if (now % T::BlockNumber::from(block_interval)).is_zero() {120 T::Currency::deposit_into_existing(&T::TreasuryAccountId::get(), <BlockInflation<T>>::get()).ok();121122 add_weight(3, 2, 12_900_000);123 }124125 consumed_weight126 }127128 }129}