blob: 3f2500daf54e0e082e0c82f0aa08f5b52599ba51 [file] [log] [blame] [view]
## Advanced Functions and Closures
This section explores some advanced features related to functions and closures,
including function pointers and returning closures.
### Function Pointers
Weve talked about how to pass closures to functions; you can also pass regular
functions to functions! This technique is useful when you want to pass a
function youve already defined rather than defining a new closure. Functions
coerce to the type `fn` (with a lowercase _f_), not to be confused with the `Fn`
closure trait. The `fn` type is called a _function pointer_. Passing functions
with function pointers will allow you to use functions as arguments to other
functions.
The syntax for specifying that a parameter is a function pointer is similar to
that of closures, as shown in Listing 20-28, where weve defined a function
`add_one` that adds 1 to its parameter. The function `do_twice` takes two
parameters: a function pointer to any function that takes an `i32` parameter
and returns an `i32`, and one `i32` value. The `do_twice` function calls the
function `f` twice, passing it the `arg` value, then adds the two function call
results together. The `main` function calls `do_twice` with the arguments
`add_one` and `5`.
<Listing number="20-28" file-name="src/main.rs" caption="Using the `fn` type to accept a function pointer as an argument">
```rust
{{#rustdoc_include ../listings/ch20-advanced-features/listing-20-28/src/main.rs}}
```
</Listing>
This code prints `The answer is: 12`. We specify that the parameter `f` in
`do_twice` is an `fn` that takes one parameter of type `i32` and returns an
`i32`. We can then call `f` in the body of `do_twice`. In `main`, we can pass
the function name `add_one` as the first argument to `do_twice`.
Unlike closures, `fn` is a type rather than a trait, so we specify `fn` as the
parameter type directly rather than declaring a generic type parameter with one
of the `Fn` traits as a trait bound.
Function pointers implement all three of the closure traits (`Fn`, `FnMut`, and
`FnOnce`), meaning you can always pass a function pointer as an argument for a
function that expects a closure. Its best to write functions using a generic
type and one of the closure traits so your functions can accept either
functions or closures.
That said, one example of where you would want to only accept `fn` and not
closures is when interfacing with external code that doesnt have closures: C
functions can accept functions as arguments, but C doesnt have closures.
As an example of where you could use either a closure defined inline or a named
function, lets look at a use of the `map` method provided by the `Iterator`
trait in the standard library. To use the `map` method to turn a vector of
numbers into a vector of strings, we could use a closure, as in Listing 20-29.
<Listing number="20-29" caption="Using a closure with the `map` method to convert numbers to strings">
```rust
{{#rustdoc_include ../listings/ch20-advanced-features/listing-20-29/src/main.rs:here}}
```
</Listing>
Or we could name a function as the argument to map instead of the closure.
Listing 20-30 shows what this would look like.
<Listing number="20-30" caption="Using the `String::to_string` method to convert numbers to strings">
```rust
{{#rustdoc_include ../listings/ch20-advanced-features/listing-20-30/src/main.rs:here}}
```
</Listing>
Note that we must use the fully qualified syntax that we talked about in
[“Advanced Traits”][advanced-traits]<!-- ignore --> because there are multiple
functions available named `to_string`.
Here, were using the `to_string` function defined in the `ToString` trait,
which the standard library has implemented for any type that implements
`Display`.
Recall from [“Enum values”][enum-values]<!-- ignore --> in Chapter 6 that the
name of each enum variant that we define also becomes an initializer function.
We can use these initializer functions as function pointers that implement the
closure traits, which means we can specify the initializer functions as
arguments for methods that take closures, as seen in Listing 20-31.
<Listing number="20-31" caption="Using an enum initializers with the `map` method to create a `Status` instance from numbers">
```rust
{{#rustdoc_include ../listings/ch20-advanced-features/listing-20-31/src/main.rs:here}}
```
</Listing>
Here we create `Status::Value` instances using each `u32` value in the range
that `map` is called on by using the initializer function of `Status::Value`.
Some people prefer this style and some people prefer to use closures. They
compile to the same code, so use whichever style is clearer to you.
### Returning Closures
Closures are represented by traits, which means you cant return closures
directly. In most cases where you might want to return a trait, you can instead
use the concrete type that implements the trait as the return value of the
function. However, you cant usually do that with closures because they dont
have a concrete type that is returnable. Youre not allowed to use the function
pointer `fn` as a return type if the closure captures any values from its scope,
for example.
Instead, you will normally use the `impl Trait` syntax we learned about in
Chapter 10. You can return any function type, using `Fn`, `FnOnce` and `FnMut`.
For example, the code in Listing 20-32 will work just fine.
<Listing number="20-32" caption="Returning a closure from a function using the `impl Trait` syntax">
```rust
{{#rustdoc_include ../listings/ch20-advanced-features/listing-20-32/src/lib.rs}}
```
</Listing>
However, as we noted in [“Closure Type Inference and
Annotation”][closure-types]<!-- ignore --> in Chapter 13, each closure is also
its own distinct type. If you need to work with multiple functions that have the
same signature but different implementations, you will need to use a trait
object for them. Consider what happens if you write code like that shown in
Listing 20-33.
<Listing file-name="src/main.rs" number="20-33" caption="Creating a `Vec<T>` of closures defined by functions that return `impl Fn`">
```rust,ignore,does_not_compile
{{#rustdoc_include ../listings/ch20-advanced-features/listing-20-33/src/main.rs}}
```
</Listing>
Here we have two functions, `returns_closure` and `returns_initialized_closure`,
which both return `impl Fn(i32) -> i32`. Notice that he closures that they
return are different, even though they implement the same type. If we try to
compile this, Rust lets us know that it wont work:
```text
{{#include ../listings/ch20-advanced-features/listing-20-33/output.txt}}
```
The error message tells us that whenever we return an `impl Trait` Rust creates
a unique _opaque type_, a type where we cannot see into the details of what Rust
constructs for us. So even though these functions both return closures that
implements the same trait, `Fn(i32) -> i32`, the opaque types Rust generates for
each are distinct. (This is similar to how Rust produces different concrete
types for distinct async blocks even when they have the same output type, as we
saw in [“Working with Any Number of Futures”][any-number-of-futures] in Chapter
17. We have seen a solution to this problem a few times now: we can use a trait
object, as in Listing 20-34.
<Listing number="20-34" caption="Creating a `Vec<T>` of closures defined by functions that return `Box<dyn Fn>` so they have the same type">
```rust
{{#rustdoc_include ../listings/ch20-advanced-features/listing-20-34/src/main.rs:here}}
```
</Listing>
This code will compile just fine. For more about trait objects, refer to the
section [“Using Trait Objects That Allow for Values of Different
Types”][using-trait-objects-that-allow-for-values-of-different-types]<!-- ignore
--> in Chapter 18.
Next, lets look at macros!
[advanced-traits]: ch20-02-advanced-traits.html#advanced-traits
[enum-values]: ch06-01-defining-an-enum.html#enum-values
[closure-types]: ch13-01-closures.html#closure-type-inference-and-annotation
[any-number-of-futures]: ch17-03-more-futures.html
[using-trait-objects-that-allow-for-values-of-different-types]: ch18-02-trait-objects.html#using-trait-objects-that-allow-for-values-of-different-types