| //! This module implements the job queue which determines the ordering in which |
| //! rustc is spawned off. It also manages the allocation of jobserver tokens to |
| //! rustc beyond the implicit token each rustc owns (i.e., the ones used for |
| //! parallel LLVM work and parallel rustc threads). |
| //! |
| //! Cargo and rustc have a somewhat non-trivial jobserver relationship with each |
| //! other, which is due to scaling issues with sharing a single jobserver |
| //! amongst what is potentially hundreds of threads of work on many-cored |
| //! systems on (at least) linux, and likely other platforms as well. |
| //! |
| //! The details of this algorithm are (also) written out in |
| //! src/librustc_jobserver/lib.rs. What follows is a description focusing on the |
| //! Cargo side of things. |
| //! |
| //! Cargo wants to complete the build as quickly as possible, fully saturating |
| //! all cores (as constrained by the -j=N) parameter. Cargo also must not spawn |
| //! more than N threads of work: the total amount of tokens we have floating |
| //! around must always be limited to N. |
| //! |
| //! It is not really possible to optimally choose which crate should build first |
| //! or last; nor is it possible to decide whether to give an additional token to |
| //! rustc first or rather spawn a new crate of work. For now, the algorithm we |
| //! implement prioritizes spawning as many crates (i.e., rustc processes) as |
| //! possible, and then filling each rustc with tokens on demand. |
| //! |
| //! The primary loop is in `drain_the_queue` below. |
| //! |
| //! We integrate with the jobserver, originating from GNU make, to make sure |
| //! that build scripts which use make to build C code can cooperate with us on |
| //! the number of used tokens and avoid overfilling the system we're on. |
| //! |
| //! The jobserver is unfortunately a very simple protocol, so we enhance it a |
| //! little when we know that there is a rustc on the other end. Via the stderr |
| //! pipe we have to rustc, we get messages such as "NeedsToken" and |
| //! "ReleaseToken" from rustc. |
| //! |
| //! "NeedsToken" indicates that a rustc is interested in acquiring a token, but |
| //! never that it would be impossible to make progress without one (i.e., it |
| //! would be incorrect for rustc to not terminate due to an unfulfilled |
| //! NeedsToken request); we do not usually fulfill all NeedsToken requests for a |
| //! given rustc. |
| //! |
| //! "ReleaseToken" indicates that a rustc is done with one of its tokens and is |
| //! ready for us to re-acquire ownership -- we will either release that token |
| //! back into the general pool or reuse it ourselves. Note that rustc will |
| //! inform us that it is releasing a token even if it itself is also requesting |
| //! tokens; is is up to us whether to return the token to that same rustc. |
| //! |
| //! The current scheduling algorithm is relatively primitive and could likely be |
| //! improved. |
| |
| use std::cell::{Cell, RefCell}; |
| use std::collections::{BTreeMap, HashMap, HashSet}; |
| use std::fmt::Write as _; |
| use std::io; |
| use std::marker; |
| use std::path::{Path, PathBuf}; |
| use std::sync::Arc; |
| use std::thread::{self, Scope}; |
| use std::time::Duration; |
| |
| use anyhow::{format_err, Context as _}; |
| use cargo_util::ProcessBuilder; |
| use jobserver::{Acquired, Client, HelperThread}; |
| use log::{debug, trace}; |
| use semver::Version; |
| |
| use super::context::OutputFile; |
| use super::job::{ |
| Freshness::{self, Dirty, Fresh}, |
| Job, |
| }; |
| use super::timings::Timings; |
| use super::{BuildContext, BuildPlan, CompileMode, Context, Unit}; |
| use crate::core::compiler::future_incompat::{ |
| self, FutureBreakageItem, FutureIncompatReportPackage, |
| }; |
| use crate::core::resolver::ResolveBehavior; |
| use crate::core::{PackageId, Shell, TargetKind}; |
| use crate::util::diagnostic_server::{self, DiagnosticPrinter}; |
| use crate::util::errors::AlreadyPrintedError; |
| use crate::util::machine_message::{self, Message as _}; |
| use crate::util::CargoResult; |
| use crate::util::{self, internal, profile}; |
| use crate::util::{Config, DependencyQueue, Progress, ProgressStyle, Queue}; |
| |
| /// This structure is backed by the `DependencyQueue` type and manages the |
| /// queueing of compilation steps for each package. Packages enqueue units of |
| /// work and then later on the entire graph is converted to DrainState and |
| /// executed. |
| pub struct JobQueue<'cfg> { |
| queue: DependencyQueue<Unit, Artifact, Job>, |
| counts: HashMap<PackageId, usize>, |
| timings: Timings<'cfg>, |
| } |
| |
| /// This structure is backed by the `DependencyQueue` type and manages the |
| /// actual compilation step of each package. Packages enqueue units of work and |
| /// then later on the entire graph is processed and compiled. |
| /// |
| /// It is created from JobQueue when we have fully assembled the crate graph |
| /// (i.e., all package dependencies are known). |
| /// |
| /// # Message queue |
| /// |
| /// Each thread running a process uses the message queue to send messages back |
| /// to the main thread. The main thread coordinates everything, and handles |
| /// printing output. |
| /// |
| /// It is important to be careful which messages use `push` vs `push_bounded`. |
| /// `push` is for priority messages (like tokens, or "finished") where the |
| /// sender shouldn't block. We want to handle those so real work can proceed |
| /// ASAP. |
| /// |
| /// `push_bounded` is only for messages being printed to stdout/stderr. Being |
| /// bounded prevents a flood of messages causing a large amount of memory |
| /// being used. |
| /// |
| /// `push` also avoids blocking which helps avoid deadlocks. For example, when |
| /// the diagnostic server thread is dropped, it waits for the thread to exit. |
| /// But if the thread is blocked on a full queue, and there is a critical |
| /// error, the drop will deadlock. This should be fixed at some point in the |
| /// future. The jobserver thread has a similar problem, though it will time |
| /// out after 1 second. |
| struct DrainState<'cfg> { |
| // This is the length of the DependencyQueue when starting out |
| total_units: usize, |
| |
| queue: DependencyQueue<Unit, Artifact, Job>, |
| messages: Arc<Queue<Message>>, |
| /// Diagnostic deduplication support. |
| diag_dedupe: DiagDedupe<'cfg>, |
| /// Count of warnings, used to print a summary after the job succeeds |
| warning_count: HashMap<JobId, WarningCount>, |
| active: HashMap<JobId, Unit>, |
| compiled: HashSet<PackageId>, |
| documented: HashSet<PackageId>, |
| scraped: HashSet<PackageId>, |
| counts: HashMap<PackageId, usize>, |
| progress: Progress<'cfg>, |
| next_id: u32, |
| timings: Timings<'cfg>, |
| |
| /// Tokens that are currently owned by this Cargo, and may be "associated" |
| /// with a rustc process. They may also be unused, though if so will be |
| /// dropped on the next loop iteration. |
| /// |
| /// Note that the length of this may be zero, but we will still spawn work, |
| /// as we share the implicit token given to this Cargo process with a |
| /// single rustc process. |
| tokens: Vec<Acquired>, |
| |
| /// rustc per-thread tokens, when in jobserver-per-rustc mode. |
| rustc_tokens: HashMap<JobId, Vec<Acquired>>, |
| |
| /// This represents the list of rustc jobs (processes) and associated |
| /// clients that are interested in receiving a token. |
| to_send_clients: BTreeMap<JobId, Vec<Client>>, |
| |
| /// The list of jobs that we have not yet started executing, but have |
| /// retrieved from the `queue`. We eagerly pull jobs off the main queue to |
| /// allow us to request jobserver tokens pretty early. |
| pending_queue: Vec<(Unit, Job, usize)>, |
| print: DiagnosticPrinter<'cfg>, |
| |
| /// How many jobs we've finished |
| finished: usize, |
| per_package_future_incompat_reports: Vec<FutureIncompatReportPackage>, |
| } |
| |
| /// Count of warnings, used to print a summary after the job succeeds |
| #[derive(Default)] |
| pub struct WarningCount { |
| /// total number of warnings |
| pub total: usize, |
| /// number of warnings that were suppressed because they |
| /// were duplicates of a previous warning |
| pub duplicates: usize, |
| /// number of fixable warnings set to `NotAllowed` |
| /// if any errors have been seen ofr the current |
| /// target |
| pub fixable: FixableWarnings, |
| } |
| |
| impl WarningCount { |
| /// If an error is seen this should be called |
| /// to set `fixable` to `NotAllowed` |
| fn disallow_fixable(&mut self) { |
| self.fixable = FixableWarnings::NotAllowed; |
| } |
| |
| /// Checks fixable if warnings are allowed |
| /// fixable warnings are allowed if no |
| /// errors have been seen for the current |
| /// target. If an error was seen `fixable` |
| /// will be `NotAllowed`. |
| fn fixable_allowed(&self) -> bool { |
| match &self.fixable { |
| FixableWarnings::NotAllowed => false, |
| _ => true, |
| } |
| } |
| } |
| |
| /// Used to keep track of how many fixable warnings there are |
| /// and if fixable warnings are allowed |
| #[derive(Default)] |
| pub enum FixableWarnings { |
| NotAllowed, |
| #[default] |
| Zero, |
| Positive(usize), |
| } |
| |
| pub struct ErrorsDuringDrain { |
| pub count: usize, |
| } |
| |
| struct ErrorToHandle { |
| error: anyhow::Error, |
| |
| /// This field is true for "interesting" errors and false for "mundane" |
| /// errors. If false, we print the above error only if it's the first one |
| /// encountered so far while draining the job queue. |
| /// |
| /// At most places that an error is propagated, we set this to false to |
| /// avoid scenarios where Cargo might end up spewing tons of redundant error |
| /// messages. For example if an i/o stream got closed somewhere, we don't |
| /// care about individually reporting every thread that it broke; just the |
| /// first is enough. |
| /// |
| /// The exception where print_always is true is that we do report every |
| /// instance of a rustc invocation that failed with diagnostics. This |
| /// corresponds to errors from Message::Finish. |
| print_always: bool, |
| } |
| |
| impl<E> From<E> for ErrorToHandle |
| where |
| anyhow::Error: From<E>, |
| { |
| fn from(error: E) -> Self { |
| ErrorToHandle { |
| error: anyhow::Error::from(error), |
| print_always: false, |
| } |
| } |
| } |
| |
| #[derive(Debug, Copy, Clone, PartialEq, Eq, Hash, PartialOrd, Ord)] |
| pub struct JobId(pub u32); |
| |
| impl std::fmt::Display for JobId { |
| fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result { |
| write!(f, "{}", self.0) |
| } |
| } |
| |
| /// A `JobState` is constructed by `JobQueue::run` and passed to `Job::run`. It includes everything |
| /// necessary to communicate between the main thread and the execution of the job. |
| /// |
| /// The job may execute on either a dedicated thread or the main thread. If the job executes on the |
| /// main thread, the `output` field must be set to prevent a deadlock. |
| pub struct JobState<'a, 'cfg> { |
| /// Channel back to the main thread to coordinate messages and such. |
| /// |
| /// When the `output` field is `Some`, care must be taken to avoid calling `push_bounded` on |
| /// the message queue to prevent a deadlock. |
| messages: Arc<Queue<Message>>, |
| |
| /// Normally output is sent to the job queue with backpressure. When the job is fresh |
| /// however we need to immediately display the output to prevent a deadlock as the |
| /// output messages are processed on the same thread as they are sent from. `output` |
| /// defines where to output in this case. |
| /// |
| /// Currently the `Shell` inside `Config` is wrapped in a `RefCell` and thus can't be passed |
| /// between threads. This means that it isn't possible for multiple output messages to be |
| /// interleaved. In the future, it may be wrapped in a `Mutex` instead. In this case |
| /// interleaving is still prevented as the lock would be held for the whole printing of an |
| /// output message. |
| output: Option<&'a DiagDedupe<'cfg>>, |
| |
| /// The job id that this state is associated with, used when sending |
| /// messages back to the main thread. |
| id: JobId, |
| |
| /// Whether or not we're expected to have a call to `rmeta_produced`. Once |
| /// that method is called this is dynamically set to `false` to prevent |
| /// sending a double message later on. |
| rmeta_required: Cell<bool>, |
| |
| // Historical versions of Cargo made use of the `'a` argument here, so to |
| // leave the door open to future refactorings keep it here. |
| _marker: marker::PhantomData<&'a ()>, |
| } |
| |
| /// Handler for deduplicating diagnostics. |
| struct DiagDedupe<'cfg> { |
| seen: RefCell<HashSet<u64>>, |
| config: &'cfg Config, |
| } |
| |
| impl<'cfg> DiagDedupe<'cfg> { |
| fn new(config: &'cfg Config) -> Self { |
| DiagDedupe { |
| seen: RefCell::new(HashSet::new()), |
| config, |
| } |
| } |
| |
| /// Emits a diagnostic message. |
| /// |
| /// Returns `true` if the message was emitted, or `false` if it was |
| /// suppressed for being a duplicate. |
| fn emit_diag(&self, diag: &str) -> CargoResult<bool> { |
| let h = util::hash_u64(diag); |
| if !self.seen.borrow_mut().insert(h) { |
| return Ok(false); |
| } |
| let mut shell = self.config.shell(); |
| shell.print_ansi_stderr(diag.as_bytes())?; |
| shell.err().write_all(b"\n")?; |
| Ok(true) |
| } |
| } |
| |
| /// Possible artifacts that can be produced by compilations, used as edge values |
| /// in the dependency graph. |
| /// |
| /// As edge values we can have multiple kinds of edges depending on one node, |
| /// for example some units may only depend on the metadata for an rlib while |
| /// others depend on the full rlib. This `Artifact` enum is used to distinguish |
| /// this case and track the progress of compilations as they proceed. |
| #[derive(Copy, Clone, Eq, PartialEq, Hash, Debug)] |
| enum Artifact { |
| /// A generic placeholder for "depends on everything run by a step" and |
| /// means that we can't start the next compilation until the previous has |
| /// finished entirely. |
| All, |
| |
| /// A node indicating that we only depend on the metadata of a compilation, |
| /// but the compilation is typically also producing an rlib. We can start |
| /// our step, however, before the full rlib is available. |
| Metadata, |
| } |
| |
| enum Message { |
| Run(JobId, String), |
| BuildPlanMsg(String, ProcessBuilder, Arc<Vec<OutputFile>>), |
| Stdout(String), |
| Stderr(String), |
| |
| // This is for general stderr output from subprocesses |
| Diagnostic { |
| id: JobId, |
| level: String, |
| diag: String, |
| fixable: bool, |
| }, |
| // This handles duplicate output that is suppressed, for showing |
| // only a count of duplicate messages instead |
| WarningCount { |
| id: JobId, |
| emitted: bool, |
| fixable: bool, |
| }, |
| // This is for warnings generated by Cargo's interpretation of the |
| // subprocess output, e.g. scrape-examples prints a warning if a |
| // unit fails to be scraped |
| Warning { |
| id: JobId, |
| warning: String, |
| }, |
| |
| FixDiagnostic(diagnostic_server::Message), |
| Token(io::Result<Acquired>), |
| Finish(JobId, Artifact, CargoResult<()>), |
| FutureIncompatReport(JobId, Vec<FutureBreakageItem>), |
| |
| // This client should get release_raw called on it with one of our tokens |
| NeedsToken(JobId), |
| |
| // A token previously passed to a NeedsToken client is being released. |
| ReleaseToken(JobId), |
| } |
| |
| impl<'a, 'cfg> JobState<'a, 'cfg> { |
| pub fn running(&self, cmd: &ProcessBuilder) { |
| self.messages.push(Message::Run(self.id, cmd.to_string())); |
| } |
| |
| pub fn build_plan( |
| &self, |
| module_name: String, |
| cmd: ProcessBuilder, |
| filenames: Arc<Vec<OutputFile>>, |
| ) { |
| self.messages |
| .push(Message::BuildPlanMsg(module_name, cmd, filenames)); |
| } |
| |
| pub fn stdout(&self, stdout: String) -> CargoResult<()> { |
| if let Some(dedupe) = self.output { |
| writeln!(dedupe.config.shell().out(), "{}", stdout)?; |
| } else { |
| self.messages.push_bounded(Message::Stdout(stdout)); |
| } |
| Ok(()) |
| } |
| |
| pub fn stderr(&self, stderr: String) -> CargoResult<()> { |
| if let Some(dedupe) = self.output { |
| let mut shell = dedupe.config.shell(); |
| shell.print_ansi_stderr(stderr.as_bytes())?; |
| shell.err().write_all(b"\n")?; |
| } else { |
| self.messages.push_bounded(Message::Stderr(stderr)); |
| } |
| Ok(()) |
| } |
| |
| /// See [`Message::Diagnostic`] and [`Message::WarningCount`]. |
| pub fn emit_diag(&self, level: String, diag: String, fixable: bool) -> CargoResult<()> { |
| if let Some(dedupe) = self.output { |
| let emitted = dedupe.emit_diag(&diag)?; |
| if level == "warning" { |
| self.messages.push(Message::WarningCount { |
| id: self.id, |
| emitted, |
| fixable, |
| }); |
| } |
| } else { |
| self.messages.push_bounded(Message::Diagnostic { |
| id: self.id, |
| level, |
| diag, |
| fixable, |
| }); |
| } |
| Ok(()) |
| } |
| |
| /// See [`Message::Warning`]. |
| pub fn warning(&self, warning: String) -> CargoResult<()> { |
| self.messages.push_bounded(Message::Warning { |
| id: self.id, |
| warning, |
| }); |
| Ok(()) |
| } |
| |
| /// A method used to signal to the coordinator thread that the rmeta file |
| /// for an rlib has been produced. This is only called for some rmeta |
| /// builds when required, and can be called at any time before a job ends. |
| /// This should only be called once because a metadata file can only be |
| /// produced once! |
| pub fn rmeta_produced(&self) { |
| self.rmeta_required.set(false); |
| self.messages |
| .push(Message::Finish(self.id, Artifact::Metadata, Ok(()))); |
| } |
| |
| pub fn future_incompat_report(&self, report: Vec<FutureBreakageItem>) { |
| self.messages |
| .push(Message::FutureIncompatReport(self.id, report)); |
| } |
| |
| /// The rustc underlying this Job is about to acquire a jobserver token (i.e., block) |
| /// on the passed client. |
| /// |
| /// This should arrange for the associated client to eventually get a token via |
| /// `client.release_raw()`. |
| pub fn will_acquire(&self) { |
| self.messages.push(Message::NeedsToken(self.id)); |
| } |
| |
| /// The rustc underlying this Job is informing us that it is done with a jobserver token. |
| /// |
| /// Note that it does *not* write that token back anywhere. |
| pub fn release_token(&self) { |
| self.messages.push(Message::ReleaseToken(self.id)); |
| } |
| } |
| |
| impl<'cfg> JobQueue<'cfg> { |
| pub fn new(bcx: &BuildContext<'_, 'cfg>) -> JobQueue<'cfg> { |
| JobQueue { |
| queue: DependencyQueue::new(), |
| counts: HashMap::new(), |
| timings: Timings::new(bcx, &bcx.roots), |
| } |
| } |
| |
| pub fn enqueue(&mut self, cx: &Context<'_, 'cfg>, unit: &Unit, job: Job) -> CargoResult<()> { |
| let dependencies = cx.unit_deps(unit); |
| let mut queue_deps = dependencies |
| .iter() |
| .filter(|dep| { |
| // Binaries aren't actually needed to *compile* tests, just to run |
| // them, so we don't include this dependency edge in the job graph. |
| // But we shouldn't filter out dependencies being scraped for Rustdoc. |
| (!dep.unit.target.is_test() && !dep.unit.target.is_bin()) |
| || dep.unit.artifact.is_true() |
| || dep.unit.mode.is_doc_scrape() |
| }) |
| .map(|dep| { |
| // Handle the case here where our `unit -> dep` dependency may |
| // only require the metadata, not the full compilation to |
| // finish. Use the tables in `cx` to figure out what kind |
| // of artifact is associated with this dependency. |
| let artifact = if cx.only_requires_rmeta(unit, &dep.unit) { |
| Artifact::Metadata |
| } else { |
| Artifact::All |
| }; |
| (dep.unit.clone(), artifact) |
| }) |
| .collect::<HashMap<_, _>>(); |
| |
| // This is somewhat tricky, but we may need to synthesize some |
| // dependencies for this target if it requires full upstream |
| // compilations to have completed. Because of pipelining, some |
| // dependency edges may be `Metadata` due to the above clause (as |
| // opposed to everything being `All`). For example consider: |
| // |
| // a (binary) |
| // â”” b (lib) |
| // â”” c (lib) |
| // |
| // Here the dependency edge from B to C will be `Metadata`, and the |
| // dependency edge from A to B will be `All`. For A to be compiled, |
| // however, it currently actually needs the full rlib of C. This means |
| // that we need to synthesize a dependency edge for the dependency graph |
| // from A to C. That's done here. |
| // |
| // This will walk all dependencies of the current target, and if any of |
| // *their* dependencies are `Metadata` then we depend on the `All` of |
| // the target as well. This should ensure that edges changed to |
| // `Metadata` propagate upwards `All` dependencies to anything that |
| // transitively contains the `Metadata` edge. |
| if unit.requires_upstream_objects() { |
| for dep in dependencies { |
| depend_on_deps_of_deps(cx, &mut queue_deps, dep.unit.clone()); |
| } |
| |
| fn depend_on_deps_of_deps( |
| cx: &Context<'_, '_>, |
| deps: &mut HashMap<Unit, Artifact>, |
| unit: Unit, |
| ) { |
| for dep in cx.unit_deps(&unit) { |
| if deps.insert(dep.unit.clone(), Artifact::All).is_none() { |
| depend_on_deps_of_deps(cx, deps, dep.unit.clone()); |
| } |
| } |
| } |
| } |
| |
| // For now we use a fixed placeholder value for the cost of each unit, but |
| // in the future this could be used to allow users to provide hints about |
| // relative expected costs of units, or this could be automatically set in |
| // a smarter way using timing data from a previous compilation. |
| self.queue.queue(unit.clone(), job, queue_deps, 100); |
| *self.counts.entry(unit.pkg.package_id()).or_insert(0) += 1; |
| Ok(()) |
| } |
| |
| /// Executes all jobs necessary to build the dependency graph. |
| /// |
| /// This function will spawn off `config.jobs()` workers to build all of the |
| /// necessary dependencies, in order. Freshness is propagated as far as |
| /// possible along each dependency chain. |
| pub fn execute(mut self, cx: &mut Context<'_, '_>, plan: &mut BuildPlan) -> CargoResult<()> { |
| let _p = profile::start("executing the job graph"); |
| self.queue.queue_finished(); |
| |
| let progress = Progress::with_style("Building", ProgressStyle::Ratio, cx.bcx.config); |
| let state = DrainState { |
| total_units: self.queue.len(), |
| queue: self.queue, |
| // 100 here is somewhat arbitrary. It is a few screenfulls of |
| // output, and hopefully at most a few megabytes of memory for |
| // typical messages. If you change this, please update the test |
| // caching_large_output, too. |
| messages: Arc::new(Queue::new(100)), |
| diag_dedupe: DiagDedupe::new(cx.bcx.config), |
| warning_count: HashMap::new(), |
| active: HashMap::new(), |
| compiled: HashSet::new(), |
| documented: HashSet::new(), |
| scraped: HashSet::new(), |
| counts: self.counts, |
| progress, |
| next_id: 0, |
| timings: self.timings, |
| tokens: Vec::new(), |
| rustc_tokens: HashMap::new(), |
| to_send_clients: BTreeMap::new(), |
| pending_queue: Vec::new(), |
| print: DiagnosticPrinter::new(cx.bcx.config), |
| finished: 0, |
| per_package_future_incompat_reports: Vec::new(), |
| }; |
| |
| // Create a helper thread for acquiring jobserver tokens |
| let messages = state.messages.clone(); |
| let helper = cx |
| .jobserver |
| .clone() |
| .into_helper_thread(move |token| { |
| messages.push(Message::Token(token)); |
| }) |
| .with_context(|| "failed to create helper thread for jobserver management")?; |
| |
| // Create a helper thread to manage the diagnostics for rustfix if |
| // necessary. |
| let messages = state.messages.clone(); |
| // It is important that this uses `push` instead of `push_bounded` for |
| // now. If someone wants to fix this to be bounded, the `drop` |
| // implementation needs to be changed to avoid possible deadlocks. |
| let _diagnostic_server = cx |
| .bcx |
| .build_config |
| .rustfix_diagnostic_server |
| .borrow_mut() |
| .take() |
| .map(move |srv| srv.start(move |msg| messages.push(Message::FixDiagnostic(msg)))); |
| |
| thread::scope( |
| move |scope| match state.drain_the_queue(cx, plan, scope, &helper) { |
| Some(err) => Err(err), |
| None => Ok(()), |
| }, |
| ) |
| } |
| } |
| |
| impl<'cfg> DrainState<'cfg> { |
| fn spawn_work_if_possible<'s>( |
| &mut self, |
| cx: &mut Context<'_, '_>, |
| jobserver_helper: &HelperThread, |
| scope: &'s Scope<'s, '_>, |
| ) -> CargoResult<()> { |
| // Dequeue as much work as we can, learning about everything |
| // possible that can run. Note that this is also the point where we |
| // start requesting job tokens. Each job after the first needs to |
| // request a token. |
| while let Some((unit, job, priority)) = self.queue.dequeue() { |
| // We want to keep the pieces of work in the `pending_queue` sorted |
| // by their priorities, and insert the current job at its correctly |
| // sorted position: following the lower priority jobs, and the ones |
| // with the same priority (since they were dequeued before the |
| // current one, we also keep that relation). |
| let idx = self |
| .pending_queue |
| .partition_point(|&(_, _, p)| p <= priority); |
| self.pending_queue.insert(idx, (unit, job, priority)); |
| if self.active.len() + self.pending_queue.len() > 1 { |
| jobserver_helper.request_token(); |
| } |
| } |
| |
| // Now that we've learned of all possible work that we can execute |
| // try to spawn it so long as we've got a jobserver token which says |
| // we're able to perform some parallel work. |
| // The `pending_queue` is sorted in ascending priority order, and we |
| // remove items from its end to schedule the highest priority items |
| // sooner. |
| while self.has_extra_tokens() && !self.pending_queue.is_empty() { |
| let (unit, job, _) = self.pending_queue.pop().unwrap(); |
| *self.counts.get_mut(&unit.pkg.package_id()).unwrap() -= 1; |
| if !cx.bcx.build_config.build_plan { |
| // Print out some nice progress information. |
| // NOTE: An error here will drop the job without starting it. |
| // That should be OK, since we want to exit as soon as |
| // possible during an error. |
| self.note_working_on(cx.bcx.config, cx.bcx.ws.root(), &unit, job.freshness())?; |
| } |
| self.run(&unit, job, cx, scope); |
| } |
| |
| Ok(()) |
| } |
| |
| fn has_extra_tokens(&self) -> bool { |
| self.active.len() < self.tokens.len() + 1 |
| } |
| |
| // The oldest job (i.e., least job ID) is the one we grant tokens to first. |
| fn pop_waiting_client(&mut self) -> (JobId, Client) { |
| // FIXME: replace this with BTreeMap::first_entry when that stabilizes. |
| let key = *self |
| .to_send_clients |
| .keys() |
| .next() |
| .expect("at least one waiter"); |
| let clients = self.to_send_clients.get_mut(&key).unwrap(); |
| let client = clients.pop().unwrap(); |
| if clients.is_empty() { |
| self.to_send_clients.remove(&key); |
| } |
| (key, client) |
| } |
| |
| // If we managed to acquire some extra tokens, send them off to a waiting rustc. |
| fn grant_rustc_token_requests(&mut self) -> CargoResult<()> { |
| while !self.to_send_clients.is_empty() && self.has_extra_tokens() { |
| let (id, client) = self.pop_waiting_client(); |
| // This unwrap is guaranteed to succeed. `active` must be at least |
| // length 1, as otherwise there can't be a client waiting to be sent |
| // on, so tokens.len() must also be at least one. |
| let token = self.tokens.pop().unwrap(); |
| self.rustc_tokens |
| .entry(id) |
| .or_insert_with(Vec::new) |
| .push(token); |
| client |
| .release_raw() |
| .with_context(|| "failed to release jobserver token")?; |
| } |
| |
| Ok(()) |
| } |
| |
| fn handle_event( |
| &mut self, |
| cx: &mut Context<'_, '_>, |
| jobserver_helper: &HelperThread, |
| plan: &mut BuildPlan, |
| event: Message, |
| ) -> Result<(), ErrorToHandle> { |
| match event { |
| Message::Run(id, cmd) => { |
| cx.bcx |
| .config |
| .shell() |
| .verbose(|c| c.status("Running", &cmd))?; |
| self.timings.unit_start(id, self.active[&id].clone()); |
| } |
| Message::BuildPlanMsg(module_name, cmd, filenames) => { |
| plan.update(&module_name, &cmd, &filenames)?; |
| } |
| Message::Stdout(out) => { |
| writeln!(cx.bcx.config.shell().out(), "{}", out)?; |
| } |
| Message::Stderr(err) => { |
| let mut shell = cx.bcx.config.shell(); |
| shell.print_ansi_stderr(err.as_bytes())?; |
| shell.err().write_all(b"\n")?; |
| } |
| Message::Diagnostic { |
| id, |
| level, |
| diag, |
| fixable, |
| } => { |
| let emitted = self.diag_dedupe.emit_diag(&diag)?; |
| if level == "warning" { |
| self.bump_warning_count(id, emitted, fixable); |
| } |
| if level == "error" { |
| let cnts = self.warning_count.entry(id).or_default(); |
| // If there is an error, the `cargo fix` message should not show |
| cnts.disallow_fixable(); |
| } |
| } |
| Message::Warning { id, warning } => { |
| cx.bcx.config.shell().warn(warning)?; |
| self.bump_warning_count(id, true, false); |
| } |
| Message::WarningCount { |
| id, |
| emitted, |
| fixable, |
| } => { |
| self.bump_warning_count(id, emitted, fixable); |
| } |
| Message::FixDiagnostic(msg) => { |
| self.print.print(&msg)?; |
| } |
| Message::Finish(id, artifact, result) => { |
| let unit = match artifact { |
| // If `id` has completely finished we remove it |
| // from the `active` map ... |
| Artifact::All => { |
| trace!("end: {:?}", id); |
| self.finished += 1; |
| if let Some(rustc_tokens) = self.rustc_tokens.remove(&id) { |
| // This puts back the tokens that this rustc |
| // acquired into our primary token list. |
| // |
| // This represents a rustc bug: it did not |
| // release all of its thread tokens but finished |
| // completely. But we want to make Cargo resilient |
| // to such rustc bugs, as they're generally not |
| // fatal in nature (i.e., Cargo can make progress |
| // still, and the build might not even fail). |
| self.tokens.extend(rustc_tokens); |
| } |
| self.to_send_clients.remove(&id); |
| self.report_warning_count( |
| cx.bcx.config, |
| id, |
| &cx.bcx.rustc().workspace_wrapper, |
| ); |
| self.active.remove(&id).unwrap() |
| } |
| // ... otherwise if it hasn't finished we leave it |
| // in there as we'll get another `Finish` later on. |
| Artifact::Metadata => { |
| trace!("end (meta): {:?}", id); |
| self.active[&id].clone() |
| } |
| }; |
| debug!("end ({:?}): {:?}", unit, result); |
| match result { |
| Ok(()) => self.finish(id, &unit, artifact, cx)?, |
| Err(_) if cx.bcx.unit_can_fail_for_docscraping(&unit) => { |
| cx.failed_scrape_units |
| .lock() |
| .unwrap() |
| .insert(cx.files().metadata(&unit)); |
| self.queue.finish(&unit, &artifact); |
| } |
| Err(error) => { |
| let msg = "The following warnings were emitted during compilation:"; |
| self.emit_warnings(Some(msg), &unit, cx)?; |
| self.back_compat_notice(cx, &unit)?; |
| return Err(ErrorToHandle { |
| error, |
| print_always: true, |
| }); |
| } |
| } |
| } |
| Message::FutureIncompatReport(id, items) => { |
| let package_id = self.active[&id].pkg.package_id(); |
| self.per_package_future_incompat_reports |
| .push(FutureIncompatReportPackage { package_id, items }); |
| } |
| Message::Token(acquired_token) => { |
| let token = acquired_token.with_context(|| "failed to acquire jobserver token")?; |
| self.tokens.push(token); |
| } |
| Message::NeedsToken(id) => { |
| trace!("queue token request"); |
| jobserver_helper.request_token(); |
| let client = cx.rustc_clients[&self.active[&id]].clone(); |
| self.to_send_clients |
| .entry(id) |
| .or_insert_with(Vec::new) |
| .push(client); |
| } |
| Message::ReleaseToken(id) => { |
| // Note that this pops off potentially a completely |
| // different token, but all tokens of the same job are |
| // conceptually the same so that's fine. |
| // |
| // self.tokens is a "pool" -- the order doesn't matter -- and |
| // this transfers ownership of the token into that pool. If we |
| // end up using it on the next go around, then this token will |
| // be truncated, same as tokens obtained through Message::Token. |
| let rustc_tokens = self |
| .rustc_tokens |
| .get_mut(&id) |
| .expect("no tokens associated"); |
| self.tokens |
| .push(rustc_tokens.pop().expect("rustc releases token it has")); |
| } |
| } |
| |
| Ok(()) |
| } |
| |
| // This will also tick the progress bar as appropriate |
| fn wait_for_events(&mut self) -> Vec<Message> { |
| // Drain all events at once to avoid displaying the progress bar |
| // unnecessarily. If there's no events we actually block waiting for |
| // an event, but we keep a "heartbeat" going to allow `record_cpu` |
| // to run above to calculate CPU usage over time. To do this we |
| // listen for a message with a timeout, and on timeout we run the |
| // previous parts of the loop again. |
| let mut events = self.messages.try_pop_all(); |
| trace!( |
| "tokens in use: {}, rustc_tokens: {:?}, waiting_rustcs: {:?} (events this tick: {})", |
| self.tokens.len(), |
| self.rustc_tokens |
| .iter() |
| .map(|(k, j)| (k, j.len())) |
| .collect::<Vec<_>>(), |
| self.to_send_clients |
| .iter() |
| .map(|(k, j)| (k, j.len())) |
| .collect::<Vec<_>>(), |
| events.len(), |
| ); |
| if events.is_empty() { |
| loop { |
| self.tick_progress(); |
| self.tokens.truncate(self.active.len() - 1); |
| match self.messages.pop(Duration::from_millis(500)) { |
| Some(message) => { |
| events.push(message); |
| break; |
| } |
| None => continue, |
| } |
| } |
| } |
| events |
| } |
| |
| /// This is the "main" loop, where Cargo does all work to run the |
| /// compiler. |
| /// |
| /// This returns an Option to prevent the use of `?` on `Result` types |
| /// because it is important for the loop to carefully handle errors. |
| fn drain_the_queue<'s>( |
| mut self, |
| cx: &mut Context<'_, '_>, |
| plan: &mut BuildPlan, |
| scope: &'s Scope<'s, '_>, |
| jobserver_helper: &HelperThread, |
| ) -> Option<anyhow::Error> { |
| trace!("queue: {:#?}", self.queue); |
| |
| // Iteratively execute the entire dependency graph. Each turn of the |
| // loop starts out by scheduling as much work as possible (up to the |
| // maximum number of parallel jobs we have tokens for). A local queue |
| // is maintained separately from the main dependency queue as one |
| // dequeue may actually dequeue quite a bit of work (e.g., 10 binaries |
| // in one package). |
| // |
| // After a job has finished we update our internal state if it was |
| // successful and otherwise wait for pending work to finish if it failed |
| // and then immediately return (or keep going, if requested by the build |
| // config). |
| let mut errors = ErrorsDuringDrain { count: 0 }; |
| // CAUTION! Do not use `?` or break out of the loop early. Every error |
| // must be handled in such a way that the loop is still allowed to |
| // drain event messages. |
| loop { |
| if errors.count == 0 || cx.bcx.build_config.keep_going { |
| if let Err(e) = self.spawn_work_if_possible(cx, jobserver_helper, scope) { |
| self.handle_error(&mut cx.bcx.config.shell(), &mut errors, e); |
| } |
| } |
| |
| // If after all that we're not actually running anything then we're |
| // done! |
| if self.active.is_empty() { |
| break; |
| } |
| |
| if let Err(e) = self.grant_rustc_token_requests() { |
| self.handle_error(&mut cx.bcx.config.shell(), &mut errors, e); |
| } |
| |
| // And finally, before we block waiting for the next event, drop any |
| // excess tokens we may have accidentally acquired. Due to how our |
| // jobserver interface is architected we may acquire a token that we |
| // don't actually use, and if this happens just relinquish it back |
| // to the jobserver itself. |
| for event in self.wait_for_events() { |
| if let Err(event_err) = self.handle_event(cx, jobserver_helper, plan, event) { |
| self.handle_error(&mut cx.bcx.config.shell(), &mut errors, event_err); |
| } |
| } |
| } |
| self.progress.clear(); |
| |
| let profile_name = cx.bcx.build_config.requested_profile; |
| // NOTE: this may be a bit inaccurate, since this may not display the |
| // profile for what was actually built. Profile overrides can change |
| // these settings, and in some cases different targets are built with |
| // different profiles. To be accurate, it would need to collect a |
| // list of Units built, and maybe display a list of the different |
| // profiles used. However, to keep it simple and compatible with old |
| // behavior, we just display what the base profile is. |
| let profile = cx.bcx.profiles.base_profile(); |
| let mut opt_type = String::from(if profile.opt_level.as_str() == "0" { |
| "unoptimized" |
| } else { |
| "optimized" |
| }); |
| if profile.debuginfo.unwrap_or(0) != 0 { |
| opt_type += " + debuginfo"; |
| } |
| |
| let time_elapsed = util::elapsed(cx.bcx.config.creation_time().elapsed()); |
| if let Err(e) = self.timings.finished(cx, &errors.to_error()) { |
| self.handle_error(&mut cx.bcx.config.shell(), &mut errors, e); |
| } |
| if cx.bcx.build_config.emit_json() { |
| let mut shell = cx.bcx.config.shell(); |
| let msg = machine_message::BuildFinished { |
| success: errors.count == 0, |
| } |
| .to_json_string(); |
| if let Err(e) = writeln!(shell.out(), "{}", msg) { |
| self.handle_error(&mut shell, &mut errors, e); |
| } |
| } |
| |
| if let Some(error) = errors.to_error() { |
| // Any errors up to this point have already been printed via the |
| // `display_error` inside `handle_error`. |
| Some(anyhow::Error::new(AlreadyPrintedError::new(error))) |
| } else if self.queue.is_empty() && self.pending_queue.is_empty() { |
| let message = format!( |
| "{} [{}] target(s) in {}", |
| profile_name, opt_type, time_elapsed |
| ); |
| if !cx.bcx.build_config.build_plan { |
| // It doesn't really matter if this fails. |
| drop(cx.bcx.config.shell().status("Finished", message)); |
| future_incompat::save_and_display_report( |
| cx.bcx, |
| &self.per_package_future_incompat_reports, |
| ); |
| } |
| |
| None |
| } else { |
| debug!("queue: {:#?}", self.queue); |
| Some(internal("finished with jobs still left in the queue")) |
| } |
| } |
| |
| fn handle_error( |
| &self, |
| shell: &mut Shell, |
| err_state: &mut ErrorsDuringDrain, |
| new_err: impl Into<ErrorToHandle>, |
| ) { |
| let new_err = new_err.into(); |
| if new_err.print_always || err_state.count == 0 { |
| crate::display_error(&new_err.error, shell); |
| if err_state.count == 0 && !self.active.is_empty() { |
| drop(shell.warn("build failed, waiting for other jobs to finish...")); |
| } |
| err_state.count += 1; |
| } else { |
| log::warn!("{:?}", new_err.error); |
| } |
| } |
| |
| // This also records CPU usage and marks concurrency; we roughly want to do |
| // this as often as we spin on the events receiver (at least every 500ms or |
| // so). |
| fn tick_progress(&mut self) { |
| // Record some timing information if `--timings` is enabled, and |
| // this'll end up being a noop if we're not recording this |
| // information. |
| self.timings.mark_concurrency( |
| self.active.len(), |
| self.pending_queue.len(), |
| self.queue.len(), |
| self.rustc_tokens.len(), |
| ); |
| self.timings.record_cpu(); |
| |
| let active_names = self |
| .active |
| .values() |
| .map(|u| self.name_for_progress(u)) |
| .collect::<Vec<_>>(); |
| drop(self.progress.tick_now( |
| self.finished, |
| self.total_units, |
| &format!(": {}", active_names.join(", ")), |
| )); |
| } |
| |
| fn name_for_progress(&self, unit: &Unit) -> String { |
| let pkg_name = unit.pkg.name(); |
| let target_name = unit.target.name(); |
| match unit.mode { |
| CompileMode::Doc { .. } => format!("{}(doc)", pkg_name), |
| CompileMode::RunCustomBuild => format!("{}(build)", pkg_name), |
| CompileMode::Test | CompileMode::Check { test: true } => match unit.target.kind() { |
| TargetKind::Lib(_) => format!("{}(test)", target_name), |
| TargetKind::CustomBuild => panic!("cannot test build script"), |
| TargetKind::Bin => format!("{}(bin test)", target_name), |
| TargetKind::Test => format!("{}(test)", target_name), |
| TargetKind::Bench => format!("{}(bench)", target_name), |
| TargetKind::ExampleBin | TargetKind::ExampleLib(_) => { |
| format!("{}(example test)", target_name) |
| } |
| }, |
| _ => match unit.target.kind() { |
| TargetKind::Lib(_) => pkg_name.to_string(), |
| TargetKind::CustomBuild => format!("{}(build.rs)", pkg_name), |
| TargetKind::Bin => format!("{}(bin)", target_name), |
| TargetKind::Test => format!("{}(test)", target_name), |
| TargetKind::Bench => format!("{}(bench)", target_name), |
| TargetKind::ExampleBin | TargetKind::ExampleLib(_) => { |
| format!("{}(example)", target_name) |
| } |
| }, |
| } |
| } |
| |
| /// Executes a job. |
| /// |
| /// Fresh jobs block until finished (which should be very fast!), Dirty |
| /// jobs will spawn a thread in the background and return immediately. |
| fn run<'s>(&mut self, unit: &Unit, job: Job, cx: &Context<'_, '_>, scope: &'s Scope<'s, '_>) { |
| let id = JobId(self.next_id); |
| self.next_id = self.next_id.checked_add(1).unwrap(); |
| |
| debug!("start {}: {:?}", id, unit); |
| |
| assert!(self.active.insert(id, unit.clone()).is_none()); |
| |
| let messages = self.messages.clone(); |
| let is_fresh = job.freshness().is_fresh(); |
| let rmeta_required = cx.rmeta_required(unit); |
| |
| let doit = move |state: JobState<'_, '_>| { |
| let mut sender = FinishOnDrop { |
| messages: &state.messages, |
| id, |
| result: None, |
| }; |
| sender.result = Some(job.run(&state)); |
| |
| // If the `rmeta_required` wasn't consumed but it was set |
| // previously, then we either have: |
| // |
| // 1. The `job` didn't do anything because it was "fresh". |
| // 2. The `job` returned an error and didn't reach the point where |
| // it called `rmeta_produced`. |
| // 3. We forgot to call `rmeta_produced` and there's a bug in Cargo. |
| // |
| // Ruling out the third, the other two are pretty common for 2 |
| // we'll just naturally abort the compilation operation but for 1 |
| // we need to make sure that the metadata is flagged as produced so |
| // send a synthetic message here. |
| if state.rmeta_required.get() && sender.result.as_ref().unwrap().is_ok() { |
| state |
| .messages |
| .push(Message::Finish(state.id, Artifact::Metadata, Ok(()))); |
| } |
| |
| // Use a helper struct with a `Drop` implementation to guarantee |
| // that a `Finish` message is sent even if our job panics. We |
| // shouldn't panic unless there's a bug in Cargo, so we just need |
| // to make sure nothing hangs by accident. |
| struct FinishOnDrop<'a> { |
| messages: &'a Queue<Message>, |
| id: JobId, |
| result: Option<CargoResult<()>>, |
| } |
| |
| impl Drop for FinishOnDrop<'_> { |
| fn drop(&mut self) { |
| let result = self |
| .result |
| .take() |
| .unwrap_or_else(|| Err(format_err!("worker panicked"))); |
| self.messages |
| .push(Message::Finish(self.id, Artifact::All, result)); |
| } |
| } |
| }; |
| |
| match is_fresh { |
| true => { |
| self.timings.add_fresh(); |
| // Running a fresh job on the same thread is often much faster than spawning a new |
| // thread to run the job. |
| doit(JobState { |
| id, |
| messages, |
| output: Some(&self.diag_dedupe), |
| rmeta_required: Cell::new(rmeta_required), |
| _marker: marker::PhantomData, |
| }); |
| } |
| false => { |
| self.timings.add_dirty(); |
| scope.spawn(move || { |
| doit(JobState { |
| id, |
| messages: messages.clone(), |
| output: None, |
| rmeta_required: Cell::new(rmeta_required), |
| _marker: marker::PhantomData, |
| }) |
| }); |
| } |
| } |
| } |
| |
| fn emit_warnings( |
| &mut self, |
| msg: Option<&str>, |
| unit: &Unit, |
| cx: &mut Context<'_, '_>, |
| ) -> CargoResult<()> { |
| let outputs = cx.build_script_outputs.lock().unwrap(); |
| let metadata = match cx.find_build_script_metadata(unit) { |
| Some(metadata) => metadata, |
| None => return Ok(()), |
| }; |
| let bcx = &mut cx.bcx; |
| if let Some(output) = outputs.get(metadata) { |
| if !output.warnings.is_empty() { |
| if let Some(msg) = msg { |
| writeln!(bcx.config.shell().err(), "{}\n", msg)?; |
| } |
| |
| for warning in output.warnings.iter() { |
| bcx.config.shell().warn(warning)?; |
| } |
| |
| if msg.is_some() { |
| // Output an empty line. |
| writeln!(bcx.config.shell().err())?; |
| } |
| } |
| } |
| |
| Ok(()) |
| } |
| |
| fn bump_warning_count(&mut self, id: JobId, emitted: bool, fixable: bool) { |
| let cnts = self.warning_count.entry(id).or_default(); |
| cnts.total += 1; |
| if !emitted { |
| cnts.duplicates += 1; |
| // Don't add to fixable if it's already been emitted |
| } else if fixable { |
| // Do not add anything to the fixable warning count if |
| // is `NotAllowed` since that indicates there was an |
| // error while building this `Unit` |
| if cnts.fixable_allowed() { |
| cnts.fixable = match cnts.fixable { |
| FixableWarnings::NotAllowed => FixableWarnings::NotAllowed, |
| FixableWarnings::Zero => FixableWarnings::Positive(1), |
| FixableWarnings::Positive(fixable) => FixableWarnings::Positive(fixable + 1), |
| }; |
| } |
| } |
| } |
| |
| /// Displays a final report of the warnings emitted by a particular job. |
| fn report_warning_count( |
| &mut self, |
| config: &Config, |
| id: JobId, |
| rustc_workspace_wrapper: &Option<PathBuf>, |
| ) { |
| let count = match self.warning_count.remove(&id) { |
| // An error could add an entry for a `Unit` |
| // with 0 warnings but having fixable |
| // warnings be disallowed |
| Some(count) if count.total > 0 => count, |
| None | Some(_) => return, |
| }; |
| let unit = &self.active[&id]; |
| let mut message = format!("`{}` ({}", unit.pkg.name(), unit.target.description_named()); |
| if unit.mode.is_rustc_test() && !(unit.target.is_test() || unit.target.is_bench()) { |
| message.push_str(" test"); |
| } else if unit.mode.is_doc_test() { |
| message.push_str(" doctest"); |
| } else if unit.mode.is_doc() { |
| message.push_str(" doc"); |
| } |
| message.push_str(") generated "); |
| match count.total { |
| 1 => message.push_str("1 warning"), |
| n => drop(write!(message, "{} warnings", n)), |
| }; |
| match count.duplicates { |
| 0 => {} |
| 1 => message.push_str(" (1 duplicate)"), |
| n => drop(write!(message, " ({} duplicates)", n)), |
| } |
| // Only show the `cargo fix` message if its a local `Unit` |
| if unit.is_local() && config.nightly_features_allowed { |
| // Do not show this if there are any errors or no fixable warnings |
| if let FixableWarnings::Positive(fixable) = count.fixable { |
| // `cargo fix` doesnt have an option for custom builds |
| if !unit.target.is_custom_build() { |
| // To make sure the correct command is shown for `clippy` we |
| // check if `RUSTC_WORKSPACE_WRAPPER` is set and pointing towards |
| // `clippy-driver`. |
| let clippy = std::ffi::OsStr::new("clippy-driver"); |
| let command = match rustc_workspace_wrapper.as_ref().and_then(|x| x.file_stem()) |
| { |
| Some(wrapper) if wrapper == clippy => "cargo clippy --fix", |
| _ => "cargo fix", |
| }; |
| let mut args = { |
| let named = unit.target.description_named(); |
| // if its a lib we need to add the package to fix |
| if unit.target.is_lib() { |
| format!("{} -p {}", named, unit.pkg.name()) |
| } else { |
| named |
| } |
| }; |
| if unit.mode.is_rustc_test() |
| && !(unit.target.is_test() || unit.target.is_bench()) |
| { |
| args.push_str(" --tests"); |
| } |
| let mut suggestions = format!("{} suggestion", fixable); |
| if fixable > 1 { |
| suggestions.push_str("s") |
| } |
| drop(write!( |
| message, |
| " (run `{command} --{args}` to apply {suggestions})" |
| )) |
| } |
| } |
| } |
| // Errors are ignored here because it is tricky to handle them |
| // correctly, and they aren't important. |
| drop(config.shell().warn(message)); |
| } |
| |
| fn finish( |
| &mut self, |
| id: JobId, |
| unit: &Unit, |
| artifact: Artifact, |
| cx: &mut Context<'_, '_>, |
| ) -> CargoResult<()> { |
| if unit.mode.is_run_custom_build() && unit.show_warnings(cx.bcx.config) { |
| self.emit_warnings(None, unit, cx)?; |
| } |
| let unlocked = self.queue.finish(unit, &artifact); |
| match artifact { |
| Artifact::All => self.timings.unit_finished(id, unlocked), |
| Artifact::Metadata => self.timings.unit_rmeta_finished(id, unlocked), |
| } |
| Ok(()) |
| } |
| |
| // This isn't super trivial because we don't want to print loads and |
| // loads of information to the console, but we also want to produce a |
| // faithful representation of what's happening. This is somewhat nuanced |
| // as a package can start compiling *very* early on because of custom |
| // build commands and such. |
| // |
| // In general, we try to print "Compiling" for the first nontrivial task |
| // run for a package, regardless of when that is. We then don't print |
| // out any more information for a package after we've printed it once. |
| fn note_working_on( |
| &mut self, |
| config: &Config, |
| ws_root: &Path, |
| unit: &Unit, |
| fresh: &Freshness, |
| ) -> CargoResult<()> { |
| if (self.compiled.contains(&unit.pkg.package_id()) |
| && !unit.mode.is_doc() |
| && !unit.mode.is_doc_scrape()) |
| || (self.documented.contains(&unit.pkg.package_id()) && unit.mode.is_doc()) |
| || (self.scraped.contains(&unit.pkg.package_id()) && unit.mode.is_doc_scrape()) |
| { |
| return Ok(()); |
| } |
| |
| match fresh { |
| // Any dirty stage which runs at least one command gets printed as |
| // being a compiled package. |
| Dirty(dirty_reason) => { |
| if let Some(reason) = dirty_reason { |
| config |
| .shell() |
| .verbose(|shell| reason.present_to(shell, unit, ws_root))?; |
| } |
| |
| if unit.mode.is_doc() { |
| self.documented.insert(unit.pkg.package_id()); |
| config.shell().status("Documenting", &unit.pkg)?; |
| } else if unit.mode.is_doc_test() { |
| // Skip doc test. |
| } else if unit.mode.is_doc_scrape() { |
| self.scraped.insert(unit.pkg.package_id()); |
| config.shell().status("Scraping", &unit.pkg)?; |
| } else { |
| self.compiled.insert(unit.pkg.package_id()); |
| if unit.mode.is_check() { |
| config.shell().status("Checking", &unit.pkg)?; |
| } else { |
| config.shell().status("Compiling", &unit.pkg)?; |
| } |
| } |
| } |
| Fresh => { |
| // If doc test are last, only print "Fresh" if nothing has been printed. |
| if self.counts[&unit.pkg.package_id()] == 0 |
| && !(unit.mode.is_doc_test() && self.compiled.contains(&unit.pkg.package_id())) |
| { |
| self.compiled.insert(unit.pkg.package_id()); |
| config.shell().verbose(|c| c.status("Fresh", &unit.pkg))?; |
| } |
| } |
| } |
| Ok(()) |
| } |
| |
| fn back_compat_notice(&self, cx: &Context<'_, '_>, unit: &Unit) -> CargoResult<()> { |
| if unit.pkg.name() != "diesel" |
| || unit.pkg.version() >= &Version::new(1, 4, 8) |
| || cx.bcx.ws.resolve_behavior() == ResolveBehavior::V1 |
| || !unit.pkg.package_id().source_id().is_registry() |
| || !unit.features.is_empty() |
| { |
| return Ok(()); |
| } |
| if !cx |
| .bcx |
| .unit_graph |
| .keys() |
| .any(|unit| unit.pkg.name() == "diesel" && !unit.features.is_empty()) |
| { |
| return Ok(()); |
| } |
| cx.bcx.config.shell().note( |
| "\ |
| This error may be due to an interaction between diesel and Cargo's new |
| feature resolver. Try updating to diesel 1.4.8 to fix this error. |
| ", |
| )?; |
| Ok(()) |
| } |
| } |
| |
| impl ErrorsDuringDrain { |
| fn to_error(&self) -> Option<anyhow::Error> { |
| match self.count { |
| 0 => None, |
| 1 => Some(format_err!("1 job failed")), |
| n => Some(format_err!("{} jobs failed", n)), |
| } |
| } |
| } |
