crates/hir_def/src/db.rs

   1 //! Defines database & queries for name resolution.
   2 use std::sync::Arc;
   3
   4 use base_db::{salsa, CrateId, SourceDatabase, Upcast};
   5 use hir_expand::{db::AstDatabase, HirFileId};
   6 use syntax::SmolStr;
   7
   8 use crate::{
   9     adt::{EnumData, StructData},
  10     attr::Attrs,
  11     body::{scope::ExprScopes, Body, BodySourceMap},
  12     data::{ConstData, FunctionData, ImplData, StaticData, TraitData, TypeAliasData},
  13     generics::GenericParams,
  14     import_map::ImportMap,
  15     item_tree::ItemTree,
  16     lang_item::{LangItemTarget, LangItems},
  17     nameres::CrateDefMap,
  18     AttrDefId, ConstId, ConstLoc, DefWithBodyId, EnumId, EnumLoc, FunctionId, FunctionLoc,
  19     GenericDefId, ImplId, ImplLoc, StaticId, StaticLoc, StructId, StructLoc, TraitId, TraitLoc,
  20     TypeAliasId, TypeAliasLoc, UnionId, UnionLoc,
  21 };
  22
  23 #[salsa::query_group(InternDatabaseStorage)]
  24 pub trait InternDatabase: SourceDatabase {
  25     #[salsa::interned]
  26     fn intern_function(&self, loc: FunctionLoc) -> FunctionId;
  27     #[salsa::interned]
  28     fn intern_struct(&self, loc: StructLoc) -> StructId;
  29     #[salsa::interned]
  30     fn intern_union(&self, loc: UnionLoc) -> UnionId;
  31     #[salsa::interned]
  32     fn intern_enum(&self, loc: EnumLoc) -> EnumId;
  33     #[salsa::interned]
  34     fn intern_const(&self, loc: ConstLoc) -> ConstId;
  35     #[salsa::interned]
  36     fn intern_static(&self, loc: StaticLoc) -> StaticId;
  37     #[salsa::interned]
  38     fn intern_trait(&self, loc: TraitLoc) -> TraitId;
  39     #[salsa::interned]
  40     fn intern_type_alias(&self, loc: TypeAliasLoc) -> TypeAliasId;
  41     #[salsa::interned]
  42     fn intern_impl(&self, loc: ImplLoc) -> ImplId;
  43 }
  44
  45 #[salsa::query_group(DefDatabaseStorage)]
  46 pub trait DefDatabase: InternDatabase + AstDatabase + Upcast<dyn AstDatabase> {
  47     #[salsa::invoke(ItemTree::item_tree_query)]
  48     fn item_tree(&self, file_id: HirFileId) -> Arc<ItemTree>;
  49
  50     #[salsa::invoke(crate_def_map_wait)]
  51     #[salsa::transparent]
  52     fn crate_def_map(&self, krate: CrateId) -> Arc<CrateDefMap>;
  53
  54     #[salsa::invoke(CrateDefMap::crate_def_map_query)]
  55     fn crate_def_map_query(&self, krate: CrateId) -> Arc<CrateDefMap>;
  56
  57     #[salsa::invoke(StructData::struct_data_query)]
  58     fn struct_data(&self, id: StructId) -> Arc<StructData>;
  59     #[salsa::invoke(StructData::union_data_query)]
  60     fn union_data(&self, id: UnionId) -> Arc<StructData>;
  61
  62     #[salsa::invoke(EnumData::enum_data_query)]
  63     fn enum_data(&self, e: EnumId) -> Arc<EnumData>;
  64
  65     #[salsa::invoke(ImplData::impl_data_query)]
  66     fn impl_data(&self, e: ImplId) -> Arc<ImplData>;
  67
  68     #[salsa::invoke(TraitData::trait_data_query)]
  69     fn trait_data(&self, e: TraitId) -> Arc<TraitData>;
  70
  71     #[salsa::invoke(TypeAliasData::type_alias_data_query)]
  72     fn type_alias_data(&self, e: TypeAliasId) -> Arc<TypeAliasData>;
  73
  74     #[salsa::invoke(FunctionData::fn_data_query)]
  75     fn function_data(&self, func: FunctionId) -> Arc<FunctionData>;
  76
  77     #[salsa::invoke(ConstData::const_data_query)]
  78     fn const_data(&self, konst: ConstId) -> Arc<ConstData>;
  79
  80     #[salsa::invoke(StaticData::static_data_query)]
  81     fn static_data(&self, konst: StaticId) -> Arc<StaticData>;
  82
  83     #[salsa::invoke(Body::body_with_source_map_query)]
  84     fn body_with_source_map(&self, def: DefWithBodyId) -> (Arc<Body>, Arc<BodySourceMap>);
  85
  86     #[salsa::invoke(Body::body_query)]
  87     fn body(&self, def: DefWithBodyId) -> Arc<Body>;
  88
  89     #[salsa::invoke(ExprScopes::expr_scopes_query)]
  90     fn expr_scopes(&self, def: DefWithBodyId) -> Arc<ExprScopes>;
  91
  92     #[salsa::invoke(GenericParams::generic_params_query)]
  93     fn generic_params(&self, def: GenericDefId) -> Arc<GenericParams>;
  94
  95     #[salsa::invoke(Attrs::attrs_query)]
  96     fn attrs(&self, def: AttrDefId) -> Attrs;
  97
  98     #[salsa::invoke(LangItems::crate_lang_items_query)]
  99     fn crate_lang_items(&self, krate: CrateId) -> Arc<LangItems>;
 100
 101     #[salsa::invoke(LangItems::lang_item_query)]
 102     fn lang_item(&self, start_crate: CrateId, item: SmolStr) -> Option<LangItemTarget>;
 103
 104     #[salsa::invoke(ImportMap::import_map_query)]
 105     fn import_map(&self, krate: CrateId) -> Arc<ImportMap>;
 106 }
 107
 108 fn crate_def_map_wait(db: &impl DefDatabase, krate: CrateId) -> Arc<CrateDefMap> {
 109     let _p = profile::span("crate_def_map:wait");
 110     db.crate_def_map_query(krate)
 111 }