Sometimes, when writing a function or data type, we may want it to work for multiple types of arguments. In Rust, we can do this with generics. Generics are called ‘parametric polymorphism’ in type theory, which means that they are types or functions that have multiple forms (‘poly’ is multiple, ‘morph’ is form) over a given parameter (‘parametric’).
Anyway, enough type theory, let’s check out some generic code. Rust’s standard library provides a type, Option<T>
, that’s generic:
# #![allow(unused_variables)] #fn main() { enum Option<T> { Some(T), None, } #}
The <T>
part, which you’ve seen a few times before, indicates that this is a generic data type. Inside the declaration of our enum
, wherever we see a T
, we substitute that type for the same type used in the generic. Here’s an example of using Option<T>
, with some extra type annotations:
# #![allow(unused_variables)] #fn main() { let x: Option<i32> = Some(5); #}
In the type declaration, we say Option<i32>
. Note how similar this looks to Option<T>
. So, in this particular Option
, T
has the value of i32
. On the right-hand side of the binding, we make a Some(T)
, where T
is 5
. Since that’s an i32
, the two sides match, and Rust is happy. If they didn’t match, we’d get an error:
let x: Option<f64> = Some(5); // error: mismatched types: expected `core::option::Option<f64>`, // found `core::option::Option<_>` (expected f64 but found integral variable)
That doesn’t mean we can’t make Option<T>
s that hold an f64
! They have to match up:
# #![allow(unused_variables)] #fn main() { let x: Option<i32> = Some(5); let y: Option<f64> = Some(5.0f64); #}
This is just fine. One definition, multiple uses.
Generics don’t have to only be generic over one type. Consider another type from Rust’s standard library that’s similar, Result<T, E>
:
# #![allow(unused_variables)] #fn main() { enum Result<T, E> { Ok(T), Err(E), } #}
This type is generic over two types: T
and E
. By the way, the capital letters can be any letter you’d like. We could define Result<T, E>
as:
# #![allow(unused_variables)] #fn main() { enum Result<A, Z> { Ok(A), Err(Z), } #}
if we wanted to. Convention says that the first generic parameter should be T
, for ‘type’, and that we use E
for ‘error’. Rust doesn’t care, however.
The Result<T, E>
type is intended to be used to return the result of a computation, and to have the ability to return an error if it didn’t work out.
We can write functions that take generic types with a similar syntax:
# #![allow(unused_variables)] #fn main() { fn takes_anything<T>(x: T) { // Do something with `x`. } #}
The syntax has two parts: the <T>
says “this function is generic over one type, T
”, and the x: T
says “x has the type T
.”
Multiple arguments can have the same generic type:
# #![allow(unused_variables)] #fn main() { fn takes_two_of_the_same_things<T>(x: T, y: T) { // ... } #}
We could write a version that takes multiple types:
# #![allow(unused_variables)] #fn main() { fn takes_two_things<T, U>(x: T, y: U) { // ... } #}
You can store a generic type in a struct
as well:
# #![allow(unused_variables)] #fn main() { struct Point<T> { x: T, y: T, } let int_origin = Point { x: 0, y: 0 }; let float_origin = Point { x: 0.0, y: 0.0 }; #}
Similar to functions, the <T>
is where we declare the generic parameters, and we then use x: T
in the type declaration, too.
When you want to add an implementation for the generic struct
, you declare the type parameter after the impl
:
# #![allow(unused_variables)] #fn main() { # struct Point<T> { # x: T, # y: T, # } # impl<T> Point<T> { fn swap(&mut self) { std::mem::swap(&mut self.x, &mut self.y); } } #}
So far you’ve seen generics that take absolutely any type. These are useful in many cases: you’ve already seen Option<T>
, and later you’ll meet universal container types like Vec<T>
. On the other hand, often you want to trade that flexibility for increased expressive power. Read about trait bounds to see why and how.
Most of the time when generics are involved, the compiler can infer the generic parameters automatically:
# #![allow(unused_variables)] #fn main() { // v must be a Vec<T> but we don't know what T is yet let mut v = Vec::new(); // v just got a bool value, so T must be bool! v.push(true); // Debug-print v println!("{:?}", v); #}
Sometimes though, the compiler needs a little help. For example, had we omitted the last line, we would get a compile error:
let v = Vec::new(); // ^^^^^^^^ cannot infer type for `T` // // note: type annotations or generic parameter binding required println!("{:?}", v);
We can solve this using either a type annotation:
# #![allow(unused_variables)] #fn main() { let v: Vec<bool> = Vec::new(); println!("{:?}", v); #}
or by binding the generic parameter T
via the so-called ‘turbofish’ ::<>
syntax:
# #![allow(unused_variables)] #fn main() { let v = Vec::<bool>::new(); println!("{:?}", v); #}
The second approach is useful in situations where we don’t want to bind the result to a variable. It can also be used to bind generic parameters in functions or methods. See Iterators § Consumers for an example.
© 2010 The Rust Project Developers
Licensed under the Apache License, Version 2.0 or the MIT license, at your option.
https://doc.rust-lang.org/book/first-edition/generics.html