store matches in database

1 files changed, 97 insertions, 0 deletions

diff --git a/planetwars-server/src/db/matches.rs b/planetwars-server/src/db/matches.rs
new file mode 100644
index 0000000..85f0631
--- /dev/null
+++ b/planetwars-server/src/db/matches.rs
@@ -0,0 +1,97 @@
+use chrono::NaiveDateTime;
+use diesel::{BelongingToDsl, RunQueryDsl};
+use diesel::{Connection, GroupedBy, PgConnection, QueryResult};
+
+use crate::schema::{match_players, matches};
+
+#[derive(Insertable)]
+#[table_name = "matches"]
+pub struct NewMatch<'a> {
+    pub log_path: &'a str,
+}
+
+#[derive(Insertable)]
+#[table_name = "match_players"]
+pub struct NewMatchPlayer {
+    /// id of the match this player is in
+    pub match_id: i32,
+    /// id of the bot behind this player
+    pub bot_id: i32,
+    /// player id within the match
+    pub player_id: i32,
+}
+
+#[derive(Queryable, Identifiable)]
+#[table_name = "matches"]
+pub struct MatchBase {
+    pub id: i32,
+    pub log_path: String,
+    pub created_at: NaiveDateTime,
+}
+
+#[derive(Queryable, Identifiable, Associations)]
+#[primary_key(match_id, player_id)]
+#[belongs_to(MatchBase, foreign_key = "match_id")]
+pub struct MatchPlayer {
+    pub match_id: i32,
+    pub bot_id: i32,
+    pub player_id: i32,
+}
+
+pub struct MatchPlayerData {
+    pub bot_id: i32,
+}
+
+pub fn create_match(
+    match_data: &NewMatch,
+    match_players: &[MatchPlayerData],
+    conn: &PgConnection,
+) -> QueryResult<i32> {
+    conn.transaction(|| {
+        let match_base = diesel::insert_into(matches::table)
+            .values(match_data)
+            .get_result::<MatchBase>(conn)?;
+
+        let match_players = match_players
+            .iter()
+            .enumerate()
+            .map(|(num, player_data)| NewMatchPlayer {
+                match_id: match_base.id,
+                bot_id: player_data.bot_id,
+                player_id: num as i32,
+            })
+            .collect::<Vec<_>>();
+
+        diesel::insert_into(match_players::table)
+            .values(&match_players)
+            .execute(conn)?;
+
+        Ok(match_base.id)
+    })
+}
+
+pub struct MatchData {
+    pub base: MatchBase,
+    pub match_players: Vec<MatchPlayer>,
+}
+
+pub fn list_matches(conn: &PgConnection) -> QueryResult<Vec<MatchData>> {
+    conn.transaction(|| {
+        let matches = matches::table.get_results::<MatchBase>(conn)?;
+
+        let match_players = MatchPlayer::belonging_to(&matches)
+            .load::<MatchPlayer>(conn)?
+            .grouped_by(&matches);
+
+        let res = matches
+            .into_iter()
+            .zip(match_players.into_iter())
+            .map(|(base, players)| MatchData {
+                base,
+                match_players: players.into_iter().collect(),
+            })
+            .collect();
+
+        Ok(res)
+    })
+}