Skip to content

Commit 68df66d

Browse files
rust-bors[bot]rust-bors[bot]
authored
Auto merge of #145186 - camsteffen:assoc-impl-kind, r=<try>
Make `AssocItem` aware of its impl kind
2 parents fce0e74 + 1d8ed9f commit 68df66d

File tree

38 files changed

+242
-209
lines changed

38 files changed

+242
-209
lines changed

‎compiler/rustc_ast_lowering/src/item.rs

Lines changed: 13 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@ use rustc_errors::{E0570, ErrorGuaranteed, struct_span_code_err};
55
use rustc_hir::attrs::AttributeKind;
66
use rustc_hir::def::{DefKind, PerNS, Res};
77
use rustc_hir::def_id::{CRATE_DEF_ID, LocalDefId};
8-
use rustc_hir::{self as hir, HirId, LifetimeSource, PredicateOrigin, find_attr};
8+
use rustc_hir::{self as hir, HirId, ImplItemImplKind, LifetimeSource, PredicateOrigin, find_attr};
99
use rustc_index::{IndexSlice, IndexVec};
1010
use rustc_middle::span_bug;
1111
use rustc_middle::ty::{ResolverAstLowering, TyCtxt};
@@ -1100,16 +1100,21 @@ impl<'hir> LoweringContext<'_, 'hir> {
11001100
owner_id: hir_id.expect_owner(),
11011101
ident: self.lower_ident(ident),
11021102
generics,
1103+
impl_kind: if is_in_trait_impl {
1104+
ImplItemImplKind::Trait {
1105+
defaultness,
1106+
trait_item_def_id: self
1107+
.resolver
1108+
.get_partial_res(i.id)
1109+
.map(|r| r.expect_full_res().opt_def_id())
1110+
.unwrap_or(None),
1111+
}
1112+
} else {
1113+
ImplItemImplKind::Inherent { vis_span: self.lower_span(i.vis.span) }
1114+
},
11031115
kind,
1104-
vis_span: self.lower_span(i.vis.span),
11051116
span: self.lower_span(i.span),
1106-
defaultness,
11071117
has_delayed_lints: !self.delayed_lints.is_empty(),
1108-
trait_item_def_id: self
1109-
.resolver
1110-
.get_partial_res(i.id)
1111-
.map(|r| r.expect_full_res().opt_def_id())
1112-
.unwrap_or(None),
11131118
};
11141119
self.arena.alloc(item)
11151120
}

‎compiler/rustc_codegen_ssa/src/codegen_attrs.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -600,7 +600,7 @@ fn codegen_fn_attrs(tcx: TyCtxt<'_>, did: LocalDefId) -> CodegenFnAttrs {
600600
fn opt_trait_item(tcx: TyCtxt<'_>, def_id: DefId) -> Option<DefId> {
601601
let impl_item = tcx.opt_associated_item(def_id)?;
602602
match impl_item.container {
603-
ty::AssocItemContainer::Impl => impl_item.trait_item_def_id,
603+
ty::AssocItemContainer::TraitImpl => impl_item.trait_item_def_id,
604604
_ => None,
605605
}
606606
}

‎compiler/rustc_hir/src/hir.rs

Lines changed: 22 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -3190,12 +3190,22 @@ pub struct ImplItem<'hir> {
31903190
pub owner_id: OwnerId,
31913191
pub generics: &'hir Generics<'hir>,
31923192
pub kind: ImplItemKind<'hir>,
3193-
pub defaultness: Defaultness,
3193+
pub impl_kind: ImplItemImplKind,
31943194
pub span: Span,
3195-
pub vis_span: Span,
31963195
pub has_delayed_lints: bool,
3197-
/// When we are in a trait impl, link to the trait-item's id.
3198-
pub trait_item_def_id: Option<DefId>,
3196+
}
3197+
3198+
#[derive(Debug, Clone, Copy, HashStable_Generic)]
3199+
pub enum ImplItemImplKind {
3200+
Inherent {
3201+
vis_span: Span,
3202+
},
3203+
Trait {
3204+
defaultness: Defaultness,
3205+
/// Item in the trait that this item implements.
3206+
/// May be `None` if there is an error.
3207+
trait_item_def_id: Option<DefId>,
3208+
},
31993209
}
32003210

