Return-position impl trait in trait (RPITIT) の実 装を雑に見てみる TaKO8Ki

Takayuki Maeda / TaKO8Ki Rust committer(compiler contributor team/diagnostic working group) 🦀 @TaKO8Ki @TaKOBKi

RPITITとは？

RPITITとは？

RPITITとは？ fn combine_vecs( v: Vec, u: Vec, ) -> impl Iterator { v.into_iter().chain(u.into_iter()).cycle() }

RPITITとは？ struct Foo { items: Vec } #[derive(Clone)] struct Bar; trait Baz { fn items(&self) -> impl Iterator; } impl Baz for Foo { fn items(&self) -> impl Iterator { self.items.iter().cloned() } }

Async fn in trait (AFIT) pub trait Trait { async fn method(&self); }

Async fn in trait (AFIT) pub trait Trait { async fn method(&self); } // desugars to pub trait Trait { fn method(&self) -> impl Future; }

RPITITがどう実装されている か？

ASTからHIRへのlowering fn foo() -> impl Bar と同じように hir::ItemKind::Opaque にloweringさ れる


ASTからHIRへのlowering pub struct OpaqueTy<'hir> { pub generics: &'hir Generics<'hir>, pub bounds: GenericBounds<'hir>, pub origin: OpaqueTyOrigin, pub lifetime_mapping: &'hir [(&'hir Lifetime, LocalDefId)], pub in_trait: bool, pub precise_capturing_args: Option<(&'hir [PreciseCapturingArg<'hir>], Span)>, }

ASTからHIRへのlowering pub struct OpaqueTy<'hir> { pub generics: &'hir Generics<'hir>, pub bounds: GenericBounds<'hir>, pub origin: OpaqueTyOrigin, pub lifetime_mapping: &'hir [(&'hir Lifetime, LocalDefId)], pub in_trait: bool, pub precise_capturing_args: Option<(&'hir [PreciseCapturingArg<'hir>], Span)>, }

ASTからHIRへのlowering let opaque_ty_item = hir::OpaqueTy { generics: this.arena.alloc(hir::Generics { params: generic_params, predicates: &[], has_where_clause_predicates: false, where_clause_span: this.lower_span(span), span: this.lower_span(span), }), bounds, origin, lifetime_mapping, in_trait, precise_capturing_args, }; compiler/rustc_ast_lowering/src/lib.rs#L1738-L1751

HIRからtyへのlowering traits/implsに新しい関連型を作ってそれを扱うようにする (hir::TyKind::OpaqueDef をopaque type A = impl B; の代わりに関連型射 影 projectionにloweringする)


HIRからtyへのlowering (traits) trait NewIntoIterator { type Item; fn into_iter(self) -> impl Iterator; } // 下のようになる (例) trait NewIntoIterator { type Item; type A: Iterator; fn into_iter(self) -> ::A; }

HIRからtyへのlowering (traits) trait NewIntoIterator { type Item; fn into_iter(self) -> impl Iterator; } // 下のようになる (例) trait NewIntoIterator { type Item; type A: Iterator; fn into_iter(self) -> ::A; }

HIRからtyへのlowering (impls) impl NewIntoIterator for Vec { type Item = u32; fn into_iter(self) -> impl Iterator { self.into_iter() } } // 下のようになる (例) impl NewIntoIterator for Vec { type Item = u32; type A = impl Iterator; fn into_iter(self) -> ::A { self.into_iter() } }

HIRからtyへのlowering (impls) impl NewIntoIterator for Vec { type Item = u32; fn into_iter(self) -> impl Iterator { self.into_iter() } } // 下のようになる (例) impl NewIntoIterator for Vec { type Item = u32; type A = impl Iterator; fn into_iter(self) -> ::A { self.into_iter() } }

HIRからtyへのlowering (traits) let trait_assoc_ty = tcx.at(span).create_def(trait_def_id, kw::Empty, DefKind::AssocTy); compiler/rustc_ty_utils/src/assoc.rs#L259

HIRからtyへのlowering (impls) let impl_assoc_ty = tcx.at(span).create_def(impl_local_def_id, kw::Empty, DefKind::AssocTy); compiler/rustc_ty_utils/src/assoc.rs#L312

HIRからtyへのlowering (impls) を少し深堀り 新しく作る関連型のvisibility (型はprivateなのかpublicなのか) やdefaultnessはどの ように決定すればいいか？


HIRからtyへのlowering (impls) を少し深堀り 親の関数から引き継いでくれば良い 
 // Copy visility of the containing function. trait_assoc_ty .visibility(tcx.visibility(fn_def_id)); // Copy defaultness of the containing function. trait_assoc_ty .defaultness(tcx.defaultness(fn_def_id)); compiler/rustc_ty_utils/src/assoc.rs#L280-L285

HIRからtyへのlowering (impls/traits) それ以外にもrustc_hir_analysis::collect::generics_of::generics_of など実行されるクエリはいくつかある。


RPITITでdynが動かないのは なぜか？

RPITITでdynが動かないのはなぜか？ RPITITやasync fnはobject safeではないからdynが使えない ref: https://doc.rust-lang.org/reference/items/traits.html#object-safety

RPITITでdynが動かないのはなぜか？ object-safetyとは、

