optimize single player search

This commit is contained in:
Robin Appelman 2024-01-20 17:25:07 +01:00
commit 3e79a76e1b
2 changed files with 36 additions and 22 deletions

View file

@ -456,6 +456,16 @@ impl Filter {
.and_where(Expr::col((Alias::new("upload_user"), Users::SteamId)).eq(uploader));
}
if !self.players.is_empty() && self.players.len() < 19 {
if self.players.len() == 1 {
let player = &self.players[0];
query
.inner_join(
Players::Table,
Expr::col((Demos::Table, Demos::Id))
.equals((Players::Table, Players::DemoId)),
)
.and_where(Expr::col((Players::Table, Players::SteamId)).eq(player));
} else {
let mut player = self.players.iter();
let mut players_arr = format!("array['{}'", player.next().unwrap());
for player in player {
@ -466,19 +476,21 @@ impl Filter {
query
.inner_join(
Players::Table,
Expr::col((Demos::Table, Demos::Id)).equals((Players::Table, Players::DemoId)),
Expr::col((Demos::Table, Demos::Id))
.equals((Players::Table, Players::DemoId)),
)
.inner_join(
Users::Table,
Expr::col((Users::Table, Users::Id)).equals((Players::Table, Players::UserId)),
)
.and_where(Expr::col((Users::Table, Users::SteamId)).is_in(self.players.clone()));
.and_where(
Expr::col((Players::Table, Players::SteamId)).is_in(self.players.clone()),
);
query.group_by_col((Demos::Table, Players::Id));
query.and_having(
Expr::cust(&players_arr)
.cast_as(Alias::new("varchar[]"))
.contained(Func::cust(ArrayAgg).arg(Expr::col((Users::Table, Users::SteamId)))),
.cast_as(Alias::new("text[]"))
.contained(
Func::cust(ArrayAgg).arg(Expr::col((Players::Table, Players::SteamId))),
),
);
}
}
}
}

View file

@ -91,6 +91,8 @@ pub enum Players {
Assists,
#[iden = "deaths"]
Deaths,
#[iden = "steam_id"]
SteamId,
}
#[allow(dead_code)]