rust/crates/ra_hir_def/src/db.rs

//! Defines database & queries for name resolution.
use std::sync::Arc;

use hir_expand::{db::AstDatabase, HirFileId};
use ra_db::{salsa, CrateId, SourceDatabase};
use ra_syntax::ast;

use crate::{
    adt::{EnumData, StructData},
    body::{scope::ExprScopes, Body, BodySourceMap},
    impls::ImplData,
    nameres::{
        raw::{ImportSourceMap, RawItems},
        CrateDefMap,
    },
    traits::{TraitData, TraitItemsIndex},
    DefWithBodyId, EnumId, FunctionLoc, ImplId, ItemLoc, ModuleId, StructOrUnionId, TraitId,
};

#[salsa::query_group(InternDatabaseStorage)]
pub trait InternDatabase: SourceDatabase {
    #[salsa::interned]
    fn intern_function(&self, loc: FunctionLoc) -> crate::FunctionId;
    #[salsa::interned]
    fn intern_struct_or_union(&self, loc: ItemLoc<ast::StructDef>) -> crate::StructOrUnionId;
    #[salsa::interned]
    fn intern_enum(&self, loc: ItemLoc<ast::EnumDef>) -> crate::EnumId;
    #[salsa::interned]
    fn intern_const(&self, loc: ItemLoc<ast::ConstDef>) -> crate::ConstId;
    #[salsa::interned]
    fn intern_static(&self, loc: ItemLoc<ast::StaticDef>) -> crate::StaticId;
    #[salsa::interned]
    fn intern_trait(&self, loc: ItemLoc<ast::TraitDef>) -> crate::TraitId;
    #[salsa::interned]
    fn intern_type_alias(&self, loc: ItemLoc<ast::TypeAliasDef>) -> crate::TypeAliasId;
    #[salsa::interned]
    fn intern_impl(&self, loc: ItemLoc<ast::ImplBlock>) -> crate::ImplId;
}

#[salsa::query_group(DefDatabase2Storage)]
pub trait DefDatabase2: InternDatabase + AstDatabase {
    #[salsa::invoke(RawItems::raw_items_with_source_map_query)]
    fn raw_items_with_source_map(
        &self,
        file_id: HirFileId,
    ) -> (Arc<RawItems>, Arc<ImportSourceMap>);

    #[salsa::invoke(RawItems::raw_items_query)]
    fn raw_items(&self, file_id: HirFileId) -> Arc<RawItems>;

    #[salsa::invoke(CrateDefMap::crate_def_map_query)]
    fn crate_def_map(&self, krate: CrateId) -> Arc<CrateDefMap>;

    #[salsa::invoke(StructData::struct_data_query)]
    fn struct_data(&self, id: StructOrUnionId) -> Arc<StructData>;

    #[salsa::invoke(EnumData::enum_data_query)]
    fn enum_data(&self, e: EnumId) -> Arc<EnumData>;

    #[salsa::invoke(ImplData::impl_data_query)]
    fn impl_data(&self, e: ImplId) -> Arc<ImplData>;

    #[salsa::invoke(TraitData::trait_data_query)]
    fn trait_data(&self, e: TraitId) -> Arc<TraitData>;

    #[salsa::invoke(TraitItemsIndex::trait_items_index)]
    fn trait_items_index(&self, module: ModuleId) -> TraitItemsIndex;

    #[salsa::invoke(Body::body_with_source_map_query)]
    fn body_with_source_map(&self, def: DefWithBodyId) -> (Arc<Body>, Arc<BodySourceMap>);

    #[salsa::invoke(Body::body_query)]
    fn body(&self, def: DefWithBodyId) -> Arc<Body>;

