Staking jobs with DRAND sorting

pallets/admin-utils/src/tests/mock.rs

@@ -8,6 +8,7 @@ use frame_support::{
 };
 use frame_system::{self as system, offchain::CreateTransactionBase};
 use frame_system::{EnsureNever, EnsureRoot, limits};
+use pallet_subtensor::EvmOriginHelper;
 use sp_consensus_aura::sr25519::AuthorityId as AuraId;
 use sp_consensus_grandpa::AuthorityList as GrandpaAuthorityList;
 use sp_core::U256;
@@ -28,7 +29,7 @@ frame_support::construct_runtime!(
         System: frame_system = 1,
         Balances: pallet_balances = 2,
         AdminUtils: crate = 3,
-        SubtensorModule: pallet_subtensor::{Pallet, Call, Storage, Event<T>, Error<T>} = 4,
+        SubtensorModule: pallet_subtensor::{Pallet, Call, Storage, Event<T>, Error<T>, Origin<T>} = 4,
         Scheduler: pallet_scheduler::{Pallet, Call, Storage, Event<T>} = 5,
         Drand: pallet_drand::{Pallet, Call, Storage, Event<T>} = 6,
         Grandpa: pallet_grandpa = 7,
@@ -151,8 +152,15 @@ parameter_types! {
     pub const LeaseDividendsDistributionInterval: u32 = 100; // 100 blocks
 }
 
+impl EvmOriginHelper<RuntimeOrigin, AccountId> for () {
+    fn make_evm_origin(_: AccountId) -> RuntimeOrigin {
+        RuntimeOrigin::none()
+    }
+}
 impl pallet_subtensor::Config for Test {
+    type EvmOriginHelper = ();
     type RuntimeEvent = RuntimeEvent;
+    type RuntimeOrigin = RuntimeOrigin;
     type RuntimeCall = RuntimeCall;
     type Currency = Balances;
     type InitialIssuance = InitialIssuance;
@@ -396,6 +404,7 @@ impl pallet_drand::Config for Test {
     type Verifier = pallet_drand::verifier::QuicknetVerifier;
     type UnsignedPriority = ConstU64<{ 1 << 20 }>;
     type HttpFetchTimeout = ConstU64<1_000>;
+    type OnPulseReceived = ();
 }
 
 impl frame_system::offchain::SigningTypes for Test {

pallets/commitments/src/mock.rs

@@ -123,6 +123,7 @@ impl pallet_drand::Config for Test {
     type Verifier = pallet_drand::verifier::QuicknetVerifier;
     type UnsignedPriority = ConstU64<{ 1 << 20 }>;
     type HttpFetchTimeout = ConstU64<1_000>;
+    type OnPulseReceived = ();
 }
 
 pub mod test_crypto {

pallets/drand/src/lib.rs

@@ -146,6 +146,24 @@ impl<T: SigningTypes> SignedPayload<T> for PulsesPayload<T::Public, BlockNumberF
     }
 }
 
+/// Callback method to signal new pulses
+pub trait OnPulseReceived {
+    /// Callback method to signal new pulses
+    fn on_pulse_received(round: RoundNumber);
+}
+
+impl OnPulseReceived for () {
+    fn on_pulse_received(_round: RoundNumber) {}
+}
+
+// Enables a chain of method calls
+impl<X: OnPulseReceived, Y: OnPulseReceived> OnPulseReceived for (X, Y) {
+    fn on_pulse_received(round: RoundNumber) {
+        X::on_pulse_received(round);
+        Y::on_pulse_received(round);
+    }
+}
+
 #[frame_support::pallet]
 pub mod pallet {
     use super::*;
@@ -174,6 +192,9 @@ pub mod pallet {
         /// complete.
         #[pallet::constant]
         type HttpFetchTimeout: Get<u64>;
+        ///
+        /// Warning: calculate weight properly when change this parameter
+        type OnPulseReceived: OnPulseReceived;
     }
 
     /// the drand beacon configuration
@@ -352,6 +373,7 @@ pub mod pallet {
             // Emit event with all new rounds
             if !new_rounds.is_empty() {
                 Self::deposit_event(Event::NewPulse { rounds: new_rounds });
+                T::OnPulseReceived::on_pulse_received(last_stored_round);
             }
 
             Ok(())

pallets/drand/src/mock.rs

@@ -102,6 +102,7 @@ impl pallet_drand_bridge::Config for Test {
     type Verifier = QuicknetVerifier;
     type UnsignedPriority = UnsignedPriority;
     type HttpFetchTimeout = ConstU64<1_000>;
+    type OnPulseReceived = ();
 }
 
 // Build genesis storage according to the mock runtime.