32013211
impl<'hir> ImplItem<'hir> {
@@ -3209,6 +3219,13 @@ impl<'hir> ImplItem<'hir> {
32093219
ImplItemId { owner_id: self.owner_id }
32103220
}
32113221

3222+
pub fn vis_span(&self) -> Option<Span> {
3223+
match self.impl_kind {
3224+
ImplItemImplKind::Trait { .. } => None,
3225+
ImplItemImplKind::Inherent { vis_span, .. } => Some(vis_span),
3226+
}
3227+
}
3228+
32123229
expect_methods_self_kind! {
32133230
expect_const, (&'hir Ty<'hir>, BodyId), ImplItemKind::Const(ty, body), (ty, *body);
32143231
expect_fn, (&FnSig<'hir>, BodyId), ImplItemKind::Fn(ty, body), (ty, *body);
@@ -4953,7 +4970,7 @@ mod size_asserts {
49534970
static_assert_size!(GenericBound<'_>, 64);
49544971
static_assert_size!(Generics<'_>, 56);
49554972
static_assert_size!(Impl<'_>, 80);
4956-
static_assert_size!(ImplItem<'_>, 96);
4973+
static_assert_size!(ImplItem<'_>, 88);
49574974
static_assert_size!(ImplItemKind<'_>, 40);
49584975
static_assert_size!(Item<'_>, 88);
49594976
static_assert_size!(ItemKind<'_>, 64);

‎compiler/rustc_hir/src/intravisit.rs

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1257,18 +1257,21 @@ pub fn walk_impl_item<'v, V: Visitor<'v>>(
12571257
owner_id: _,
12581258
ident,
12591259
ref generics,
1260+
ref impl_kind,
12601261
ref kind,
1261-
ref defaultness,
12621262
span: _,
1263-
vis_span: _,
12641263
has_delayed_lints: _,
1265-
trait_item_def_id: _,
12661264
} = *impl_item;
12671265

12681266
try_visit!(visitor.visit_ident(ident));
12691267
try_visit!(visitor.visit_generics(generics));
1270-
try_visit!(visitor.visit_defaultness(defaultness));
12711268
try_visit!(visitor.visit_id(impl_item.hir_id()));
1269+
match impl_kind {
1270+
ImplItemImplKind::Inherent { vis_span: _ } => {}
1271+
ImplItemImplKind::Trait { defaultness, trait_item_def_id: _ } => {
1272+
try_visit!(visitor.visit_defaultness(defaultness));
1273+
}
1274+
}
12721275
match *kind {
12731276
ImplItemKind::Const(ref ty, body) => {
12741277
try_visit!(visitor.visit_ty_unambig(ty));

‎compiler/rustc_hir_analysis/src/check/check.rs

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1009,7 +1009,7 @@ pub(crate) fn check_item_type(tcx: TyCtxt<'_>, def_id: LocalDefId) -> Result<(),
10091009
res = res.and(check_associated_item(tcx, def_id));
10101010
let assoc_item = tcx.associated_item(def_id);
10111011
match assoc_item.container {
1012-
ty::AssocItemContainer::Impl => {}
1012+
ty::AssocItemContainer::InherentImpl | ty::AssocItemContainer::TraitImpl => {}
10131013
ty::AssocItemContainer::Trait => {
10141014
res = res.and(check_trait_item(tcx, def_id));
10151015
}
@@ -1026,7 +1026,7 @@ pub(crate) fn check_item_type(tcx: TyCtxt<'_>, def_id: LocalDefId) -> Result<(),
10261026
res = res.and(check_associated_item(tcx, def_id));
10271027
let assoc_item = tcx.associated_item(def_id);
10281028
match assoc_item.container {
1029-
ty::AssocItemContainer::Impl => {}
1029+
ty::AssocItemContainer::InherentImpl | ty::AssocItemContainer::TraitImpl => {}
10301030
ty::AssocItemContainer::Trait => {
10311031
res = res.and(check_trait_item(tcx, def_id));
10321032
}
@@ -1043,7 +1043,7 @@ pub(crate) fn check_item_type(tcx: TyCtxt<'_>, def_id: LocalDefId) -> Result<(),
10431043

10441044
let assoc_item = tcx.associated_item(def_id);
10451045
let has_type = match assoc_item.container {
1046-
ty::AssocItemContainer::Impl => true,
1046+
ty::AssocItemContainer::InherentImpl | ty::AssocItemContainer::TraitImpl => true,
10471047
ty::AssocItemContainer::Trait => {
10481048
tcx.ensure_ok().explicit_item_bounds(def_id);
10491049
tcx.ensure_ok().explicit_item_self_bounds(def_id);

‎compiler/rustc_hir_analysis/src/check/compare_impl_item.rs

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1449,7 +1449,9 @@ fn compare_self_type<'tcx>(
14491449

14501450
let self_string = |method: ty::AssocItem| {
14511451
let untransformed_self_ty = match method.container {
1452-
ty::AssocItemContainer::Impl => impl_trait_ref.self_ty(),
1452+
ty::AssocItemContainer::InherentImpl | ty::AssocItemContainer::TraitImpl => {
1453+
impl_trait_ref.self_ty()
1454+
}
14531455
ty::AssocItemContainer::Trait => tcx.types.self_param,
14541456
};
14551457
let self_arg_ty = tcx.fn_sig(method.def_id).instantiate_identity().input(0);
@@ -2458,8 +2460,11 @@ fn param_env_with_gat_bounds<'tcx>(
24582460

24592461
for impl_ty in impl_tys_to_install {
24602462
let trait_ty = match impl_ty.container {
2463+
ty::AssocItemContainer::InherentImpl => bug!(),
24612464
ty::AssocItemContainer::Trait => impl_ty,
2462-
ty::AssocItemContainer::Impl => tcx.associated_item(impl_ty.trait_item_def_id.unwrap()),
2465+
ty::AssocItemContainer::TraitImpl => {
2466+
tcx.associated_item(impl_ty.trait_item_def_id.unwrap())
2467+
}
24632468
};
24642469

24652470
let mut bound_vars: smallvec::SmallVec<[ty::BoundVariableKind; 8]> =

‎compiler/rustc_hir_analysis/src/check/wfcheck.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -949,7 +949,7 @@ pub(crate) fn check_associated_item(
949949

950950
let self_ty = match item.container {
951951
ty::AssocItemContainer::Trait => tcx.types.self_param,
952-
ty::AssocItemContainer::Impl => {
952+
ty::AssocItemContainer::InherentImpl | ty::AssocItemContainer::TraitImpl => {
953953
tcx.type_of(item.container_id(tcx)).instantiate_identity()
954954
}
955955
};

‎compiler/rustc_hir_analysis/src/collect/type_of.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -198,7 +198,7 @@ pub(super) fn type_of(tcx: TyCtxt<'_>, def_id: LocalDefId) -> ty::EarlyBinder<'_
198198
}
199199
}
200200
ImplItemKind::Type(ty) => {
201-
if tcx.impl_trait_ref(tcx.hir_get_parent_item(hir_id)).is_none() {
201+
if let ImplItemImplKind::Inherent { .. } = item.impl_kind {
202202
check_feature_inherent_assoc_ty(tcx, item.span);
203203
}
204204

‎compiler/rustc_hir_typeck/src/fn_ctxt/_impl.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1033,7 +1033,7 @@ impl<'a, 'tcx> FnCtxt<'a, 'tcx> {
10331033
self.set_tainted_by_errors(e);
10341034
}
10351035
}
1036-
ty::AssocItemContainer::Impl => {
1036+
ty::AssocItemContainer::InherentImpl | ty::AssocItemContainer::TraitImpl => {
10371037
if segments.len() == 1 {
10381038
// `<T>::assoc` will end up here, and so
10391039
// can `T::assoc`. If this came from an

‎compiler/rustc_hir_typeck/src/lib.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -290,7 +290,7 @@ fn infer_type_if_missing<'tcx>(fcx: &FnCtxt<'_, 'tcx>, node: Node<'tcx>) -> Opti
290290
{
291291
if let Some(item) = tcx.opt_associated_item(def_id.into())
292292
&& let ty::AssocKind::Const { .. } = item.kind
293-
&& let ty::AssocItemContainer::Impl = item.container
293+
&& let ty::AssocItemContainer::TraitImpl = item.container
294294
&& let Some(trait_item_def_id) = item.trait_item_def_id
295295
{
296296
let impl_def_id = item.container_id(tcx);

0 commit comments

Comments
 (0)