    #[salsa::invoke(ExprScopes::expr_scopes_query)]
    fn expr_scopes(&self, def: DefWithBodyId) -> Arc<ExprScopes>;
}
Move ids to hir_def crate 2019-10-30 10:10:38 +00:00			`//! Defines database & queries for name resolution.`
move raw_items to hir_def 2019-10-30 13:12:55 +00:00			`use std::sync::Arc;`
Move ids to hir_def crate 2019-10-30 10:10:38 +00:00
move raw_items to hir_def 2019-10-30 13:12:55 +00:00			`use hir_expand::{db::AstDatabase, HirFileId};`
Move CrateDefMap to hir_def 2019-10-31 15:45:10 +00:00			`use ra_db::{salsa, CrateId, SourceDatabase};`
Move ids to hir_def crate 2019-10-30 10:10:38 +00:00			`use ra_syntax::ast;`

move struct & enum data to hir_def 2019-10-31 13:40:36 +00:00			`use crate::{`
			`adt::{EnumData, StructData},`
Move body queries to hir_def 2019-11-14 14:37:22 +00:00			`body::{scope::ExprScopes, Body, BodySourceMap},`
Move traits to hir_def 2019-11-20 11:22:06 +00:00			`impls::ImplData,`
Move CrateDefMap to hir_def 2019-10-31 15:45:10 +00:00			`nameres::{`
			`raw::{ImportSourceMap, RawItems},`
			`CrateDefMap,`
			`},`
Move traits to hir_def 2019-11-20 11:22:06 +00:00			`traits::{TraitData, TraitItemsIndex},`
Next gen IDs for functions The current system with AstIds has two primaraly drawbacks: * It is possible to manufacture IDs out of thin air. For example, it's possible to create IDs for items which are not considered in CrateDefMap due to cfg. Or it is possible to mixup structs and unions, because they share ID space. * Getting the ID of a parent requires a secondary index. Instead, the plan is to pursue the more traditional approach, where each items stores the id of the parent declaration. This makes `FromSource` more awkward, but also more correct: now, to get from an AST to HIR, we first do this recursively for the parent item, and the just search the children of the parent for the matching def 2019-11-20 13:03:59 +00:00			`DefWithBodyId, EnumId, FunctionLoc, ImplId, ItemLoc, ModuleId, StructOrUnionId, TraitId,`
move struct & enum data to hir_def 2019-10-31 13:40:36 +00:00			`};`
move raw_items to hir_def 2019-10-30 13:12:55 +00:00
Move ids to hir_def crate 2019-10-30 10:10:38 +00:00			`#[salsa::query_group(InternDatabaseStorage)]`
			`pub trait InternDatabase: SourceDatabase {`
			`#[salsa::interned]`
Next gen IDs for functions The current system with AstIds has two primaraly drawbacks: * It is possible to manufacture IDs out of thin air. For example, it's possible to create IDs for items which are not considered in CrateDefMap due to cfg. Or it is possible to mixup structs and unions, because they share ID space. * Getting the ID of a parent requires a secondary index. Instead, the plan is to pursue the more traditional approach, where each items stores the id of the parent declaration. This makes `FromSource` more awkward, but also more correct: now, to get from an AST to HIR, we first do this recursively for the parent item, and the just search the children of the parent for the matching def 2019-11-20 13:03:59 +00:00			`fn intern_function(&self, loc: FunctionLoc) -> crate::FunctionId;`
Move ids to hir_def crate 2019-10-30 10:10:38 +00:00			`#[salsa::interned]`
Add ImplId 2019-11-15 16:14:50 +00:00			`fn intern_struct_or_union(&self, loc: ItemLoc<ast::StructDef>) -> crate::StructOrUnionId;`
move raw_items to hir_def 2019-10-30 13:12:55 +00:00			`#[salsa::interned]`
Add ImplId 2019-11-15 16:14:50 +00:00			`fn intern_enum(&self, loc: ItemLoc<ast::EnumDef>) -> crate::EnumId;`
Move ids to hir_def crate 2019-10-30 10:10:38 +00:00			`#[salsa::interned]`
Add ImplId 2019-11-15 16:14:50 +00:00			`fn intern_const(&self, loc: ItemLoc<ast::ConstDef>) -> crate::ConstId;`
Move ids to hir_def crate 2019-10-30 10:10:38 +00:00			`#[salsa::interned]`
Add ImplId 2019-11-15 16:14:50 +00:00			`fn intern_static(&self, loc: ItemLoc<ast::StaticDef>) -> crate::StaticId;`
Move ids to hir_def crate 2019-10-30 10:10:38 +00:00			`#[salsa::interned]`
Add ImplId 2019-11-15 16:14:50 +00:00			`fn intern_trait(&self, loc: ItemLoc<ast::TraitDef>) -> crate::TraitId;`
Move ids to hir_def crate 2019-10-30 10:10:38 +00:00			`#[salsa::interned]`
Add ImplId 2019-11-15 16:14:50 +00:00			`fn intern_type_alias(&self, loc: ItemLoc<ast::TypeAliasDef>) -> crate::TypeAliasId;`
			`#[salsa::interned]`
			`fn intern_impl(&self, loc: ItemLoc<ast::ImplBlock>) -> crate::ImplId;`
