aboutsummaryrefslogtreecommitdiff
path: root/crates/ra_hir/src/ty.rs
diff options
context:
space:
mode:
Diffstat (limited to 'crates/ra_hir/src/ty.rs')
-rw-r--r--crates/ra_hir/src/ty.rs33
1 files changed, 24 insertions, 9 deletions
diff --git a/crates/ra_hir/src/ty.rs b/crates/ra_hir/src/ty.rs
index 179ebddee..67fcc5346 100644
--- a/crates/ra_hir/src/ty.rs
+++ b/crates/ra_hir/src/ty.rs
@@ -862,14 +862,15 @@ impl<'a, D: HirDatabase> InferenceContext<'a, D> {
862 } 862 }
863 863
864 fn resolve_all(mut self) -> InferenceResult { 864 fn resolve_all(mut self) -> InferenceResult {
865 let mut tv_stack = Vec::new();
865 let mut expr_types = mem::replace(&mut self.type_of_expr, ArenaMap::default()); 866 let mut expr_types = mem::replace(&mut self.type_of_expr, ArenaMap::default());
866 for ty in expr_types.values_mut() { 867 for ty in expr_types.values_mut() {
867 let resolved = self.resolve_ty_completely(mem::replace(ty, Ty::Unknown)); 868 let resolved = self.resolve_ty_completely(&mut tv_stack, mem::replace(ty, Ty::Unknown));
868 *ty = resolved; 869 *ty = resolved;
869 } 870 }
870 let mut pat_types = mem::replace(&mut self.type_of_pat, ArenaMap::default()); 871 let mut pat_types = mem::replace(&mut self.type_of_pat, ArenaMap::default());
871 for ty in pat_types.values_mut() { 872 for ty in pat_types.values_mut() {
872 let resolved = self.resolve_ty_completely(mem::replace(ty, Ty::Unknown)); 873 let resolved = self.resolve_ty_completely(&mut tv_stack, mem::replace(ty, Ty::Unknown));
873 *ty = resolved; 874 *ty = resolved;
874 } 875 }
875 InferenceResult { 876 InferenceResult {
@@ -1014,13 +1015,20 @@ impl<'a, D: HirDatabase> InferenceContext<'a, D> {
1014 /// by their known types. All types returned by the infer_* functions should 1015 /// by their known types. All types returned by the infer_* functions should
1015 /// be resolved as far as possible, i.e. contain no type variables with 1016 /// be resolved as far as possible, i.e. contain no type variables with
1016 /// known type. 1017 /// known type.
1017 fn resolve_ty_as_possible(&mut self, ty: Ty) -> Ty { 1018 fn resolve_ty_as_possible(&mut self, tv_stack: &mut Vec<TypeVarId>, ty: Ty) -> Ty {
1018 ty.fold(&mut |ty| match ty { 1019 ty.fold(&mut |ty| match ty {
1019 Ty::Infer(tv) => { 1020 Ty::Infer(tv) => {
1020 let inner = tv.to_inner(); 1021 let inner = tv.to_inner();
1022 if tv_stack.contains(&inner) {
1023 // recursive type
1024 return tv.fallback_value();
1025 }
1021 if let Some(known_ty) = self.var_unification_table.probe_value(inner).known() { 1026 if let Some(known_ty) = self.var_unification_table.probe_value(inner).known() {
1022 // known_ty may contain other variables that are known by now 1027 // known_ty may contain other variables that are known by now
1023 self.resolve_ty_as_possible(known_ty.clone()) 1028 tv_stack.push(inner);
1029 let result = self.resolve_ty_as_possible(tv_stack, known_ty.clone());
1030 tv_stack.pop();
1031 result
1024 } else { 1032 } else {
1025 ty 1033 ty
1026 } 1034 }
@@ -1049,13 +1057,20 @@ impl<'a, D: HirDatabase> InferenceContext<'a, D> {
1049 1057
1050 /// Resolves the type completely; type variables without known type are 1058 /// Resolves the type completely; type variables without known type are
1051 /// replaced by Ty::Unknown. 1059 /// replaced by Ty::Unknown.
1052 fn resolve_ty_completely(&mut self, ty: Ty) -> Ty { 1060 fn resolve_ty_completely(&mut self, tv_stack: &mut Vec<TypeVarId>, ty: Ty) -> Ty {
1053 ty.fold(&mut |ty| match ty { 1061 ty.fold(&mut |ty| match ty {
1054 Ty::Infer(tv) => { 1062 Ty::Infer(tv) => {
1055 let inner = tv.to_inner(); 1063 let inner = tv.to_inner();
1064 if tv_stack.contains(&inner) {
1065 // recursive type
1066 return tv.fallback_value();
1067 }
1056 if let Some(known_ty) = self.var_unification_table.probe_value(inner).known() { 1068 if let Some(known_ty) = self.var_unification_table.probe_value(inner).known() {
1057 // known_ty may contain other variables that are known by now 1069 // known_ty may contain other variables that are known by now
1058 self.resolve_ty_completely(known_ty.clone()) 1070 tv_stack.push(inner);
1071 let result = self.resolve_ty_completely(tv_stack, known_ty.clone());
1072 tv_stack.pop();
1073 result
1059 } else { 1074 } else {
1060 tv.fallback_value() 1075 tv.fallback_value()
1061 } 1076 }
@@ -1070,7 +1085,7 @@ impl<'a, D: HirDatabase> InferenceContext<'a, D> {
1070 let name = path.as_ident().cloned().unwrap_or_else(Name::self_param); 1085 let name = path.as_ident().cloned().unwrap_or_else(Name::self_param);
1071 if let Some(scope_entry) = self.scopes.resolve_local_name(expr, name) { 1086 if let Some(scope_entry) = self.scopes.resolve_local_name(expr, name) {
1072 let ty = self.type_of_pat.get(scope_entry.pat())?; 1087 let ty = self.type_of_pat.get(scope_entry.pat())?;
1073 let ty = self.resolve_ty_as_possible(ty.clone()); 1088 let ty = self.resolve_ty_as_possible(&mut vec![], ty.clone());
1074 return Some(ty); 1089 return Some(ty);
1075 }; 1090 };
1076 }; 1091 };
@@ -1239,7 +1254,7 @@ impl<'a, D: HirDatabase> InferenceContext<'a, D> {
1239 // use a new type variable if we got Ty::Unknown here 1254 // use a new type variable if we got Ty::Unknown here
1240 let ty = self.insert_type_vars_shallow(ty); 1255 let ty = self.insert_type_vars_shallow(ty);
1241 self.unify(&ty, expected); 1256 self.unify(&ty, expected);
1242 let ty = self.resolve_ty_as_possible(ty); 1257 let ty = self.resolve_ty_as_possible(&mut vec![], ty);
1243 self.write_pat_ty(pat, ty.clone()); 1258 self.write_pat_ty(pat, ty.clone());
1244 ty 1259 ty
1245 } 1260 }
@@ -1538,7 +1553,7 @@ impl<'a, D: HirDatabase> InferenceContext<'a, D> {
1538 // use a new type variable if we got Ty::Unknown here 1553 // use a new type variable if we got Ty::Unknown here
1539 let ty = self.insert_type_vars_shallow(ty); 1554 let ty = self.insert_type_vars_shallow(ty);
1540 self.unify(&ty, &expected.ty); 1555 self.unify(&ty, &expected.ty);
1541 let ty = self.resolve_ty_as_possible(ty); 1556 let ty = self.resolve_ty_as_possible(&mut vec![], ty);
1542 self.write_expr_ty(tgt_expr, ty.clone()); 1557 self.write_expr_ty(tgt_expr, ty.clone());
1543 ty 1558 ty
1544 } 1559 }