worker/jobs: Inline enqueue_sync_to_index() fn

Author: Turbo87

src/admin/delete_crate.rs

@@ -97,8 +97,11 @@ pub async fn run(opts: Opts) -> anyhow::Result<()> {
             };
 
             info!("{name}: Enqueuing index sync jobs…");
-            if let Err(error) = jobs::enqueue_sync_to_index(name, conn) {
-                warn!("{name}: Failed to enqueue index sync jobs: {error}");
+            if let Err(error) = jobs::SyncToGitIndex::new(name).enqueue(conn) {
+                warn!("{name}: Failed to enqueue SyncToGitIndex job: {error}");
+            }
+            if let Err(error) = jobs::SyncToSparseIndex::new(name).enqueue(conn) {
+                warn!("{name}: Failed to enqueue SyncToSparseIndex job: {error}");
             }
 
             info!("{name}: Enqueuing DeleteCrateFromStorage job…");

src/admin/delete_version.rs

@@ -5,6 +5,7 @@ use crate::tasks::spawn_blocking;
 use crate::worker::jobs;
 use crate::{admin::dialoguer, db, schema::versions};
 use anyhow::Context;
+use crates_io_worker::BackgroundJob;
 use diesel::{Connection, ExpressionMethods, QueryDsl};
 use diesel_async::async_connection_wrapper::AsyncConnectionWrapper;
 
@@ -102,8 +103,11 @@ pub async fn run(opts: Opts) -> anyhow::Result<()> {
         })?;
 
         info!(%crate_name, "Enqueuing index sync jobs");
-        if let Err(error) = jobs::enqueue_sync_to_index(crate_name, conn) {
-            warn!(%crate_name, ?error, "Failed to enqueue index sync jobs");
+        if let Err(error) = jobs::SyncToGitIndex::new(crate_name).enqueue(conn) {
+            warn!(%crate_name, ?error, "Failed to enqueue SyncToGitIndex job");
+        }
+        if let Err(error) = jobs::SyncToSparseIndex::new(crate_name).enqueue(conn) {
+            warn!(%crate_name, ?error, "Failed to enqueue SyncToSparseIndex job");
         }
 
         Ok(opts)

src/admin/yank_version.rs

@@ -3,8 +3,7 @@ use crate::db;
 use crate::models::{Crate, Version};
 use crate::schema::versions;
 use crate::tasks::spawn_blocking;
-use crate::worker::jobs;
-use crate::worker::jobs::UpdateDefaultVersion;
+use crate::worker::jobs::{SyncToGitIndex, SyncToSparseIndex, UpdateDefaultVersion};
 use crates_io_worker::BackgroundJob;
 use diesel::prelude::*;
 
@@ -63,8 +62,8 @@ fn yank(opts: Opts, conn: &mut PgConnection) -> anyhow::Result<()> {
         .set(versions::yanked.eq(true))
         .execute(conn)?;
 
-    jobs::enqueue_sync_to_index(&krate.name, conn)?;
-
+    SyncToGitIndex::new(&krate.name).enqueue(conn)?;
+    SyncToSparseIndex::new(&krate.name).enqueue(conn)?;
     UpdateDefaultVersion::new(krate.id).enqueue(conn)?;
 
     Ok(())

src/controllers/krate/publish.rs

@@ -442,7 +442,8 @@ pub async fn publish(app: AppState, req: BytesRequest) -> AppResult<Json<GoodCra
                 ))
                 .map_err(|e| internal(format!("failed to upload crate: {e}")))?;
 
-            jobs::enqueue_sync_to_index(&krate.name, conn)?;
+            jobs::SyncToGitIndex::new(&krate.name).enqueue(conn)?;
+            jobs::SyncToSparseIndex::new(&krate.name).enqueue(conn)?;
 
             SendPublishNotificationsJob::new(version.id).enqueue(conn)?;
 

src/controllers/version/metadata.rs

@@ -29,7 +29,7 @@ use crate::tasks::spawn_blocking;
 use crate::util::diesel::Conn;
 use crate::util::errors::{bad_request, custom, version_not_found, AppResult};
 use crate::views::{EncodableDependency, EncodableVersion};
-use crate::worker::jobs::{self, UpdateDefaultVersion};
+use crate::worker::jobs::{SyncToGitIndex, SyncToSparseIndex, UpdateDefaultVersion};
 
 use super::version_and_crate;
 
@@ -233,7 +233,8 @@ pub fn perform_version_yank_update(
     };
     insert_version_owner_action(conn, version.id, user.id, api_token_id, action)?;
 
-    jobs::enqueue_sync_to_index(&krate.name, conn)?;
+    SyncToGitIndex::new(&krate.name).enqueue(conn)?;
+    SyncToSparseIndex::new(&krate.name).enqueue(conn)?;
     UpdateDefaultVersion::new(krate.id).enqueue(conn)?;
 
     Ok(())

src/tests/worker/git.rs

@@ -2,6 +2,7 @@ use crate::models::Crate;
 use crate::tests::builders::PublishBuilder;
 use crate::tests::util::{RequestHelper, TestApp};
 use crate::worker::jobs;
+use crates_io_worker::BackgroundJob;
 use diesel::prelude::*;
 use http::StatusCode;
 
@@ -51,7 +52,8 @@ async fn index_smoke_test() {
         let krate: Crate = assert_ok!(Crate::by_name("serde").first(conn));
         assert_ok!(diesel::delete(crates::table.find(krate.id)).execute(conn));
 
-        assert_ok!(jobs::enqueue_sync_to_index("serde", conn));
+        assert_ok!(jobs::SyncToGitIndex::new("serde").enqueue(conn));
+        assert_ok!(jobs::SyncToSparseIndex::new("serde").enqueue(conn));
     });
 
     app.run_pending_background_jobs().await;

src/worker/jobs/mod.rs

@@ -1,7 +1,3 @@
-use crate::util::diesel::Conn;
-use crates_io_worker::{BackgroundJob, EnqueueError};
-use std::fmt::Display;
-
 mod archive_version_downloads;
 mod daily_db_maintenance;
 mod delete_crate;
@@ -32,23 +28,3 @@ pub use self::send_publish_notifications::SendPublishNotificationsJob;
 pub use self::sync_admins::SyncAdmins;
 pub use self::typosquat::CheckTyposquat;
 pub use self::update_default_version::UpdateDefaultVersion;
-
-/// Enqueue both index sync jobs (git and sparse) for a crate, unless they
-/// already exist in the background job queue.
-///
-/// Note that there are currently no explicit tests for this functionality,
-/// since our test suite only allows us to use a single database connection
-/// and the background worker queue locking only work when using multiple
-/// connections.
-#[instrument(name = "swirl.enqueue", skip_all, fields(message = "sync_to_index", krate = %krate))]
-pub fn enqueue_sync_to_index<T: Display>(
-    krate: T,
-    conn: &mut impl Conn,
-) -> Result<(), EnqueueError> {
-    let krate = krate.to_string();
-
-    SyncToGitIndex::new(krate.clone()).enqueue(conn)?;
-    SyncToSparseIndex::new(krate).enqueue(conn)?;
-
-    Ok(())
-}