| //! Defines the set of legal keys that can be used in queries. |
| |
| use crate::infer::canonical::Canonical; |
| use crate::mir; |
| use crate::traits; |
| use crate::ty::fast_reject::SimplifiedType; |
| use crate::ty::subst::{GenericArg, SubstsRef}; |
| use crate::ty::{self, Ty, TyCtxt}; |
| use rustc_hir::def_id::{CrateNum, DefId, LocalDefId, LOCAL_CRATE}; |
| use rustc_query_system::query::DefaultCacheSelector; |
| use rustc_span::symbol::Symbol; |
| use rustc_span::{Span, DUMMY_SP}; |
| |
| /// The `Key` trait controls what types can legally be used as the key |
| /// for a query. |
| pub trait Key { |
| type CacheSelector; |
| |
| /// Given an instance of this key, what crate is it referring to? |
| /// This is used to find the provider. |
| fn query_crate(&self) -> CrateNum; |
| |
| /// In the event that a cycle occurs, if no explicit span has been |
| /// given for a query with key `self`, what span should we use? |
| fn default_span(&self, tcx: TyCtxt<'_>) -> Span; |
| } |
| |
| impl<'tcx> Key for ty::InstanceDef<'tcx> { |
| type CacheSelector = DefaultCacheSelector; |
| |
| fn query_crate(&self) -> CrateNum { |
| LOCAL_CRATE |
| } |
| |
| fn default_span(&self, tcx: TyCtxt<'_>) -> Span { |
| tcx.def_span(self.def_id()) |
| } |
| } |
| |
| impl<'tcx> Key for ty::Instance<'tcx> { |
| type CacheSelector = DefaultCacheSelector; |
| |
| fn query_crate(&self) -> CrateNum { |
| LOCAL_CRATE |
| } |
| |
| fn default_span(&self, tcx: TyCtxt<'_>) -> Span { |
| tcx.def_span(self.def_id()) |
| } |
| } |
| |
| impl<'tcx> Key for mir::interpret::GlobalId<'tcx> { |
| type CacheSelector = DefaultCacheSelector; |
| |
| fn query_crate(&self) -> CrateNum { |
| self.instance.query_crate() |
| } |
| |
| fn default_span(&self, tcx: TyCtxt<'_>) -> Span { |
| self.instance.default_span(tcx) |
| } |
| } |
| |
| impl<'tcx> Key for mir::interpret::LitToConstInput<'tcx> { |
| type CacheSelector = DefaultCacheSelector; |
| |
| fn query_crate(&self) -> CrateNum { |
| LOCAL_CRATE |
| } |
| |
| fn default_span(&self, _tcx: TyCtxt<'_>) -> Span { |
| DUMMY_SP |
| } |
| } |
| |
| impl Key for CrateNum { |
| type CacheSelector = DefaultCacheSelector; |
| |
| fn query_crate(&self) -> CrateNum { |
| *self |
| } |
| fn default_span(&self, _: TyCtxt<'_>) -> Span { |
| DUMMY_SP |
| } |
| } |
| |
| impl Key for LocalDefId { |
| type CacheSelector = DefaultCacheSelector; |
| |
| fn query_crate(&self) -> CrateNum { |
| self.to_def_id().query_crate() |
| } |
| fn default_span(&self, tcx: TyCtxt<'_>) -> Span { |
| self.to_def_id().default_span(tcx) |
| } |
| } |
| |
| impl Key for DefId { |
| type CacheSelector = DefaultCacheSelector; |
| |
| fn query_crate(&self) -> CrateNum { |
| self.krate |
| } |
| fn default_span(&self, tcx: TyCtxt<'_>) -> Span { |
| tcx.def_span(*self) |
| } |
| } |
| |
| impl Key for (DefId, DefId) { |
| type CacheSelector = DefaultCacheSelector; |
| |
| fn query_crate(&self) -> CrateNum { |
| self.0.krate |
| } |
| fn default_span(&self, tcx: TyCtxt<'_>) -> Span { |
| self.1.default_span(tcx) |
| } |
| } |
| |
| impl Key for (CrateNum, DefId) { |
| type CacheSelector = DefaultCacheSelector; |
| |
| fn query_crate(&self) -> CrateNum { |
| self.0 |
| } |
| fn default_span(&self, tcx: TyCtxt<'_>) -> Span { |
| self.1.default_span(tcx) |
| } |
| } |
| |
| impl Key for (DefId, SimplifiedType) { |
| type CacheSelector = DefaultCacheSelector; |
| |
| fn query_crate(&self) -> CrateNum { |
| self.0.krate |
| } |
| fn default_span(&self, tcx: TyCtxt<'_>) -> Span { |
| self.0.default_span(tcx) |
| } |
| } |
| |
| impl<'tcx> Key for SubstsRef<'tcx> { |
| type CacheSelector = DefaultCacheSelector; |
| |
| fn query_crate(&self) -> CrateNum { |
| LOCAL_CRATE |
| } |
| fn default_span(&self, _: TyCtxt<'_>) -> Span { |
| DUMMY_SP |
| } |
| } |
| |
| impl<'tcx> Key for (DefId, SubstsRef<'tcx>) { |
| type CacheSelector = DefaultCacheSelector; |
| |
| fn query_crate(&self) -> CrateNum { |
| self.0.krate |
| } |
| fn default_span(&self, tcx: TyCtxt<'_>) -> Span { |
| self.0.default_span(tcx) |
| } |
| } |
| |
| impl<'tcx> Key for (ty::ParamEnv<'tcx>, ty::PolyTraitRef<'tcx>) { |
| type CacheSelector = DefaultCacheSelector; |
| |
| fn query_crate(&self) -> CrateNum { |
| self.1.def_id().krate |
| } |
| fn default_span(&self, tcx: TyCtxt<'_>) -> Span { |
| tcx.def_span(self.1.def_id()) |
| } |
| } |
| |
| impl<'tcx> Key for (&'tcx ty::Const<'tcx>, mir::Field) { |
| type CacheSelector = DefaultCacheSelector; |
| |
| fn query_crate(&self) -> CrateNum { |
| LOCAL_CRATE |
| } |
| fn default_span(&self, _: TyCtxt<'_>) -> Span { |
| DUMMY_SP |
| } |
| } |
| |
| impl<'tcx> Key for ty::PolyTraitRef<'tcx> { |
| type CacheSelector = DefaultCacheSelector; |
| |
| fn query_crate(&self) -> CrateNum { |
| self.def_id().krate |
| } |
| fn default_span(&self, tcx: TyCtxt<'_>) -> Span { |
| tcx.def_span(self.def_id()) |
| } |
| } |
| |
| impl<'tcx> Key for GenericArg<'tcx> { |
| type CacheSelector = DefaultCacheSelector; |
| |
| fn query_crate(&self) -> CrateNum { |
| LOCAL_CRATE |
| } |
| fn default_span(&self, _: TyCtxt<'_>) -> Span { |
| DUMMY_SP |
| } |
| } |
| |
| impl<'tcx> Key for &'tcx ty::Const<'tcx> { |
| type CacheSelector = DefaultCacheSelector; |
| |
| fn query_crate(&self) -> CrateNum { |
| LOCAL_CRATE |
| } |
| fn default_span(&self, _: TyCtxt<'_>) -> Span { |
| DUMMY_SP |
| } |
| } |
| |
| impl<'tcx> Key for Ty<'tcx> { |
| type CacheSelector = DefaultCacheSelector; |
| |
| fn query_crate(&self) -> CrateNum { |
| LOCAL_CRATE |
| } |
| fn default_span(&self, _: TyCtxt<'_>) -> Span { |
| DUMMY_SP |
| } |
| } |
| |
| impl<'tcx> Key for ty::ParamEnv<'tcx> { |
| type CacheSelector = DefaultCacheSelector; |
| |
| fn query_crate(&self) -> CrateNum { |
| LOCAL_CRATE |
| } |
| fn default_span(&self, _: TyCtxt<'_>) -> Span { |
| DUMMY_SP |
| } |
| } |
| |
| impl<'tcx, T: Key> Key for ty::ParamEnvAnd<'tcx, T> { |
| type CacheSelector = DefaultCacheSelector; |
| |
| fn query_crate(&self) -> CrateNum { |
| self.value.query_crate() |
| } |
| fn default_span(&self, tcx: TyCtxt<'_>) -> Span { |
| self.value.default_span(tcx) |
| } |
| } |
| |
| impl<'tcx> Key for traits::Environment<'tcx> { |
| type CacheSelector = DefaultCacheSelector; |
| |
| fn query_crate(&self) -> CrateNum { |
| LOCAL_CRATE |
| } |
| fn default_span(&self, _: TyCtxt<'_>) -> Span { |
| DUMMY_SP |
| } |
| } |
| |
| impl Key for Symbol { |
| type CacheSelector = DefaultCacheSelector; |
| |
| fn query_crate(&self) -> CrateNum { |
| LOCAL_CRATE |
| } |
| fn default_span(&self, _tcx: TyCtxt<'_>) -> Span { |
| DUMMY_SP |
| } |
| } |
| |
| /// Canonical query goals correspond to abstract trait operations that |
| /// are not tied to any crate in particular. |
| impl<'tcx, T> Key for Canonical<'tcx, T> { |
| type CacheSelector = DefaultCacheSelector; |
| |
| fn query_crate(&self) -> CrateNum { |
| LOCAL_CRATE |
| } |
| |
| fn default_span(&self, _tcx: TyCtxt<'_>) -> Span { |
| DUMMY_SP |
| } |
| } |
| |
| impl Key for (Symbol, u32, u32) { |
| type CacheSelector = DefaultCacheSelector; |
| |
| fn query_crate(&self) -> CrateNum { |
| LOCAL_CRATE |
| } |
| |
| fn default_span(&self, _tcx: TyCtxt<'_>) -> Span { |
| DUMMY_SP |
| } |
| } |