RPITITでdynが動かないのはなぜか？ trait NotObjectSafe { const CONST: i32 = 1; // ERROR: cannot have associated const fn foo() {} // ERROR: associated function without Sized fn returns(&self) -> Self; // ERROR: Self in return type fn typed(&self, x: T) {} // ERROR: has generic type parameters fn nested(self: Rc>) {} // ERROR: nested receiver not yet supported }

RPITITでdynが動かないのはなぜか？ rust-lang/rfcs#255 で定義されているように Trait objectのmethodが呼び出されるタイミングではなく、trait objectが生 成されるときにobject safetyを強制するようになった。これによってTrait objectのobject safetyに関して詳細なエラーが得られるようになった。

RPITITでdynが動かないのはなぜか？ trait NotObjectSafe { fn foo() {} } struct S; impl NotObjectSafe for S { fn foo() { } } fn main() { let obj: Box = Box::new(S); }

RPITITでdynが動かないのはなぜか？ error[E0038]: the trait `NotObjectSafe` cannot be made into an object --> src/main.rs:12:14 | 12 | let obj: Box = Box::new(S); // ERROR | ^^^^^^^^^^^^^^^^^^^^^^ `NotObjectSafe` cannot be made into an object | note: for a trait to be "object safe" it needs to allow building a vtable to allow the call to be resolvable dynamically; for more information visit --> src/main.rs:2:8 | 1 | trait NotObjectSafe { | ------------- this trait cannot be made into an object... 2 | fn foo() {} // ERROR: associated function without Sized | ^^^ ...because associated function `foo` has no `self` parameter = help: only type `S` implements the trait, consider using it directly instead

RPITITでdynが動かないのはなぜか？ error[E0038]: the trait `NotObjectSafe` cannot be made into an object --> src/main.rs:12:14 | 12 | let obj: Box = Box::new(S); // ERROR | ^^^^^^^^^^^^^^^^^^^^^^ `NotObjectSafe` cannot be made into an object | note: for a trait to be "object safe" it needs to allow building a vtable to allow the call to be resolvable dynamically; for more information visit --> src/main.rs:2:8 | 1 | trait NotObjectSafe { | ------------- this trait cannot be made into an object... 2 | fn foo() {} // ERROR: associated function without Sized | ^^^ ...because associated function `foo` has no `self` parameter = help: only type `S` implements the trait, consider using it directly instead

RPITITでdynを動かしたい

RPITITでdynを動かしたい リリース記事ではtrait-variant crateがdynamic dispatchに対応する予定ら しい。 > We plan to provide utilities that enable dynamic dispatch in an upcoming version of the trait-variant crate. ref: https://blog.rust-lang.org/2023/12/21/async-fn-rpit-in-traits.html

RPITITでdynを動かしたい まだdynamic dispatchに対応したバージョンがリリースされていないので対応 方針を軽く探ってみる。 rust-lang/impl-trait-utils#34

RPITITでdynを動かしたい pub trait AsyncIter { type Item; async fn next(&mut self) -> Option; async fn size_hint(&self) -> Option; }

RPITITでdynを動かしたい pub trait AsyncIter { type Item; // async fn next(&mut self) -> Option; fn next(&mut self) -> impl Future>; // async fn size_hint(&self) -> Option; fn size_hint(&self) -> impl Future>; }

RPITITでdynを動かしたい pub trait AsyncIter { type Item; type Next: Future>; // fn next(&mut self) -> impl Future> fn next(&mut self) -> impl Future>; type SizeHint: Future>; // fn size_hint(&self) -> impl Future>; fn size_hint(&self) -> Option; }

RPITITでdynを動かしたい pub struct DynAsyncIter <'data, Item> { fatptr: FatPtr<'data, Item>, } trait ErasedAsyncIter { type Item; fn next<'me>(&'me mut self) -> Pin> + 'me>>; fn size_hint<'me>(&'me self) -> Pin> + 'me>>; }

RPITITでdynを動かしたい impl<'data, Item> DynAsyncIter<'data, Item> { pub fn new(value: T) -> DynAsyncIter<'data, Item> where T: AsyncIter + 'data, Item: 'data, { let b: Box> = Box::new(value); let raw: *mut dyn ErasedAsyncIter = Box::into_raw(b); unsafe { DynAsyncIter { fatptr: FatPtr::new(raw).tagged(), } } } // .. }

RPITITでdynを動かしたい union FatPtr<'data, Item> { raw: *mut (dyn ErasedAsyncIter + 'data), usizes: (usize, usize), } impl<'data, Item> FatPtr<'data, Item> { // ... unsafe fn tagged(self) -> Self { let (data, vtable) = self.usizes; FatPtr { usizes: (data | 1, vtable), } } } usizesフィールドの1番目の要素 としてvtableへのポインタが 入っている

RPITITでdynを動かしたい #[trait_variant (dyn)] pub trait AsyncIter { type Item; async fn next(&mut self) -> Option; async fn size_hint(&self) -> Option; } trait_variant(dyn)によってこ れらのtrait/structが実装され るようにするのが現状の方針ら しい

まとめ ● ASTから hir::ItemKind::Opaque にloweringされるところから始まり、RPITIT 動きが何となく理解できた。
 ● RPITITやasync fnはobject safeではないのでdynが使えない。 ● dynamic dispatchを使うためには今のところ async-trait proc macro が必要だが、将来的に trait-variant が対応する予定がある。

Links ● https://github.com/rust-lang/rust/pull/122103
 ● https://github.com/rust-lang/rust/pull/101224
 ● https://rust-lang.github.io/rfcs/3425-return-position-impl-trait-in-traits.ht ml
 ● https://github.com/rust-lang/rfcs/blob/master/text/0255-object-safety.md
 ● https://github.com/rust-lang/impl-trait-utils/issues/34