forked from projectmoon/tenebrous-dicebot
Use ON CONFLICT and transactions where appropriate.
This commit is contained in:
parent
7eee16961e
commit
a3b39ee42c
|
@ -51,16 +51,13 @@ impl Rooms for Database {
|
|||
}
|
||||
|
||||
async fn insert_room_info(&self, info: &RoomInfo) -> Result<(), DataError> {
|
||||
//Clear out old info first, because we want this to be an "upsert."
|
||||
sqlx::query("DELETE FROM room_info where room_id = ?")
|
||||
.bind(&info.room_id)
|
||||
.execute(&self.conn)
|
||||
.await
|
||||
.ok();
|
||||
|
||||
sqlx::query(r#"INSERT INTO room_info (room_id, room_name) VALUES (?, ?)"#)
|
||||
sqlx::query(
|
||||
r#"INSERT INTO room_info (room_id, room_name) VALUES (?, ?)
|
||||
ON CONFLICT(room_id) DO UPDATE SET room_name = ?"#,
|
||||
)
|
||||
.bind(&info.room_id)
|
||||
.bind(&info.room_name)
|
||||
.bind(&info.room_name)
|
||||
.execute(&self.conn)
|
||||
.await?;
|
||||
|
||||
|
@ -107,12 +104,10 @@ impl Rooms for Database {
|
|||
}
|
||||
|
||||
async fn add_user_to_room(&self, username: &str, room_id: &str) -> Result<(), DataError> {
|
||||
// This is here because it is possible to process a bunch of
|
||||
// user join/leave events at once, and we don't want to cause
|
||||
// constraint violation errors.
|
||||
self.remove_user_from_room(username, room_id).await.ok();
|
||||
|
||||
sqlx::query("INSERT INTO room_users (room_id, username) VALUES (?, ?)")
|
||||
sqlx::query(
|
||||
"INSERT INTO room_users (room_id, username) VALUES (?, ?)
|
||||
ON CONFLICT DO NOTHING",
|
||||
)
|
||||
.bind(room_id)
|
||||
.bind(username)
|
||||
.execute(&self.conn)
|
||||
|
@ -134,16 +129,19 @@ impl Rooms for Database {
|
|||
async fn clear_info(&self, room_id: &str) -> Result<(), DataError> {
|
||||
// We do not clear event history here, because if we rejoin a
|
||||
// room, we would re-process events we've already seen.
|
||||
let mut tx = self.conn.begin().await?;
|
||||
|
||||
sqlx::query("DELETE FROM room_info where room_id = ?")
|
||||
.bind(room_id)
|
||||
.execute(&self.conn)
|
||||
.execute(&mut tx)
|
||||
.await?;
|
||||
|
||||
sqlx::query("DELETE FROM room_users where room_id = ?")
|
||||
.bind(room_id)
|
||||
.execute(&self.conn)
|
||||
.execute(&mut tx)
|
||||
.await?;
|
||||
|
||||
tx.commit().await?;
|
||||
Ok(())
|
||||
}
|
||||
}
|
||||
|
|
Loading…
Reference in New Issue