non-compiling current changes

proof_of_stake/src/epoched_new.rs

@@ -248,6 +248,55 @@ where
         let key = self.storage_prefix.push(&"data".to_owned()).unwrap();
         NestedMap::open(key)
     }
+
+    pub fn init_at_genesis<S>(
+        &self,
+        storage: &mut S,
+        value: Data,
+        current_epoch: Epoch,
+    ) -> storage_api::Result<()>
+    where
+        S: StorageWrite + for<'iter> StorageRead<'iter>,
+    {
+        self.init(storage, value, current_epoch, 0)
+    }
+
+    /// Initialize new nested data at the given epoch offset.
+    pub fn init<S>(
+        &self,
+        storage: &mut S,
+        data: Data,
+        current_epoch: Epoch,
+        offset: u64,
+    ) -> storage_api::Result<()>
+    where
+        S: StorageWrite + for<'iter> StorageRead<'iter>,
+    {
+        let key = self.get_last_update_storage_key();
+        storage.write(&key, current_epoch)?;
+
+        self.set_at_epoch(storage, data, current_epoch, offset)
+    }
+
+    fn set_at_epoch<S>(
+        &self,
+        storage: &mut S,
+        data: Data,
+        current_epoch: Epoch,
+        offset: u64,
+    ) -> storage_api::Result<()>
+    where
+        S: StorageWrite + for<'iter> StorageRead<'iter>,
+    {
+        let data_handler = self.get_data_handler();
+        let epoch = current_epoch + offset;
+        let _prev = data_handler.insert(storage, epoch, data)?;
+        Ok(())
+    }
+
+    fn get_last_update_storage_key(&self) -> storage::Key {
+        self.storage_prefix.push(&"last_update".to_owned()).unwrap()
+    }
 }
 
 impl<Data, FutureEpochs, const NUM_PAST_EPOCHS: u64>