sp_trie/
recorder_ext.rs

1// This file is part of Substrate.
2
3// Copyright (C) Parity Technologies (UK) Ltd.
4// SPDX-License-Identifier: Apache-2.0
5
6// Licensed under the Apache License, Version 2.0 (the "License");
7// you may not use this file except in compliance with the License.
8// You may obtain a copy of the License at
9//
10// 	http://www.apache.org/licenses/LICENSE-2.0
11//
12// Unless required by applicable law or agreed to in writing, software
13// distributed under the License is distributed on an "AS IS" BASIS,
14// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15// See the License for the specific language governing permissions and
16// limitations under the License.
17
18//! Extension for the default recorder.
19
20use crate::RawStorageProof;
21use alloc::{collections::BTreeSet, vec::Vec};
22use trie_db::{Recorder, TrieLayout};
23
24/// Convenience extension for the `Recorder` struct.
25///
26/// Used to deduplicate some logic.
27pub trait RecorderExt<L: TrieLayout>
28where
29	Self: Sized,
30{
31	/// Convert the recorder into a `BTreeSet`.
32	fn into_set(self) -> BTreeSet<Vec<u8>>;
33
34	/// Convert the recorder into a `RawStorageProof`, avoiding duplicate nodes.
35	fn into_raw_storage_proof(self) -> RawStorageProof {
36		// The recorder may record the same trie node multiple times,
37		// and we don't want duplicate nodes in our proofs
38		// => let's deduplicate it by collecting to a BTreeSet first
39		self.into_set().into_iter().collect()
40	}
41}
42
43impl<L: TrieLayout> RecorderExt<L> for Recorder<L> {
44	fn into_set(mut self) -> BTreeSet<Vec<u8>> {
45		self.drain().into_iter().map(|record| record.data).collect::<BTreeSet<_>>()
46	}
47}
sp_trie/recorder_ext.rs

sp_trie/
recorder_ext.rs
