From f6c4fdde7d0c1e051d9d3b3dc592428da9aaf791 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?K=C3=A9vin=20Commaille?= <zecakeh@tedomum.fr>
Date: Tue, 27 Apr 2021 10:57:38 +0200
Subject: [PATCH] matrix-sdk: Implement EncodeKey for EventType

---
 matrix_sdk_base/src/store/sled_store/mod.rs | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/matrix_sdk_base/src/store/sled_store/mod.rs b/matrix_sdk_base/src/store/sled_store/mod.rs
index 3076ec15..ac815683 100644
--- a/matrix_sdk_base/src/store/sled_store/mod.rs
+++ b/matrix_sdk_base/src/store/sled_store/mod.rs
@@ -134,6 +134,12 @@ impl EncodeKey for (&str, &str, &str) {
     }
 }
 
+impl EncodeKey for EventType {
+    fn encode(&self) -> Vec<u8> {
+        self.as_str().encode()
+    }
+}
+
 #[derive(Clone)]
 pub struct SledStore {
     path: Option<PathBuf>,
@@ -495,7 +501,7 @@ impl SledStore {
     ) -> Result<Option<AnySyncStateEvent>> {
         Ok(self
             .room_state
-            .get((room_id.as_str(), event_type.to_string().as_str(), state_key).encode())?
+            .get((room_id.as_str(), event_type.as_str(), state_key).encode())?
             .map(|e| self.deserialize_event(&e))
             .transpose()?)
     }
@@ -594,7 +600,7 @@ impl SledStore {
     ) -> Result<Option<AnyBasicEvent>> {
         Ok(self
             .account_data
-            .get(event_type.to_string().as_str().encode())?
+            .get(event_type.encode())?
             .map(|m| self.deserialize_event(&m))
             .transpose()?)
     }