mirror of
https://github.com/rust-lang/rust.git
synced 2026-04-27 18:57:42 +03:00
Nicholas Nethercote
2bc5167e99
`QueryEngine` is a struct with one function pointer per query. This commit removes it by moving each function pointer into the relevant query's vtable, which is already a collection of function pointers for that query. This makes things simpler.
78 lines
2.6 KiB
Rust
78 lines
2.6 KiB
Rust
//! Support for serializing the dep-graph and reloading it.
|
|
|
|
// tidy-alphabetical-start
|
|
#![allow(internal_features)]
|
|
#![feature(core_intrinsics)]
|
|
#![feature(min_specialization)]
|
|
#![feature(rustc_attrs)]
|
|
#![feature(try_blocks)]
|
|
// tidy-alphabetical-end
|
|
|
|
use rustc_data_structures::sync::AtomicU64;
|
|
use rustc_middle::dep_graph;
|
|
use rustc_middle::queries::{self, ExternProviders, Providers};
|
|
use rustc_middle::query::on_disk_cache::{CacheEncoder, EncodedDepNodeIndex, OnDiskCache};
|
|
use rustc_middle::query::plumbing::{QuerySystem, QuerySystemFns, QueryVTable};
|
|
use rustc_middle::query::{AsLocalKey, QueryCache, QueryMode};
|
|
use rustc_middle::ty::TyCtxt;
|
|
use rustc_span::Span;
|
|
|
|
pub use crate::dep_kind_vtables::make_dep_kind_vtables;
|
|
pub use crate::job::{QueryJobMap, break_query_cycles, print_query_stack};
|
|
pub use crate::plumbing::{collect_active_jobs_from_all_queries, query_key_hash_verify_all};
|
|
use crate::plumbing::{encode_all_query_results, try_mark_green};
|
|
use crate::profiling_support::QueryKeyStringCache;
|
|
pub use crate::profiling_support::alloc_self_profile_query_strings;
|
|
use crate::values::Value;
|
|
|
|
#[macro_use]
|
|
mod plumbing;
|
|
|
|
mod dep_kind_vtables;
|
|
mod error;
|
|
mod execution;
|
|
mod job;
|
|
mod profiling_support;
|
|
mod values;
|
|
|
|
/// Trait that knows how to look up the [`QueryVTable`] for a particular query.
|
|
///
|
|
/// This trait allows some per-query code to be defined in generic functions
|
|
/// with a trait bound, instead of having to be defined inline within a macro
|
|
/// expansion.
|
|
///
|
|
/// There is one macro-generated implementation of this trait for each query,
|
|
/// on the type `rustc_query_impl::query_impl::$name::VTableGetter`.
|
|
trait GetQueryVTable<'tcx> {
|
|
type Cache: QueryCache + 'tcx;
|
|
|
|
fn query_vtable(tcx: TyCtxt<'tcx>) -> &'tcx QueryVTable<'tcx, Self::Cache>;
|
|
}
|
|
|
|
pub fn query_system<'tcx>(
|
|
local_providers: Providers,
|
|
extern_providers: ExternProviders,
|
|
on_disk_cache: Option<OnDiskCache>,
|
|
incremental: bool,
|
|
) -> QuerySystem<'tcx> {
|
|
QuerySystem {
|
|
arenas: Default::default(),
|
|
query_vtables: make_query_vtables(incremental),
|
|
on_disk_cache,
|
|
fns: QuerySystemFns {
|
|
local_providers,
|
|
extern_providers,
|
|
encode_query_results: encode_all_query_results,
|
|
try_mark_green,
|
|
},
|
|
jobs: AtomicU64::new(1),
|
|
}
|
|
}
|
|
|
|
rustc_middle::rustc_with_all_queries! { define_queries! }
|
|
|
|
pub fn provide(providers: &mut rustc_middle::util::Providers) {
|
|
providers.hooks.alloc_self_profile_query_strings = alloc_self_profile_query_strings;
|
|
providers.hooks.query_key_hash_verify_all = query_key_hash_verify_all;
|
|
}
|