rustc_query_impl/
lib.rs

1//! Support for serializing the dep-graph and reloading it.
2
3// tidy-alphabetical-start
4#![allow(internal_features)]
5#![feature(core_intrinsics)]
6#![feature(min_specialization)]
7#![feature(rustc_attrs)]
8#![feature(try_blocks)]
9// tidy-alphabetical-end
10
11use rustc_data_structures::sync::{AtomicU64, Lock};
12use rustc_middle::dep_graph;
13use rustc_middle::queries::{ExternProviders, Providers};
14use rustc_middle::query::on_disk_cache::OnDiskCache;
15use rustc_middle::query::{QueryCache, QuerySystem, QueryVTable};
16use rustc_middle::ty::TyCtxt;
17
18pub use crate::dep_kind_vtables::make_dep_kind_vtables;
19pub use crate::execution::{CollectActiveJobsKind, collect_active_query_jobs};
20pub use crate::job::{QueryJobMap, break_query_cycle, print_query_stack};
21
22mod dep_kind_vtables;
23mod error;
24mod execution;
25mod handle_cycle_error;
26mod job;
27mod plumbing;
28mod profiling_support;
29mod query_impl;
30
31/// Trait that knows how to look up the [`QueryVTable`] for a particular query.
32///
33/// This trait allows some per-query code to be defined in generic functions
34/// with a trait bound, instead of having to be defined inline within a macro
35/// expansion.
36///
37/// There is one macro-generated implementation of this trait for each query,
38/// on the type `rustc_query_impl::query_impl::$name::VTableGetter`.
39trait GetQueryVTable<'tcx> {
40    type Cache: QueryCache + 'tcx;
41
42    fn query_vtable(tcx: TyCtxt<'tcx>) -> &'tcx QueryVTable<'tcx, Self::Cache>;
43}
44
45pub fn query_system<'tcx>(
46    local_providers: Providers,
47    extern_providers: ExternProviders,
48    on_disk_cache: Option<OnDiskCache>,
49    incremental: bool,
50) -> QuerySystem<'tcx> {
51    QuerySystem {
52        arenas: Default::default(),
53        query_vtables: query_impl::make_query_vtables(incremental),
54        side_effects: Default::default(),
55        on_disk_cache,
56        local_providers,
57        extern_providers,
58        jobs: AtomicU64::new(1),
59        cycle_handler_nesting: Lock::new(0),
60    }
61}
62
63pub fn provide(providers: &mut rustc_middle::util::Providers) {
64    providers.hooks.alloc_self_profile_query_strings =
65        profiling_support::alloc_self_profile_query_strings;
66    providers.hooks.verify_query_key_hashes = plumbing::verify_query_key_hashes;
67    providers.hooks.encode_query_values = plumbing::encode_query_values;
68}
rustc_query_impl/lib.rs

rustc_query_impl/
lib.rs
