Skip to content

Commit 02916a3

Browse files
matthiaskrgrmatthiaskrgr
authored
Rollup merge of #127145 - compiler-errors:as_lang_item, r=lcnr
Add `as_lang_item` to `LanguageItems`, new trait solver Add `as_lang_item` which turns `DefId` into a `TraitSolverLangItem` in the new trait solver, so we can turn the large chain of if statements in `assemble_builtin_impl_candidates` into a match instead. r? lcnr
2 parents 444a0ff + 5a83751 commit 02916a3

File tree

6 files changed

+164
-91
lines changed

6 files changed

+164
-91
lines changed

‎compiler/rustc_hir/src/lang_items.rs

+12-1
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@ use crate::def_id::DefId;
1111
use crate::{MethodKind, Target};
1212

1313
use rustc_ast as ast;
14+
use rustc_data_structures::fx::FxIndexMap;
1415
use rustc_data_structures::stable_hasher::{HashStable, StableHasher};
1516
use rustc_macros::{Decodable, Encodable, HashStable_Generic};
1617
use rustc_span::symbol::{kw, sym, Symbol};
@@ -23,14 +24,19 @@ pub struct LanguageItems {
2324
/// Mappings from lang items to their possibly found [`DefId`]s.
2425
/// The index corresponds to the order in [`LangItem`].
2526
items: [Option<DefId>; std::mem::variant_count::<LangItem>()],
27+
reverse_items: FxIndexMap<DefId, LangItem>,
2628
/// Lang items that were not found during collection.
2729
pub missing: Vec<LangItem>,
2830
}
2931

