remove Cancelable from Module API, part 2

This commit is contained in:
Aleksey Kladov 2019-01-15 18:30:58 +03:00
parent ca52cf1ecd
commit c159e414b4
7 changed files with 28 additions and 33 deletions

View File

@ -97,14 +97,14 @@ impl Module {
} }
/// Returns the crate this module is part of. /// Returns the crate this module is part of.
pub fn krate(&self, db: &impl HirDatabase) -> Cancelable<Option<Crate>> { pub fn krate(&self, db: &impl HirDatabase) -> Option<Crate> {
self.krate_impl(db) self.krate_impl(db)
} }
/// Topmost parent of this module. Every module has a `crate_root`, but some /// Topmost parent of this module. Every module has a `crate_root`, but some
/// might be missing `krate`. This can happen if a module's file is not included /// might be missing `krate`. This can happen if a module's file is not included
/// in the module tree of any target in Cargo.toml. /// in the module tree of any target in Cargo.toml.
pub fn crate_root(&self, db: &impl HirDatabase) -> Cancelable<Module> { pub fn crate_root(&self, db: &impl HirDatabase) -> Module {
self.crate_root_impl(db) self.crate_root_impl(db)
} }
@ -114,23 +114,23 @@ impl Module {
} }
/// Iterates over all child modules. /// Iterates over all child modules.
pub fn children(&self, db: &impl HirDatabase) -> Cancelable<impl Iterator<Item = Module>> { pub fn children(&self, db: &impl HirDatabase) -> impl Iterator<Item = Module> {
self.children_impl(db) self.children_impl(db)
} }
/// Finds a parent module. /// Finds a parent module.
pub fn parent(&self, db: &impl HirDatabase) -> Cancelable<Option<Module>> { pub fn parent(&self, db: &impl HirDatabase) -> Option<Module> {
self.parent_impl(db) self.parent_impl(db)
} }
pub fn path_to_root(&self, db: &impl HirDatabase) -> Cancelable<Vec<Module>> { pub fn path_to_root(&self, db: &impl HirDatabase) -> Vec<Module> {
let mut res = vec![self.clone()]; let mut res = vec![self.clone()];
let mut curr = self.clone(); let mut curr = self.clone();
while let Some(next) = curr.parent(db)? { while let Some(next) = curr.parent(db) {
res.push(next.clone()); res.push(next.clone());
curr = next curr = next
} }
Ok(res) res
} }
/// Returns a `ModuleScope`: a set of items, visible in this module. /// Returns a `ModuleScope`: a set of items, visible in this module.

View File

