worker/jobs: Inline enqueue_sync_to_index() fn

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(())
-}