fix(publish): Block until it is in index

Originally, crates.io would block on publish requests until the publish
was complete, giving `cargo publish` this behavior by extension.  When
crates.io switched to asynchronous publishing, this intermittently broke
people's workflows when publishing multiple crates.  I say interittent
because it usually works until it doesn't and it is unclear why to the
end user because it will be published by the time they check.  In the
end, callers tend to either put in timeouts (and pray), poll the
server's API, or use `crates-index` crate to poll the index.

This isn't sufficient because
- For any new interested party, this is a pit of failure they'll fall
  into
- crates-index has re-implemented index support incorrectly in the past,
  currently doesn't handle auth, doesn't support `git-cli`, etc.
- None of these previous options work if we were to implement
  workspace-publish support (rust-lang#1169)
- The new sparse registry might increase the publish times, making the
  delay easier to hit manually
- The new sparse registry goes through CDNs so checking the server's API
  might not be sufficient
- Once the sparse registry is available, crates-index users will find
  out when the package is ready in git but it might not be ready through
  the sparse registry because of CDNs

So now `cargo` will block until it sees the package in the index.
- This is checking via the index instead of server APIs in case there
  are propagation delays.  This has the side effect of being noisy
  because of all of the "Updating index" messages.
- This is done unconditionally because cargo used to block and that
  didn't seem to be a problem, blocking by default is the less error
  prone case, and there doesn't seem to be enough justification for a
  "don't block" flag.

Fixes rust-lang#9507

Author: epage

crates/cargo-test-support/src/compare.rs

@@ -197,6 +197,7 @@ fn substitute_macros(input: &str) -> String {
         ("[MIGRATING]", "   Migrating"),
         ("[EXECUTABLE]", "  Executable"),
         ("[SKIPPING]", "    Skipping"),
+        ("[WAITING]", "     Waiting"),
     ];
     let mut result = input.to_owned();
     for &(pat, subst) in &macros {

src/cargo/ops/registry.rs

@@ -18,9 +18,13 @@ use termcolor::Color::Green;
 use termcolor::ColorSpec;
 
 use crate::core::dependency::DepKind;
+use crate::core::dependency::Dependency;
 use crate::core::manifest::ManifestMetadata;
+use crate::core::registry::PackageRegistry;
+use crate::core::registry::Registry as _;
 use crate::core::resolver::CliFeatures;
 use crate::core::source::Source;
+use crate::core::QueryKind;
 use crate::core::{Package, SourceId, Workspace};
 use crate::ops;
 use crate::ops::Packages;
@@ -183,6 +187,10 @@ pub fn publish(ws: &Workspace<'_>, opts: &PublishOpts<'_>) -> CargoResult<()> {
         reg_id,
         opts.dry_run,
     )?;
+    if !opts.dry_run {
+        let timeout = std::time::Duration::from_secs(300);
+        wait_for_publish(opts.config, reg_id, pkg, timeout)?;
+    }
 
     Ok(())
 }
@@ -374,6 +382,61 @@ fn transmit(
     Ok(())
 }
 
+fn wait_for_publish(
+    config: &Config,
+    registry_src: SourceId,
+    pkg: &Package,
+    timeout: std::time::Duration,
+) -> CargoResult<()> {
+    let version_req = format!("={}", pkg.version());
+    let query = Dependency::parse(pkg.name(), Some(&version_req), registry_src)?;
+
+    let now = std::time::Instant::now();
+    let sleep_time = std::time::Duration::from_secs(1);
+    let mut logged = false;
+    loop {
+        let registry_description;
+        {
+            let mut registry = PackageRegistry::new(config)?;
+            let _lock = config.acquire_package_cache_lock()?;
+            registry.lock_patches();
+            registry_description = registry.describe_source(registry_src);
+
+            let summaries = loop {
+                // Exact to avoid returning all for path/git
+                match registry.query_vec(&query, QueryKind::Exact) {
+                    std::task::Poll::Ready(res) => {
+                        break res?;
+                    }
+                    std::task::Poll::Pending => registry.block_until_ready()?,
+                }
+            };
+            if !summaries.is_empty() {
+                break;
+            }
+        }
+
+        if timeout < now.elapsed() {
+            config.shell().warn(format!(
+                "timed out waiting for publishing of {} to complete",
+                pkg.name(),
+            ))?;
+            break;
+        }
+
+        if !logged {
+            config.shell().status(
+                "Waiting",
+                format!("on {} to finish publish", registry_description),
+            )?;
+            logged = true;
+        }
+        std::thread::sleep(sleep_time);
+    }
+
+    Ok(())
+}
+
 /// Returns the index and token from the config file for the given registry.
 ///
 /// `registry` is typically the registry specified on the command-line. If

tests/testsuite/publish.rs

@@ -2051,7 +2051,7 @@ error: package ID specification `bar` did not match any packages
 }
 
 #[cargo_test]