Move ids to hir_def crate 2019-10-30 10:10:38 +00:00			`}`
move raw_items to hir_def 2019-10-30 13:12:55 +00:00
			`#[salsa::query_group(DefDatabase2Storage)]`
			`pub trait DefDatabase2: InternDatabase + AstDatabase {`
			`#[salsa::invoke(RawItems::raw_items_with_source_map_query)]`
			`fn raw_items_with_source_map(`
			`&self,`
			`file_id: HirFileId,`
			`) -> (Arc<RawItems>, Arc<ImportSourceMap>);`

			`#[salsa::invoke(RawItems::raw_items_query)]`
			`fn raw_items(&self, file_id: HirFileId) -> Arc<RawItems>;`
move struct & enum data to hir_def 2019-10-31 13:40:36 +00:00
Move CrateDefMap to hir_def 2019-10-31 15:45:10 +00:00			`#[salsa::invoke(CrateDefMap::crate_def_map_query)]`
			`fn crate_def_map(&self, krate: CrateId) -> Arc<CrateDefMap>;`

move struct & enum data to hir_def 2019-10-31 13:40:36 +00:00			`#[salsa::invoke(StructData::struct_data_query)]`
Unfork struct and union ids 2019-11-09 12:34:00 +00:00			`fn struct_data(&self, id: StructOrUnionId) -> Arc<StructData>;`
Move CrateDefMap to hir_def 2019-10-31 15:45:10 +00:00
move struct & enum data to hir_def 2019-10-31 13:40:36 +00:00			`#[salsa::invoke(EnumData::enum_data_query)]`
			`fn enum_data(&self, e: EnumId) -> Arc<EnumData>;`
Move body queries to hir_def 2019-11-14 14:37:22 +00:00
Remove old impls infrastructure 2019-11-15 18:28:00 +00:00			`#[salsa::invoke(ImplData::impl_data_query)]`
			`fn impl_data(&self, e: ImplId) -> Arc<ImplData>;`

Move traits to hir_def 2019-11-20 11:22:06 +00:00			`#[salsa::invoke(TraitData::trait_data_query)]`
			`fn trait_data(&self, e: TraitId) -> Arc<TraitData>;`

			`#[salsa::invoke(TraitItemsIndex::trait_items_index)]`
			`fn trait_items_index(&self, module: ModuleId) -> TraitItemsIndex;`

Move body queries to hir_def 2019-11-14 14:37:22 +00:00			`#[salsa::invoke(Body::body_with_source_map_query)]`
			`fn body_with_source_map(&self, def: DefWithBodyId) -> (Arc<Body>, Arc<BodySourceMap>);`

			`#[salsa::invoke(Body::body_query)]`
			`fn body(&self, def: DefWithBodyId) -> Arc<Body>;`

			`#[salsa::invoke(ExprScopes::expr_scopes_query)]`
			`fn expr_scopes(&self, def: DefWithBodyId) -> Arc<ExprScopes>;`
move raw_items to hir_def 2019-10-30 13:12:55 +00:00			`}`