Support trait method call autoimports

1 files changed, 5 insertions, 1 deletions

diff --git a/crates/ra_hir/src/code_model.rs b/crates/ra_hir/src/code_model.rs
index 73158b8bd..140b3a87f 100644
--- a/crates/ra_hir/src/code_model.rs
+++ b/crates/ra_hir/src/code_model.rs
@@ -548,6 +548,10 @@ impl Function {
         let mut validator = ExprValidator::new(self.id, infer, sink);
         validator.validate_body(db);
     }
+
+    pub fn container(self, db: &impl DefDatabase) -> AssocContainerId {
+        self.id.lookup(db).container
+    }
 }
 
 #[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
@@ -699,7 +703,7 @@ impl AssocItem {
 
     pub fn container(self, db: &impl DefDatabase) -> AssocContainerId {
         match self {
-            AssocItem::Function(f) => f.id.lookup(db).container,
+            AssocItem::Function(f) => f.container(db),
             AssocItem::Const(c) => c.id.lookup(db).container,
             AssocItem::TypeAlias(t) => t.id.lookup(db).container,
         }