diff --git a/crates/desktop/artcraft/src/services/storyteller/threads/storyteller_task_polling_thread.rs b/crates/desktop/artcraft/src/services/storyteller/threads/storyteller_task_polling_thread.rs index c076f31231..d068b9cf68 100644 --- a/crates/desktop/artcraft/src/services/storyteller/threads/storyteller_task_polling_thread.rs +++ b/crates/desktop/artcraft/src/services/storyteller/threads/storyteller_task_polling_thread.rs @@ -125,10 +125,10 @@ async fn polling_loop( TaskStatus::CompleteSuccess => continue, _ => {} // Fall-through. } - + update_task_status(UpdateTaskArgs { db: task_database.get_connection(), - task_id: task.id.clone(), + task_id: &task.id, status: TaskStatus::CompleteSuccess, }).await?; diff --git a/crates/schema/database/sqlite_tasks/src/queries/update_task_status.rs b/crates/schema/database/sqlite_tasks/src/queries/update_task_status.rs index 5534260b3a..7b864f70c3 100644 --- a/crates/schema/database/sqlite_tasks/src/queries/update_task_status.rs +++ b/crates/schema/database/sqlite_tasks/src/queries/update_task_status.rs @@ -1,3 +1,5 @@ +use log::info; +use sqlx::Execute; use crate::connection::TaskDbConnection; use crate::error::SqliteTasksError; use enums::tauri::tasks::task_status::TaskStatus; @@ -5,7 +7,7 @@ use tokens::tokens::sqlite::tasks::TaskId; pub struct UpdateTaskArgs<'a> { pub db: &'a TaskDbConnection, - pub task_id: TaskId, + pub task_id: &'a TaskId, pub status: TaskStatus, } @@ -17,17 +19,23 @@ pub async fn update_task_status( let task_id_temp = args.task_id.as_str(); let status_temp = args.status.to_str(); + info!("Update task id: {}, status: {}", task_id_temp, status_temp); + // TODO(bt,2025-07-15): We can't use a LIMIT without a certain compiler flag ? let query = sqlx::query!(r#" UPDATE tasks SET task_status = ? WHERE id = ? "#, - task_id_temp, status_temp, + task_id_temp, ); - let _r = query.execute(args.db.get_pool()).await?; + info!("query: {:?}", query.sql()); + + let res = query.execute(args.db.get_pool()).await?; + + info!("Updated rows: {}", res.rows_affected()); Ok(()) }