crypto: Allow session to be saved in a batched way.

matrix_sdk_crypto/src/machine.rs

@@ -358,7 +358,7 @@ impl OlmMachine {
                     }
                 };
 
-                if let Err(e) = self.store.save_session(session).await {
+                if let Err(e) = self.store.save_sessions(&[session]).await {
                     error!("Failed to store newly created Olm session {}", e);
                     continue;
                 }
@@ -739,7 +739,7 @@ impl OlmMachine {
             // Decryption was successful, save the new ratchet state of the
             // session that was used to decrypt the message.
             trace!("Saved the new session state for {}", sender);
-            self.store.save_session(session).await?;
+            self.store.save_sessions(&[session]).await?;
         }
 
         Ok(plaintext)
@@ -804,7 +804,7 @@ impl OlmMachine {
             let plaintext = session.decrypt(message).await?;
 
             // Save the new ratcheted state of the session.
-            self.store.save_session(session).await?;
+            self.store.save_sessions(&[session]).await?;
             plaintext
         };
 
@@ -1055,7 +1055,7 @@ impl OlmMachine {
             .unwrap_or_else(|_| panic!(format!("Can't serialize {} to canonical JSON", payload)));
 
         let ciphertext = session.encrypt(&plaintext).await.to_tuple();
-        self.store.save_session(session).await?;
+        self.store.save_sessions(&[session]).await?;
 
         let message_type: usize = ciphertext.0.into();
 

matrix_sdk_crypto/src/store/memorystore.rs

@@ -52,8 +52,11 @@ impl CryptoStore for MemoryStore {
         Ok(())
     }
 
-    async fn save_session(&mut self, session: Session) -> Result<()> {
+    async fn save_sessions(&mut self, sessions: &[Session]) -> Result<()> {
-        self.sessions.add(session).await;
+        for session in sessions {
+            self.sessions.add(session.clone()).await;
+        }
+
         Ok(())
     }
 
@@ -125,7 +128,7 @@ mod test {
         assert!(store.load_account().await.unwrap().is_none());
         store.save_account(account).await.unwrap();
 
-        store.save_session(session.clone()).await.unwrap();
+        store.save_sessions(&[session.clone()]).await.unwrap();
 
         let sessions = store
             .get_sessions(&session.sender_key)

matrix_sdk_crypto/src/store/mod.rs

@@ -75,12 +75,12 @@ pub trait CryptoStore: Debug + Send + Sync {
     /// * `account` - The account that should be stored.
     async fn save_account(&mut self, account: Account) -> Result<()>;
 
-    /// Save the given session in the store.
+    /// Save the given sessions in the store.
     ///
     /// # Arguments
     ///
-    /// * `session` - The session that should be stored.
+    /// * `session` - The sessions that should be stored.
-    async fn save_session(&mut self, session: Session) -> Result<()>;
+    async fn save_sessions(&mut self, session: &[Session]) -> Result<()>;
 
     /// Get all the sessions that belong to the given sender key.
     ///

matrix_sdk_crypto/src/store/sqlite.rs

@@ -527,7 +527,9 @@ impl CryptoStore for SqliteStore {
         Ok(())
     }
 
-    async fn save_session(&mut self, session: Session) -> Result<()> {
+    async fn save_sessions(&mut self, sessions: &[Session]) -> Result<()> {
+        // TODO turn this into a transaction
+        for session in sessions {
             self.lazy_load_sessions(&session.sender_key).await?;
             self.sessions.add(session.clone()).await;
 
@@ -553,6 +555,7 @@ impl CryptoStore for SqliteStore {
             .bind(&pickle)
             .execute(&mut *connection)
             .await?;
+        }
 
         Ok(())
     }
@@ -806,14 +809,14 @@ mod test {
         let (mut store, _dir) = get_store(None).await;
         let (account, session) = get_account_and_session().await;
 
-        assert!(store.save_session(session.clone()).await.is_err());
+        assert!(store.save_sessions(&[session.clone()]).await.is_err());
 
         store
             .save_account(account.clone())
             .await
             .expect("Can't save account");
 
-        store.save_session(session).await.unwrap();
+        store.save_sessions(&[session]).await.unwrap();
     }
 
     #[tokio::test]
@@ -824,7 +827,7 @@ mod test {
             .save_account(account.clone())
             .await
             .expect("Can't save account");
-        store.save_session(session.clone()).await.unwrap();
+        store.save_sessions(&[session.clone()]).await.unwrap();
 
         let sessions = store
             .load_sessions_for(&session.sender_key)
@@ -846,7 +849,7 @@ mod test {
             .save_account(account.clone())
             .await
             .expect("Can't save account");
-        store.save_session(session).await.unwrap();
+        store.save_sessions(&[session]).await.unwrap();
 
         let sessions = store.get_sessions(&sender_key).await.unwrap().unwrap();
         let sessions_lock = sessions.lock().await;