Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Static async fn in traits #3185

Merged
merged 8 commits into from
Dec 7, 2021
Merged
Changes from 6 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
258 changes: 258 additions & 0 deletions text/3185-static-async-fn-in-trait.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,258 @@
# Draft RFC: Static async fn in traits
tmandry marked this conversation as resolved.
Show resolved Hide resolved

- Feature Name: `async_fn_in_traits`
- Start Date: 2021-10-13
- RFC PR: [rust-lang/rfcs#3185](https://github.com/rust-lang/rfcs/pull/3185)
- Rust Issue: [rust-lang/rust#0000](https://github.com/rust-lang/rust/issues/0000)

# Summary
[summary]: #summary

Support `async fn` in traits that can be called via static dispatch. These will desugar to an anonymous associated type.

# Motivation
[motivation]: #motivation

Async/await allows users to write asynchronous code much easier than they could before. However, it doesn't play nice with other core language features that make Rust the great language it is, like traits.

In this RFC we will begin the process of integrating these two features and smoothing over a wrinkle that async Rust users have been working around since async/await stabilized nearly 3 years ago.

# Guide-level explanation
[guide-level-explanation]: #guide-level-explanation

You can write `async fn` in traits and trait impls. For example:

```rust
trait Service {
async fn request(&self, key: i32) -> Response;
}

struct MyService {
db: Database
}

impl Service for MyService {
async fn request(&self, key: i32) -> Response {
Response {
contents: self.db.query(key).await.to_string()
}
}
}
```

This is useful for writing generic async code.

Currently, if you use an `async fn` in a trait, that trait is not `dyn` safe. If you need to use dynamic dispatch combined with async functions, you can use the [`async-trait`] crate. We expect to extend the language to support this use case in the future.

Note that if a function in a trait is written as an `async fn`, it must also be written as an `async fn` in your implementation of that trait. With the above trait, you could not write this:
Copy link

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Is there a possibility that this restriction could be later lifted in a backwards-compatible way?

This is important, because with the explicit impl Future you're able to run some preliminary logic before constructing the final future, which makes it possible for the future not to capture some of the input arguments, potentially making the future Send where it wouldn't be otherwise.

In regular async functions, it's sort of an implementation detail of the function whether it's implemented as an actual async fn or as an fn -> impl Future — while this fact "leaks" to rustdoc and elsewhere, it can be more or less flipped back and forth without breaking the public interface of the function. Whereas this design would "bake in", at the trait level, the fact that implementation functions must use the async fn syntax, even though the users of the trait don't actually care since they invoke a method and get a future from it either way.

So my concern is, would anyone actually use the async fn syntax when defining a trait, knowing that this forces an unnecessary requirement on all implementations?

Copy link

@rushmorem rushmorem Oct 31, 2021

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@bugaevc IMHO, as you have already alluded to, these function signatures are not interchangeable and I don't think they should be treated as such. At a glance async fn tells me that it is fully async and not awaiting has no side effects at all. fn -> impl Future, on the other hand, may or may not have side effects. Not awaiting it may also block the thread or not. To cut the long story short, I think these function signatures provide different guarantees. A fact we might want to make explicit.

Copy link
Member

@scottmcm scottmcm Nov 2, 2021

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

One way that this could happen would be for impls to infer the associated type's values.

That would be nice for things other than async fn too -- one could imagine

impl Iterator for Foo {
    fn next(&mut self) -> Option<i32> { ... }
}

just working without needing to say type Output = i32;, treating the associated type kinda like it was a TAIT.

(With TAIT it could almost be written something like

type BlahBlah = impl Any;
impl Iterator for Foo {
    type Output = BlahBlah;
    fn next(&mut self) -> Option<i32> { ... }
}

which is noisier without really adding anything particularly useful to the reader.)

Copy link
Member

@ibraheemdev ibraheemdev Nov 2, 2021

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Yes, I had the same idea. Inferring associated types would be very nice outside of async traits as well. The downside is that when reading a trait impl, you might have to do a bit more searching to identify a given associated type.

Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

We can allow this later by adding return position impl Trait in traits and making this feature interoperable with it (i.e., redefine the async fn in traits desugaring in terms of return position impl Trait). I think we should do so, because as you and @cramertj point out, library authors will want to know they aren't backing themselves into a corner by using this feature.


```rust
impl Service for MyService {
fn request(&self, key: i32) -> impl Future<Output = Response> {
async {
...
}
}
}
```

Doing so will give you an "expected async fn" error. If you need to do this for some reason, you can use an associated type in the trait:

```rust
trait Service {
type RequestFut<'a>: Future<Output = Response>
where
Self: 'a;
fn request(&self, key: i32) -> RequestFut;
}

impl Service for MyService {
type RequestFut<'a> = impl Future + 'a
where
Self: 'a;
fn request<'a>(&'a self, key: i32) -> RequestFut<'a> {
async { ... }
}
}
```

Note that in the impl we are setting the value of the associated type to `impl Future`, because async blocks produce unnameable opaque types. The associated type is also generic over a lifetime `'a`, which allows it to capture the `&'a self` reference passed by the caller.

# Reference-level explanation
[reference-level-explanation]: #reference-level-explanation

## New syntax

We introduce the `async fn` sugar into traits and impls. No changes to the grammar are needed because the Rust grammar already support this construction, but async functions result in compilation errors in later phases of the compiler.

```rust
trait Example {
async fn method(&self);
}

impl Example for ExampleType {
async fn method(&self);
}
```

## Semantic rules

When an async function is present in a trait or trait impl...

### The trait is not considered dyn safe

This limitation is expected to be lifted in future RFCs.

### Both the trait and its impls must use `async` syntax

It is not legal to use an async function in a trait and a "desugared" function in an impl.

## Equivalent desugaring

### Trait

Async functions in a trait desugar to an associated function that returns a generic associated type (GAT):

* Just as with [ordinary async functions](https://rust-lang.github.io/rfcs/2394-async_await.html#lifetime-capture-in-the-anonymous-future), the GAT has a generic parameter for every generic parameter that appears on the fn, along with implicit lifetime parameters.
* The GAT has the complete set of where clauses that appear on the `fn`, including any implied bounds.
* The GAT is "anonymous", meaning that its name is an internal symbol that cannot be referred to directly. (In the examples, we will use `$` to represent this name.)


```rust
trait Example {
async fn method<P0..Pn>(&self)
where
WC0..WCn;
}

// Becomes:

trait Example {
type $<'me, P0..Pn>: Future<Output = ()>
where
WC0..WCn, // Explicit where clauses
Self: 'me; // Implied bound from `&self` parameter

fn method<P0..Pn>(&self) -> Self::$<'_, P0..Pn>
where
WC0..WCn;
}
```

`async fn` that appear in impls are desugared in the same general way as an [existing async function](https://doc.rust-lang.org/reference/items/functions.html#async-functions), but with some slight differences:

* The value of the associated type `$` is equal to an `impl Future` type, rather than the `impl Future` being the return type of the function
* The function returns `Self::$<...>` with all the appropriate generic parameters

Otherwise, the desugaring is the same. The body of the function becomes an `async move { ... }` block that both (a) captures all parameters and (b) contains the body expression.

```rust
impl Example for ExampleType {
async fn method<P0..Pn>(&self) {
...
}
}

impl Example for ExampleType {
type $<'me, P0..Pn> = impl Future<Output = ()> + 'me
where
WC0..WCn, // Explicit where clauses
Self: 'me; // Implied bound from `&self` parameter

fn method<P0..Pn>(&self) -> Self::$<'_, P0..Pn> {
async move { ... }
}
}
```

# Rationale and alternatives
[rationale-and-alternatives]: #rationale-and-alternatives

## Why are we adding this RFC now?

This RFC represents the least controversial addition to async/await that we could add right now. It was not added before due to limitations in the compiler that have now been lifted – namely, support for [Generic Associated Types][gat] and [Type Alias Impl Trait][tait].

[gat]: https://github.com/rust-lang/generic-associated-types-initiative
[tait]: https://github.com/rust-lang/rust/issues/63063
kennytm marked this conversation as resolved.
Show resolved Hide resolved

## Why are the result traits not dyn safe?

Supporting async fn and dyn is a complex topic -- you can read the details on the [dyn traits](https://rust-lang.github.io/async-fundamentals-initiative/evaluation/challenges/dyn_traits.html) page of the async fundamentals evaluation doc.

## Can we add support for dyn later?

Yes, nothing in this RFC precludes us from making traits containing async functions dyn safe, presuming that we can overcome the obstacles inherent in the design space.

## What are users using today and why don't we just do that?

Users in the ecosystem have worked around the lack of support for this feature with the [async-trait] proc macro, which desugars into `Box<dyn Future>`s instead of anonymous associated types. This has the disadvantage of requiring users to use `Box<dyn>` along with all the [performance implications] of that, which prevent some use cases. It is also not suitable for users like [embassy](https://github.com/embassy-rs/embassy), which aim to support the "no-std" ecosystem.

[async-trait]: https://github.com/dtolnay/async-trait
[performance implications]: https://rust-lang.github.io/wg-async-foundations/vision/submitted_stories/status_quo/barbara_benchmarks_async_trait.html

## Will anyone use async-trait crate once this RFC lands?

The async-trait crate will continue to be useful after this RFC, because it allows traits to remain `dyn`-safe. This is a limitation in the current design that we plan to address in the future.

# Prior art
[prior-art]: #prior-art

## The `async-trait` crate

The most common way to use `async fn` in traits is to use the [`async-trait`] crate. This crate takes a different approach to the one described in this RFC. Async functions are converted into ordinary trait functions that return `Box<dyn Future>` rather than using an associated type. This means that the resulting traits are dyn safe and avoids a dependency on generic associated types, but it also has two downsides:

* Requires a box allocation on every trait function call; while this is often no big deal, it can be prohibitive for some applications.
* Requires the trait to state up front whether the resulting futures are `Send` or not. The [`async-trait`] crate defaults to `Send` and users write `#[async_trait(?Send)]` to disable this default.

Since the async function support in this RFC means that traits are not dyn safe, we do not expect it to completely displace uses of the `#[async_trait]` crate.

[`async-trait`]: https://crates.io/crates/async-trait

## The real-async-trait crate

The [`real-async-trait`] lowers `async fn` to use GATs and impl Trait, roughly as described in this RFC.

[`real-async-trait`]: https://crates.io/crates/real-async-trait

# Unresolved questions
[unresolved-questions]: #unresolved-questions

- What parts of the design do you expect to resolve through the RFC process before this gets merged?
- What parts of the design do you expect to resolve through the implementation of this feature before stabilization?
- What related issues do you consider out of scope for this RFC that could be addressed in the future independently of the solution that comes out of this RFC?

# Future possibilities
[future-possibilities]: #future-possibilities

## Dyn compatibility

It is not a breaking change for traits to become dyn safe. We expect to make traits with async functions dyn safe, but doing so requires overcoming a number of interesting challenges, as described in the [async fundamentals evaluation doc][eval].

## Impl trait in traits

The [impl trait initiative] is expecting to propose "impl trait in traits" (see the [explainer](https://rust-lang.github.io/impl-trait-initiative/explainer/rpit_trait.html) for a brief summary). This RFC is compatible with the proposed design.
tmandry marked this conversation as resolved.
Show resolved Hide resolved

## Allowing sugared and desugared forms

In the current proposal, `async fn`s in traits must be implemented using `async fn`. Using a desugared form is not allowed, which can preclude implementations from doing things like doing some work at call time before returning a future. It would also be backwards-incompatible for library authors to move between the sugared and desugared form.

Once impl trait in traits is supported, we can redefine the desugaring of `async fn` in traits in terms of that feature (similar to how `async fn` is desugared for free functions). That provides a clear path to allowing the desugared form to be used interchangeably with the `async fn` form. In other words, you should be able to write the following:

```rust
trait Example {
async fn method(&self);
}

impl Example for ExampleType {
fn method(&self) -> impl Future<Output = ()> + '_ {}
}
```

It could also be made backward-compatible for the trait to change between the sugared and desugared form.

## Ability to name the type of the returned future

This RFC does not propose any means to name the future that results from an `async fn`. That is expected to be covered in a future RFC from the [impl trait initiative]; you can read more about the [proposed design](https://rust-lang.github.io/impl-trait-initiative/explainer/rpit_names.html) in the explainer.

[eval]: https://rust-lang.github.io/async-fundamentals-initiative/evaluation.html
[impl trait initiative]: https://rust-lang.github.io/impl-trait-initiative/