match prefix to next_key

Signed-off-by: muraca <mmuraca247@gmail.com>
paritytech · Jan 25, 2023 · 6fb5d13 · 6fb5d13
1 parent b26116e
commit 6fb5d13
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/frame/support/src/storage/mod.rs b/frame/support/src/storage/mod.rs
@@ -1792,6 +1792,7 @@ mod test {
 			assert_eq!(FooDoubleMap::contains_prefix(0), false);
 
 			assert_ok!(FooDoubleMap::try_append(1, 1, 4));
+			assert_ok!(FooDoubleMap::try_append(2, 1, 4));
 			assert!(FooDoubleMap::iter_prefix_values(1).next().is_some());
 			assert!(FooDoubleMap::contains_prefix(1));
 			FooDoubleMap::remove(1, 1);

diff --git a/frame/support/src/storage/unhashed.rs b/frame/support/src/storage/unhashed.rs
@@ -155,7 +155,10 @@ pub fn clear_prefix(
 }
 
 pub fn contains_prefix(prefix: &[u8]) -> bool {
-	sp_io::storage::next_key(prefix).is_some()
+	match sp_io::storage::next_key(prefix) {
+		Some(key) => key.starts_with(prefix),
+		None => false,
+	}
 }
 
 /// Get a Vec of bytes from storage.