Use insert and update on duplicate instead of replace (#421)

geofmureithi · Sep 21, 2024 · d348630 · d348630
1 parent 5d84489
commit d348630
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/packages/apalis-sql/src/mysql.rs b/packages/apalis-sql/src/mysql.rs
@@ -177,13 +177,14 @@ impl<T: DeserializeOwned + Send + Unpin + Job + Sync + 'static> MysqlStorage<T>
  let worker_type = T::NAME;
  let storage_name = std::any::type_name::<Self>();
  let query =
- "REPLACE INTO workers (id, worker_type, storage_name, layers, last_seen) VALUES (?, ?, ?, ?, ?);";
+ "INSERT INTO workers (id, worker_type, storage_name, layers, last_seen) VALUES (?, ?, ?, ?, ?) ON DUPLICATE KEY UPDATE id = ?;";
  sqlx::query(query)
  .bind(worker_id.to_string())
  .bind(worker_type)
  .bind(storage_name)
  .bind(std::any::type_name::<Service>())
  .bind(last_seen)
+ .bind(worker_id.to_string())
  .execute(&mut *tx)
  .await?;
  Ok(())