fixup! When consuming a compat refresh token, consume others in the session

Author: sandhose

crates/storage-pg/.sqlx/query-ebd54e61ac1310d3a79721ac2e769ae35beca0088d462ec678ad15d1b1edfb0e.json renamed to crates/storage-pg/.sqlx/query-f75e44b528234dac708640ad9a111f3f6b468a91bf0d5b574795bf8c80605f19.json

@@ -205,6 +205,7 @@ impl CompatRefreshTokenRepository for PgCompatRefreshTokenRepository<'_> {
                 UPDATE compat_refresh_tokens
                 SET consumed_at = $2
                 WHERE compat_session_id = $1
+                  AND consumed_at IS NULL
             "#,
             Uuid::from(compat_refresh_token.session_id),
             consumed_at,
@@ -213,7 +214,15 @@ impl CompatRefreshTokenRepository for PgCompatRefreshTokenRepository<'_> {
         .execute(&mut *self.conn)
         .await?;
 
-        DatabaseError::ensure_affected_rows(&res, 1)?;
+        // This can affect multiple rows in case we've imported refresh tokens
+        // from Synapse. What we care about is that it at least affected one,
+        // which is what we're checking here
+        if res.rows_affected() == 0 {
+            return Err(DatabaseError::RowsAffected {
+                expected: 1,
+                actual: 0,
+            });
+        }
 
         let compat_refresh_token = compat_refresh_token
             .consume(consumed_at)