add union to code_model

1 files changed, 5 insertions, 1 deletions

diff --git a/crates/ra_hir/src/nameres/collector.rs b/crates/ra_hir/src/nameres/collector.rs
index c615d80c3..621236551 100644
--- a/crates/ra_hir/src/nameres/collector.rs
+++ b/crates/ra_hir/src/nameres/collector.rs
@@ -6,7 +6,7 @@ use ra_db::FileId;
 use ra_syntax::ast;
 
 use crate::{
-    Function, Module, Struct, Enum, Const, Static, Trait, TypeAlias,
+    Function, Module, Struct, Union, Enum, Const, Static, Trait, TypeAlias,
     DefDatabase, HirFileId, Name, Path,
     KnownName,
     nameres::{
@@ -495,6 +495,10 @@ where
                 let s = def!(Struct, ast_id);
                 PerNs::both(s, s)
             }
+            raw::DefKind::Union(ast_id) => {
+                let s = def!(Union, ast_id);
+                PerNs::both(s, s)
+            }
             raw::DefKind::Enum(ast_id) => PerNs::types(def!(Enum, ast_id)),
             raw::DefKind::Const(ast_id) => PerNs::values(def!(Const, ast_id)),
             raw::DefKind::Static(ast_id) => PerNs::values(def!(Static, ast_id)),