1 //! The home of `HirDatabase`, which is the Salsa database containing all the
2 //! type inference-related queries.
6 use base_db::{impl_intern_key, salsa, CrateId, Upcast};
10 layout::{Layout, LayoutError, TargetDataLayout},
11 AdtId, BlockId, ConstId, ConstParamId, DefWithBodyId, EnumVariantId, FunctionId, GenericDefId,
12 ImplId, LifetimeParamId, LocalFieldId, TypeOrConstParamId, VariantId,
14 use la_arena::ArenaMap;
15 use smallvec::SmallVec;
19 consteval::{ComputedExpr, ConstEvalError},
20 method_resolution::{InherentImpls, TraitImpls, TyFingerprint},
21 Binders, CallableDefId, FnDefId, GenericArg, ImplTraitId, InferenceResult, Interner, PolyFnSig,
22 QuantifiedWhereClause, ReturnTypeImplTraits, Substitution, TraitRef, Ty, TyDefId, ValueTyDefId,
24 use hir_expand::name::Name;
26 #[salsa::query_group(HirDatabaseStorage)]
27 pub trait HirDatabase: DefDatabase + Upcast<dyn DefDatabase> {
28 #[salsa::invoke(infer_wait)]
30 fn infer(&self, def: DefWithBodyId) -> Arc<InferenceResult>;
32 #[salsa::invoke(crate::infer::infer_query)]
33 fn infer_query(&self, def: DefWithBodyId) -> Arc<InferenceResult>;
35 #[salsa::invoke(crate::lower::ty_query)]
36 #[salsa::cycle(crate::lower::ty_recover)]
37 fn ty(&self, def: TyDefId) -> Binders<Ty>;
39 #[salsa::invoke(crate::lower::value_ty_query)]
40 fn value_ty(&self, def: ValueTyDefId) -> Binders<Ty>;
42 #[salsa::invoke(crate::lower::impl_self_ty_query)]
43 #[salsa::cycle(crate::lower::impl_self_ty_recover)]
44 fn impl_self_ty(&self, def: ImplId) -> Binders<Ty>;
46 #[salsa::invoke(crate::lower::const_param_ty_query)]
47 fn const_param_ty(&self, def: ConstParamId) -> Ty;
49 #[salsa::invoke(crate::consteval::const_eval_variant_query)]
50 #[salsa::cycle(crate::consteval::const_eval_recover)]
51 fn const_eval(&self, def: ConstId) -> Result<ComputedExpr, ConstEvalError>;
53 #[salsa::invoke(crate::consteval::const_eval_query_variant)]
54 #[salsa::cycle(crate::consteval::const_eval_variant_recover)]
55 fn const_eval_variant(&self, def: EnumVariantId) -> Result<ComputedExpr, ConstEvalError>;
57 #[salsa::invoke(crate::lower::impl_trait_query)]
58 fn impl_trait(&self, def: ImplId) -> Option<Binders<TraitRef>>;
60 #[salsa::invoke(crate::lower::field_types_query)]
61 fn field_types(&self, var: VariantId) -> Arc<ArenaMap<LocalFieldId, Binders<Ty>>>;
63 #[salsa::invoke(crate::layout::layout_of_adt_query)]
64 #[salsa::cycle(crate::layout::layout_of_adt_recover)]
65 fn layout_of_adt(&self, def: AdtId, subst: Substitution) -> Result<Layout, LayoutError>;
67 #[salsa::invoke(crate::layout::target_data_layout_query)]
68 fn target_data_layout(&self, krate: CrateId) -> Arc<TargetDataLayout>;
70 #[salsa::invoke(crate::lower::callable_item_sig)]
71 fn callable_item_signature(&self, def: CallableDefId) -> PolyFnSig;
73 #[salsa::invoke(crate::lower::return_type_impl_traits)]
74 fn return_type_impl_traits(
77 ) -> Option<Arc<Binders<ReturnTypeImplTraits>>>;
79 #[salsa::invoke(crate::lower::generic_predicates_for_param_query)]
80 #[salsa::cycle(crate::lower::generic_predicates_for_param_recover)]
81 fn generic_predicates_for_param(
84 param_id: TypeOrConstParamId,
85 assoc_name: Option<Name>,
86 ) -> Arc<[Binders<QuantifiedWhereClause>]>;
88 #[salsa::invoke(crate::lower::generic_predicates_query)]
89 fn generic_predicates(&self, def: GenericDefId) -> Arc<[Binders<QuantifiedWhereClause>]>;
91 #[salsa::invoke(crate::lower::trait_environment_query)]
92 fn trait_environment(&self, def: GenericDefId) -> Arc<crate::TraitEnvironment>;
94 #[salsa::invoke(crate::lower::generic_defaults_query)]
95 #[salsa::cycle(crate::lower::generic_defaults_recover)]
96 fn generic_defaults(&self, def: GenericDefId) -> Arc<[Binders<GenericArg>]>;
98 #[salsa::invoke(InherentImpls::inherent_impls_in_crate_query)]
99 fn inherent_impls_in_crate(&self, krate: CrateId) -> Arc<InherentImpls>;
101 #[salsa::invoke(InherentImpls::inherent_impls_in_block_query)]
102 fn inherent_impls_in_block(&self, block: BlockId) -> Option<Arc<InherentImpls>>;
104 /// Collects all crates in the dependency graph that have impls for the
105 /// given fingerprint. This is only used for primitive types and types
106 /// annotated with `rustc_has_incoherent_inherent_impls`; for other types
107 /// we just look at the crate where the type is defined.
108 #[salsa::invoke(crate::method_resolution::incoherent_inherent_impl_crates)]
109 fn incoherent_inherent_impl_crates(
113 ) -> SmallVec<[CrateId; 2]>;
115 #[salsa::invoke(TraitImpls::trait_impls_in_crate_query)]
116 fn trait_impls_in_crate(&self, krate: CrateId) -> Arc<TraitImpls>;
118 #[salsa::invoke(TraitImpls::trait_impls_in_block_query)]
119 fn trait_impls_in_block(&self, krate: BlockId) -> Option<Arc<TraitImpls>>;
121 #[salsa::invoke(TraitImpls::trait_impls_in_deps_query)]
122 fn trait_impls_in_deps(&self, krate: CrateId) -> Arc<TraitImpls>;
124 // Interned IDs for Chalk integration
126 fn intern_callable_def(&self, callable_def: CallableDefId) -> InternedCallableDefId;
128 fn intern_type_or_const_param_id(
130 param_id: TypeOrConstParamId,
131 ) -> InternedTypeOrConstParamId;
133 fn intern_lifetime_param_id(&self, param_id: LifetimeParamId) -> InternedLifetimeParamId;
135 fn intern_impl_trait_id(&self, id: ImplTraitId) -> InternedOpaqueTyId;
137 fn intern_closure(&self, id: (DefWithBodyId, ExprId)) -> InternedClosureId;
139 fn intern_generator(&self, id: (DefWithBodyId, ExprId)) -> InternedGeneratorId;
141 #[salsa::invoke(chalk_db::associated_ty_data_query)]
142 fn associated_ty_data(&self, id: chalk_db::AssocTypeId) -> Arc<chalk_db::AssociatedTyDatum>;
144 #[salsa::invoke(chalk_db::trait_datum_query)]
145 fn trait_datum(&self, krate: CrateId, trait_id: chalk_db::TraitId)
146 -> Arc<chalk_db::TraitDatum>;
148 #[salsa::invoke(chalk_db::struct_datum_query)]
152 struct_id: chalk_db::AdtId,
153 ) -> Arc<chalk_db::StructDatum>;
155 #[salsa::invoke(chalk_db::impl_datum_query)]
156 fn impl_datum(&self, krate: CrateId, impl_id: chalk_db::ImplId) -> Arc<chalk_db::ImplDatum>;
158 #[salsa::invoke(chalk_db::fn_def_datum_query)]
159 fn fn_def_datum(&self, krate: CrateId, fn_def_id: FnDefId) -> Arc<chalk_db::FnDefDatum>;
161 #[salsa::invoke(chalk_db::fn_def_variance_query)]
162 fn fn_def_variance(&self, fn_def_id: FnDefId) -> chalk_db::Variances;
164 #[salsa::invoke(chalk_db::adt_variance_query)]
165 fn adt_variance(&self, adt_id: chalk_db::AdtId) -> chalk_db::Variances;
167 #[salsa::invoke(chalk_db::associated_ty_value_query)]
168 fn associated_ty_value(
171 id: chalk_db::AssociatedTyValueId,
172 ) -> Arc<chalk_db::AssociatedTyValue>;
174 #[salsa::invoke(crate::traits::normalize_projection_query)]
175 #[salsa::transparent]
176 fn normalize_projection(
178 projection: crate::ProjectionTy,
179 env: Arc<crate::TraitEnvironment>,
182 #[salsa::invoke(trait_solve_wait)]
183 #[salsa::transparent]
187 goal: crate::Canonical<crate::InEnvironment<crate::Goal>>,
188 ) -> Option<crate::Solution>;
190 #[salsa::invoke(crate::traits::trait_solve_query)]
191 fn trait_solve_query(
194 goal: crate::Canonical<crate::InEnvironment<crate::Goal>>,
195 ) -> Option<crate::Solution>;
197 #[salsa::invoke(chalk_db::program_clauses_for_chalk_env_query)]
198 fn program_clauses_for_chalk_env(
201 env: chalk_ir::Environment<Interner>,
202 ) -> chalk_ir::ProgramClauses<Interner>;
205 fn infer_wait(db: &dyn HirDatabase, def: DefWithBodyId) -> Arc<InferenceResult> {
206 let _p = profile::span("infer:wait").detail(|| match def {
207 DefWithBodyId::FunctionId(it) => db.function_data(it).name.to_string(),
208 DefWithBodyId::StaticId(it) => db.static_data(it).name.clone().to_string(),
209 DefWithBodyId::ConstId(it) => {
210 db.const_data(it).name.clone().unwrap_or_else(Name::missing).to_string()
212 DefWithBodyId::VariantId(it) => {
213 db.enum_data(it.parent).variants[it.local_id].name.to_string()
220 db: &dyn HirDatabase,
222 goal: crate::Canonical<crate::InEnvironment<crate::Goal>>,
223 ) -> Option<crate::Solution> {
224 let _p = profile::span("trait_solve::wait");
225 db.trait_solve_query(krate, goal)
229 fn hir_database_is_object_safe() {
230 fn _assert_object_safe(_: &dyn HirDatabase) {}
233 #[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
234 pub struct InternedTypeOrConstParamId(salsa::InternId);
235 impl_intern_key!(InternedTypeOrConstParamId);
237 #[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
238 pub struct InternedLifetimeParamId(salsa::InternId);
239 impl_intern_key!(InternedLifetimeParamId);
241 #[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
242 pub struct InternedConstParamId(salsa::InternId);
243 impl_intern_key!(InternedConstParamId);
245 #[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
246 pub struct InternedOpaqueTyId(salsa::InternId);
247 impl_intern_key!(InternedOpaqueTyId);
249 #[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
250 pub struct InternedClosureId(salsa::InternId);
251 impl_intern_key!(InternedClosureId);
253 #[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
254 pub struct InternedGeneratorId(salsa::InternId);
255 impl_intern_key!(InternedGeneratorId);
257 /// This exists just for Chalk, because Chalk just has a single `FnDefId` where
258 /// we have different IDs for struct and enum variant constructors.
259 #[derive(Debug, Clone, Copy, PartialEq, Eq, Hash, Ord, PartialOrd)]
260 pub struct InternedCallableDefId(salsa::InternId);
261 impl_intern_key!(InternedCallableDefId);