@ -66,21 +66,21 @@ impl Module {
Some((file_id, src)) Some((file_id, src))
} }
pub(crate) fn krate_impl(&self, db: &impl HirDatabase) -> Cancelable<Option<Crate>> { pub(crate) fn krate_impl(&self, db: &impl HirDatabase) -> Option<Crate> {
let root = self.crate_root(db)?; let root = self.crate_root(db);
let loc = root.def_id.loc(db); let loc = root.def_id.loc(db);
let file_id = loc.source_item_id.file_id.as_original_file(); let file_id = loc.source_item_id.file_id.as_original_file();
let crate_graph = db.crate_graph(); let crate_graph = db.crate_graph();
let crate_id = ctry!(crate_graph.crate_id_for_crate_root(file_id)); let crate_id = crate_graph.crate_id_for_crate_root(file_id)?;
Ok(Some(Crate::new(crate_id))) Some(Crate::new(crate_id))
} }
pub(crate) fn crate_root_impl(&self, db: &impl HirDatabase) -> Cancelable<Module> { pub(crate) fn crate_root_impl(&self, db: &impl HirDatabase) -> Module {
let loc = self.def_id.loc(db); let loc = self.def_id.loc(db);
let module_tree = db.module_tree(loc.source_root_id); let module_tree = db.module_tree(loc.source_root_id);
let module_id = loc.module_id.crate_root(&module_tree); let module_id = loc.module_id.crate_root(&module_tree);
Ok(Module::from_module_id(db, loc.source_root_id, module_id)) Module::from_module_id(db, loc.source_root_id, module_id)
} }
/// Finds a child module with the specified name. /// Finds a child module with the specified name.
@ -92,7 +92,7 @@ impl Module {
} }
/// Iterates over all child modules. /// Iterates over all child modules.
pub fn children_impl(&self, db: &impl HirDatabase) -> Cancelable<impl Iterator<Item = Module>> { pub fn children_impl(&self, db: &impl HirDatabase) -> impl Iterator<Item = Module> {
// FIXME this should be implementable without collecting into a vec, but // FIXME this should be implementable without collecting into a vec, but
// it's kind of hard since the iterator needs to keep a reference to the // it's kind of hard since the iterator needs to keep a reference to the
// module tree. // module tree.
@ -103,18 +103,14 @@ impl Module {
.children(&module_tree) .children(&module_tree)
.map(|(_, module_id)| Module::from_module_id(db, loc.source_root_id, module_id)) .map(|(_, module_id)| Module::from_module_id(db, loc.source_root_id, module_id))
.collect::<Vec<_>>(); .collect::<Vec<_>>();
Ok(children.into_iter()) children.into_iter()
} }
pub fn parent_impl(&self, db: &impl HirDatabase) -> Cancelable<Option<Module>> { pub fn parent_impl(&self, db: &impl HirDatabase) -> Option<Module> {
let loc = self.def_id.loc(db); let loc = self.def_id.loc(db);
let module_tree = db.module_tree(loc.source_root_id); let module_tree = db.module_tree(loc.source_root_id);
let parent_id = ctry!(loc.module_id.parent(&module_tree)); let parent_id = loc.module_id.parent(&module_tree)?;
Ok(Some(Module::from_module_id( Some(Module::from_module_id(db, loc.source_root_id, parent_id))
db,
loc.source_root_id,
parent_id,
)))
} }
/// Returns a `ModuleScope`: a set of items, visible in this module. /// Returns a `ModuleScope`: a set of items, visible in this module.
@ -132,10 +128,10 @@ impl Module {
) -> Cancelable<PerNs<DefId>> { ) -> Cancelable<PerNs<DefId>> {
let mut curr_per_ns = PerNs::types( let mut curr_per_ns = PerNs::types(
match path.kind { match path.kind {
PathKind::Crate => self.crate_root(db)?, PathKind::Crate => self.crate_root(db),
PathKind::Self_ | PathKind::Plain => self.clone(), PathKind::Self_ | PathKind::Plain => self.clone(),
PathKind::Super => { PathKind::Super => {
if let Some(p) = self.parent(db)? { if let Some(p) = self.parent(db) {
p p
} else { } else {
return Ok(PerNs::none()); return Ok(PerNs::none());

View File

@ -217,7 +217,7 @@ impl DefId {
/// Returns the containing crate. /// Returns the containing crate.
pub fn krate(&self, db: &impl HirDatabase) -> Cancelable<Option<Crate>> { pub fn krate(&self, db: &impl HirDatabase) -> Cancelable<Option<Crate>> {
Ok(self.module(db)?.krate(db)?) Ok(self.module(db)?.krate(db))
} }
/// Returns the containing impl block, if this is an impl item. /// Returns the containing impl block, if this is an impl item.

View File

@ -75,7 +75,7 @@ impl CrateImplBlocks {
} }
} }
for child in module.children(db)? { for child in module.children(db) {
self.collect_recursive(db, child)?; self.collect_recursive(db, child)?;
} }

View File

@ -99,16 +99,16 @@ impl db::RootDatabase {
impl db::RootDatabase { impl db::RootDatabase {
/// Returns `Vec` for the same reason as `parent_module` /// Returns `Vec` for the same reason as `parent_module`
pub(crate) fn crate_for(&self, file_id: FileId) -> Cancelable<Vec<CrateId>> { pub(crate) fn crate_for(&self, file_id: FileId) -> Vec<CrateId> {
let module = match source_binder::module_from_file_id(self, file_id) { let module = match source_binder::module_from_file_id(self, file_id) {
Some(it) => it, Some(it) => it,
None => return Ok(Vec::new()), None => return Vec::new(),
}; };
let krate = match module.krate(self)? { let krate = match module.krate(self) {
Some(it) => it, Some(it) => it,
None => return Ok(Vec::new()), None => return Vec::new(),
}; };
Ok(vec![krate.crate_id()]) vec![krate.crate_id()]
} }
pub(crate) fn find_all_refs( pub(crate) fn find_all_refs(
&self, &self,

View File

@ -419,7 +419,7 @@ impl Analysis {
/// Returns crates this file belongs too. /// Returns crates this file belongs too.
pub fn crate_for(&self, file_id: FileId) -> Cancelable<Vec<CrateId>> { pub fn crate_for(&self, file_id: FileId) -> Cancelable<Vec<CrateId>> {
self.with_db(|db| db.crate_for(file_id))? self.with_db(|db| db.crate_for(file_id))
} }
/// Returns the root file of the given crate. /// Returns the root file of the given crate.

View File

@ -80,7 +80,6 @@ fn runnable_mod(db: &RootDatabase, file_id: FileId, module: &ast::Module) -> Opt
// FIXME: thread cancellation instead of `.ok`ing // FIXME: thread cancellation instead of `.ok`ing
let path = module let path = module
.path_to_root(db) .path_to_root(db)
.ok()?
.into_iter() .into_iter()
.rev() .rev()
.filter_map(|it| it.name(db)) .filter_map(|it| it.name(db))