-fn delayed_publish_errors() {
+fn wait_for_publish() {
     // Counter for number of tries before the package is "published"
     let arc: Arc<Mutex<u32>> = Arc::new(Mutex::new(0));
     let arc2 = arc.clone();
@@ -2107,13 +2107,16 @@ Use the --token command-line flag to remove this warning.
 See [..]
 [PACKAGING] delay v0.0.1 ([CWD])
 [UPLOADING] delay v0.0.1 ([CWD])
+[UPDATING] `dummy-registry` index
+[WAITING] on registry `dummy-registry` to finish publish
+[UPDATING] `dummy-registry` index
 ",
         )
         .run();
 
-    // Check nothing has touched the responder
+    // Verify the responder has been pinged
     let lock = arc2.lock().unwrap();
-    assert_eq!(*lock, 0);
+    assert_eq!(*lock, 2);
     drop(lock);
 
     let p = project()
@@ -2131,23 +2134,6 @@ See [..]
         .file("src/main.rs", "fn main() {}")
         .build();
 
-    p.cargo("build -Z sparse-registry")
-        .masquerade_as_nightly_cargo(&["sparse-registry"])
-        .with_status(101)
-        .with_stderr(
-            "\
-[UPDATING] [..]
-[ERROR] no matching package named `delay` found
-location searched: registry `crates-io`
-required by package `foo v0.0.1 ([..]/foo)`
-",
-        )
-        .run();
-
-    let lock = arc2.lock().unwrap();
-    assert_eq!(*lock, 1);
-    drop(lock);
-
     p.cargo("build -Z sparse-registry")
         .masquerade_as_nightly_cargo(&["sparse-registry"])
         .with_status(0)
@@ -2158,7 +2144,7 @@ required by package `foo v0.0.1 ([..]/foo)`
 /// the responder twice per cargo invocation. If that ever gets changed
 /// this test will need to be changed accordingly.
 #[cargo_test]
-fn delayed_publish_errors_underscore() {
+fn wait_for_publish_underscore() {
     // Counter for number of tries before the package is "published"
     let arc: Arc<Mutex<u32>> = Arc::new(Mutex::new(0));
     let arc2 = arc.clone();
@@ -2214,13 +2200,17 @@ Use the --token command-line flag to remove this warning.
 See [..]
 [PACKAGING] delay_with_underscore v0.0.1 ([CWD])
 [UPLOADING] delay_with_underscore v0.0.1 ([CWD])
+[UPDATING] `dummy-registry` index
+[WAITING] on registry `dummy-registry` to finish publish
+[UPDATING] `dummy-registry` index
 ",
         )
         .run();
 
-    // Check nothing has touched the responder
+    // Verify the repsponder has been pinged
     let lock = arc2.lock().unwrap();
-    assert_eq!(*lock, 0);
+    // NOTE: package names with - or _ hit the responder twice per cargo invocation
+    assert_eq!(*lock, 3);
     drop(lock);
 
     let p = project()
@@ -2238,24 +2228,6 @@ See [..]
         .file("src/main.rs", "fn main() {}")
         .build();
 
-    p.cargo("build -Z sparse-registry")
-        .masquerade_as_nightly_cargo(&["sparse-registry"])
-        .with_status(101)
-        .with_stderr(
-            "\
-[UPDATING] [..]
-[ERROR] no matching package named `delay_with_underscore` found
-location searched: registry `crates-io`
-required by package `foo v0.0.1 ([..]/foo)`
-",
-        )
-        .run();
-
-    let lock = arc2.lock().unwrap();
-    // package names with - or _ hit the responder twice per cargo invocation
-    assert_eq!(*lock, 2);
-    drop(lock);
-
     p.cargo("build -Z sparse-registry")
         .masquerade_as_nightly_cargo(&["sparse-registry"])
         .with_status(0)