matrix/authentication-service
1
0
Fork 0
mirror of https://github.com/matrix-org/matrix-authentication-service.git synced 2025-07-29 22:01:14 +03:00
Code Activity

storage: do less joins in compat sessions

Browse Source
This commit is contained in:
Quentin Gliech
2023-01-10 18:49:35 +01:00
parent 35787aa072
commit 920869b583
11 changed files with 616 additions and 542 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
crates/graphql/src/model/compat_sessions.rs
View File

@ -15,7 +15,7 @@
use anyhow::Context as _;
use async_graphql::{Context, Description, Object, ID};
use chrono::{DateTime, Utc};
use mas_storage::{user::UserRepository, Repository};
use mas_storage::{compat::lookup_compat_session, user::UserRepository, Repository};
use sqlx::PgPool;
use url::Url;
@ -94,7 +94,17 @@ impl CompatSsoLogin {
}
/// The compat session which was started by this login.
async fn session(&self) -> Option<CompatSession> {
self.0.session().cloned().map(CompatSession)
async fn session(
&self,
ctx: &Context<'_>,
) -> Result<Option<CompatSession>, async_graphql::Error> {
let Some(session_id) = self.0.session_id() else { return Ok(None) };
let mut conn = ctx.data::<PgPool>()?.acquire().await?;
let session = lookup_compat_session(&mut conn, session_id)
.await?
.context("Could not load compat session")?;
Ok(Some(CompatSession(session)))
}
}