martin/kosyncrs
1
0
Fork 0
Code Issues Pull requests Projects Releases 3 Packages Wiki Activity

do not use UPSERT

Browse source
This commit is contained in:
Martin Brodbeck 2025-01-16 14:25:01 +01:00
parent eab0e2bac8
commit fd4603ba42
1 changed files with 39 additions and 18 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

39
src/main.rs
View file

@ -87,9 +87,7 @@ async fn main() {
let mut bind_with = "127.0.0.1:".to_owned();
bind_with += &listen_port;
let listener = tokio::net::TcpListener::bind(bind_with)
.await
.unwrap();
let listener = tokio::net::TcpListener::bind(bind_with).await.unwrap();
axum::serve(listener, app).await.unwrap();
}
@ -190,17 +188,25 @@ async fn update_progress(
.unwrap()
.as_secs();
let row: (i64,) = sqlx::query_as("SELECT id FROM users WHERE username = $1")
let user_row: (i64,) = sqlx::query_as("SELECT id FROM users WHERE username = $1")
.bind(&username)
.fetch_one(&db_pool)
.await
.unwrap();
let user_id = row.0;
let user_id = user_row.0;
sqlx::query("INSERT INTO progresses (user_id, document, progress, percentage, device, device_id, timestamp) VALUES ($1, $2, $3, $4, $5, $6, $7) \
ON CONFLICT (user_id, document) DO UPDATE \
SET user_id = $1, document = $2, progress = $3, percentage = $4, device = $5, device_id = $6, timestamp = $7")
let mut tx = db_pool.begin().await.unwrap();
let doc_row = sqlx::query("SELECT 1 FROM progresses WHERE user_id = $1 AND document = $2")
.bind(user_id)
.bind(&payload.document)
.fetch_optional(&mut *tx)
.await
.unwrap();
if doc_row.is_none() {
sqlx::query("INSERT INTO progresses (user_id, document, progress, percentage, device, device_id, timestamp) VALUES ($1, $2, $3, $4, $5, $6, $7)")
.bind(user_id)
.bind(&payload.document)
.bind(&payload.progress)
@ -208,9 +214,24 @@ async fn update_progress(
.bind(&payload.device)
.bind(&payload.device_id)
.bind(timestamp as i64)
.execute(&db_pool)
.execute(&mut *tx)
.await
.unwrap();
} else {
sqlx::query("UPDATE progresses SET progress = $3, percentage = $4, device = $5, device_id = $6, timestamp = $7 WHERE user_id = $1 AND document = $2")
.bind(user_id)
.bind(&payload.document)
.bind(&payload.progress)
.bind(&payload.percentage)
.bind(&payload.device)
.bind(&payload.device_id)
.bind(timestamp as i64)
.execute(&mut *tx)
.await
.unwrap();
}
tx.commit().await.unwrap();
StatusCode::OK
}