From 414f81774d3b88dac6e49bd875600e470bc118a1 Mon Sep 17 00:00:00 2001
From: Liu-Cheng Xu <xuliuchengxlc@gmail.com>
Date: Mon, 23 Aug 2021 19:16:24 +0800
Subject: [PATCH] Add NextKeys and QueuedKeys for session module (#291)

* Add NextKeys and QueuedKeys for session module

* Fix fmt
---
 src/frame/session.rs | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/src/frame/session.rs b/src/frame/session.rs
index 7971455893..2faa6ae979 100644
--- a/src/frame/session.rs
+++ b/src/frame/session.rs
@@ -72,6 +72,22 @@ pub struct ValidatorsStore<T: Session> {
     pub _runtime: PhantomData<T>,
 }
 
+/// The queued keys for the next session.
+#[derive(Encode, Store, Debug)]
+pub struct QueuedKeysStore<T: Session> {
+    #[store(returns = Vec<(<T as Session>::ValidatorId, T::Keys)>)]
+    /// Marker for the runtime
+    pub _runtime: PhantomData<T>,
+}
+
+/// The next session keys for a validator.
+#[derive(Encode, Store, Debug)]
+pub struct NextKeysStore<'a, T: Session> {
+    #[store(returns = Option<<T as Session>::Keys>)]
+    /// The validator account.
+    pub validator_id: &'a <T as Session>::ValidatorId,
+}
+
 default_impl!(ValidatorsStore);
 
 /// Set the session keys for a validator.