]> git.lizzy.rs Git - rust.git/blob - src/librustc/ty/query/plumbing.rs
Ensure StorageDead is created even if variable initialization fails
[rust.git] / src / librustc / ty / query / plumbing.rs
1 // Copyright 2012-2015 The Rust Project Developers. See the COPYRIGHT
2 // file at the top-level directory of this distribution and at
3 // http://rust-lang.org/COPYRIGHT.
4 //
5 // Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
6 // http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
7 // <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
8 // option. This file may not be copied, modified, or distributed
9 // except according to those terms.
10
11 //! The implementation of the query system itself. Defines the macros
12 //! that generate the actual methods on tcx which find and execute the
13 //! provider, manage the caches, and so forth.
14
15 use dep_graph::{DepNodeIndex, DepNode, DepKind, DepNodeColor};
16 use errors::DiagnosticBuilder;
17 use errors::Level;
18 use errors::Diagnostic;
19 use errors::FatalError;
20 use ty::tls;
21 use ty::{TyCtxt};
22 use ty::query::Query;
23 use ty::query::config::{QueryConfig, QueryDescription};
24 use ty::query::job::{QueryJob, QueryResult, QueryInfo};
25 use ty::item_path;
26
27 use util::common::{profq_msg, ProfileQueriesMsg, QueryMsg};
28
29 use rustc_data_structures::fx::{FxHashMap};
30 use rustc_data_structures::sync::{Lrc, Lock};
31 use std::mem;
32 use std::ptr;
33 use std::collections::hash_map::Entry;
34 use syntax_pos::Span;
35 use syntax::codemap::DUMMY_SP;
36
37 pub struct QueryCache<'tcx, D: QueryConfig<'tcx> + ?Sized> {
38     pub(super) results: FxHashMap<D::Key, QueryValue<D::Value>>,
39     pub(super) active: FxHashMap<D::Key, QueryResult<'tcx>>,
40 }
41
42 pub(super) struct QueryValue<T> {
43     pub(super) value: T,
44     pub(super) index: DepNodeIndex,
45 }
46
47 impl<T> QueryValue<T> {
48     pub(super) fn new(value: T,
49                       dep_node_index: DepNodeIndex)
50                       -> QueryValue<T> {
51         QueryValue {
52             value,
53             index: dep_node_index,
54         }
55     }
56 }
57
58 impl<'tcx, M: QueryConfig<'tcx>> QueryCache<'tcx, M> {
59     pub(super) fn new() -> QueryCache<'tcx, M> {
60         QueryCache {
61             results: FxHashMap(),
62             active: FxHashMap(),
63         }
64     }
65 }
66
67 // If enabled, send a message to the profile-queries thread
68 macro_rules! profq_msg {
69     ($tcx:expr, $msg:expr) => {
70         if cfg!(debug_assertions) {
71             if $tcx.sess.profile_queries() {
72                 profq_msg($tcx.sess, $msg)
73             }
74         }
75     }
76 }
77
78 // If enabled, format a key using its debug string, which can be
79 // expensive to compute (in terms of time).
80 macro_rules! profq_query_msg {
81     ($query:expr, $tcx:expr, $key:expr) => {{
82         let msg = if cfg!(debug_assertions) {
83             if $tcx.sess.profile_queries_and_keys() {
84                 Some(format!("{:?}", $key))
85             } else { None }
86         } else { None };
87         QueryMsg {
88             query: $query,
89             msg,
90         }
91     }}
92 }
93
94 /// A type representing the responsibility to execute the job in the `job` field.
95 /// This will poison the relevant query if dropped.
96 pub(super) struct JobOwner<'a, 'tcx: 'a, Q: QueryDescription<'tcx> + 'a> {
97     cache: &'a Lock<QueryCache<'tcx, Q>>,
98     key: Q::Key,
99     job: Lrc<QueryJob<'tcx>>,
100 }
101
102 impl<'a, 'tcx, Q: QueryDescription<'tcx>> JobOwner<'a, 'tcx, Q> {
103     /// Either gets a JobOwner corresponding the the query, allowing us to
104     /// start executing the query, or it returns with the result of the query.
105     /// If the query is executing elsewhere, this will wait for it.
106     /// If the query panicked, this will silently panic.
107     ///
108     /// This function is inlined because that results in a noticeable speedup
109     /// for some compile-time benchmarks.
110     #[inline(always)]
111     pub(super) fn try_get(
112         tcx: TyCtxt<'a, 'tcx, '_>,
113         span: Span,
114         key: &Q::Key,
115     ) -> TryGetJob<'a, 'tcx, Q> {
116         let cache = Q::query_cache(tcx);
117         loop {
118             let mut lock = cache.borrow_mut();
119             if let Some(value) = lock.results.get(key) {
120                 profq_msg!(tcx, ProfileQueriesMsg::CacheHit);
121                 let result = Ok((value.value.clone(), value.index));
122                 return TryGetJob::JobCompleted(result);
123             }
124             let job = match lock.active.entry((*key).clone()) {
125                 Entry::Occupied(entry) => {
126                     match *entry.get() {
127                         QueryResult::Started(ref job) => job.clone(),
128                         QueryResult::Poisoned => FatalError.raise(),
129                     }
130                 }
131                 Entry::Vacant(entry) => {
132                     // No job entry for this query. Return a new one to be started later
133                     return tls::with_related_context(tcx, |icx| {
134                         let info = QueryInfo {
135                             span,
136                             query: Q::query(key.clone()),
137                         };
138                         let job = Lrc::new(QueryJob::new(info, icx.query.clone()));
139                         let owner = JobOwner {
140                             cache,
141                             job: job.clone(),
142                             key: (*key).clone(),
143                         };
144                         entry.insert(QueryResult::Started(job));
145                         TryGetJob::NotYetStarted(owner)
146                     })
147                 }
148             };
149             mem::drop(lock);
150
151             if let Err(cycle) = job.await(tcx, span) {
152                 return TryGetJob::JobCompleted(Err(cycle));
153             }
154         }
155     }
156
157     /// Completes the query by updating the query cache with the `result`,
158     /// signals the waiter and forgets the JobOwner, so it won't poison the query
159     pub(super) fn complete(self, result: &Q::Value, dep_node_index: DepNodeIndex) {
160         // We can move out of `self` here because we `mem::forget` it below
161         let key = unsafe { ptr::read(&self.key) };
162         let job = unsafe { ptr::read(&self.job) };
163         let cache = self.cache;
164
165         // Forget ourself so our destructor won't poison the query
166         mem::forget(self);
167
168         let value = QueryValue::new(result.clone(), dep_node_index);
169         {
170             let mut lock = cache.borrow_mut();
171             lock.active.remove(&key);
172             lock.results.insert(key, value);
173         }
174
175         job.signal_complete();
176     }
177
178     /// Executes a job by changing the ImplicitCtxt to point to the
179     /// new query job while it executes. It returns the diagnostics
180     /// captured during execution and the actual result.
181     pub(super) fn start<'lcx, F, R>(
182         &self,
183         tcx: TyCtxt<'_, 'tcx, 'lcx>,
184         compute: F)
185     -> (R, Vec<Diagnostic>)
186     where
187         F: for<'b> FnOnce(TyCtxt<'b, 'tcx, 'lcx>) -> R
188     {
189         // The TyCtxt stored in TLS has the same global interner lifetime
190         // as `tcx`, so we use `with_related_context` to relate the 'gcx lifetimes
191         // when accessing the ImplicitCtxt
192         let r = tls::with_related_context(tcx, move |current_icx| {
193             // Update the ImplicitCtxt to point to our new query job
194             let new_icx = tls::ImplicitCtxt {
195                 tcx,
196                 query: Some(self.job.clone()),
197                 layout_depth: current_icx.layout_depth,
198                 task: current_icx.task,
199             };
200
201             // Use the ImplicitCtxt while we execute the query
202             tls::enter_context(&new_icx, |_| {
203                 compute(tcx)
204             })
205         });
206
207         // Extract the diagnostic from the job
208         let diagnostics = mem::replace(&mut *self.job.diagnostics.lock(), Vec::new());
209
210         (r, diagnostics)
211     }
212 }
213
214 impl<'a, 'tcx, Q: QueryDescription<'tcx>> Drop for JobOwner<'a, 'tcx, Q> {
215     fn drop(&mut self) {
216         // Poison the query so jobs waiting on it panic
217         self.cache.borrow_mut().active.insert(self.key.clone(), QueryResult::Poisoned);
218         // Also signal the completion of the job, so waiters
219         // will continue execution
220         self.job.signal_complete();
221     }
222 }
223
224 #[derive(Clone)]
225 pub struct CycleError<'tcx> {
226     /// The query and related span which uses the cycle
227     pub(super) usage: Option<(Span, Query<'tcx>)>,
228     pub(super) cycle: Vec<QueryInfo<'tcx>>,
229 }
230
231 /// The result of `try_get_lock`
232 pub(super) enum TryGetJob<'a, 'tcx: 'a, D: QueryDescription<'tcx> + 'a> {
233     /// The query is not yet started. Contains a guard to the cache eventually used to start it.
234     NotYetStarted(JobOwner<'a, 'tcx, D>),
235
236     /// The query was already completed.
237     /// Returns the result of the query and its dep node index
238     /// if it succeeded or a cycle error if it failed
239     JobCompleted(Result<(D::Value, DepNodeIndex), CycleError<'tcx>>),
240 }
241
242 impl<'a, 'gcx, 'tcx> TyCtxt<'a, 'gcx, 'tcx> {
243     pub(super) fn report_cycle(self, CycleError { usage, cycle: stack }: CycleError<'gcx>)
244         -> DiagnosticBuilder<'a>
245     {
246         assert!(!stack.is_empty());
247
248         let fix_span = |span: Span, query: &Query<'gcx>| {
249             self.sess.codemap().def_span(query.default_span(self, span))
250         };
251
252         // Disable naming impls with types in this path, since that
253         // sometimes cycles itself, leading to extra cycle errors.
254         // (And cycle errors around impls tend to occur during the
255         // collect/coherence phases anyhow.)
256         item_path::with_forced_impl_filename_line(|| {
257             let span = fix_span(stack[1 % stack.len()].span, &stack[0].query);
258             let mut err = struct_span_err!(self.sess,
259                                            span,
260                                            E0391,
261                                            "cycle detected when {}",
262                                            stack[0].query.describe(self));
263
264             for i in 1..stack.len() {
265                 let query = &stack[i].query;
266                 let span = fix_span(stack[(i + 1) % stack.len()].span, query);
267                 err.span_note(span, &format!("...which requires {}...", query.describe(self)));
268             }
269
270             err.note(&format!("...which again requires {}, completing the cycle",
271                               stack[0].query.describe(self)));
272
273             if let Some((span, query)) = usage {
274                 err.span_note(fix_span(span, &query),
275                               &format!("cycle used when {}", query.describe(self)));
276             }
277
278             return err
279         })
280     }
281
282     pub fn try_print_query_stack() {
283         eprintln!("query stack during panic:");
284
285         tls::with_context_opt(|icx| {
286             if let Some(icx) = icx {
287                 let mut current_query = icx.query.clone();
288                 let mut i = 0;
289
290                 while let Some(query) = current_query {
291                     let mut db = DiagnosticBuilder::new(icx.tcx.sess.diagnostic(),
292                         Level::FailureNote,
293                         &format!("#{} [{}] {}",
294                                  i,
295                                  query.info.query.name(),
296                                  query.info.query.describe(icx.tcx)));
297                     db.set_span(icx.tcx.sess.codemap().def_span(query.info.span));
298                     icx.tcx.sess.diagnostic().force_print_db(db);
299
300                     current_query = query.parent.clone();
301                     i += 1;
302                 }
303             }
304         });
305
306         eprintln!("end of query stack");
307     }
308
309     /// Try to read a node index for the node dep_node.
310     /// A node will have an index, when it's already been marked green, or when we can mark it
311     /// green. This function will mark the current task as a reader of the specified node, when
312     /// the a node index can be found for that node.
313     pub(super) fn try_mark_green_and_read(self, dep_node: &DepNode) -> Option<DepNodeIndex> {
314         match self.dep_graph.node_color(dep_node) {
315             Some(DepNodeColor::Green(dep_node_index)) => {
316                 self.dep_graph.read_index(dep_node_index);
317                 Some(dep_node_index)
318             }
319             Some(DepNodeColor::Red) => {
320                 None
321             }
322             None => {
323                 // try_mark_green (called below) will panic when full incremental
324                 // compilation is disabled. If that's the case, we can't try to mark nodes
325                 // as green anyway, so we can safely return None here.
326                 if !self.dep_graph.is_fully_enabled() {
327                     return None;
328                 }
329                 match self.dep_graph.try_mark_green(self.global_tcx(), &dep_node) {
330                     Some(dep_node_index) => {
331                         debug_assert!(self.dep_graph.is_green(&dep_node));
332                         self.dep_graph.read_index(dep_node_index);
333                         Some(dep_node_index)
334                     }
335                     None => {
336                         None
337                     }
338                 }
339             }
340         }
341     }
342
343     fn try_get_with<Q: QueryDescription<'gcx>>(
344         self,
345         span: Span,
346         key: Q::Key)
347     -> Result<Q::Value, CycleError<'gcx>>
348     {
349         debug!("ty::queries::{}::try_get_with(key={:?}, span={:?})",
350                Q::NAME,
351                key,
352                span);
353
354         profq_msg!(self,
355             ProfileQueriesMsg::QueryBegin(
356                 span.data(),
357                 profq_query_msg!(Q::NAME, self, key),
358             )
359         );
360
361         let job = match JobOwner::try_get(self, span, &key) {
362             TryGetJob::NotYetStarted(job) => job,
363             TryGetJob::JobCompleted(result) => {
364                 return result.map(|(v, index)| {
365                     self.dep_graph.read_index(index);
366                     v
367                 })
368             }
369         };
370
371         // Fast path for when incr. comp. is off. `to_dep_node` is
372         // expensive for some DepKinds.
373         if !self.dep_graph.is_fully_enabled() {
374             let null_dep_node = DepNode::new_no_params(::dep_graph::DepKind::Null);
375             return self.force_query_with_job::<Q>(key, job, null_dep_node).map(|(v, _)| v);
376         }
377
378         let dep_node = Q::to_dep_node(self, &key);
379
380         if dep_node.kind.is_anon() {
381             profq_msg!(self, ProfileQueriesMsg::ProviderBegin);
382
383             let res = job.start(self, |tcx| {
384                 tcx.dep_graph.with_anon_task(dep_node.kind, || {
385                     Q::compute(tcx.global_tcx(), key)
386                 })
387             });
388
389             profq_msg!(self, ProfileQueriesMsg::ProviderEnd);
390             let ((result, dep_node_index), diagnostics) = res;
391
392             self.dep_graph.read_index(dep_node_index);
393
394             self.queries.on_disk_cache
395                 .store_diagnostics_for_anon_node(dep_node_index, diagnostics);
396
397             job.complete(&result, dep_node_index);
398
399             return Ok(result);
400         }
401
402         if !dep_node.kind.is_input() {
403             if let Some(dep_node_index) = self.try_mark_green_and_read(&dep_node) {
404                 profq_msg!(self, ProfileQueriesMsg::CacheHit);
405                 return self.load_from_disk_and_cache_in_memory::<Q>(key,
406                                                                     job,
407                                                                     dep_node_index,
408                                                                     &dep_node)
409             }
410         }
411
412         match self.force_query_with_job::<Q>(key, job, dep_node) {
413             Ok((result, dep_node_index)) => {
414                 self.dep_graph.read_index(dep_node_index);
415                 Ok(result)
416             }
417             Err(e) => Err(e)
418         }
419     }
420
421     fn load_from_disk_and_cache_in_memory<Q: QueryDescription<'gcx>>(
422         self,
423         key: Q::Key,
424         job: JobOwner<'a, 'gcx, Q>,
425         dep_node_index: DepNodeIndex,
426         dep_node: &DepNode
427     ) -> Result<Q::Value, CycleError<'gcx>>
428     {
429         // Note this function can be called concurrently from the same query
430         // We must ensure that this is handled correctly
431
432         debug_assert!(self.dep_graph.is_green(dep_node));
433
434         // First we try to load the result from the on-disk cache
435         let result = if Q::cache_on_disk(key.clone()) &&
436                         self.sess.opts.debugging_opts.incremental_queries {
437             let prev_dep_node_index =
438                 self.dep_graph.prev_dep_node_index_of(dep_node);
439             let result = Q::try_load_from_disk(self.global_tcx(),
440                                                     prev_dep_node_index);
441
442             // We always expect to find a cached result for things that
443             // can be forced from DepNode.
444             debug_assert!(!dep_node.kind.can_reconstruct_query_key() ||
445                             result.is_some(),
446                             "Missing on-disk cache entry for {:?}",
447                             dep_node);
448             result
449         } else {
450             // Some things are never cached on disk.
451             None
452         };
453
454         let result = if let Some(result) = result {
455             result
456         } else {
457             // We could not load a result from the on-disk cache, so
458             // recompute.
459
460             // The diagnostics for this query have already been
461             // promoted to the current session during
462             // try_mark_green(), so we can ignore them here.
463             let (result, _) = job.start(self, |tcx| {
464                 // The dep-graph for this computation is already in
465                 // place
466                 tcx.dep_graph.with_ignore(|| {
467                     Q::compute(tcx, key)
468                 })
469             });
470             result
471         };
472
473         // If -Zincremental-verify-ich is specified, re-hash results from
474         // the cache and make sure that they have the expected fingerprint.
475         if self.sess.opts.debugging_opts.incremental_verify_ich {
476             use rustc_data_structures::stable_hasher::{StableHasher, HashStable};
477             use ich::Fingerprint;
478
479             assert!(Some(self.dep_graph.fingerprint_of(dep_node_index)) ==
480                     self.dep_graph.prev_fingerprint_of(dep_node),
481                     "Fingerprint for green query instance not loaded \
482                         from cache: {:?}", dep_node);
483
484             debug!("BEGIN verify_ich({:?})", dep_node);
485             let mut hcx = self.create_stable_hashing_context();
486             let mut hasher = StableHasher::new();
487
488             result.hash_stable(&mut hcx, &mut hasher);
489
490             let new_hash: Fingerprint = hasher.finish();
491             debug!("END verify_ich({:?})", dep_node);
492
493             let old_hash = self.dep_graph.fingerprint_of(dep_node_index);
494
495             assert!(new_hash == old_hash, "Found unstable fingerprints \
496                 for {:?}", dep_node);
497         }
498
499         if self.sess.opts.debugging_opts.query_dep_graph {
500             self.dep_graph.mark_loaded_from_cache(dep_node_index, true);
501         }
502
503         job.complete(&result, dep_node_index);
504
505         Ok(result)
506     }
507
508     fn force_query_with_job<Q: QueryDescription<'gcx>>(
509         self,
510         key: Q::Key,
511         job: JobOwner<'_, 'gcx, Q>,
512         dep_node: DepNode)
513     -> Result<(Q::Value, DepNodeIndex), CycleError<'gcx>> {
514         // If the following assertion triggers, it can have two reasons:
515         // 1. Something is wrong with DepNode creation, either here or
516         //    in DepGraph::try_mark_green()
517         // 2. Two distinct query keys get mapped to the same DepNode
518         //    (see for example #48923)
519         assert!(!self.dep_graph.dep_node_exists(&dep_node),
520                 "Forcing query with already existing DepNode.\n\
521                     - query-key: {:?}\n\
522                     - dep-node: {:?}",
523                 key, dep_node);
524
525         profq_msg!(self, ProfileQueriesMsg::ProviderBegin);
526         let res = job.start(self, |tcx| {
527             if dep_node.kind.is_eval_always() {
528                 tcx.dep_graph.with_eval_always_task(dep_node,
529                                                     tcx,
530                                                     key,
531                                                     Q::compute)
532             } else {
533                 tcx.dep_graph.with_task(dep_node,
534                                         tcx,
535                                         key,
536                                         Q::compute)
537             }
538         });
539         profq_msg!(self, ProfileQueriesMsg::ProviderEnd);
540
541         let ((result, dep_node_index), diagnostics) = res;
542
543         if self.sess.opts.debugging_opts.query_dep_graph {
544             self.dep_graph.mark_loaded_from_cache(dep_node_index, false);
545         }
546
547         if dep_node.kind != ::dep_graph::DepKind::Null {
548             self.queries.on_disk_cache
549                 .store_diagnostics(dep_node_index, diagnostics);
550         }
551
552         job.complete(&result, dep_node_index);
553
554         Ok((result, dep_node_index))
555     }
556
557     /// Ensure that either this query has all green inputs or been executed.
558     /// Executing query::ensure(D) is considered a read of the dep-node D.
559     ///
560     /// This function is particularly useful when executing passes for their
561     /// side-effects -- e.g., in order to report errors for erroneous programs.
562     ///
563     /// Note: The optimization is only available during incr. comp.
564     pub(super) fn ensure_query<Q: QueryDescription<'gcx>>(self, key: Q::Key) -> () {
565         let dep_node = Q::to_dep_node(self, &key);
566
567         // Ensuring an "input" or anonymous query makes no sense
568         assert!(!dep_node.kind.is_anon());
569         assert!(!dep_node.kind.is_input());
570         if self.try_mark_green_and_read(&dep_node).is_none() {
571             // A None return from `try_mark_green_and_read` means that this is either
572             // a new dep node or that the dep node has already been marked red.
573             // Either way, we can't call `dep_graph.read()` as we don't have the
574             // DepNodeIndex. We must invoke the query itself. The performance cost
575             // this introduces should be negligible as we'll immediately hit the
576             // in-memory cache, or another query down the line will.
577             let _ = self.get_query::<Q>(DUMMY_SP, key);
578         }
579     }
580
581     #[allow(dead_code)]
582     fn force_query<Q: QueryDescription<'gcx>>(
583         self,
584         key: Q::Key,
585         span: Span,
586         dep_node: DepNode
587     ) -> Result<(Q::Value, DepNodeIndex), CycleError<'gcx>> {
588         // We may be concurrently trying both execute and force a query
589         // Ensure that only one of them runs the query
590         let job = match JobOwner::try_get(self, span, &key) {
591             TryGetJob::NotYetStarted(job) => job,
592             TryGetJob::JobCompleted(result) => return result,
593         };
594         self.force_query_with_job::<Q>(key, job, dep_node)
595     }
596
597     pub(super) fn try_get_query<Q: QueryDescription<'gcx>>(
598         self,
599         span: Span,
600         key: Q::Key,
601     ) -> Result<Q::Value, DiagnosticBuilder<'a>> {
602         match self.try_get_with::<Q>(span, key) {
603             Ok(e) => Ok(e),
604             Err(e) => Err(self.report_cycle(e)),
605         }
606     }
607
608     pub(super) fn get_query<Q: QueryDescription<'gcx>>(
609         self,
610         span: Span,
611         key: Q::Key,
612     ) -> Q::Value {
613         self.try_get_query::<Q>(span, key).unwrap_or_else(|mut e| {
614             e.emit();
615             Q::handle_cycle_error(self)
616         })
617     }
618 }
619
620 macro_rules! handle_cycle_error {
621     ([][$this: expr]) => {{
622         Value::from_cycle_error($this.global_tcx())
623     }};
624     ([fatal_cycle$(, $modifiers:ident)*][$this:expr]) => {{
625         $this.sess.abort_if_errors();
626         unreachable!();
627     }};
628     ([$other:ident$(, $modifiers:ident)*][$($args:tt)*]) => {
629         handle_cycle_error!([$($modifiers),*][$($args)*])
630     };
631 }
632
633 macro_rules! define_queries {
634     (<$tcx:tt>
635      $($(#[$attr:meta])*
636        [$($modifiers:tt)*] fn $name:ident: $node:ident($K:ty) -> $V:ty,)*) => {
637
638         use std::mem;
639         #[cfg(parallel_queries)]
640         use ty::query::job::QueryResult;
641         use rustc_data_structures::sync::Lock;
642         use {
643             rustc_data_structures::stable_hasher::HashStable,
644             rustc_data_structures::stable_hasher::StableHasherResult,
645             rustc_data_structures::stable_hasher::StableHasher,
646             ich::StableHashingContext
647         };
648
649         define_queries_struct! {
650             tcx: $tcx,
651             input: ($(([$($modifiers)*] [$($attr)*] [$name]))*)
652         }
653
654         impl<$tcx> Queries<$tcx> {
655             pub fn new(
656                 providers: IndexVec<CrateNum, Providers<$tcx>>,
657                 on_disk_cache: OnDiskCache<'tcx>,
658             ) -> Self {
659                 Queries {
660                     providers,
661                     on_disk_cache,
662                     $($name: Lock::new(QueryCache::new())),*
663                 }
664             }
665
666             #[cfg(parallel_queries)]
667             pub fn collect_active_jobs(&self) -> Vec<Lrc<QueryJob<$tcx>>> {
668                 let mut jobs = Vec::new();
669
670                 // We use try_lock here since we are only called from the
671                 // deadlock handler, and this shouldn't be locked
672                 $(for v in self.$name.try_lock().unwrap().active.values() {
673                     match *v {
674                         QueryResult::Started(ref job) => jobs.push(job.clone()),
675                         _ => (),
676                     }
677                 })*
678
679                 return jobs;
680             }
681         }
682
683         #[allow(bad_style)]
684         #[derive(Copy, Clone, Debug, PartialEq, Eq, Hash)]
685         pub enum Query<$tcx> {
686             $($(#[$attr])* $name($K)),*
687         }
688
689         impl<$tcx> Query<$tcx> {
690             pub fn name(&self) -> &'static str {
691                 match *self {
692                     $(Query::$name(_) => stringify!($name),)*
693                 }
694             }
695
696             pub fn describe(&self, tcx: TyCtxt) -> String {
697                 let (r, name) = match *self {
698                     $(Query::$name(key) => {
699                         (queries::$name::describe(tcx, key), stringify!($name))
700                     })*
701                 };
702                 if tcx.sess.verbose() {
703                     format!("{} [{}]", r, name)
704                 } else {
705                     r
706                 }
707             }
708
709             // FIXME(eddyb) Get more valid Span's on queries.
710             pub fn default_span(&self, tcx: TyCtxt<'_, $tcx, '_>, span: Span) -> Span {
711                 if !span.is_dummy() {
712                     return span;
713                 }
714                 // The def_span query is used to calculate default_span,
715                 // so exit to avoid infinite recursion
716                 match *self {
717                     Query::def_span(..) => return span,
718                     _ => ()
719                 }
720                 match *self {
721                     $(Query::$name(key) => key.default_span(tcx),)*
722                 }
723             }
724         }
725
726         impl<'a, $tcx> HashStable<StableHashingContext<'a>> for Query<$tcx> {
727             fn hash_stable<W: StableHasherResult>(&self,
728                                                 hcx: &mut StableHashingContext<'a>,
729                                                 hasher: &mut StableHasher<W>) {
730                 mem::discriminant(self).hash_stable(hcx, hasher);
731                 match *self {
732                     $(Query::$name(key) => key.hash_stable(hcx, hasher),)*
733                 }
734             }
735         }
736
737         pub mod queries {
738             use std::marker::PhantomData;
739
740             $(#[allow(bad_style)]
741             pub struct $name<$tcx> {
742                 data: PhantomData<&$tcx ()>
743             })*
744         }
745
746         // This module and the functions in it exist only to provide a
747         // predictable symbol name prefix for query providers. This is helpful
748         // for analyzing queries in profilers.
749         pub(super) mod __query_compute {
750             $(#[inline(never)]
751             pub fn $name<F: FnOnce() -> R, R>(f: F) -> R {
752                 f()
753             })*
754         }
755
756         $(impl<$tcx> QueryConfig<$tcx> for queries::$name<$tcx> {
757             type Key = $K;
758             type Value = $V;
759
760             const NAME: &'static str = stringify!($name);
761         }
762
763         impl<$tcx> QueryAccessors<$tcx> for queries::$name<$tcx> {
764             fn query(key: Self::Key) -> Query<'tcx> {
765                 Query::$name(key)
766             }
767
768             fn query_cache<'a>(tcx: TyCtxt<'a, $tcx, '_>) -> &'a Lock<QueryCache<$tcx, Self>> {
769                 &tcx.queries.$name
770             }
771
772             #[allow(unused)]
773             fn to_dep_node(tcx: TyCtxt<'_, $tcx, '_>, key: &Self::Key) -> DepNode {
774                 use dep_graph::DepConstructor::*;
775
776                 DepNode::new(tcx, $node(*key))
777             }
778
779             #[inline]
780             fn compute(tcx: TyCtxt<'_, 'tcx, '_>, key: Self::Key) -> Self::Value {
781                 __query_compute::$name(move || {
782                     let provider = tcx.queries.providers[key.query_crate()].$name;
783                     provider(tcx.global_tcx(), key)
784                 })
785             }
786
787             fn handle_cycle_error(tcx: TyCtxt<'_, 'tcx, '_>) -> Self::Value {
788                 handle_cycle_error!([$($modifiers)*][tcx])
789             }
790         }
791
792         impl<'a, $tcx, 'lcx> queries::$name<$tcx> {
793             /// Ensure that either this query has all green inputs or been executed.
794             /// Executing query::ensure(D) is considered a read of the dep-node D.
795             ///
796             /// This function is particularly useful when executing passes for their
797             /// side-effects -- e.g., in order to report errors for erroneous programs.
798             ///
799             /// Note: The optimization is only available during incr. comp.
800             pub fn ensure(tcx: TyCtxt<'a, $tcx, 'lcx>, key: $K) -> () {
801                 tcx.ensure_query::<queries::$name>(key);
802             }
803         })*
804
805         #[derive(Copy, Clone)]
806         pub struct TyCtxtAt<'a, 'gcx: 'a+'tcx, 'tcx: 'a> {
807             pub tcx: TyCtxt<'a, 'gcx, 'tcx>,
808             pub span: Span,
809         }
810
811         impl<'a, 'gcx, 'tcx> Deref for TyCtxtAt<'a, 'gcx, 'tcx> {
812             type Target = TyCtxt<'a, 'gcx, 'tcx>;
813             fn deref(&self) -> &Self::Target {
814                 &self.tcx
815             }
816         }
817
818         impl<'a, $tcx, 'lcx> TyCtxt<'a, $tcx, 'lcx> {
819             /// Return a transparent wrapper for `TyCtxt` which uses
820             /// `span` as the location of queries performed through it.
821             pub fn at(self, span: Span) -> TyCtxtAt<'a, $tcx, 'lcx> {
822                 TyCtxtAt {
823                     tcx: self,
824                     span
825                 }
826             }
827
828             $($(#[$attr])*
829             pub fn $name(self, key: $K) -> $V {
830                 self.at(DUMMY_SP).$name(key)
831             })*
832         }
833
834         impl<'a, $tcx, 'lcx> TyCtxtAt<'a, $tcx, 'lcx> {
835             $($(#[$attr])*
836             pub fn $name(self, key: $K) -> $V {
837                 self.tcx.get_query::<queries::$name>(self.span, key)
838             })*
839         }
840
841         define_provider_struct! {
842             tcx: $tcx,
843             input: ($(([$($modifiers)*] [$name] [$K] [$V]))*)
844         }
845
846         impl<$tcx> Copy for Providers<$tcx> {}
847         impl<$tcx> Clone for Providers<$tcx> {
848             fn clone(&self) -> Self { *self }
849         }
850     }
851 }
852
853 macro_rules! define_queries_struct {
854     (tcx: $tcx:tt,
855      input: ($(([$($modifiers:tt)*] [$($attr:tt)*] [$name:ident]))*)) => {
856         pub(crate) struct Queries<$tcx> {
857             /// This provides access to the incr. comp. on-disk cache for query results.
858             /// Do not access this directly. It is only meant to be used by
859             /// `DepGraph::try_mark_green()` and the query infrastructure.
860             pub(crate) on_disk_cache: OnDiskCache<'tcx>,
861
862             providers: IndexVec<CrateNum, Providers<$tcx>>,
863
864             $($(#[$attr])*  $name: Lock<QueryCache<$tcx, queries::$name<$tcx>>>,)*
865         }
866     };
867 }
868
869 macro_rules! define_provider_struct {
870     (tcx: $tcx:tt,
871      input: ($(([$($modifiers:tt)*] [$name:ident] [$K:ty] [$R:ty]))*)) => {
872         pub struct Providers<$tcx> {
873             $(pub $name: for<'a> fn(TyCtxt<'a, $tcx, $tcx>, $K) -> $R,)*
874         }
875
876         impl<$tcx> Default for Providers<$tcx> {
877             fn default() -> Self {
878                 $(fn $name<'a, $tcx>(_: TyCtxt<'a, $tcx, $tcx>, key: $K) -> $R {
879                     bug!("tcx.{}({:?}) unsupported by its crate",
880                          stringify!($name), key);
881                 })*
882                 Providers { $($name),* }
883             }
884         }
885     };
886 }
887
888
889 /// The red/green evaluation system will try to mark a specific DepNode in the
890 /// dependency graph as green by recursively trying to mark the dependencies of
891 /// that DepNode as green. While doing so, it will sometimes encounter a DepNode
892 /// where we don't know if it is red or green and we therefore actually have
893 /// to recompute its value in order to find out. Since the only piece of
894 /// information that we have at that point is the DepNode we are trying to
895 /// re-evaluate, we need some way to re-run a query from just that. This is what
896 /// `force_from_dep_node()` implements.
897 ///
898 /// In the general case, a DepNode consists of a DepKind and an opaque
899 /// GUID/fingerprint that will uniquely identify the node. This GUID/fingerprint
900 /// is usually constructed by computing a stable hash of the query-key that the
901 /// DepNode corresponds to. Consequently, it is not in general possible to go
902 /// back from hash to query-key (since hash functions are not reversible). For
903 /// this reason `force_from_dep_node()` is expected to fail from time to time
904 /// because we just cannot find out, from the DepNode alone, what the
905 /// corresponding query-key is and therefore cannot re-run the query.
906 ///
907 /// The system deals with this case letting `try_mark_green` fail which forces
908 /// the root query to be re-evaluated.
909 ///
910 /// Now, if force_from_dep_node() would always fail, it would be pretty useless.
911 /// Fortunately, we can use some contextual information that will allow us to
912 /// reconstruct query-keys for certain kinds of DepNodes. In particular, we
913 /// enforce by construction that the GUID/fingerprint of certain DepNodes is a
914 /// valid DefPathHash. Since we also always build a huge table that maps every
915 /// DefPathHash in the current codebase to the corresponding DefId, we have
916 /// everything we need to re-run the query.
917 ///
918 /// Take the `mir_validated` query as an example. Like many other queries, it
919 /// just has a single parameter: the DefId of the item it will compute the
920 /// validated MIR for. Now, when we call `force_from_dep_node()` on a dep-node
921 /// with kind `MirValidated`, we know that the GUID/fingerprint of the dep-node
922 /// is actually a DefPathHash, and can therefore just look up the corresponding
923 /// DefId in `tcx.def_path_hash_to_def_id`.
924 ///
925 /// When you implement a new query, it will likely have a corresponding new
926 /// DepKind, and you'll have to support it here in `force_from_dep_node()`. As
927 /// a rule of thumb, if your query takes a DefId or DefIndex as sole parameter,
928 /// then `force_from_dep_node()` should not fail for it. Otherwise, you can just
929 /// add it to the "We don't have enough information to reconstruct..." group in
930 /// the match below.
931 pub fn force_from_dep_node<'a, 'gcx, 'lcx>(tcx: TyCtxt<'a, 'gcx, 'lcx>,
932                                            dep_node: &DepNode)
933                                            -> bool {
934     use hir::def_id::LOCAL_CRATE;
935
936     // We must avoid ever having to call force_from_dep_node() for a
937     // DepNode::CodegenUnit:
938     // Since we cannot reconstruct the query key of a DepNode::CodegenUnit, we
939     // would always end up having to evaluate the first caller of the
940     // `codegen_unit` query that *is* reconstructible. This might very well be
941     // the `compile_codegen_unit` query, thus re-codegenning the whole CGU just
942     // to re-trigger calling the `codegen_unit` query with the right key. At
943     // that point we would already have re-done all the work we are trying to
944     // avoid doing in the first place.
945     // The solution is simple: Just explicitly call the `codegen_unit` query for
946     // each CGU, right after partitioning. This way `try_mark_green` will always
947     // hit the cache instead of having to go through `force_from_dep_node`.
948     // This assertion makes sure, we actually keep applying the solution above.
949     debug_assert!(dep_node.kind != DepKind::CodegenUnit,
950                   "calling force_from_dep_node() on DepKind::CodegenUnit");
951
952     if !dep_node.kind.can_reconstruct_query_key() {
953         return false
954     }
955
956     macro_rules! def_id {
957         () => {
958             if let Some(def_id) = dep_node.extract_def_id(tcx) {
959                 def_id
960             } else {
961                 // return from the whole function
962                 return false
963             }
964         }
965     };
966
967     macro_rules! krate {
968         () => { (def_id!()).krate }
969     };
970
971     macro_rules! force {
972         ($query:ident, $key:expr) => {
973             {
974                 use $crate::util::common::{ProfileQueriesMsg, profq_msg};
975
976                 profq_msg!(tcx,
977                     ProfileQueriesMsg::QueryBegin(
978                         DUMMY_SP.data(),
979                         profq_query_msg!(::ty::query::queries::$query::NAME, tcx, $key),
980                     )
981                 );
982
983                 match tcx.force_query::<::ty::query::queries::$query>($key, DUMMY_SP, *dep_node) {
984                     Ok(_) => {},
985                     Err(e) => {
986                         tcx.report_cycle(e).emit();
987                     }
988                 }
989             }
990         }
991     };
992
993     // FIXME(#45015): We should try move this boilerplate code into a macro
994     //                somehow.
995     match dep_node.kind {
996         // These are inputs that are expected to be pre-allocated and that
997         // should therefore always be red or green already
998         DepKind::AllLocalTraitImpls |
999         DepKind::Krate |
1000         DepKind::CrateMetadata |
1001         DepKind::HirBody |
1002         DepKind::Hir |
1003
1004         // This are anonymous nodes
1005         DepKind::TraitSelect |
1006
1007         // We don't have enough information to reconstruct the query key of
1008         // these
1009         DepKind::IsCopy |
1010         DepKind::IsSized |
1011         DepKind::IsFreeze |
1012         DepKind::NeedsDrop |
1013         DepKind::Layout |
1014         DepKind::ConstEval |
1015         DepKind::InstanceSymbolName |
1016         DepKind::MirShim |
1017         DepKind::BorrowCheckKrate |
1018         DepKind::Specializes |
1019         DepKind::ImplementationsOfTrait |
1020         DepKind::TypeParamPredicates |
1021         DepKind::CodegenUnit |
1022         DepKind::CompileCodegenUnit |
1023         DepKind::FulfillObligation |
1024         DepKind::VtableMethods |
1025         DepKind::EraseRegionsTy |
1026         DepKind::ConstValueToAllocation |
1027         DepKind::NormalizeProjectionTy |
1028         DepKind::NormalizeTyAfterErasingRegions |
1029         DepKind::DropckOutlives |
1030         DepKind::EvaluateObligation |
1031         DepKind::TypeOpEq |
1032         DepKind::TypeOpSubtype |
1033         DepKind::TypeOpProvePredicate |
1034         DepKind::TypeOpNormalizeTy |
1035         DepKind::TypeOpNormalizePredicate |
1036         DepKind::TypeOpNormalizePolyFnSig |
1037         DepKind::TypeOpNormalizeFnSig |
1038         DepKind::SubstituteNormalizeAndTestPredicates |
1039         DepKind::InstanceDefSizeEstimate |
1040         DepKind::ProgramClausesForEnv |
1041
1042         // This one should never occur in this context
1043         DepKind::Null => {
1044             bug!("force_from_dep_node() - Encountered {:?}", dep_node)
1045         }
1046
1047         // These are not queries
1048         DepKind::CoherenceCheckTrait |
1049         DepKind::ItemVarianceConstraints => {
1050             return false
1051         }
1052
1053         DepKind::RegionScopeTree => { force!(region_scope_tree, def_id!()); }
1054
1055         DepKind::Coherence => { force!(crate_inherent_impls, LOCAL_CRATE); }
1056         DepKind::CoherenceInherentImplOverlapCheck => {
1057             force!(crate_inherent_impls_overlap_check, LOCAL_CRATE)
1058         },
1059         DepKind::PrivacyAccessLevels => { force!(privacy_access_levels, LOCAL_CRATE); }
1060         DepKind::MirBuilt => { force!(mir_built, def_id!()); }
1061         DepKind::MirConstQualif => { force!(mir_const_qualif, def_id!()); }
1062         DepKind::MirConst => { force!(mir_const, def_id!()); }
1063         DepKind::MirValidated => { force!(mir_validated, def_id!()); }
1064         DepKind::MirOptimized => { force!(optimized_mir, def_id!()); }
1065
1066         DepKind::BorrowCheck => { force!(borrowck, def_id!()); }
1067         DepKind::MirBorrowCheck => { force!(mir_borrowck, def_id!()); }
1068         DepKind::UnsafetyCheckResult => { force!(unsafety_check_result, def_id!()); }
1069         DepKind::UnsafeDeriveOnReprPacked => { force!(unsafe_derive_on_repr_packed, def_id!()); }
1070         DepKind::Reachability => { force!(reachable_set, LOCAL_CRATE); }
1071         DepKind::MirKeys => { force!(mir_keys, LOCAL_CRATE); }
1072         DepKind::CrateVariances => { force!(crate_variances, LOCAL_CRATE); }
1073         DepKind::AssociatedItems => { force!(associated_item, def_id!()); }
1074         DepKind::TypeOfItem => { force!(type_of, def_id!()); }
1075         DepKind::GenericsOfItem => { force!(generics_of, def_id!()); }
1076         DepKind::PredicatesOfItem => { force!(predicates_of, def_id!()); }
1077         DepKind::ExplicitPredicatesOfItem => { force!(explicit_predicates_of, def_id!()); }
1078         DepKind::InferredOutlivesOf => { force!(inferred_outlives_of, def_id!()); }
1079         DepKind::InferredOutlivesCrate => { force!(inferred_outlives_crate, LOCAL_CRATE); }
1080         DepKind::SuperPredicatesOfItem => { force!(super_predicates_of, def_id!()); }
1081         DepKind::TraitDefOfItem => { force!(trait_def, def_id!()); }
1082         DepKind::AdtDefOfItem => { force!(adt_def, def_id!()); }
1083         DepKind::ImplTraitRef => { force!(impl_trait_ref, def_id!()); }
1084         DepKind::ImplPolarity => { force!(impl_polarity, def_id!()); }
1085         DepKind::FnSignature => { force!(fn_sig, def_id!()); }
1086         DepKind::CoerceUnsizedInfo => { force!(coerce_unsized_info, def_id!()); }
1087         DepKind::ItemVariances => { force!(variances_of, def_id!()); }
1088         DepKind::IsConstFn => { force!(is_const_fn, def_id!()); }
1089         DepKind::IsForeignItem => { force!(is_foreign_item, def_id!()); }
1090         DepKind::SizedConstraint => { force!(adt_sized_constraint, def_id!()); }
1091         DepKind::DtorckConstraint => { force!(adt_dtorck_constraint, def_id!()); }
1092         DepKind::AdtDestructor => { force!(adt_destructor, def_id!()); }
1093         DepKind::AssociatedItemDefIds => { force!(associated_item_def_ids, def_id!()); }
1094         DepKind::InherentImpls => { force!(inherent_impls, def_id!()); }
1095         DepKind::TypeckBodiesKrate => { force!(typeck_item_bodies, LOCAL_CRATE); }
1096         DepKind::TypeckTables => { force!(typeck_tables_of, def_id!()); }
1097         DepKind::UsedTraitImports => { force!(used_trait_imports, def_id!()); }
1098         DepKind::HasTypeckTables => { force!(has_typeck_tables, def_id!()); }
1099         DepKind::SymbolName => { force!(def_symbol_name, def_id!()); }
1100         DepKind::SpecializationGraph => { force!(specialization_graph_of, def_id!()); }
1101         DepKind::ObjectSafety => { force!(is_object_safe, def_id!()); }
1102         DepKind::TraitImpls => { force!(trait_impls_of, def_id!()); }
1103         DepKind::CheckMatch => { force!(check_match, def_id!()); }
1104
1105         DepKind::ParamEnv => { force!(param_env, def_id!()); }
1106         DepKind::DescribeDef => { force!(describe_def, def_id!()); }
1107         DepKind::DefSpan => { force!(def_span, def_id!()); }
1108         DepKind::LookupStability => { force!(lookup_stability, def_id!()); }
1109         DepKind::LookupDeprecationEntry => {
1110             force!(lookup_deprecation_entry, def_id!());
1111         }
1112         DepKind::ConstIsRvaluePromotableToStatic => {
1113             force!(const_is_rvalue_promotable_to_static, def_id!());
1114         }
1115         DepKind::RvaluePromotableMap => { force!(rvalue_promotable_map, def_id!()); }
1116         DepKind::ImplParent => { force!(impl_parent, def_id!()); }
1117         DepKind::TraitOfItem => { force!(trait_of_item, def_id!()); }
1118         DepKind::IsReachableNonGeneric => { force!(is_reachable_non_generic, def_id!()); }
1119         DepKind::IsUnreachableLocalDefinition => {
1120             force!(is_unreachable_local_definition, def_id!());
1121         }
1122         DepKind::IsMirAvailable => { force!(is_mir_available, def_id!()); }
1123         DepKind::ItemAttrs => { force!(item_attrs, def_id!()); }
1124         DepKind::CodegenFnAttrs => { force!(codegen_fn_attrs, def_id!()); }
1125         DepKind::FnArgNames => { force!(fn_arg_names, def_id!()); }
1126         DepKind::RenderedConst => { force!(rendered_const, def_id!()); }
1127         DepKind::DylibDepFormats => { force!(dylib_dependency_formats, krate!()); }
1128         DepKind::IsPanicRuntime => { force!(is_panic_runtime, krate!()); }
1129         DepKind::IsCompilerBuiltins => { force!(is_compiler_builtins, krate!()); }
1130         DepKind::HasGlobalAllocator => { force!(has_global_allocator, krate!()); }
1131         DepKind::ExternCrate => { force!(extern_crate, def_id!()); }
1132         DepKind::LintLevels => { force!(lint_levels, LOCAL_CRATE); }
1133         DepKind::InScopeTraits => { force!(in_scope_traits_map, def_id!().index); }
1134         DepKind::ModuleExports => { force!(module_exports, def_id!()); }
1135         DepKind::IsSanitizerRuntime => { force!(is_sanitizer_runtime, krate!()); }
1136         DepKind::IsProfilerRuntime => { force!(is_profiler_runtime, krate!()); }
1137         DepKind::GetPanicStrategy => { force!(panic_strategy, krate!()); }
1138         DepKind::IsNoBuiltins => { force!(is_no_builtins, krate!()); }
1139         DepKind::ImplDefaultness => { force!(impl_defaultness, def_id!()); }
1140         DepKind::CheckItemWellFormed => { force!(check_item_well_formed, def_id!()); }
1141         DepKind::CheckTraitItemWellFormed => { force!(check_trait_item_well_formed, def_id!()); }
1142         DepKind::CheckImplItemWellFormed => { force!(check_impl_item_well_formed, def_id!()); }
1143         DepKind::ReachableNonGenerics => { force!(reachable_non_generics, krate!()); }
1144         DepKind::NativeLibraries => { force!(native_libraries, krate!()); }
1145         DepKind::PluginRegistrarFn => { force!(plugin_registrar_fn, krate!()); }
1146         DepKind::DeriveRegistrarFn => { force!(derive_registrar_fn, krate!()); }
1147         DepKind::CrateDisambiguator => { force!(crate_disambiguator, krate!()); }
1148         DepKind::CrateHash => { force!(crate_hash, krate!()); }
1149         DepKind::OriginalCrateName => { force!(original_crate_name, krate!()); }
1150         DepKind::ExtraFileName => { force!(extra_filename, krate!()); }
1151
1152         DepKind::AllTraitImplementations => {
1153             force!(all_trait_implementations, krate!());
1154         }
1155
1156         DepKind::DllimportForeignItems => {
1157             force!(dllimport_foreign_items, krate!());
1158         }
1159         DepKind::IsDllimportForeignItem => {
1160             force!(is_dllimport_foreign_item, def_id!());
1161         }
1162         DepKind::IsStaticallyIncludedForeignItem => {
1163             force!(is_statically_included_foreign_item, def_id!());
1164         }
1165         DepKind::NativeLibraryKind => { force!(native_library_kind, def_id!()); }
1166         DepKind::LinkArgs => { force!(link_args, LOCAL_CRATE); }
1167
1168         DepKind::ResolveLifetimes => { force!(resolve_lifetimes, krate!()); }
1169         DepKind::NamedRegion => { force!(named_region_map, def_id!().index); }
1170         DepKind::IsLateBound => { force!(is_late_bound_map, def_id!().index); }
1171         DepKind::ObjectLifetimeDefaults => {
1172             force!(object_lifetime_defaults_map, def_id!().index);
1173         }
1174
1175         DepKind::Visibility => { force!(visibility, def_id!()); }
1176         DepKind::DepKind => { force!(dep_kind, krate!()); }
1177         DepKind::CrateName => { force!(crate_name, krate!()); }
1178         DepKind::ItemChildren => { force!(item_children, def_id!()); }
1179         DepKind::ExternModStmtCnum => { force!(extern_mod_stmt_cnum, def_id!()); }
1180         DepKind::GetLangItems => { force!(get_lang_items, LOCAL_CRATE); }
1181         DepKind::DefinedLangItems => { force!(defined_lang_items, krate!()); }
1182         DepKind::MissingLangItems => { force!(missing_lang_items, krate!()); }
1183         DepKind::VisibleParentMap => { force!(visible_parent_map, LOCAL_CRATE); }
1184         DepKind::MissingExternCrateItem => {
1185             force!(missing_extern_crate_item, krate!());
1186         }
1187         DepKind::UsedCrateSource => { force!(used_crate_source, krate!()); }
1188         DepKind::PostorderCnums => { force!(postorder_cnums, LOCAL_CRATE); }
1189
1190         DepKind::Freevars => { force!(freevars, def_id!()); }
1191         DepKind::MaybeUnusedTraitImport => {
1192             force!(maybe_unused_trait_import, def_id!());
1193         }
1194         DepKind::MaybeUnusedExternCrates => { force!(maybe_unused_extern_crates, LOCAL_CRATE); }
1195         DepKind::StabilityIndex => { force!(stability_index, LOCAL_CRATE); }
1196         DepKind::AllTraits => { force!(all_traits, LOCAL_CRATE); }
1197         DepKind::AllCrateNums => { force!(all_crate_nums, LOCAL_CRATE); }
1198         DepKind::ExportedSymbols => { force!(exported_symbols, krate!()); }
1199         DepKind::CollectAndPartitionMonoItems => {
1200             force!(collect_and_partition_mono_items, LOCAL_CRATE);
1201         }
1202         DepKind::IsCodegenedItem => { force!(is_codegened_item, def_id!()); }
1203         DepKind::OutputFilenames => { force!(output_filenames, LOCAL_CRATE); }
1204
1205         DepKind::TargetFeaturesWhitelist => { force!(target_features_whitelist, LOCAL_CRATE); }
1206
1207         DepKind::Features => { force!(features_query, LOCAL_CRATE); }
1208
1209         DepKind::ProgramClausesFor => { force!(program_clauses_for, def_id!()); }
1210         DepKind::WasmCustomSections => { force!(wasm_custom_sections, krate!()); }
1211         DepKind::WasmImportModuleMap => { force!(wasm_import_module_map, krate!()); }
1212         DepKind::ForeignModules => { force!(foreign_modules, krate!()); }
1213
1214         DepKind::UpstreamMonomorphizations => {
1215             force!(upstream_monomorphizations, krate!());
1216         }
1217         DepKind::UpstreamMonomorphizationsFor => {
1218             force!(upstream_monomorphizations_for, def_id!());
1219         }
1220     }
1221
1222     true
1223 }
1224
1225
1226 // FIXME(#45015): Another piece of boilerplate code that could be generated in
1227 //                a combined define_dep_nodes!()/define_queries!() macro.
1228 macro_rules! impl_load_from_cache {
1229     ($($dep_kind:ident => $query_name:ident,)*) => {
1230         impl DepNode {
1231             // Check whether the query invocation corresponding to the given
1232             // DepNode is eligible for on-disk-caching.
1233             pub fn cache_on_disk(&self, tcx: TyCtxt) -> bool {
1234                 use ty::query::queries;
1235                 use ty::query::QueryDescription;
1236
1237                 match self.kind {
1238                     $(DepKind::$dep_kind => {
1239                         let def_id = self.extract_def_id(tcx).unwrap();
1240                         queries::$query_name::cache_on_disk(def_id)
1241                     })*
1242                     _ => false
1243                 }
1244             }
1245
1246             // This is method will execute the query corresponding to the given
1247             // DepNode. It is only expected to work for DepNodes where the
1248             // above `cache_on_disk` methods returns true.
1249             // Also, as a sanity check, it expects that the corresponding query
1250             // invocation has been marked as green already.
1251             pub fn load_from_on_disk_cache(&self, tcx: TyCtxt) {
1252                 match self.kind {
1253                     $(DepKind::$dep_kind => {
1254                         debug_assert!(tcx.dep_graph
1255                                          .node_color(self)
1256                                          .map(|c| c.is_green())
1257                                          .unwrap_or(false));
1258
1259                         let def_id = self.extract_def_id(tcx).unwrap();
1260                         let _ = tcx.$query_name(def_id);
1261                     })*
1262                     _ => {
1263                         bug!()
1264                     }
1265                 }
1266             }
1267         }
1268     }
1269 }
1270
1271 impl_load_from_cache!(
1272     TypeckTables => typeck_tables_of,
1273     MirOptimized => optimized_mir,
1274     UnsafetyCheckResult => unsafety_check_result,
1275     BorrowCheck => borrowck,
1276     MirBorrowCheck => mir_borrowck,
1277     MirConstQualif => mir_const_qualif,
1278     SymbolName => def_symbol_name,
1279     ConstIsRvaluePromotableToStatic => const_is_rvalue_promotable_to_static,
1280     CheckMatch => check_match,
1281     TypeOfItem => type_of,
1282     GenericsOfItem => generics_of,
1283     PredicatesOfItem => predicates_of,
1284     UsedTraitImports => used_trait_imports,
1285     CodegenFnAttrs => codegen_fn_attrs,
1286     SpecializationGraph => specialization_graph_of,
1287 );