3032
impl LanguageItems {
3133
/// Construct an empty collection of lang items and no missing ones.
3234
pub fn new() -> Self {
33-
Self { items: [None; std::mem::variant_count::<LangItem>()], missing: Vec::new() }
35+
Self {
36+
items: [None; std::mem::variant_count::<LangItem>()],
37+
reverse_items: FxIndexMap::default(),
38+
missing: Vec::new(),
39+
}
3440
}
3541

3642
pub fn get(&self, item: LangItem) -> Option<DefId> {
@@ -39,6 +45,11 @@ impl LanguageItems {
3945

4046
pub fn set(&mut self, item: LangItem, def_id: DefId) {
4147
self.items[item as usize] = Some(def_id);
48+
self.reverse_items.insert(def_id, item);
49+
}
50+
51+
pub fn from_def_id(&self, def_id: DefId) -> Option<LangItem> {
52+
self.reverse_items.get(&def_id).copied()
4253
}
4354

4455
pub fn iter(&self) -> impl Iterator<Item = (LangItem, DefId)> + '_ {

‎compiler/rustc_middle/src/middle/lang_items.rs

+4
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,10 @@ impl<'tcx> TyCtxt<'tcx> {
2727
self.lang_items().get(lang_item) == Some(def_id)
2828
}
2929

30+
pub fn as_lang_item(self, def_id: DefId) -> Option<LangItem> {
31+
self.lang_items().from_def_id(def_id)
32+
}
33+
3034
/// Given a [`DefId`] of one of the [`Fn`], [`FnMut`] or [`FnOnce`] traits,
3135
/// returns a corresponding [`ty::ClosureKind`].
3236
/// For any other [`DefId`] return `None`.

‎compiler/rustc_middle/src/ty/context.rs

+64-45
Original file line numberDiff line numberDiff line change
@@ -366,6 +366,10 @@ impl<'tcx> Interner for TyCtxt<'tcx> {
366366
self.is_lang_item(def_id, trait_lang_item_to_lang_item(lang_item))
367367
}
368368

369+
fn as_lang_item(self, def_id: DefId) -> Option<TraitSolverLangItem> {
370+
lang_item_to_trait_lang_item(self.lang_items().from_def_id(def_id)?)
371+
}
372+
369373
fn associated_type_def_ids(self, def_id: DefId) -> impl IntoIterator<Item = DefId> {
370374
self.associated_items(def_id)
371375
.in_definition_order()
@@ -522,14 +526,6 @@ impl<'tcx> Interner for TyCtxt<'tcx> {
522526
self.trait_def(trait_def_id).implement_via_object
523527
}
524528

525-
fn fn_trait_kind_from_def_id(self, trait_def_id: DefId) -> Option<ty::ClosureKind> {
526-
self.fn_trait_kind_from_def_id(trait_def_id)
527-
}
528-
529-
fn async_fn_trait_kind_from_def_id(self, trait_def_id: DefId) -> Option<ty::ClosureKind> {
530-
self.async_fn_trait_kind_from_def_id(trait_def_id)
531-
}
532-
533529
fn supertrait_def_ids(self, trait_def_id: DefId) -> impl IntoIterator<Item = DefId> {
534530
self.supertrait_def_ids(trait_def_id)
535531
}
@@ -573,46 +569,69 @@ impl<'tcx> Interner for TyCtxt<'tcx> {
573569
}
574570
}
575571

576-
fn trait_lang_item_to_lang_item(lang_item: TraitSolverLangItem) -> LangItem {
577-
match lang_item {
578-
TraitSolverLangItem::AsyncDestruct => LangItem::AsyncDestruct,
579-
TraitSolverLangItem::AsyncFnKindHelper => LangItem::AsyncFnKindHelper,
580-
TraitSolverLangItem::AsyncFnKindUpvars => LangItem::AsyncFnKindUpvars,
581-
TraitSolverLangItem::AsyncFnOnceOutput => LangItem::AsyncFnOnceOutput,
582-
TraitSolverLangItem::AsyncIterator => LangItem::AsyncIterator,
583-
TraitSolverLangItem::CallOnceFuture => LangItem::CallOnceFuture,
584-
TraitSolverLangItem::CallRefFuture => LangItem::CallRefFuture,
585-
TraitSolverLangItem::Clone => LangItem::Clone,
586-
TraitSolverLangItem::Copy => LangItem::Copy,
587-
TraitSolverLangItem::Coroutine => LangItem::Coroutine,
588-
TraitSolverLangItem::CoroutineReturn => LangItem::CoroutineReturn,
589-
TraitSolverLangItem::CoroutineYield => LangItem::CoroutineYield,
590-
TraitSolverLangItem::Destruct => LangItem::Destruct,
591-
TraitSolverLangItem::DiscriminantKind => LangItem::DiscriminantKind,
592-
TraitSolverLangItem::DynMetadata => LangItem::DynMetadata,
593-
TraitSolverLangItem::EffectsMaybe => LangItem::EffectsMaybe,
594-
TraitSolverLangItem::EffectsIntersection => LangItem::EffectsIntersection,
595-
TraitSolverLangItem::EffectsIntersectionOutput => LangItem::EffectsIntersectionOutput,
596-
TraitSolverLangItem::EffectsNoRuntime => LangItem::EffectsNoRuntime,
597-
TraitSolverLangItem::EffectsRuntime => LangItem::EffectsRuntime,
598-
TraitSolverLangItem::FnPtrTrait => LangItem::FnPtrTrait,
599-
TraitSolverLangItem::FusedIterator => LangItem::FusedIterator,
600-
TraitSolverLangItem::Future => LangItem::Future,
601-
TraitSolverLangItem::FutureOutput => LangItem::FutureOutput,
602-
TraitSolverLangItem::Iterator => LangItem::Iterator,
603-
TraitSolverLangItem::Metadata => LangItem::Metadata,
604-
TraitSolverLangItem::Option => LangItem::Option,
605-
TraitSolverLangItem::PointeeTrait => LangItem::PointeeTrait,
606-
TraitSolverLangItem::PointerLike => LangItem::PointerLike,
607-
TraitSolverLangItem::Poll => LangItem::Poll,
608-
TraitSolverLangItem::Sized => LangItem::Sized,
609-
TraitSolverLangItem::TransmuteTrait => LangItem::TransmuteTrait,
610-
TraitSolverLangItem::Tuple => LangItem::Tuple,
611-
TraitSolverLangItem::Unpin => LangItem::Unpin,
612-
TraitSolverLangItem::Unsize => LangItem::Unsize,
572+
macro_rules! bidirectional_lang_item_map {
573+
($($name:ident),+ $(,)?) => {
574+
fn trait_lang_item_to_lang_item(lang_item: TraitSolverLangItem) -> LangItem {
575+
match lang_item {
576+
$(TraitSolverLangItem::$name => LangItem::$name,)+
577+
}
578+
}
579+
580+
fn lang_item_to_trait_lang_item(lang_item: LangItem) -> Option<TraitSolverLangItem> {
581+
Some(match lang_item {
582+
$(LangItem::$name => TraitSolverLangItem::$name,)+
583+
_ => return None,
584+
})
585+
}
613586
}
614587
}
615588

589+
bidirectional_lang_item_map! {
590+
// tidy-alphabetical-start
591+
AsyncDestruct,
592+
AsyncFn,
593+
AsyncFnKindHelper,
594+
AsyncFnKindUpvars,
595+
AsyncFnMut,
596+
AsyncFnOnce,
597+
AsyncFnOnceOutput,
598+
AsyncIterator,
599+
CallOnceFuture,
600+
CallRefFuture,
601+
Clone,
602+
Copy,
603+
Coroutine,
604+
CoroutineReturn,
605+
CoroutineYield,
606+
Destruct,
607+
DiscriminantKind,
608+
DynMetadata,
609+
EffectsIntersection,
610+
EffectsIntersectionOutput,
611+
EffectsMaybe,
612+
EffectsNoRuntime,
613+
EffectsRuntime,
614+
Fn,
615+
FnMut,
616+
FnOnce,
617+
FnPtrTrait,
618+
FusedIterator,
619+
Future,
620+
FutureOutput,
621+
Iterator,
622+
Metadata,
623+
Option,
624+
PointeeTrait,
625+
PointerLike,
626+
Poll,
627+
Sized,
628+
TransmuteTrait,
629+
Tuple,
630+
Unpin,
631+
Unsize,
632+
// tidy-alphabetical-end
633+
}
634+
616635
impl<'tcx> rustc_type_ir::inherent::DefId<TyCtxt<'tcx>> for DefId {
617636
fn as_local(self) -> Option<LocalDefId> {
618637
self.as_local()

‎compiler/rustc_next_trait_solver/src/solve/assembly/mod.rs

+76-41
Original file line numberDiff line numberDiff line change
@@ -387,48 +387,83 @@ where
387387
G::consider_auto_trait_candidate(self, goal)
388388
} else if cx.trait_is_alias(trait_def_id) {
389389
G::consider_trait_alias_candidate(self, goal)
390-
} else if cx.is_lang_item(trait_def_id, TraitSolverLangItem::Sized) {
391-
G::consider_builtin_sized_candidate(self, goal)
392-
} else if cx.is_lang_item(trait_def_id, TraitSolverLangItem::Copy)
393-
|| cx.is_lang_item(trait_def_id, TraitSolverLangItem::Clone)
394-
{
395-
G::consider_builtin_copy_clone_candidate(self, goal)
396-
} else if cx.is_lang_item(trait_def_id, TraitSolverLangItem::PointerLike) {
397-
G::consider_builtin_pointer_like_candidate(self, goal)
398-
} else if cx.is_lang_item(trait_def_id, TraitSolverLangItem::FnPtrTrait) {
399-
G::consider_builtin_fn_ptr_trait_candidate(self, goal)
400-
} else if let Some(kind) = self.cx().fn_trait_kind_from_def_id(trait_def_id) {
401-
G::consider_builtin_fn_trait_candidates(self, goal, kind)
402-
} else if let Some(kind) = self.cx().async_fn_trait_kind_from_def_id(trait_def_id) {
403-
G::consider_builtin_async_fn_trait_candidates(self, goal, kind)
404-
} else if cx.is_lang_item(trait_def_id, TraitSolverLangItem::AsyncFnKindHelper) {
405-
G::consider_builtin_async_fn_kind_helper_candidate(self, goal)
406-
} else if cx.is_lang_item(trait_def_id, TraitSolverLangItem::Tuple) {
407-
G::consider_builtin_tuple_candidate(self, goal)
408-
} else if cx.is_lang_item(trait_def_id, TraitSolverLangItem::PointeeTrait) {
409-
G::consider_builtin_pointee_candidate(self, goal)
410-
} else if cx.is_lang_item(trait_def_id, TraitSolverLangItem::Future) {
411-
G::consider_builtin_future_candidate(self, goal)
412-
} else if cx.is_lang_item(trait_def_id, TraitSolverLangItem::Iterator) {
413-
G::consider_builtin_iterator_candidate(self, goal)
414-
} else if cx.is_lang_item(trait_def_id, TraitSolverLangItem::FusedIterator) {
415-
G::consider_builtin_fused_iterator_candidate(self, goal)
416-
} else if cx.is_lang_item(trait_def_id, TraitSolverLangItem::AsyncIterator) {
417-
G::consider_builtin_async_iterator_candidate(self, goal)
418-
} else if cx.is_lang_item(trait_def_id, TraitSolverLangItem::Coroutine) {
419-
G::consider_builtin_coroutine_candidate(self, goal)
420-
} else if cx.is_lang_item(trait_def_id, TraitSolverLangItem::DiscriminantKind) {
421-
G::consider_builtin_discriminant_kind_candidate(self, goal)
422-
} else if cx.is_lang_item(trait_def_id, TraitSolverLangItem::AsyncDestruct) {
423-
G::consider_builtin_async_destruct_candidate(self, goal)
424-
} else if cx.is_lang_item(trait_def_id, TraitSolverLangItem::Destruct) {
425-
G::consider_builtin_destruct_candidate(self, goal)
426-
} else if cx.is_lang_item(trait_def_id, TraitSolverLangItem::TransmuteTrait) {
427-
G::consider_builtin_transmute_candidate(self, goal)
428-
} else if cx.is_lang_item(trait_def_id, TraitSolverLangItem::EffectsIntersection) {
429-
G::consider_builtin_effects_intersection_candidate(self, goal)
430390
} else {
431-
Err(NoSolution)
391+
match cx.as_lang_item(trait_def_id) {
392+
Some(TraitSolverLangItem::Sized) => G::consider_builtin_sized_candidate(self, goal),
393+
Some(TraitSolverLangItem::Copy | TraitSolverLangItem::Clone) => {
394+
G::consider_builtin_copy_clone_candidate(self, goal)
395+
}
396+
Some(TraitSolverLangItem::Fn) => {
397+
G::consider_builtin_fn_trait_candidates(self, goal, ty::ClosureKind::Fn)
398+
}
399+
Some(TraitSolverLangItem::FnMut) => {
400+
G::consider_builtin_fn_trait_candidates(self, goal, ty::ClosureKind::FnMut)
401+
}
402+
Some(TraitSolverLangItem::FnOnce) => {
403+
G::consider_builtin_fn_trait_candidates(self, goal, ty::ClosureKind::FnOnce)
404+
}
405+
Some(TraitSolverLangItem::AsyncFn) => {
406+
G::consider_builtin_async_fn_trait_candidates(self, goal, ty::ClosureKind::Fn)
407+
}
408+
Some(TraitSolverLangItem::AsyncFnMut) => {
409+
G::consider_builtin_async_fn_trait_candidates(
410+
self,
411+
goal,
412+
ty::ClosureKind::FnMut,
413+
)
414+
}
415+
Some(TraitSolverLangItem::AsyncFnOnce) => {
416+
G::consider_builtin_async_fn_trait_candidates(
417+
self,
418+
goal,
419+
ty::ClosureKind::FnOnce,
420+
)
421+
}
422+
Some(TraitSolverLangItem::PointerLike) => {
423+
G::consider_builtin_pointer_like_candidate(self, goal)
424+
}
425+
Some(TraitSolverLangItem::FnPtrTrait) => {
426+
G::consider_builtin_fn_ptr_trait_candidate(self, goal)
427+
}
428+
Some(TraitSolverLangItem::AsyncFnKindHelper) => {
429+
G::consider_builtin_async_fn_kind_helper_candidate(self, goal)
430+
}
431+
Some(TraitSolverLangItem::Tuple) => G::consider_builtin_tuple_candidate(self, goal),
432+
Some(TraitSolverLangItem::PointeeTrait) => {
433+
G::consider_builtin_pointee_candidate(self, goal)
434+
}
435+
Some(TraitSolverLangItem::Future) => {
436+
G::consider_builtin_future_candidate(self, goal)
437+
}
438+
Some(TraitSolverLangItem::Iterator) => {
439+
G::consider_builtin_iterator_candidate(self, goal)
440+
}
441+
Some(TraitSolverLangItem::FusedIterator) => {
442+
G::consider_builtin_fused_iterator_candidate(self, goal)
443+
}
444+
Some(TraitSolverLangItem::AsyncIterator) => {
445+
G::consider_builtin_async_iterator_candidate(self, goal)
446+
}
447+
Some(TraitSolverLangItem::Coroutine) => {
448+
G::consider_builtin_coroutine_candidate(self, goal)
449+
}
450+
Some(TraitSolverLangItem::DiscriminantKind) => {
451+
G::consider_builtin_discriminant_kind_candidate(self, goal)
452+
}
453+
Some(TraitSolverLangItem::AsyncDestruct) => {
454+
G::consider_builtin_async_destruct_candidate(self, goal)
455+
}
456+
Some(TraitSolverLangItem::Destruct) => {
457+
G::consider_builtin_destruct_candidate(self, goal)
458+
}
459+
Some(TraitSolverLangItem::TransmuteTrait) => {
460+
G::consider_builtin_transmute_candidate(self, goal)
461+
}
462+
Some(TraitSolverLangItem::EffectsIntersection) => {
463+
G::consider_builtin_effects_intersection_candidate(self, goal)
464+
}
465+
_ => Err(NoSolution),
466+
}
432467
};
433468

434469
candidates.extend(result);

‎compiler/rustc_type_ir/src/interner.rs

+2-4
Original file line numberDiff line numberDiff line change
@@ -220,6 +220,8 @@ pub trait Interner:
220220

221221
fn is_lang_item(self, def_id: Self::DefId, lang_item: TraitSolverLangItem) -> bool;
222222

223+
fn as_lang_item(self, def_id: Self::DefId) -> Option<TraitSolverLangItem>;
224+
223225
fn associated_type_def_ids(self, def_id: Self::DefId) -> impl IntoIterator<Item = Self::DefId>;
224226

225227
fn for_each_relevant_impl(
@@ -245,10 +247,6 @@ pub trait Interner:
245247

246248
fn trait_may_be_implemented_via_object(self, trait_def_id: Self::DefId) -> bool;
247249

248-
fn fn_trait_kind_from_def_id(self, trait_def_id: Self::DefId) -> Option<ty::ClosureKind>;
249-
250-
fn async_fn_trait_kind_from_def_id(self, trait_def_id: Self::DefId) -> Option<ty::ClosureKind>;
251-
252250
fn supertrait_def_ids(self, trait_def_id: Self::DefId)
253251
-> impl IntoIterator<Item = Self::DefId>;
254252

‎compiler/rustc_type_ir/src/lang_items.rs

+6
Original file line numberDiff line numberDiff line change
@@ -3,8 +3,11 @@
33
pub enum TraitSolverLangItem {
44
// tidy-alphabetical-start
55
AsyncDestruct,
6+
AsyncFn,
67
AsyncFnKindHelper,
78
AsyncFnKindUpvars,
9+
AsyncFnMut,
10+
AsyncFnOnce,
811
AsyncFnOnceOutput,
912
AsyncIterator,
1013
CallOnceFuture,
@@ -22,6 +25,9 @@ pub enum TraitSolverLangItem {
2225
EffectsMaybe,
2326
EffectsNoRuntime,
2427
EffectsRuntime,
28+
Fn,
29+
FnMut,
30+
FnOnce,
2531
FnPtrTrait,
2632
FusedIterator,
2733
Future,

0 commit comments

Comments
 (0)