- 1 Getting Started
- 2 Programming a Guessing Game
- 3 Comming Programming Concepts
- 4 Ownership
- 5 Using Structs
- 6 Enums and Pattern Matching
- 7 Packages Crates and Modules
- 8 Common Collections
- 9 Error Handling
- 10 Generic Types Traits Lifetimes
- 11 How to Write Tests
- 12 Building a Command Line Program
- 13 Functional Language Features
- 14 More About Cargo
- 15 Smart Pointers
Install
curl --proto '=https' --tlsv1.3 https://sh.rustup.rs -sSf | sh
Check, update, uninstall
rustc --version
rustup update
rustup self uninstall
It’s traditional when learning a new language to write a little program that prints the text Hello, world!
to the screen, so we’ll do the same here!
We suggest making a projects directory
in your home directory and keeping all your projects there
mkdir ~/projects
cd ~/projects
mkdir hello_world
cd hello_world
Next, make a new source file and call it main.rs.
fn main() {
println!("Hello, world!");
}
Compile
rustc main.rs
./main
These lines define a function named main
. The function body is wrapped in {}
fn main() {
}
If you want to stick to a standard style across Rust projects, you can use an automatic formatter tool called
rustfmt
- Rust style is to indent with
four spaces
, not a tab - println! calls a
Rust macro
- we pass string as an argument to
println!
- we end the line with a semicolon
(;)
Before running a Rust program, you must compile it using rustc
rustc main.rs
Cargo
is Rust’s build system and package manager
.
Let’s create a new project using Cargo
cargo new hello_cargo
cd hello_cargo
Filename: Cargo.toml - Cargo’s configuration file.
[package]
name = "hello_cargo"
version = "0.1.0"
edition = "2021"
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
[dependencies]
Build project
cargo build
Because the default build is a debug build, Cargo puts the binary in a directory named debug. You can run the executable with this command:
./target/debug/hello_cargo
We can also use cargo run to compile the code and then run
the resultant executable all in one
command:
cargo run
This command quickly checks
your code to make sure it compiles but doesn’t produce an executable
cargo check
When your project is finally ready for release, you can use command to compile it with optimizations.
cargo --build release
We’ll implement a classic beginner programming problem: a guessing game. Here’s how it works: the program will generate a random integer between 1 and 100. It will then prompt the player to enter a guess. After a guess is entered, the program will indicate whether the guess is too low or too high. If the guess is correct, the game will print a congratulatory message and exit.
Go to the projects directory and make a new project
using Cargo
cargo new guessing_game
cd guessing_game
The first part of the guessing game program will ask for user input, process that input, and check that the input is in the expected form.
use std::io; //standard library, known as std
fn main() { //the entry point
println!("Guess the number!"); //macro that prints a string to the screen
println!("Please input your guess."); // macro println!
let mut guess = String::new();
io::stdin()
.read_line(&mut guess)
.expect("Failed to read line");
println!("You guessed: {guess}");
}
Next, we’ll create a variable to store the user input, like this:
let mut guess = String::new();
To make a variable mutable, we add mut before the variable name:
let apples = 5; // immutable
let mut bananas = 5; // mutable
In full, the let mut guess = String::new(); line has created a mutable variable that is currently bound to a new, empty instance of a String:
let mut guess
will introduce amutable variable
namedguess
- equal sign
(=)
tells Rust we want tobind
something to the variable ::
syntax in the::new
line indicates that new is an associated function of theString
type
Call the stdin
function from the io
module, which will allow us to handle user input
:
io::stdin()
.read_line(&mut guess)
If we hadn’t
imported
the io library withuse std::io
; at the beginning of the program, we could still use the function by writing this function call asstd::io::stdin
.
Next, the line .read_line(&mut guess)
calls the read_line
method on the standard input handle to get input
from the user
The &
indicates that this argument is a reference
, which gives you a way to let multiple parts of your code access one piece of data without needing to copy that data into memory multiple times.
One long line
is difficult to read, so it’s best to divide
it
The next part is this method
.expect("Failed to read line");
As mentioned earlier, read_line
puts whatever the user enters into the string we pass to it, but it also returns a Result
value - enum
.
Result’s variants are Ok
and Err
.
An instance of Result
has an expect
method:
- if
Err
value, expect will cause the program to crash and display message - if
Ok
value, expect will take the return value and return just it
If you don’t call
expect
, the program will compile, but you’ll get awarning
There’s only one more line to discuss in the code so far:
println!("You guessed: {guess}");
The
{}
set of curly brackets is aplaceholder
: think of {} as little crab pincers that hold a value in place.
Printing a variable and the result of an expression in one call to println! would look like this:
let x = 5;
let y = 10;
println!("x = {x} and y + 2 = {}", y + 2); // "x = 5 and y = 12"
Let’s test the first part of the guessing game.
cargo run
Next, we need to generate a secret number that the user will try to guess.
The rand
crate is a library crate, which contains code that is intended
[dependencies]
rand = "0.8.5"
The specifier 0.8.5 is actually shorthand for ^0.8.5, which means any version that is at least 0.8.5 but below 0.9.0.
When you build
a project for the first time
, Cargo figures out all the versions of the dependencies that fit the criteria and then writes them to the Cargo.lock
file.
When you do want to update a crate, Cargo provides the command update
, which will ignore the Cargo.lock
file and figure out all the latest versions that fit your specifications in Cargo.toml.
cargo update
Let’s start using rand to generate a number to guess.
use std::io;
use rand::Rng; //trait defines methods that random number generators implement
fn main() {
println!("Guess the number!");
let secret_number = rand::thread_rng().gen_range(1..=100);
println!("The secret number is: {secret_number}");
println!("Please input your guess.");
let mut guess = String::new();
io::stdin()
.read_line(&mut guess)
.expect("Failed to read line");
println!("You guessed: {guess}");
}
Another neat feature of Cargo is that running the
cargo doc --open
command will build documentation provided by all your dependencies locally and open it in your browser
We call the rand::thread_rng
function that gives us the particular random number generator we’re going to use: one that is local to the current thread of execution and is seeded by the operating system. Then we call the gen_range
method on the random number generator. This method is defined by the Rng trait
that we brought into scope with the use rand::Rng;
statement.
Now that we have user input and a random number, we can compare them.
use rand::Rng;
use std::cmp::Ordering;
use std::io;
fn main() {
// --snip--
println!("You guessed: {guess}");
match guess.cmp(&secret_number) {
Ordering::Less => println!("Too small!"),
Ordering::Greater => println!("Too big!"),
Ordering::Equal => println!("You win!"),
}
}
First we add another use statement, bringing a type called std::cmp::Ordering
into scope from the standard library. The Ordering type is another enum
and has the variants Less, Greater, and Equal
.
The cmp
method compares two values and can be called on anything that can be compared. It takes a reference
to whatever you want to compare with: here it’s comparing guess
to secret_number
.
We use a match
expression to decide what to do next based on which variant of Ordering
was returned from the call to cmp
with the values in guess
and secret_number
.
The core of the error states that there are
mismatched types
. Rust has a strong, static type system. However, it also has type inference. When we wrotelet mut guess = String::new()
, Rust was able to infer that guess should be aString
and didn’t make us write the type.
Ultimately, we want to convert
the String
the program reads as input into a real number type
so we can compare it numerically to the secret number:
// --snip--
let mut guess = String::new();
io::stdin()
.read_line(&mut guess)
.expect("Failed to read line");
let guess: u32 = guess.trim().parse().expect("Please type a number!"); //convert str to u32
println!("You guessed: {guess}");
match guess.cmp(&secret_number) {
Ordering::Less => println!("Too small!"),
Ordering::Greater => println!("Too big!"),
Ordering::Equal => println!("You win!"),
}
Rust allows us to
shadow
the previous value of guess with a new one.Shadowing
lets usreuse
the guess variable name rather than forcing us to create two unique variables, such asguess_str
andguess
The parse
method on strings converts a string to another type
. Here, we use it to convert from a string to a number
. We need to tell Rust the exact number type we want by using let guess: u32
.
The loop keyword creates an infinite loop.
// --snip--
println!("The secret number is: {secret_number}");
loop {
println!("Please input your guess.");
// --snip--
match guess.cmp(&secret_number) {
Ordering::Less => println!("Too small!"),
Ordering::Greater => println!("Too big!"),
Ordering::Equal => println!("You win!"),
}
}
}
We switch from an expect call to a match expression to move from crashing on an error to handling the error.
// --snip--
io::stdin()
.read_line(&mut guess)
.expect("Failed to read line");
let guess: u32 = match guess.trim().parse() {
Ok(num) => num,
Err(_) => continue,
};
println!("You guessed: {guess}");
// --snip--
This project was a hands-on way to introduce you to many new Rust concepts: let, match, functions, the use of external crates, and more.
This chapter covers concepts that appear in almost every programming language and how they work in Rust.
By default, variables are immutable
When a variable is immutable, once a value is bound to a name, you can’t change
that value.
fn main() {
let x = 5;
println!("The value of x is: {x}");
x = 6;
println!("The value of x is: {x}");
}
Although variables are immutable by default, you can make them mutable by adding mut
in front of the variable name
fn main() {
let mut x = 5;
println!("The value of x is: {x}");
x = 6;
println!("The value of x is: {x}");
}
Like immutable variables
, constants are values that are bound to a name and are not allowed to change
, but there are a few differences
between constants and variables.
aren’t allowed to use mut
with constants.- may be set only to a
constant expression
, not the result of a value that could only be computed at runtime.
const THREE_HOURS_IN_SECONDS: u32 = 60 * 60 * 3;
Rust’s naming convention for constants is to use all uppercase with underscores
between words
You can declare a new variable with the same name
as a previous variable - the first variable is shadowed
by the second
We can shadow a variable by using the same variable’s name and repeating the use of the let
keyword as follows
fn main() {
let x = 5;
let x = x + 1;
{
let x = x * 2;
println!("The value of x in the inner scope is: {x}");
}
println!("The value of x is: {x}");
}
Shadowing is different
from marking a variable as mut
because
- we’ll get a compile-time error if we accidentally try to reassign to this variable without using the
let
keyword - we’re effectively creating a new variable when we use the
let
keyword again, we can change the type of the value but reuse the same name
let spaces = " "; // str
let spaces = spaces.len(); // int
Every value in Rust is of a certain data type
, which tells Rust what kind of data is being specified so it knows how to work with that data.
Keep in mind that Rust is a statically typed language
let guess: u32 = "42".parse().expect("Not a number!");
A scalar type represents a single value. Rust has four primary scalar types:
- integers
- floating point numbers
- booleans
- characters
An integer is a number without a fractional component i8 i16 i32 i64 i128
, u8 u16 u32 u64 u128
the isize
and usize
types depend on the architecture of the computer
Rust also has two primitive types for floating-point numbers, which are numbers with decimal points. Rust’s floating-point types are f32
and f64
fn main() {
let x = 2.0; // f64
let y: f32 = 3.0; // f32
}
fn main() {
// addition
let sum = 5 + 10;
// subtraction
let difference = 95.5 - 4.3;
// multiplication
let product = 4 * 30;
// division
let quotient = 56.7 / 32.2;
let truncated = -5 / 3; // Results in -1
// remainder
let remainder = 43 % 5;
}
As in most other programming languages, a Boolean type in Rust has two possible values: true
and false
.
fn main() {
let t = true;
let f: bool = false; // with explicit type annotation
}
Rust’s char
type is the language’s most primitive alphabetic type.
fn main() {
let c = 'z';
let z: char = 'ℤ'; // with explicit type annotation
let heart_eyed_cat = '😻';
}
Compound types can group multiple values into one type. Rust has two primitive compound types: tuples
and arrays
.
A tuple
is a general way of grouping together a number of values with a variety of types into one
compound type.
fn main() {
let tup: (i32, f64, u8) = (500, 6.4, 1);
}
The variable tup binds to the entire tuple because a tuple is considered a single compound element. To get the individual values out of a tuple, we can use pattern matching to destructure a tuple value, like this:
fn main() {
let tup = (500, 6.4, 1);
let (x, y, z) = tup;
println!("The value of y is: {y}");
}
We can also access a tuple element directly by using a period (.)
followed by the index of the value we want to access
fn main() {
let x: (i32, f64, u8) = (500, 6.4, 1);
let five_hundred = x.0;
let six_point_four = x.1;
let one = x.2;
}
The tuple without any values has a special name, unit
.
Unlike a tuple, every element of an array must have the same type
. Unlike arrays in some other languages, arrays in Rust have a fixed length
.
fn main() {
let a = [1, 2, 3, 4, 5];
let a: [i32; 5];
let a = [3; 5];
}
A vector
is a similar collection type provided by the standard library that is allowed to grow or shrink
in size
An array is a single chunk of memory of a known, fixed
size that can be allocated
on the stack
. You can access elements of an array using indexing
fn main() {
let a = [1, 2, 3, 4, 5];
let first = a[0];
let second = a[1];
}
Rust will check
that the index you’ve specified is less than
the array length
. If the index is greater than or equal to the length, Rust will panic
.
This is an example of Rust’s memory safety principles
in action. In many low-level languages, this kind of check is not done, and when you provide an incorrect
index, invalid memory can be accessed
.
main
function, which is the entry point
of many programs. You’ve also seen the fn
keyword, which allows you to declare new functions.
Rust code uses snake case
as the conventional style
for function and variable names, in which all letters are lowercase and underscores separate words
fn main() {
println!("Hello, world!");
another_function();
}
fn another_function() {
println!("Another function.");
}
We can define functions to have parameters
fn main() {
another_function(5);
}
fn another_function(x: i32) { //The type of x is specified as i32
println!("The value of x is: {x}");
}
In function signatures, you must declare the type
of each parameter
When defining multiple parameters
, separate the parameter declarations with commas
Rust is an expression-based
language
- Statements are instructions that perform some action and do not return a value
- Expressions evaluate to a resultant value
Creating a variable and assigning a value to it with the let keyword is a statement
fn main() {
let y = 6;
}
Statements do not return values. Therefore, you can’t assign a let
statement to another variable, as the following code tries to do; you’ll get an error
fn main() {
let x = (let y = 6);
}
The let y = 6
statement does not return
a value, so there isn’t anything for x
to bind to
Expressions
evaluate to a value and make up most of the rest of the code
that you’ll write in Rust. Calling a function is an expression. Calling a macro is an expression. A new scope block created with curly brackets is an expression
Expressions
do not include ending semicolons
. If you add a semicolon
to the end of an expression, you turn
it into a statement
, and it will then not return a value.
{
let x = 3;
x + 1
}
We don’t name
return values, but we must declare
their type
after an arrow (->)
.
In Rust, the return value
of the function is synonymous with the value of the final expression
in the block of the body of a function.
You can return early
from a function by using the return
fn five() -> i32 { // declare only return Type
5 // no ; or return word
}
But if we place a semicolon
at the end of the line containing x + 1, changing
it from an expression
to a statement
- wll be error statements don’t evaluate to a value, which is expressed by (), the unit type
fn plus_one(x: i32) -> i32 {
x + 1; // error here, return unit type - ()
}
// So we’re doing something complicated here, long enough that we need
// multiple lines of comments to do it! Whew! Hopefully, this comment will
// explain what’s going on.
fn main() {
// I’m feeling lucky today
let lucky_number = 7;
}
The most common constructs that let you control the flow of execution of Rust code are if
expressions and loops
.
fn main() {
let number = 3;
if number < 5 {
println!("condition was true");
} else {
println!("condition was false");
}
}
Unlike languages such as Ruby and JavaScript, Rust will not automatically
try to convert
non-Boolean
types to a Boolean
. You must be explicit
and always provide if with a Boolean as its condition
.
fn main() {
let number = 3;
if number != 0 {
println!("number was something other than zero");
}
}
Using too many else if
expressions can clutter
your code, so if you have more than one - use match
Because if
is an expression
, we can use it on the right side
of a let
statement to assign the outcome to a variable
fn main() {
let condition = true;
let number = if condition { 5 } else { 6 };
println!("The value of number is: {number}");
}
Rust has three kinds of loops: loop
, while
, and for
. Let’s try each one.
You can place the break
keyword within the loop to tell the program when to stop
We also used continue
in the guessing game, which in a loop tells the program to skip over any remaining code in this iteration of the loop and go to the next iteration.
You can add the value
you want returned after the break
expression you use to stop the loop
fn main() {
let mut counter = 0;
let result = loop {
counter += 1;
if counter == 10 {
break counter * 2;
}
};
println!("The result is {result}");
}
You can optionally specify a loop label
on a loop that you can then use with break
or continue
fn main() {
let mut count = 0;
'counting_up: loop {
println!("count = {count}");
let mut remaining = 10;
loop {
println!("remaining = {remaining}");
if remaining == 9 {
break; // break internal loop
}
if count == 2 {
break 'counting_up; // break external loop
}
remaining -= 1;
}
count += 1;
}
println!("End count = {count}");
}
A program will often need
to evaluate a condition
within a loop. While the condition is true, the loop runs.
fn main() {
let mut number = 3;
while number != 0 {
println!("{number}!");
number -= 1;
}
println!("LIFTOFF!!!");
}
You can use a for
loop and execute some code for each item
in a collection
.
fn main() {
let a = [10, 20, 30, 40, 50];
for element in a {
println!("the value is: {element}");
}
}
Ownership is a set of rules that govern how a Rust program manages
memory.
The stack
stores values in the order it gets them and removes the values in the opposite order - last in, first out
.
Like plates. Adding or removing plates from the middle or bottom wouldn’t work as well! Adding data is called pushing onto the stack, and removing data is called popping off the stack.
The heap
is less organized: when you put data on the heap, you request a certain amount of space. The memory allocator finds an empty spot in the heap that is big enough, marks it as being in use, and returns a pointer, which is the address of that location - allocating
Like Restaurant. When you enter, you state the number of people in your group, and the host finds an empty table that fits everyone and leads you there. If someone in your group comes late, they can ask where you’ve been seated to find you.
- Each value in Rust has an owner
- There can only be one owner at a time
- When the owner goes out of scope, the value will be dropped
fn main() {
{ // s is not valid here, it’s not yet declared
let s = "hello"; // s is valid from this point forward
// do stuff with s
} // this scope is now over, and s is no longer valid
}
Rust has a second string type String
This type manages data allocated on the heap and as such is able to store an amount of text that is unknown to us at compile time.
fn main() {
let mut s = String::from("hello");
s.push_str(", world!"); // push_str() appends a literal to a String
println!("{}", s); // This will print `hello, world!`
}
With the String type, in order to support a mutable, growable piece of text, we need to allocate an amount of memory on the heap, unknown at compile time, to hold the contents. This means:
- The memory must be requested from the memory allocator at runtime.
- We need a way of returning this memory to the allocator when we’re done with our String.
The memory is automatically returned drop
once the variable that owns it goes out of scope
.
fn main() {
{
let s = String::from("hello"); // s is valid from this point forward
// do stuff with s
} // this scope is now over, and s is no
// longer valid
}
To ensure memory safety, after the line let s2 = s1;
, Rust considers s1
as no longer valid. Therefore, Rust doesn’t need to free anything
when s1 goes out of scope.
fn main() {
let s1 = String::from("hello");
let s2 = s1;
println!("{}, world!", s1);
}
Instead of being called a shallow copy, it’s known as a move
. In this example, we would say that s1 was moved into s2
.
If we do want to deeply copy
the heap data of the String, not just the stack data, we can use a common method called clone
.
fn main() {
let s1 = String::from("hello");
let s2 = s1.clone();
println!("s1 = {}, s2 = {}", s1, s2);
}
Integers that have a known size at compile time
are stored entirely on the stack
, so copies of the actual values are quick to make.
fn main() {
let x = 5;
let y = x;
println!("x = {}, y = {}", x, y);
}
Types implement the Copy
- All the integer types, such as u32.
- The Boolean type, bool, with values true and false.
- All the floating-point types, such as f64.
- The character type, char.
- Tuples, if they only contain types that also implement Copy. For example, (i32, i32) implements Copy, but (i32, String) does not.
The mechanics of passing a value to a function are similar to those when assigning a value to a variable
. Passing a variable to a function will move or copy
, just as assignment does.
fn main() {
let s = String::from("hello"); // s comes into scope
takes_ownership(s); // s's value moves into the function...
// ... and so is no longer valid here
let x = 5; // x comes into scope
makes_copy(x); // x would move into the function,
// but i32 is Copy, so it's okay to still
// use x afterward
} // Here, x goes out of scope, then s. But because s's value was moved, nothing
// special happens.
fn takes_ownership(some_string: String) { // some_string comes into scope
println!("{}", some_string);
} // Here, some_string goes out of scope and `drop` is called. The backing
// memory is freed.
fn makes_copy(some_integer: i32) { // some_integer comes into scope
println!("{}", some_integer);
} // Here, some_integer goes out of scope. Nothing special happens.
Returning values can also transfer ownership
.
fn main() {
let s1 = gives_ownership(); // gives_ownership moves its return
// value into s1
let s2 = String::from("hello"); // s2 comes into scope
let s3 = takes_and_gives_back(s2); // s2 is moved into
// takes_and_gives_back, which also
// moves its return value into s3
} // Here, s3 goes out of scope and is dropped. s2 was moved, so nothing
// happens. s1 goes out of scope and is dropped.
fn gives_ownership() -> String { // gives_ownership will move its
// return value into the function
// that calls it
let some_string = String::from("yours"); // some_string comes into scope
some_string // some_string is returned and
// moves out to the calling
// function
}
// This function takes a String and returns one
fn takes_and_gives_back(a_string: String) -> String { // a_string comes into
// scope
a_string // a_string is returned and moves out to the calling function
}
A reference
is like a pointer
in that it’s an address we can follow to access the data stored at that address.
Function that has a reference to an object as a parameter instead
of taking ownership of the value
fn main() {
let s1 = String::from("hello");
let len = calculate_length(&s1);
println!("The length of '{}' is {}.", s1, len);
}
fn calculate_length(s: &String) -> usize {
s.len()
}
Pass &s1
into calculate_length
and, in its definition, we take &String
rather than String
.
The opposite of referencing by using
&
is dereferencing, which is accomplished with the dereference operator,*
.
We call the action of creating a reference borrowing
. As in real life, if a person owns something, you can borrow it from them. When you’re done, you have to give it back. You don’t own it
.
Just as variables are immutable by default, so are references. We’re not allowed to modify something we have a reference to.
First we change s
to be mut
. Then we create a mutable reference with &mut
s where we call the change function, and update the function signature to accept a mutable reference with some_string: &mut String
. This makes it very clear that the change function will mutate the value it borrows
.
fn main() {
let mut s = String::from("hello");
change(&mut s);
}
fn change(some_string: &mut String) {
some_string.push_str(", world");
}
Mutable references have one big restriction: if you have a mutable reference to a value, you
can have no other references to that value
-data races
Rust enforces a similar rule for combining mutable and immutable references.
fn main() {
let mut s = String::from("hello");
let r1 = &s; // no problem
let r2 = &s; // no problem
let r3 = &mut s; // BIG PROBLEM
println!("{}, {}, and {}", r1, r2, r3);
}
The scopes of the immutable references r1
and r2
end after the println!
where they are last used, which is before the mutable reference r3
is created.
fn main() {
let mut s = String::from("hello");
let r1 = &s; // no problem
let r2 = &s; // no problem
println!("{} and {}", r1, r2);
// variables r1 and r2 will not be used after this point
let r3 = &mut s; // no problem
println!("{}", r3);
}
In languages with pointers, it’s easy to erroneously create a dangling pointer
— a pointer that references a location in memory that may have been given to someone else
— by freeing some memory while preserving a pointer to that memory.
fn main() {
let reference_to_nothing = dangle();
}
fn dangle() -> &String { // dangle returns a reference to a String
let s = String::from("hello"); // s is a new String
&s // we return a reference to the String, s
} // Here, s goes out of scope, and is dropped. Its memory goes away.
// Danger!
The solution here is to return the String
directly:
fn main() {
let string = no_dangle();
}
fn no_dangle() -> String {
let s = String::from("hello");
s
}
Let’s recap what we’ve discussed about references:
- At any given time, you can have either one mutable reference or any number of immutable references.
- References must always be valid.
Slices let you reference a contiguous sequence of elements in a collection rather than the whole collection. A slice is a kind of reference
, so it does not have ownership
.
The problem: Having to worry about the index in word getting out of sync with the data in s is tedious and error prone!
fn first_word(s: &String) -> usize {
let bytes = s.as_bytes();
for (i, &item) in bytes.iter().enumerate() {
if item == b' ' {
return i;
}
}
s.len()
}
fn main() {
let mut s = String::from("hello world");
let word = first_word(&s); // word will get the value 5
s.clear(); // this empties the String, making it equal to ""
// word still has the value 5 here, but there's no more string that
// we could meaningfully use the value 5 with. word is now totally invalid!
}
A string slice is a reference to part of a String
, and it looks like this:
fn main() {
let s = String::from("hello world");
let hello = &s[0..5];
let world = &s[6..11];
let slice = &s[..2]; //if you want to start at index 0
let slice = &s[3..]; //if your slice includes the last byte
let slice = &s[..]; //entire slice
}
Recall from the borrowing rules that if we have an immutable reference to something, we cannot also take a mutable reference
. Because clear
needs to truncate the String
, it needs to get a mutable reference
. The println!
after the call to clear uses the reference in word, so the immutable reference must still be active at that point
. Rust disallows the mutable reference in clear and the immutable reference in word from existing at the same time
, and compilation fails.
fn first_word(s: &String) -> &str {
let bytes = s.as_bytes();
for (i, &item) in bytes.iter().enumerate() {
if item == b' ' {
return &s[0..i];
}
}
&s[..]
}
fn main() {
let mut s = String::from("hello world");
let word = first_word(&s);
s.clear(); // error!
println!("the first word is: {}", word);
}
Knowing that you can take slices of literals and String values leads us to one more improvement on first_word, and that’s its signature:
fn first_word(s: &String) -> &str {
A more experienced Rustacean would write the signature shown in Listing 4-9 instead because it allows us to use the same function on both &String values and &str values.
fn first_word(s: &str) -> &str {
Just as we might want to refer to part of a string, we might want to refer to part of an array
#![allow(unused)]
fn main() {
let a = [1, 2, 3, 4, 5];
let slice = &a[1..3];
assert_eq!(slice, &[2, 3]);
}
The concepts of ownership, borrowing, and slices ensure memory safety in Rust programs at compile time. The Rust language gives you control over your memory usage in the same way as other systems programming languages, but having the owner of data automatically clean up that data when the owner goes out of scope means you don’t have to write and debug extra code to get this control.
A struct, or structure
, is a custom data type that lets you package together and name multiple related values that make up a meaningful group.
struct User {
active: bool,
username: String,
email: String,
sign_in_count: u64,
}
fn main() {}
To get a specific value from a struct, we use dot notation.
fn main() {
let mut user1 = User {
active: true,
username: String::from("someusername123"),
email: String::from("someone@example.com"),
sign_in_count: 1,
};
user1.email = String::from("anotheremail@example.com");
}
Because the parameter names and the struct field names are exactly the same in we can use the field init shorthand syntax
fn build_user(email: String, username: String) -> User {
User {
active: true,
username,
email,
sign_in_count: 1,
}
}
It’s often useful to create a new instance of a struct that includes most of the values from another instance, but changes some. You can do this using struct update syntax
. The syntax ..
specifies that the remaining fields not explicitly set should have the same value as the fields in the given instance.
fn main() {
// --snip--
let user2 = User {
email: String::from("another@example.com"),
..user1
};
}
Rust also supports structs that look similar to tuples, called tuple structs
.
struct Color(i32, i32, i32);
struct Point(i32, i32, i32);
fn main() {
let black = Color(0, 0, 0);
let origin = Point(0, 0, 0);
}
You can also define structs that don’t have any fields!
struct AlwaysEqual;
fn main() {
let subject = AlwaysEqual;
}
In the User struct definition in Listing 5-1, we used the owned String type rather than the &str string slice type. This is a deliberate choice because we want each instance of this struct to own all of its data and for that data to be valid for as long as the entire struct is valid.
To understand when we might want to use structs, let’s write a program that calculates the area of a rectangle. We’ll start by using single variables, and then refactor the program until we’re using structs instead.
fn main() {
let width1 = 30;
let height1 = 50;
println!(
"The area of the rectangle is {} square pixels.",
area(width1, height1)
);
}
fn area(width: u32, height: u32) -> u32 {
width * height
}
In one way, this program is better. Tuples let us add a bit of structure, and we’re now passing just one argument. But in another way, this version is less clear: tuples don’t name their elements, so we have to index into the parts of the tuple, making our calculation less obvious.
fn main() {
let rect1 = (30, 50);
println!(
"The area of the rectangle is {} square pixels.",
area(rect1)
);
}
fn area(dimensions: (u32, u32)) -> u32 {
dimensions.0 * dimensions.1
}
We use structs to add meaning by labeling the data. We can transform the tuple we’re using into a struct with a name for the whole as well as names for the parts
struct Rectangle {
width: u32,
height: u32,
}
fn main() {
let rect1 = Rectangle {
width: 30,
height: 50,
};
println!(
"The area of the rectangle is {} square pixels.",
area(&rect1)
);
}
fn area(rectangle: &Rectangle) -> u32 {
rectangle.width * rectangle.height
}
Rust does include functionality to print out debugging information, but we have to explicitly opt in to make that functionality available for our struct. To do that, we add the outer attribute #[derive(Debug)]
just before the struct definition
#[derive(Debug)]
struct Rectangle {
width: u32,
height: u32,
}
fn main() {
let rect1 = Rectangle {
width: 30,
height: 50,
};
println!("rect1 is {:?}", rect1); // {:?} used for print debug info
}
Another way to print out a value using the Debug
format is to use the dbg! macro
#[derive(Debug)]
struct Rectangle {
width: u32,
height: u32,
}
fn main() {
let scale = 2;
let rect1 = Rectangle {
width: dbg!(30 * scale),
height: 50,
};
dbg!(&rect1);
}
Methods
are similar to functions: we declare them with the fn keyword and a name, they can have parameters and a return value, and they contain some code that’s run when the method is called from somewhere else.
Let’s change the area function that has a Rectangle
instance as a parameter and instead make an area
method defined on the Rectangle struct
.
To define the function within the context of Rectangle
, we start an impl
(implementation) block for Rectangle. Everything within this impl block will be associated
with the Rectangle type
#[derive(Debug)]
struct Rectangle {
width: u32,
height: u32,
}
impl Rectangle {
fn area(&self) -> u32 {
self.width * self.height
}
}
fn main() {
let rect1 = Rectangle {
width: 30,
height: 50,
};
println!(
"The area of the rectangle is {} square pixels.",
rect1.area()
);
}
In the signature for area, we use &self
instead of rectangle: &Rectangle
. The &self
is actually short for self: &Self
. Within an impl
block, the type Self is an alias
for the type that the impl block is for.
Methods like this are called getters
, and Rust does not implement them automatically for struct fields as some other languages do.
impl Rectangle {
fn width(&self) -> bool {
self.width > 0
}
}
fn main() {
let rect1 = Rectangle {
width: 30,
height: 50,
};
if rect1.width() {
println!("The rectangle has a nonzero width; it is {}", rect1.width);
}
}
In C and C++, two different operators are used for calling methods: you use
.
if you’re calling a method on the object directly and->
if you’re calling the method on a pointer to the object and need to dereference the pointer first. In other words, ifobject
is a pointer,object->something()
is similar to(*object).something()
. Rust doesn’t have an equivalent to the-> operator;
instead, Rust has a feature calledautomatic referencing and dereferencing
. Calling methods is one of the few places in Rust that has this behavior.
Let’s practice using methods by implementing a second method on the Rectangle
struct. This time we want an instance of Rectangle to take another instance of Rectangle and return true
if the second Rectangle can fit completely within self
(the first Rectangle); otherwise, it should return false
.
#[derive(Debug)]
struct Rectangle {
width: u32,
height: u32,
}
impl Rectangle {
fn area(&self) -> u32 {
self.width * self.height
}
fn can_hold(&self, other: &Rectangle) -> bool {
self.width > other.width && self.height > other.height
}
}
fn main() {
let rect1 = Rectangle {
width: 30,
height: 50,
};
let rect2 = Rectangle {
width: 10,
height: 40,
};
let rect3 = Rectangle {
width: 60,
height: 45,
};
println!("Can rect1 hold rect2? {}", rect1.can_hold(&rect2));
println!("Can rect1 hold rect3? {}", rect1.can_hold(&rect3));
}
All functions defined within an impl
block are called associated functions because
they’re associated with the type named after the impl. We can define associated functions that don’t have self
as their first parameter (and thus are not methods) because they don’t need an instance
of the type to work with. We’ve already used one function like this: the String::from
function that’s defined on the String type.
#[derive(Debug)]
struct Rectangle {
width: u32,
height: u32,
}
impl Rectangle {
fn square(size: u32) -> Self {
Self {
width: size,
height: size,
}
}
}
fn main() {
let sq = Rectangle::square(3); // call this associated function
}
Each struct is allowed to have multiple impl
blocks.
impl Rectangle {
fn area(&self) -> u32 {
self.width * self.height
}
}
impl Rectangle {
fn can_hold(&self, other: &Rectangle) -> bool {
self.width > other.width && self.height > other.height
}
}
Structs let you create custom types that are meaningful for your domain. By using structs, you can keep associated pieces of data connected to each other and name each piece to make your code clear. In impl blocks, you can define functions that are associated with your type, and methods are a kind of associated function that let you specify the behavior that instances of your structs have.
In this chapter, we’ll look at enumerations
, also referred to as enums
. Enums allow you to define a type by enumerating its possible variants
.
Where structs give you a way of grouping together related fields and data, like a Rectangle
with its width and height, enums
give you a way of saying a value is one of a possible set of values
. For example, we may want to say that Rectangle
is one of a set of possible shapes that also includes Circle
and Triangle
We can express this concept in code by defining an IpAddrKind enumeration and listing the possible kinds an IP address can be, V4
and V6
.
enum IpAddrKind {
V4,
V6,
}
fn main() {
let four = IpAddrKind::V4; // We can create instances of each
let six = IpAddrKind::V6; // of the two variants of `IpAddrKind`
route(IpAddrKind::V4); // we can call this function
route(IpAddrKind::V6); // with either variant: V4, V6
}
fn route(ip_kind: IpAddrKind) {} // for instance we can define a function that takes any IpAddrKind
Representing the same concept using just an enum is more concise: rather than an enum inside a struct, we can put data directly into each enum variant.
fn main() {
enum IpAddr {
V4(String),
V6(String),
}
let home = IpAddr::V4(String::from("127.0.0.1")); // one liners to difine
let loopback = IpAddr::V6(String::from("::1")); // home and loopback
}
There’s another advantage to using an enum rather than a struct: each variant can have different types and amounts of associated data
. Version four IP addresses will always have four numeric components
that will have values between 0 and 255. If we wanted to store V4 addresses as four u8 values but still express V6 addresses as one String value, we wouldn’t be able to with a struct
.
enum IpAddr {
V4(u8, u8, u8, u8),
V6(String),
}
let home = IpAddr::V4(127, 0, 0, 1);
let loopback = IpAddr::V6(String::from("::1"));
Let’s look at how the standard library
defines IpAddr
: it has the exact enum
and variants
that we’ve defined and used, but it embeds the address data inside the variants in the form of two different structs
struct Ipv4Addr {
// --snip--
}
struct Ipv6Addr {
// --snip--
}
enum IpAddr {
V4(Ipv4Addr),
V6(Ipv6Addr),
}
Let’s look at another example of an enum: this one has a wide variety of types
embedded in its variants.
enum Message {
Quit,
Move { x: i32, y: i32 },
Write(String),
ChangeColor(i32, i32, i32),
}
There is one more similarity between enums and structs: just as we’re able to define methods on structs using impl
, we’re also able to define methods
on enums. Here’s a method named call
that we could define on our Message enum:
fn main() {
enum Message {
Quit,
Move { x: i32, y: i32 },
Write(String),
ChangeColor(i32, i32, i32),
}
impl Message {
fn call(&self) {
// method body would be defined here
}
}
let m = Message::Write(String::from("hello"));
m.call();
}
This section explores a case study of Option
, which is another enum
defined by the standard library. The Option type encodes the very common scenario in which a value could be something
or it could be nothing
.
Rust doesn’t have the null
feature that many other languages have.
enum Option<T> {
None,
Some(T),
}
<T>
means that the Some
variant of the Option
enum can hold one
piece of data of any type
let some_number = Some(5);
let some_char = Some('e');
let absent_number: Option<i32> = None;
Option<T>
and T
(where T can be any type) are different types
, the compiler won’t let us use an Option<T>
value as if it were definitely a valid value
let x: i8 = 5;
let y: Option<i8> = Some(5);
let sum = x + y;
Rust has an extremely powerful control flow construct called match
that allows you to compare a value against a series of patterns and then execute code based on which pattern matches.
Function that takes an unknown US coin and, in a similar way as the counting machine, determines which coin it is and returns its value in cents
enum Coin {
Penny,
Nickel,
Dime,
Quarter,
}
fn value_in_cents(coin: Coin) -> u8 {
match coin {
Coin::Penny => 1,
Coin::Nickel => 5,
Coin::Dime => 10,
Coin::Quarter => 25,
}
}
If you want to run multiple lines
of code in a match arm, you must use curly brackets, and the comma following the arm is then optional
fn value_in_cents(coin: Coin) -> u8 {
match coin {
Coin::Penny => {
println!("Lucky penny!");
1
}
Coin::Nickel => 5,
Coin::Dime => 10,
Coin::Quarter => 25,
}
}
Another useful feature of match
arms is that they can bind to the parts of the values
that match the pattern. In the match expression for this code, we add a variable called state
to the pattern that matches values of the variant Coin::Quarter
#[derive(Debug)]
enum UsState {
Alabama,
Alaska,
// --snip--
}
enum Coin {
Penny,
Nickel,
Dime,
Quarter(UsState),
}
fn value_in_cents(coin: Coin) -> u8 {
match coin {
Coin::Penny => 1,
Coin::Nickel => 5,
Coin::Dime => 10,
Coin::Quarter(state) => {
println!("State quarter from {:?}!", state); //{:?} debug
25
}
}
}
fn main() {
value_in_cents(Coin::Quarter(UsState::Alaska));
}
We can also handle Option<T>
using match
fn plus_one(x: Option<i32>) -> Option<i32> {
match x {
None => None,
Some(i) => Some(i + 1),
}
}
let five = Some(5);
let six = plus_one(five);
let none = plus_one(None);
There’s one other aspect of match we need to discuss: the arms patterns
must cover all possibilities
. This will NOT
work
fn plus_one(x: Option<i32>) -> Option<i32> {
match x {
Some(i) => Some(i + 1),
}
}
Using enums, we can also take special actions for a few particular values, but for all other
values take one default action
.
let dice_roll = 9;
match dice_roll {
3 => add_fancy_hat(), // if 3 add fancy
7 => remove_fancy_hat(), // if 7 remove fancy
other => move_player(other), // default for all other
}
fn add_fancy_hat() {}
fn remove_fancy_hat() {}
fn move_player(num_spaces: u8) {}
Let’s change the rules of the game: now, if you roll anything other than a 3 or a 7
, you must roll again
. We no longer need to use the catch-all value, so we can change our code to use _
instead of the variable named other
let dice_roll = 9;
match dice_roll {
3 => add_fancy_hat(),
7 => remove_fancy_hat(),
_ => reroll(), // reroll if not 7 or 3
}
fn add_fancy_hat() {}
fn remove_fancy_hat() {}
fn reroll() {}
Finally, we’ll change the rules of the game one more time so that nothing
else happens on your turn if you roll anything
other than a 3 or a 7.
let dice_roll = 9;
match dice_roll {
3 => add_fancy_hat(),
7 => remove_fancy_hat(),
_ => (), // do nothing if not 3 or 7
}
fn add_fancy_hat() {}
fn remove_fancy_hat() {}
The if let
syntax lets you combine if and let into a less verbose
way to handle values that match one
pattern while ignoring the rest.
let config_max = Some(3u8);
match config_max {
Some(max) => println!("The maximum is configured to be {}", max),
_ => (),
}
Instead
, we could write this in a shorter way using if let
.
let config_max = Some(3u8);
if let Some(max) = config_max {
println!("The maximum is configured to be {}", max);
}
If we wanted to count
all non-quarter coins
we see while also announcing the state of the quarters, we could do that with a match
expression, like this
let mut count = 0;
match coin {
Coin::Quarter(state) => println!("State quarter from {:?}!", state),
_ => count += 1,
}
Or we could use an if let
and else
expression
let mut count = 0;
if let Coin::Quarter(state) = coin {
println!("State quarter from {:?}!", state);
} else {
count += 1;
}
We’ve now covered how to use enums to create custom types that can be one of a set of enumerated values. We’ve shown how the standard library’s Option<T>
type helps you use the type system to prevent errors. When enum values have data inside them, you can use match
or if let
to extract and use those values, depending on how many cases you need to handle.
Rust has a number of features
that allow you to manage
your code’s organization, including which details are exposed, which details are private, and what names are in each scope in your programs. These features, sometimes collectively referred to as the module system
, include:
Packages
: A Cargo feature that lets you build, test, and share cratesCrates
: A tree of modules that produces a library or executableModules and use
: Let you control the organization, scope, and privacy of pathsPaths
: A way of naming an item, such as a struct, function, or module
A crate
is the smallest amount of code that the Rust compiler considers at a time
.
A crate can come in one of two forms: a binary crate
or a library crate
.
Binary crates
are programs you can compile to an executable
that you can run, such as a command-line program or a server - have function main
Library crates
don’t have main
and define functionality intended to be shared with multiple projects
$ cargo new my-project
Created binary (application) `my-project` package
$ ls my-project
Cargo.toml
src
$ ls my-project/src
main.rs
-
Start from the crate root
: When compiling a crate, the compiler first looks in the crate root file (usuallysrc/lib.rs
for a library crate orsrc/main.rs
for a binary crate) for code to compile. -
Declaring modules
: In the crate root file, you can declare new modules; say, you declare a “garden” module withmod garden;
. The compiler will look for the module’s code in these places:
-- Inline, within curly brackets that replace the semicolon followingmod garden
-- In the filesrc/garden.rs
-- In the filesrc/garden/mod.rs
-
Declaring submodules
: In any file other than the crate root, you can declare submodules. For example, you might declaremod vegetables;
in src/garden.rs. The compiler will look for the submodule’s code within the directory named for the parent module in these places:
-- Inline, directly followingmod vegetables
, within curly brackets instead of the semicolon
-- In the filesrc/garden/vegetables.rs
-- In the filesrc/garden/vegetables/mod.rs
-
Paths to code in modules
: Once a module is part of your crate, you can refer to code in that module from anywhere else in that same crate, as long as the privacy rules allow, using the path to the code. For example, an Asparagus type in the garden vegetables module would be found atcrate::garden::vegetables::Asparagus
. -
Private vs public
: Code within a module is private from its parent modules by default. To make a module public, declare it withpub mod
instead ofmod
. To make items within a public module public as well, use pub before their declarations. -
The
use
keyword: Within a scope, the use keyword creates shortcuts to items to reduce repetition of long paths. In any scope that can refer tocrate::garden::vegetables::Asparagus
, you can create a shortcut with usecrate::garden::vegetables::Asparagus;
and from then on you only need to writeAsparagus
to make use of that type in the scope.
Here we create a binary crate named backyard
that illustrates these rules. The crate’s directory, also named backyard
, contains these files and directories:
backyard
├── Cargo.lock
├── Cargo.toml
└── src
├── garden
│ └── vegetables.rs
├── garden.rs
└── main.rs
The crate root file in this case is src/main.rs
, and it contains:
use crate::garden::vegetables::Asparagus;
pub mod garden;
fn main() {
let plant = Asparagus {};
println!("I'm growing {:?}!", plant);
}
The pub mod garden;
line tells the compiler to include the code it finds in src/garden.rs
, which is:
pub mod vegetables;
Here, pub mod vegetables;
means the code in src/garden/vegetables.rs
is included too. That code is:
#[derive(Debug)]
pub struct Asparagus {}
Modules
let us organize code within a crate for readability and easy reuse. Modules also allow us to control the privacy
of items, because code within a module is private by default
.
In the restaurant industry
, some parts of a restaurant are referred to as front of house
and others as back of house
. Front of house is where customers are; this encompasses where the hosts seat customers, servers take orders and payment, and bartenders make drinks. Back of house is where the chefs and cooks work in the kitchen, dishwashers clean up, and managers do administrative work.
Create a new library
named restaurant by running cargo new restaurant --lib
.
Filename: src/lib.rs
:
mod front_of_house {
mod hosting {
fn add_to_waitlist() {}
fn seat_at_table() {}
}
mod serving {
fn take_order() {}
fn serve_order() {}
fn take_payment() {}
}
}
We define a module with the mod keyword
followed by the name of the module (in this case, front_of_house
).
src/main.rs
and src/lib.rs
are called crate roots
. The reason for their name is that the contents of either of these two files form a module named crate
at the root of the crate’s module structure, known as the module tree
.
crate
└── front_of_house
├── hosting
│ ├── add_to_waitlist
│ └── seat_at_table
└── serving
├── take_order
├── serve_order
└── take_payment
The module tree might remind you of the filesystem’s directory tree
on your computer; this is a very apt
comparison!
A path can take two forms:
- An
absolute path
is the full path starting from a crate root; for code from an external crate, the absolute path begins with the crate name, and for code from the current crate, it starts with the literal crate. - A
relative path
starts from the current module and usesself
,super
, or an identifier in the current module.
Our preference in general is to specify absolute paths because it’s more likely we’ll want to move code definitions and item calls independently of each other.
Both absolute and relative paths are followed by one or more identifiers separated by double colons (::)
.
mod front_of_house {
mod hosting {
fn add_to_waitlist() {}
}
}
pub fn eat_at_restaurant() {
// Absolute path
crate::front_of_house::hosting::add_to_waitlist();
// Relative path
front_of_house::hosting::add_to_waitlist();
}
We have the correct
paths
for the hostingmodule
and theadd_to_waitlist
function, but Rust won’t let us use them because itdoesn’t have access
to the private sections.
We want the eat_at_restaurant
function in the parent module to have access to the add_to_waitlist
function in the child module, so we mark the hosting module with the pub
keyword
mod front_of_house {
pub mod hosting {
fn add_to_waitlist() {}
}
}
pub fn eat_at_restaurant() {
// Absolute path
crate::front_of_house::hosting::add_to_waitlist();
// Relative path
front_of_house::hosting::add_to_waitlist();
}
The
pub
keyword on a module only lets code in itsancestor modules
refer to it, not access itsinner code
Let’s also make the add_to_waitlist
function public
by adding the pub
keyword before its definition
mod front_of_house {
pub mod hosting {
pub fn add_to_waitlist() {}
}
}
pub fn eat_at_restaurant() {
// Absolute path
crate::front_of_house::hosting::add_to_waitlist();
// Relative path
front_of_house::hosting::add_to_waitlist();
}
The module tree should be defined in src/lib.rs. Then, any public items can be used in the binary crate by starting paths with the name of the package. The binary crate becomes a user of the library crate just like a completely external crate would use the library crate: it can only use the public API. This helps you design a good API; not only are you the author, you’re also a client!
We can construct relative paths that begin in the parent module, rather than the current module or the crate root, by using super
at the start of the path.
fn deliver_order() {}
mod back_of_house {
fn fix_incorrect_order() {
cook_order();
super::deliver_order();
}
fn cook_order() {}
}
We can also use pub
to designate structs
and enums
as public
, but there are a few details extra to the usage of pub with structs and enums.
We’ve defined a public
back_of_house::Breakfast
struct with a public
toast
field but a private
seasonal_fruit
field
mod back_of_house {
pub struct Breakfast {
pub toast: String,
seasonal_fruit: String,
}
impl Breakfast {
pub fn summer(toast: &str) -> Breakfast {
Breakfast {
toast: String::from(toast),
seasonal_fruit: String::from("peaches"),
}
}
}
}
pub fn eat_at_restaurant() {
// Order a breakfast in the summer with Rye toast
let mut meal = back_of_house::Breakfast::summer("Rye");
// Change our mind about what bread we'd like
meal.toast = String::from("Wheat");
println!("I'd like {} toast please", meal.toast);
// The next line won't compile if we uncomment it; we're not allowed
// to see or modify the seasonal fruit that comes with the meal
// meal.seasonal_fruit = String::from("blueberries");
}
In contrast, if we make an enum
public
, all of its variants are then public
. We only need the pub
before the enum keyword
mod back_of_house {
pub enum Appetizer {
Soup,
Salad,
}
}
pub fn eat_at_restaurant() {
let order1 = back_of_house::Appetizer::Soup;
let order2 = back_of_house::Appetizer::Salad;
}
We bring the crate::front_of_house::hosting
module into the scope of the eat_at_restaurant
function so we only have to specify hosting::add_to_waitlist
to call the add_to_waitlist
function in eat_at_restaurant
** Note that use only creates the shortcut for the particular scope
in which the use
occurs.**
mod front_of_house {
pub mod hosting {
pub fn add_to_waitlist() {}
}
}
use crate::front_of_house::hosting;
pub fn eat_at_restaurant() {
hosting::add_to_waitlist();
}
Adding
use
and apath
in a scope is similar to creating asymbolic link
in the filesystem.
Specifying the parent module
when calling the function makes it clear that the function isn’t locally defined
while still minimizing repetition of the full path.
On the other hand, when bringing in structs, enums, and other items
with use
, it’s idiomatic to specify the full path
.
use std::collections::HashMap;
fn main() {
let mut map = HashMap::new();
map.insert(1, 2);
}
The exception to this idiom is if we’re bringing two items with the same name into scope with use statements, because Rust doesn’t allow that.
There’s another solution to the problem of bringing two types of the same name into the same scope with use
: after the path, we can specify as
and a new local name, or alias, for the type.
use std::fmt::Result;
use std::io::Result as IoResult;
fn function1() -> Result {
// --snip--
}
fn function2() -> IoResult<()> {
// --snip--
}
We can combine pub
and use
. This technique is called re-exporting
because we’re bringing an item into scope but also making that item available for others to bring into their scope.
mod front_of_house {
pub mod hosting {
pub fn add_to_waitlist() {}
}
}
pub use crate::front_of_house::hosting;
pub fn eat_at_restaurant() {
hosting::add_to_waitlist();
}
To bring rand
definitions into the scope of our package, we added a use line starting with the name of the crate
use rand::Rng;
fn main() {
let secret_number = rand::thread_rng().gen_range(1..=100);
}
with HashMap
we would use this line:
use std::collections::HashMap;
We can use nested paths to bring items into scope in one line
use std::{cmp::Ordering, io};
use std::io::{self, Write};
If we want to bring all
public items
use std::collections::*;
We’ll extract modules
into files
instead of having all the modules defined in the crate root file.
Filename: src/lib.rs
mod front_of_house;
pub use crate::front_of_house::hosting;
pub fn eat_at_restaurant() {
hosting::add_to_waitlist();
}
Filename: src/front_of_house.rs
pub mod hosting;
Filename: src/front_of_house/hosting.rs
pub fn add_to_waitlist() {}
Rust lets you split a package
into multiple crates
and a crate into modules
so you can refer to items defined in one module from another module. You can do this by specifying absolute
or relative
paths. These paths can be brought into scope with a use
statement so you can use a shorter path for multiple uses of the item in that scope. Module code is private by default
, but you can make definitions public by adding the pub
keyword.
Most other data types represent one specific value, but collections can contain multiple values. Unlike the built-in array and tuple types, the data these collections point to is stored on the heap, which means the amount of data does not need to be known at compile time and can grow or shrink as the program runs.
- A
vector
allows you to store a variable number of values next to each other. - A
string
is a collection of characters. We’ve mentioned the String type previously, but in this chapter we’ll talk about it in depth. - A
hash map
allows you to associate a value with a particular key. It’s a particular implementation of the more general data structure called a map.
The first collection type we’ll look at is Vec<T>
, also known as a vector
.
To create a new empty vector, we call the Vec::new
function,
let v: Vec<i32> = Vec::new();
let v = vec![1, 2, 3];
To create a vector and then add elements to it, we can use the push
method,
let mut v = Vec::new();
v.push(5);
v.push(6);
There are two
ways to reference
a value stored in a vector: via indexing
or using the get
let v = vec![1, 2, 3, 4, 5];
let third: &i32 = &v[2]; // using index
println!("The third element is {third}");
let third: Option<&i32> = v.get(2); // using get
match third {
Some(third) => println!("The third element is {third}"),
None => println!("There is no third element."),
}
Indexing
- this method is best used when you want your program to crash if there’s an attempt to access an element past the end of the vector.
When the get
method is passed an index that is outside the vector, it returns None without panicking
.
To access each element
in a vector in turn, we would iterate through all of the elements rather than use indices to access one at a time with for loop
let v = vec![100, 32, 57];
for i in &v {
println!("{i}");
}
We can also iterate
over mutable
references
let mut v = vec![100, 32, 57];
for i in &mut v {
*i += 50;
}
When we need one type
to represent elements of different types
, we can define and use an enum
enum SpreadsheetCell {
Int(i32),
Float(f64),
Text(String),
}
let row = vec![
SpreadsheetCell::Int(3),
SpreadsheetCell::Text(String::from("blue")),
SpreadsheetCell::Float(10.12),
];
Like any other struct, a vector is freed when it goes out of scope
{
let v = vec![1, 2, 3, 4];
// do stuff with v
} // <- v goes out of scope and is freed here
New Rustaceans commonly get stuck on strings for a combination of three reasons:
- Rust’s propensity for exposing possible errors
- strings being a more complicated data structure than many programmers give them credit for
- and UTF-8
Rust has only one string type in the core language, which is the string slice str
that is usually seen in its borrowed form &str
The String
type, which is provided by Rust’s standard library
rather than coded into the core language, is a growable
, mutable
, owned
, UTF-8
encoded string type.
Many of the same operations
available with Vec<T>
are available with String
as well, because String is actually implemented as a wrapper around a vector
of bytes with some extra guarantees, restrictions, and capabilities.
let mut s = String::new();
Often, we’ll have some initial data
that we want to start the string with. For that, we use
the to_string
method, which is available on any type that implements the Display
trait
let data = "initial contents";
let s = data.to_string();
// the method also works on a literal directly:
let s = "initial contents".to_string();
We can also
use the function String::from
to create a String from a string literal.
let s = String::from("initial contents");
In this case,
String::from
andto_string
do thesame
thing, so which you choose is a matter of style and readability.
A String can grow
in size and its contents can change, just like the contents of a Vec<T>
, if you push
more data into it. In addition, you can conveniently use the +
operator or the format!
macro to concatenate String values.
We can grow
a String by using the push_str
method to
let mut s = String::from("foo");
s.push_str("bar");
The push_str method takes a string slice and
don’t take ownership
of the parameter
The push
method takes a single character
as a parameter and adds it to the String.
let mut s = String::from("lo");
s.push('l');
Often, you’ll want to combine two
existing strings. One way to do so is to use
the +
operator
let s1 = String::from("Hello, ");
let s2 = String::from("world!");
let s3 = s1 + &s2; // note s1 has been moved here and can no longer be used
The string s3 will contain Hello, world!. The reason
s1
isno longer valid
after the addition, and the reason we used areference to s2
, has to do with the signature of themethod
that’s called when we use the + operator. The+
operatoruses
theadd
method
If we need to concatenate multiple strings, the behavior of the + operator gets unwieldy.
For more complicated string
combining, we can instead use
the format!
macro:
let s1 = String::from("tic");
let s2 = String::from("tac");
let s3 = String::from("toe");
let s = format!("{s1}-{s2}-{s3}");
Rust strings don’t support indexing
A String
is a wrapper over a Vec<u8>
.
Sometimes UTF-8
stores as 1 byte
, sometimes with Unicode scalar value as 2 bytes
.
To avoid returning an unexpected value
and causing bugs
that might not be discovered immediately, Rust doesn’t compile this code at all and prevents misunderstandings early in the development process.
Another point about UTF-8
is that there are actually three
relevant ways to look
at strings from Rust’s perspective: as bytes
, scalar values
, and grapheme clusters
A final reason Rust
doesn’t allow us to index
into a String to get a character is that indexing operations are expected to always take constant time (O(1)). But it isn’t possible to guarantee that performance with a String, becauseRust would have to walk through the contents from the beginning to the index
to determinehow many valid
characters there were.
Indexing
into a string is often a bad idea
because it’s not clear
what the return type of the string-indexing operation should be
: a byte value, a character, a grapheme cluster, or a string slice.
The best way
to operate on pieces of strings is to be explicit
about whether you want characters or bytes
. For individual Unicode scalar values, use the chars
method.
for c in "Зд".chars() {
println!("{c}");
}
Alternatively, the bytes
method returns each raw byte
for b in "Зд".bytes() {
println!("{b}");
}
To summarize, strings are complicated
. Different programming languages make different choices about how to present this complexity to the programmer. Rust has chosen to make the correct handling of String data the default
behavior for all Rust programs, which means programmers have to put more thought into handling UTF-8
data upfront
The type HashMap<K, V>
stores a mapping of keys
of type K to values
of type V using a hashing function, which determines how it places these keys and values into memory.
Hash maps are useful
when you want to look up data not by using an index
, as you can with vectors, but by using a key
that can be of any type.
One way to create an empty hash map is using new and adding elements with insert. Just like vectors, hash maps store their data on the heap.
use std::collections::HashMap;
let mut scores = HashMap::new();
scores.insert(String::from("Blue"), 10);
scores.insert(String::from("Yellow"), 50);
we need to first
use the HashMap from the collections
portion of the standard library.
We can get a value
out of the hash map by providing its key to the get
method
use std::collections::HashMap;
let mut scores = HashMap::new();
scores.insert(String::from("Blue"), 10);
scores.insert(String::from("Yellow"), 50);
let team_name = String::from("Blue");
let score = scores.get(&team_name).copied().unwrap_or(0);
Here,
score
will have the value that’s associated with theBlue
team, and the result will be10
. Theget
method returns anOption<&V>
; if there’sno value
for that key in the hash map, get will returnNone
. This program handles theOption
by callingcopied
to get anOption<i32>
rather than anOption<&i32>
, thenunwrap_or
to set score to zero if scores doesn't have an entry for the key.
We can iterate
over each key/value pair in a hash map
in a similar manner as we do with vectors, using a for loop
use std::collections::HashMap;
let mut scores = HashMap::new();
scores.insert(String::from("Blue"), 10);
scores.insert(String::from("Yellow"), 50);
for (key, value) in &scores {
println!("{key}: {value}");
}
For types that implement the Copy trait, like i32
, the values are copied into the hash
map. For owned
values like String
, the values will be moved
and the hash map will be the owner
of those values
use std::collections::HashMap;
let field_name = String::from("Favorite color");
let field_value = String::from("Blue");
let mut map = HashMap::new();
map.insert(field_name, field_value);
// field_name and field_value are invalid at this point, try using them and
// see what compiler error you get!
We
aren’t able to use
the variablesfield_name
andfield_value
after they’ve been moved into the hash map with the call to insert.
When you want to change the data in a hash map, you have to decide how to handle the case when a key already has a value assigned.
Overwriting a Value
use std::collections::HashMap;
let mut scores = HashMap::new();
scores.insert(String::from("Blue"), 10);
scores.insert(String::from("Blue"), 25);
println!("{:?}", scores);
If we insert a key and a value into a hash map and then insert that same key with a different value, the value associated with that key will be replaced
Adding a Key and Value Only If a Key Isn’t Present
use std::collections::HashMap;
let mut scores = HashMap::new();
scores.insert(String::from("Blue"), 10);
scores.entry(String::from("Yellow")).or_insert(50);
scores.entry(String::from("Blue")).or_insert(50);
println!("{:?}", scores);
if the key does exist in the hash map, the existing value should remain the way it is. If the key doesn’t exist, insert it and a value for it.
Updating a Value Based on the Old Value
use std::collections::HashMap;
let text = "hello world wonderful world";
let mut map = HashMap::new();
for word in text.split_whitespace() {
let count = map.entry(word).or_insert(0);
*count += 1;
}
println!("{:?}", map);
We use a hash map with the words as keys and increment the value to keep track of how many times we’ve seen that word
By default, HashMap uses a hashing function called SipHash
Rust groups errors into two
major categories: recoverable
and unrecoverable errors
. For a recoverable error, such as a file not found
error, we most likely just want to report
the problem to the user and retry the operation. Unrecoverable errors are always symptoms of bugs
, like trying to access a location beyond the end of an array
, and so we want to immediately stop
the program.
Rust doesn’t have exceptions
. Instead, it has the type Result<T, E>
for recoverable
errors and the panic!
macro that stops execution when the program encounters an unrecoverable error
.
By default, when a panic occurs, the program starts unwinding, which means Rust walks back up the stack and cleans up the data from each function it encounters. However, this walking back and cleanup is a lot of work. Rust, therefore, allows you to choose the alternative of immediately aborting, which ends the program without cleaning up.
//Cargo.toml file
[profile.release]
panic = 'abort'
Sometimes, bad things happen in your code, and there’s nothing you can do about it. In these cases, Rust has the panic!
macro
fn main() {
panic!("crash and burn");
}
We can set the RUST_BACKTRACE
environment variable to get a backtrace of exactly what happened to cause the error. A backtrace is a list of all the functions that have been called to get to this point
.
RUST_BACKTRACE=1 cargo run
Debug symbols are
enabled by default
when using cargo build or cargo run without the--release
flag, as we have here.
Most errors aren’t serious enough to require the program to stop entirely.
enum Result<T, E> {
Ok(T),
Err(E),
}
We need to add
to the code to take different actions depending
on the value
File::open returns
use std::fs::File;
fn main() {
let greeting_file_result = File::open("hello.txt");
let greeting_file = match greeting_file_result {
Ok(file) => file, // return inner file
Err(error) => panic!("Problem opening the file: {:?}", error), // panic
};
}
We want to take different actions
for different failure reasons: if File::open failed
because the file doesn’t exist, we want to create
the file and return the handle to the new file
use std::fs::File;
use std::io::ErrorKind;
fn main() {
let greeting_file_result = File::open("hello.txt");
let greeting_file = match greeting_file_result {
Ok(file) => file,
Err(error) => match error.kind() { // different kinds of errors
ErrorKind::NotFound => match File::create("hello.txt") { // if not found try to create file
Ok(fc) => fc,
Err(e) => panic!("Problem creating the file: {:?}", e), // panic if failed
},
other_error => {
panic!("Problem opening the file: {:?}", other_error); // in other kind panic
}
},
};
}
The match
expression is very useful but also very much a primitive
. Here’s another way
to write the same logic this time using closures
and the unwrap_or_else
method
use std::fs::File;
use std::io::ErrorKind;
fn main() {
let greeting_file = File::open("hello.txt").unwrap_or_else(|error| {
if error.kind() == ErrorKind::NotFound {
File::create("hello.txt").unwrap_or_else(|error| {
panic!("Problem creating the file: {:?}", error);
})
} else {
panic!("Problem opening the file: {:?}", error);
}
});
}
If the Result
value is the Ok
variant, unwrap
will return
the value inside the Ok
. If the Result is the Err
variant, unwrap will call the panic!
macro for us.
use std::fs::File;
fn main() {
let greeting_file = File::open("hello.txt").unwrap();
}
Similarly, the expect
method lets us also choose
the panic!
error message. Using expect instead of unwrap
and providing good error messages can convey your intent and make tracking
down the source of a panic easier
.
use std::fs::File;
fn main() {
let greeting_file = File::open("hello.txt")
.expect("hello.txt should be included in this project");
}
In production-quality code, most Rustaceans choose
expect rather than unwrap
When a function’s implementation calls something that might fail
, instead of handling the error within the function itself, you can return the error
to the calling code
so that it can decide what to do. This is known as propagating
use std::fs::File;
use std::io::{self, Read};
fn read_username_from_file() -> Result<String, io::Error> {
let username_file_result = File::open("hello.txt");
let mut username_file = match username_file_result {
Ok(file) => file,
Err(e) => return Err(e),
};
let mut username = String::new();
match username_file.read_to_string(&mut username) {
Ok(_) => Ok(username), // return String if all ok
Err(e) => Err(e), // return Error if cant read
}
}
It’s up to the calling code to decide what to do with those values. If the calling code gets an Err value, it could call panic! and crash the program, use a default username, or look up the username from somewhere other than a file, for example.
This pattern of propagating
errors is so common in Rust
that Rust provides the question mark operator ?
to make this easier.
fn read_username_from_file() -> Result<String, io::Error> {
let mut username_file = File::open("hello.txt")?;
let mut username = String::new();
username_file.read_to_string(&mut username)?;
Ok(username)
}
error type
received isconverted
into the error type defined in thereturn type
of thecurrent function
- function returnsone error type
to represent all the ways a function might fail
The ?
operator eliminates a lot of boilerplate and makes
this function’s implementation simpler
.
fn read_username_from_file() -> Result<String, io::Error> {
let mut username = String::new();
File::open("hello.txt")?.read_to_string(&mut username)?;
Ok(username)
}
If the value is an
Err
, the Err will bereturned
from thewhole function
as if we had used thereturn
keyword so the error value gets propagated to the calling code.
Reading a file
into a string is a fairly common operation
, so the standard library provides the convenient fs::read_to_string
fn read_username_from_file() -> Result<String, io::Error> {
fs::read_to_string("hello.txt")
}
The ?
operator can only be used
in functions whose return
type is compatible
with the value the ? is used on.
The error message
also mentioned that ? can be used
with Option<T>
values as well.
fn last_char_of_first_line(text: &str) -> Option<char> {
text.lines().next()?.chars().last()
}
As with using ? on Result, you can only
use ?
onOption
in a function thatreturns an Option
Luckily, main can also return a Result<(), E>
use std::error::Error;
use std::fs::File;
fn main() -> Result<(), Box<dyn Error>> {
let greeting_file = File::open("hello.txt")?;
Ok(())
}
you can read
Box<dyn Error>
to meanany kind of error
When a main
function returns
a Result<(), E>
, the executable will exit with a value of 0
if main returns Ok(())
and will exit with a nonzero
value if main returns an Err
value.
You could call panic!
for any error
situation, whether there’s a possible way to recover or not, but then you’re making
the decision that a situation
is unrecoverable
on behalf of the calling code. When you choose to return a Result
value, you give
the calling code options
.
When you’re writing an example
to illustrate some concept, also including robust error-handling code
can make the example less clear
.
Similarly, the unwrap and expect
methods are very handy
when prototyping
, before you’re ready to decide how to handle errors.
If a method call fails in a test
, you’d want the whole test to fail
, even if that method isn’t the functionality under test.
If you can ensure by manually inspecting the code that you’ll never have an Err variant
, it’s perfectly acceptable to call unwrap, and even better to document the reason
you think you’ll never have an Err variant in the expect
text
let home: IpAddr = "127.0.0.1"
.parse()
.expect("Hardcoded IP address should be valid");
In cases where continuing could be insecure or harmful
, the best choice
might be to call panic!
and alert the person using your library to the bug in their code so they can fix it during development.
However, when failure is expected
, it’s more appropriate to return a Result
than to make a panic! call.
When your code performs an operation that could put a user at risk
if it’s called using invalid values
, your code should verify the values are valid first
and panic
if the values aren’t valid.
you can use
Rust’s type system
(and thus the type checking done by the compiler) to domany of the checks
for you.
Let’s take the idea of using Rust’s type system
to ensure we have a valid value
one step further and look at creating a custom type for validation.
One way
to do this would be to parse the guess as an i32 instead of only a u32 to allow potentially negative numbers, and then add a check everywhere
for the number being in range
Instead
, we can make a new type
and put the validations in a function to create an instance of the type rather than repeating the validations everywhere.
pub struct Guess { //define a struct that has a field named value that holds an i32
value: i32,
}
impl Guess {
pub fn new(value: i32) -> Guess { // creates instances of Guess values
if value < 1 || value > 100 { // check if value correct
panic!("Guess value must be between 1 and 100, got {}.", value); // panic if not
}
Guess { value } // return if all ok
}
pub fn value(&self) -> i32 { // getter, because value is private
self.value
}
}
A function that has a parameter or returns only numbers between 1 and 100 could then declare in its signature that it takes or returns a
Guess rather than an i32
and wouldn’t need to do anyadditional checks
in its body.
Rust’s error handling features are designed to help you write more robust code. The panic! macro signals that your program is in a state it can’t handle and lets you tell the process to stop instead of trying to proceed with invalid or incorrect values. The Result enum uses Rust’s type system to indicate that operations might fail in a way that your code could recover from. You can use Result to tell code that calls your code that it needs to handle potential success or failure as well. Using panic! and Result in the appropriate situations will make your code more reliable in the face of inevitable problems.
Every programming language has tools for effectively handling the duplication of concepts. In Rust, one such tool is generics
: abstract stand-ins for concrete types or other properties.
To eliminate this duplication, we’ll create an abstraction
by defining a function
that operates on any list of integers passed in a parameter.
fn largest(list: &[i32]) -> &i32 {
let mut largest = &list[0];
for item in list {
if item > largest {
largest = item;
}
}
largest
}
fn main() {
let number_list = vec![34, 50, 25, 100, 65];
let result = largest(&number_list);
println!("The largest number is {}", result);
assert_eq!(*result, 100);
let number_list = vec![102, 34, 6000, 89, 54, 2, 43, 8];
let result = largest(&number_list);
println!("The largest number is {}", result);
assert_eq!(*result, 6000);
}
In summary, here are the steps we took to change the code:
- Identify duplicate code.
- Extract the duplicate code into the body of the function and specify the inputs and return values of that code in the function signature.
- Update the two instances of duplicated code to call the function instead.
We use generics
to create definitions for items like function
signatures or structs, which we can then use with many different concrete data types.
When we use a parameter
in the body of the function, we have to declare
the parameter name in the signature
so the compiler knows what that name means.
fn largest<T>(list: &[T]) -> &T {
We can also define structs
to use a generic type parameter in one or more fields using the <>
syntax.
struct Point<T> {
x: T,
y: T,
}
fn main() {
let integer = Point { x: 5, y: 10 };
let float = Point { x: 1.0, y: 4.0 };
}
Note that because we’ve used only
one generic
type to definePoint<T>
, this definition says that the Point struct is generic oversome type T
, and the fields x and y areboth that same type
, whatever that type may be.
To define a Point struct where x and y are both generics
but could have different types
, we can use multiple generic type
parameters.
struct Point<T, U> {
x: T,
y: U,
}
fn main() {
let both_integer = Point { x: 5, y: 10 };
let both_float = Point { x: 1.0, y: 4.0 };
let integer_and_float = Point { x: 5, y: 4.0 };
}
We can define enums to hold generic data
types in their variants.
enum Option<T> {
Some(T),
None,
}
Enums can use multiple generic
types as well.
enum Result<T, E> {
Ok(T),
Err(E),
}
We can implement methods on structs and enums and use generic types
in their definitions
, too.
struct Point<T> {
x: T,
y: T,
}
impl<T> Point<T> { // using generic data type <T>
fn x(&self) -> &T {
&self.x
}
}
fn main() {
let p = Point { x: 5, y: 10 };
println!("p.x = {}", p.x());
}
Generic type parameters in a struct definition aren’t always the same
as those you use in that same struct’s method
signatures.
struct Point<X1, Y1> {
x: X1,
y: Y1,
}
impl<X1, Y1> Point<X1, Y1> {
fn mixup<X2, Y2>(self, other: Point<X2, Y2>) -> Point<X1, Y2> {
Point {
x: self.x,
y: other.y,
}
}
}
fn main() {
let p1 = Point { x: 5, y: 10.4 };
let p2 = Point { x: "Hello", y: 'c' };
let p3 = p1.mixup(p2);
println!("p3.x = {}, p3.y = {}", p3.x, p3.y);
}
In main, we’ve defined a Point that has an i32 for x (with value 5) and an f64 for y (with value 10.4). The p2 variable is a Point struct that has a string slice for x (with value "Hello") and a char for y (with value c). Calling mixup on p1 with the argument p2 gives us p3, which will have an i32 for x, because x came from p1. The p3 variable will have a char for y, because y came from p2. The println! macro call will print p3.x = 5, p3.y = c.
Rust accomplishes this by performing monomorphization
of the code using generics at compile time. Monomorphization is the process of turning generic code
into specific code by filling
in the concrete types
that are used when compiled.
A trait defines functionality
a particular type has and can share with other types
.
A type’s behavior consists of the methods
we can call on that type
. Different types
share the same behavior
if we can call the same methods
on all of those types. Trait
definitions are a way to group method
signatures together to define a set of behaviors necessary to accomplish some purpose
.
We want to make a media aggregator library crate
named aggregator that can display summaries
of data that might be stored in a NewsArticle or Tweet instance. To do this, we need a summary from each type
, and we’ll request that summary by calling a summarize method
on an instance.
pub trait Summary {
fn summarize(&self) -> String;
}
Now that we’ve defined the desired signatures of the Summary trait’s methods
, we can implement it on the types in our media aggregator.
pub trait Summary {
fn summarize(&self) -> String;
}
pub struct NewsArticle {
pub headline: String,
pub location: String,
pub author: String,
pub content: String,
}
impl Summary for NewsArticle { // trait Summary for NewsArticle
fn summarize(&self) -> String {
format!("{}, by {} ({})", self.headline, self.author, self.location)
}
}
pub struct Tweet {
pub username: String,
pub content: String,
pub reply: bool,
pub retweet: bool,
}
impl Summary for Tweet { // trait Summary for Tweet
fn summarize(&self) -> String {
format!("{}: {}", self.username, self.content)
}
}
Implementing a trait on a type is similar to implementing regular methods. The difference is that after impl, we put the trait name we want to implement, then use the for keyword, and then specify the name of the type we want to implement the trait for.
Now call the trait
methods on instances of NewsArticle and Tweet in the same way we call regular methods
.
use aggregator::{Summary, Tweet};
fn main() {
let tweet = Tweet {
username: String::from("horse_ebooks"),
content: String::from(
"of course, as you probably already know, people",
),
reply: false,
retweet: false,
};
println!("1 new tweet: {}", tweet.summarize());
}
But we
can’t implement external traits on external types
. For example, we can’t implement the Display trait on Vec within our aggregator crate, because Display and Vec are both defined in the standard library and aren’t local to our aggregator crate. This restriction is part of a property calledcoherence
, and more specifically theorphan rule
, so named because the parent type is not present. This rule ensures thatother people’s code can’t break your code
and vice versa. Without the rule, two crates could implement the same trait for the same type, and Rust wouldn’t know which implementation to use.
Sometimes it’s useful to have default behavior for some or all of the methods
in a trait instead of requiring implementations for all methods on every type.
pub trait Summary {
fn summarize(&self) -> String {
String::from("(Read more...)")
}
}
impl Summary for NewsArticle {}
Default implementations can call other methods in the same trait, even if those other methods don’t have a default implementation.
pub trait Summary { // trait with 2 methods
fn summarize_author(&self) -> String;
fn summarize(&self) -> String {
format!("(Read more from {}...)", self.summarize_author())
}
}
impl Summary for Tweet { // define only summarize_author()
fn summarize_author(&self) -> String {
format!("@{}", self.username)
}
}
// can use in main default trait summarize()
println!("1 new tweet: {}", tweet.summarize());
Now that you know how to define and implement traits
, we can explore how to use traits to define functions
that accept many different types
. To do this, we use the impl Trait syntax.
pub fn notify(item: &impl Summary) {
println!("Breaking news! {}", item.summarize());
}
Instead of a concrete type for the item parameter, we specify the impl keyword and the trait name.
The impl Trait syntax
works for straightforward cases but is actually syntax sugar
for a longer form known as a trait bound
;
pub fn notify<T: Summary>(item: &T) {
println!("Breaking news! {}", item.summarize());
}
Using
impl Trait
is appropriateif we want
this function to allow item1 and item2 to havedifferent types
Say we wanted notify to use display formatting as well as summarize on item: we specify in the notify definition that item must implement both Display and Summary
pub fn notify(item: &(impl Summary + Display)) {
// or generic
pub fn notify<T: Summary + Display>(item: &T) {
Each generic has its own trait bounds, so functions with multiple generic type parameters can contain lots of trait bound
information between the function’s name and its parameter list. specifying trait bounds inside a where clause
fn some_function<T: Display + Clone, U: Clone + Debug>(t: &T, u: &U) -> i32 {
// with where
fn some_function<T, U>(t: &T, u: &U) -> i32
where
T: Display + Clone,
U: Clone + Debug,
{
We can also use the impl Trait syntax in the return position
to return a value of some type that implements a trait
fn returns_summarizable() -> impl Summary {
Tweet {
username: String::from("horse_ebooks"),
content: String::from(
"of course, as you probably already know, people",
),
reply: false,
retweet: false,
}
}
By using impl Summary for the return type, we specify that the
returns_summarizable
function returns some type that implements theSummary trait
without naming the concrete type. you can only use impl Trait if you’re returning asingle type
.
By using a trait bound with an impl block that uses generic type parameters, we can implement methods conditionally for types that implement the specified traits.
use std::fmt::Display;
struct Pair<T> {
x: T,
y: T,
}
impl<T> Pair<T> { // always implements the new function to return a new instance of Pair<T>
fn new(x: T, y: T) -> Self {
Self { x, y }
}
}
impl<T: Display + PartialOrd> Pair<T> { // implements the cmp_display method only if its inner type T implements the PartialOrd and Display
fn cmp_display(&self) {
if self.x >= self.y {
println!("The largest member is x = {}", self.x);
} else {
println!("The largest member is y = {}", self.y);
}
}
}
Implementations of a trait on any type that
satisfies the trait bounds
are calledblanket implementations
and are extensively used in the Rust standard library.
For example, the standard library implements the ToString trait on any type that implements the Display trait.
impl<T: Display> ToString for T {
// --snip--
}
Rather than ensuring that a type has the behavior we want, lifetimes ensure
that references are valid as long as we need
them to be.
The main aim of lifetimes is to prevent dangling references
, which cause a program to reference data other
than the data it’s intended to reference
.
fn main() {
let r;
{
let x = 5;
r = &x; // asign r to &x
}
println!("r: {}", r); // x is no longer valid - out of scope
}
The Rust compiler has a borrow checker
that compares
scopes to determine whether all borrows are valid
.
fn main() {
let r; // ---------+-- 'a
// |
{ // |
let x = 5; // -+-- 'b |
r = &x; // | |
} // -+ |
// |
println!("r: {}", r); // |
} // ---------+
Here, we’ve annotated the lifetime of
r
with'a
and the lifetime ofx
with'b
. As you can see, the inner'b
block is much smaller than the outer'a
lifetime block
Fixes the code so it doesn’t have a dangling reference
fn main() {
let x = 5; // ----------+-- 'b
// |
let r = &x; // --+-- 'a |
// | |
println!("r: {}", r); // | |
// --+ |
} // ----------+
Here,
x
has the lifetime'b
, which in this case is larger than'a
We’ll write a function that returns the longer of two string slices.
fn main() {
let string1 = String::from("abcd");
let string2 = "xyz";
let result = longest(string1.as_str(), string2);
println!("The longest string is {}", result);
}
Note that we want the function to take string slices, which are references, rather than strings, because we don’t want the longest function to take ownership of its parameters.
If we try to implement the longest function, it won’t compile.
fn longest(x: &str, y: &str) -> &str {
if x.len() > y.len() {
x
} else {
y
}
}
To
fix
this error, we’lladd generic lifetime parameters
that define the relationship between the references so the borrow checker can perform its analysis.
Lifetime annotations don’t change how long any of the references live
. Rather, they describe the relationships of the lifetimes
of multiple references to each other without affecting the lifetimes.
&i32 // a reference
&'a i32 // a reference with an explicit lifetime
&'a mut i32 // a mutable reference with an explicit lifetime
Declare the generic lifetime parameters
inside angle brackets
between the function name and the parameter list, just as we did with generic type parameters
.
fn longest<'a>(x: &'a str, y: &'a str) -> &'a str {
if x.len() > y.len() {
x
} else {
y
}
}
So far, the structs we’ve defined all hold owned types
. We can define structs to hold references
, but in that case we would need to add a lifetime annotation
on every reference in the struct’s definition.
struct ImportantExcerpt<'a> {
part: &'a str,
}
Lifetimes on function or method parameters are called input lifetimes
, and lifetimes on return values are called output lifetimes
.
The first rule
is that the compiler assigns a lifetime parameter to each parameter
that’s a reference. fn foo<'a>(x: &'a i32);
The second rule
is that, if there is exactly one input
lifetime parameter, that lifetime is assigned to all output lifetime
parameters: fn foo<'a>(x: &'a i32) -> &'a i32
The third rule
is that, if there are multiple input lifetime parameters, but one of them is &self or &mut self
because this is a method, the lifetime of self is assigned to all output lifetime
parameters.
When we implement methods
on a struct with lifetimes, we use
the same syntax as that of generic type parameters
shown
//no lifetime
impl<'a> ImportantExcerpt<'a> {
fn level(&self) -> i32 {
3
}
}
//third lifetime elision rule
impl<'a> ImportantExcerpt<'a> {
fn announce_and_return_part(&self, announcement: &str) -> &str {
println!("Attention please: {}", announcement);
self.part
}
}
'static
, which denotes that the affected reference
can live
for the entire duration of the program
. All string
literals have the 'static
lifetime, which we can annotate as follows:
let s: &'static str = "I have a static lifetime.";
Let’s briefly look at the syntax of specifying generic type parameters, trait bounds, and lifetimes all in one
function!
use std::fmt::Display;
fn longest_with_an_announcement<'a, T>(
x: &'a str,
y: &'a str,
ann: T,
) -> &'a str
where
T: Display,
{
println!("Announcement! {}", ann);
if x.len() > y.len() {
x
} else {
y
}
}
This is the longest function from Listing 10-21 that returns the longer of two string slices. But now it has an extra parameter named ann of the generic type T, which can be filled in by any type that implements the Display trait as specified by the where clause. This extra parameter will be printed using {}, which is why the Display trait bound is necessary. Because lifetimes are a type of generic, the declarations of the lifetime parameter 'a and the generic type parameter T go in the same list inside the angle brackets after the function name.
In his 1972 essay “The Humble Programmer,” Edsger W. Dijkstra said that “Program testing can be a very effective way to show the presence of bugs, but it is hopelessly inadequate for showing their absence.”
3 actions
- Set up any needed data or state.
- Run the code you want to test.
- Assert the results are what you expect.
Example; cargo test
to run test
#[cfg(test)]
mod tests {
#[test] // this attribute indicates this is a test function
fn it_works() {
let result = 2 + 2;
assert_eq!(result, 4); //macro to assert that result, which contains the result of adding 2 and 2, equals 4
}
}
The assert!
macro, provided by the standard library, is useful when you want to ensure that some condition in a test evaluates to true
#[derive(Debug)]
struct Rectangle {
width: u32,
height: u32,
}
impl Rectangle {
fn can_hold(&self, other: &Rectangle) -> bool {
self.width > other.width && self.height > other.height
}
}
#[cfg(test)]
mod tests {
use super::*; // anything we define in the outer module is available to this tests module.
#[test]
fn larger_can_hold_smaller() {
let larger = Rectangle {
width: 8,
height: 7,
};
let smaller = Rectangle {
width: 5,
height: 1,
};
assert!(larger.can_hold(&smaller));
}
#[test]
fn smaller_cannot_hold_larger() {
let larger = Rectangle {
width: 8,
height: 7,
};
let smaller = Rectangle {
width: 5,
height: 1,
};
assert!(!smaller.can_hold(&larger));
}
}
These macros compare two arguments for equality or inequality
, respectively. They’ll also print the two values
if the assertion fails
, which makes it easier to see why the test failed;
pub fn add_two(a: i32) -> i32 {
a + 2
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn it_adds_two() {
assert_eq!(4, add_two(2));
}
}
Any arguments specified after the required arguments
are passed along to the format!
macro
#[test]
fn greeting_contains_name() {
let result = greeting("Carol");
assert!(
result.contains("Carol"),
"Greeting did not contain name, value was `{}`",
result
);
}
In addition to checking return values, it’s important to check that our code handles error conditions as we expect.
pub struct Guess {
value: i32,
}
impl Guess {
pub fn new(value: i32) -> Guess {
if value < 1 || value > 100 {
panic!("Guess value must be between 1 and 100, got {}.", value);
}
Guess { value }
}
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
// The test passes if the code inside the function panics
#[should_panic(expected = "less than or equal to 100")]
fn greater_than_100() {
Guess::new(200);
}
}
Use Result<T, E>
and return an Err
instead of panicking
#[cfg(test)]
mod tests {
#[test]
fn it_works() -> Result<(), String> {
if 2 + 2 == 4 {
Ok(())
} else {
Err(String::from("two plus two does not equal four"))
}
}
}
Some
command line options go
to cargo test
, and some go to the resulting test binary
. To separate
these two types of arguments, you list the arguments that go to cargo test
followed by the separator --
and then the ones that go to the test binary.
When you run multiple tests, by default
they run in parallel
using threads, meaning they finish running faster and you get feedback quicker.
cargo test -- --test-threads=1
By default
, if a test passes, Rust’s test library captures anything
printed to standard output. To show println!
use
cargo test -- --show-output
Running Single Tests
cargo test one_hundred
Filtering to Run Multiple Tests. Two of our tests’ names contain add
, we can run those two by running:
cargo test add
Ignoring Some Tests Unless Specifically Requested
#[test]
#[ignore]
fn expensive_test() {
// code that takes an hour to run
}
The Rust community thinks about tests in terms of two main categories
: unit tests
and integration tests
. Unit tests
are small
and more focused, testing one module
in isolation at a time, and can test private interfaces
. Integration tests
are entirely external
to your library and use your code in the same way any other external code would, using only the public interface
and potentially exercising multiple modules per test
.
You’ll put unit tests in the src directory in each file with the code that they’re testing. The convention is to create a module named tests in each file to contain the test functions and to annotate the module with cfg(test).
The Tests Module and #[cfg(test)]
#[cfg(test)]
mod tests { // create module tests
#[test]
fn it_works() {
let result = 2 + 2;
assert_eq!(result, 4);
}
}
Testing Private Functions
pub fn add_two(a: i32) -> i32 {
internal_adder(a, 2)
}
fn internal_adder(a: i32, b: i32) -> i32 { // without pub
a + b
}
#[cfg(test)]
mod tests {
use super::*; // bring all fn to scope
#[test]
fn internal() {
assert_eq!(4, internal_adder(2, 2));
}
}
There’s debate within the testing community about whether or not private functions should be tested directly
In Rust, integration tests are entirely external to your library.They use your library in the same way any other code would, which means they can only call functions that are part of your library’s public API.
The tests Directory
adder
├── Cargo.lock
├── Cargo.toml
├── src
│ └── lib.rs
└── tests
└── integration_test.rs
use adder; // bring lib to tests
#[test]
fn it_adds_two() {
assert_eq!(4, adder::add_two(2));
}
We don’t need to annotate any code in
tests/integration_test.rs
with#[cfg(test)]
Integration Tests for Binary Crates
If our project is a binary crate that only contains a src/main.rs file and doesn’t have a src/lib.rs file, we
can’t create integration tests
in the tests directory and bring functions defined in the src/main.rs file into scope with a use statement.Only library crates expose functions
that other crates can use; binary crates are meant to be run on their own.
Our grep
project will combine a number of concepts you’ve learned so far:
- Organizing code (using what you learned about modules in Chapter 7)
- Using vectors and strings (collections, Chapter 8)
- Handling errors (Chapter 9)
- Using traits and lifetimes where appropriate (Chapter 10)
- Writing tests (Chapter 11)
cargo new minigrep
cd minigrep
cargo run -- searchstring example-filename.txt
Two hyphens to indicate the following arguments are for our program rather than for cargo, a string to search for, and a path to a file to search in
The code in allows your minigrep program to read any command line arguments passed to it and then collect the values into a vector.
use std::env;
fn main() {
let args: Vec<String> = env::args().collect();
dbg!(args);
}
Now we need to save the values of the two arguments
in variables
so we can use the values throughout the rest of the program.
use std::env;
fn main() {
let args: Vec<String> = env::args().collect();
let query = &args[1];
let file_path = &args[2];
println!("Searching for {}", query);
println!("In file {}", file_path);
}
With the text in place, edit src/main.rs and add code to read the file
use std::env;
use std::fs;
fn main() {
// --snip--
println!("In file {}", file_path);
let contents = fs::read_to_string(file_path)
.expect("Should have been able to read the file");
println!("With text:\n{contents}");
}
The code read and then printed the contents of the file. But the code has a
few flaws
. At the moment, themain
function hasmultiple responsibilities
: generally, functions are clearer and easier to maintain if each function is responsible foronly one idea
. The other problem is that we’renot handling errors
as well as we could.
To improve our program, we’ll fix four problems
- our main function now performs
two tasks
- best to
group
theconfiguration variables
into one structure to make their purpose clear - reading a file can fail in a
number of ways
- it would be best if all the
error-handling
codewere in one place
so future maintainers had only one place to consult the code
Rust community has developed guidelines
for splitting the separate concerns:
- Split your program into a
main.rs
and alib.rs
and move your program’s logic to lib.rs. - As long as your command
line parsing logic is small
, it can remainin main.rs
. - When the command line
parsing logic starts getting complicated
, extract it from main.rs and move it tolib.rs
.
The responsibilities
that remain in the main function
after this process should be limited to the following:
Calling
the command lineparsing logic
with the argument valuesSetting up
any otherconfiguration
Calling a run
function in lib.rsHandling the error
if run returns an error
Because you
can’t test the main
function directly, this structure lets you test all of your program’s logic by moving it into functions in lib.rs. The code that remains inmain.rs will be small
enough toverify
its correctnessby reading it
.
Prepare for moving the command line parsing logic to src/lib.rs
fn main() {
let args: Vec<String> = env::args().collect();
let (query, file_path) = parse_config(&args);
// --snip--
}
fn parse_config(args: &[String]) -> (&str, &str) {
let query = &args[1];
let file_path = &args[2];
(query, file_path)
}
At the moment, we’re returning a tuple
, but then we immediately break
that tuple into individual parts
again. This is a sign
that perhaps we don’t
have the right abstraction
yet.
Another indicator
that shows there’s room for improvement is the config
part of parse_config
, which implies that the two values
we return are related
and are both part of one configuration
value.
fn main() {
let args: Vec<String> = env::args().collect();
let config = parse_config(&args);
println!("Searching for {}", config.query);
println!("In file {}", config.file_path);
let contents = fs::read_to_string(config.file_path)
.expect("Should have been able to read the file");
// --snip--
}
struct Config {
query: String,
file_path: String,
}
fn parse_config(args: &[String]) -> Config {
let query = args[1].clone(); // need new references
let file_path = args[2].clone(); // clone not efficient but easy
Config { query, file_path }
}
We’ve added a
struct
namedConfig
defined to have fields namedquery
andfile_path
. The signature ofparse_config
now indicates that it returns aConfig
value.
Purpose of the parse_config
function is to create
a Config instance
, we can change
parse_config from a plain function
to a function named new
that is associated with the Config
struct.
fn main() {
let args: Vec<String> = env::args().collect();
let config = Config::new(&args);
// --snip--
}
// --snip--
impl Config {
fn new(args: &[String]) -> Config {
let query = args[1].clone();
let file_path = args[2].clone();
Config { query, file_path }
}
}
We’ve updated main where we were calling
parse_config
to instead callConfig::new
.
Improving the Error Message
// --snip--
fn new(args: &[String]) -> Config {
if args.len() < 3 {
panic!("not enough arguments");
}
// --snip--
However, we also have extraneous information we don’t want to give to our users.
Returning a Result
Instead of Calling panic!
impl Config {
fn build(args: &[String]) -> Result<Config, &'static str> {
if args.len() < 3 {
return Err("not enough arguments");
}
let query = args[1].clone();
let file_path = args[2].clone();
Ok(Config { query, file_path })
}
}
change the function name from
new
tobuild
because many programmers expect new functions to never fail. Returning an Err value from Config::build allows the main function to handle the Result value returned from the build function and exit the process more cleanly in the error case.
Calling Config::build
and Handling Errors
use std::process;
fn main() {
let args: Vec<String> = env::args().collect();
let config = Config::build(&args).unwrap_or_else(|err| { // closure
println!("Problem parsing arguments: {err}"); // with |err| as argument
process::exit(1); // stop the program and return the number
});
// --snip--
A nonzero exit status is a convention to signal to the process that called our program that the program exited with an error state.
We’ll extract a function named run
that will hold
all the logic
currently in the main
function that isn’t involved with
setting up configuration
or handling errors
.
fn main() {
// --snip--
println!("Searching for {}", config.query);
println!("In file {}", config.file_path);
run(config);
}
fn run(config: Config) {
let contents = fs::read_to_string(config.file_path)
.expect("Should have been able to read the file");
println!("With text:\n{contents}");
}
// --snip--
The run function now contains all the remaining logic from main, starting from reading the file. The run function takes the Config instance as an argument.
With the remaining program logic separated into the run
function, we can improve
the error handling
, as we did with Config::build
use std::error::Error;
// --snip--
fn run(config: Config) -> Result<(), Box<dyn Error>> { // change return
let contents = fs::read_to_string(config.file_path)?;
println!("With text:\n{contents}");
Ok(()) // idiomatic way to indicate
// that we’re calling run for its side effects only;
}
- We changed the return type of the run function to
Result<(), Box<dyn Error>>
.Box<dyn Error>
means the function willreturn
a type that implements theError trait
, but wedon’t have to specify
what particulartype
the return value will be.- We’ve removed the call to
expect
in favor of the?
operator. Rather thanpanic!
on an error,?
willreturn
theerror value
from the current function for the caller to handle.- The
run
function nowreturns
anOk
value in the success case.
We’ll check for errors and handle them using a technique similar to one we used with Config::build
fn main() {
// --snip--
println!("Searching for {}", config.query);
println!("In file {}", config.file_path);
if let Err(e) = run(config) {
println!("Application error: {e}");
process::exit(1);
}
}
We use
if let
rather thanunwrap_or_else
to check whetherrun
returns anErr
value and callprocess::exit(1)
if it does.
Let’s move
all the code that isn’t the main function from src/main.rs to src/lib.rs
:
- The run function definition
- The relevant use statements
- The definition of Config
- The Config::build function definition
use std::error::Error;
use std::fs;
pub struct Config {
pub query: String,
pub file_path: String,
}
impl Config {
pub fn build(args: &[String]) -> Result<Config, &'static str> {
// --snip--
}
}
pub fn run(config: Config) -> Result<(), Box<dyn Error>> {
// --snip--
}
We’ve made liberal use of the
pub
keyword: onConfig
, on its fields and itsbuild
method, and on therun
function. We now have a library crate that has a public API we can test!
Now we need to bring the code we moved to src/lib.rs
into the scope
of the binary crate in src/main.rs
use std::env;
use std::process;
use minigrep::Config;
fn main() {
// --snip--
if let Err(e) = minigrep::run(config) {
// --snip--
}
}
In this section, we’ll add the searching logic to the minigrep program using the test-driven development (TDD) process with the following steps:
- Write a test that fails and run it to make sure it fails for the reason you expect.
- Write or modify just enough code to make the new test pass.
- Refactor the code you just added or changed and make sure the tests continue to pass.
- Repeat from step 1!
The test function specifies the behavior we want the search
function to have: it will take a query and the text to search, and it will return only the lines from the text that contain the query.
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn one_result() {
let query = "duct";
let contents = "\
Rust:
safe, fast, productive.
Pick three.";
assert_eq!(vec!["safe, fast, productive."], search(query, contents));
}
}
In accordance with TDD principles, we’ll add just enough code to get the test to compile and run by adding a definition of the search
function that always returns an empty vector
pub fn search<'a>(query: &str, contents: &'a str) -> Vec<&'a str> {
vec![]
}
with 'a we indicate that the returned vector should contain string slices that reference slices of the argument contents (rather than the argument query). if the compiler assumes we’re making string slices of query rather than contents, it will do its safety checking incorrectly
Currently, our test is failing because we always return
an empty vector. To fix that and implement search
, our program needs to follow these steps:
- Iterate through each line of the contents.
- Check whether the line contains our query string.
- If it does, add it to the list of values we’re returning.
- If it doesn’t, do nothing.
- Return the list of results that match.
Iterating Through Lines with the lines Method
pub fn search<'a>(query: &str, contents: &'a str) -> Vec<&'a str> {
for line in contents.lines() {
// do something with line
}
}
The lines method returns an
iterator
.
Searching Each Line for the Query
pub fn search<'a>(query: &str, contents: &'a str) -> Vec<&'a str> {
for line in contents.lines() {
if line.contains(query) {
// do something with line
}
}
}
Storing Matching Lines
pub fn search<'a>(query: &str, contents: &'a str) -> Vec<&'a str> {
let mut results = Vec::new(); // using vector for matching lines
for line in contents.lines() {
if line.contains(query) {
results.push(line);
}
}
results // return matchiing lines
}
Now that the search
function is working and tested, we need to call search from our run
function.
pub fn run(config: Config) -> Result<(), Box<dyn Error>> {
let contents = fs::read_to_string(config.file_path)?;
for line in search(&config.query, &contents) {
println!("{line}");
}
Ok(())
}
We’re still using a for loop to
return
each line fromsearch
and print it.
We’ll improve minigrep
by adding an extra feature: an option for case-insensitive searching that the user can turn on via an environment variable.
We first add a new search_case_insensitive function that will be called when the environment variable has a value. We’ll continue to follow the TDD process, so the first step is again to write a failing test.
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn case_sensitive() {
let query = "duct";
let contents = "\
Rust:
safe, fast, productive.
Pick three.
Duct tape."; // not match also
assert_eq!(vec!["safe, fast, productive."], search(query, contents));
}
#[test]
fn case_insensitive() {
let query = "rUsT";
let contents = "\
Rust:
safe, fast, productive.
Pick three.
Trust me.";
assert_eq!(
vec!["Rust:", "Trust me."],
search_case_insensitive(query, contents)
);
}
}
The search_case_insensitive
function will be almost the same as the search
function.
pub fn search_case_insensitive<'a>(
query: &str,
contents: &'a str,
) -> Vec<&'a str> {
let query = query.to_lowercase(); // shadowed
let mut results = Vec::new();
for line in contents.lines() {
if line.to_lowercase().contains(&query) {
results.push(line);
}
}
results
}
Note that
query
is now aString
rather than a string slice, because callingto_lowercase
creates new data rather than referencing existing data.
We’ll add a configuration option to the Config struct to switch between case-sensitive and case-insensitive search.
pub struct Config {
pub query: String,
pub file_path: String,
pub ignore_case: bool,
}
Next, we need the run
function to check the ignore_case field’s value and use that to decide whether to call the search
function or the search_case_insensitive
function
pub fn run(config: Config) -> Result<(), Box<dyn Error>> {
let contents = fs::read_to_string(config.file_path)?;
let results = if config.ignore_case {
search_case_insensitive(&config.query, &contents)
} else {
search(&config.query, &contents)
};
for line in results {
println!("{line}");
}
Ok(())
}
Finally, we need to check for the environment variable.
use std::env;
// --snip--
impl Config {
pub fn build(args: &[String]) -> Result<Config, &'static str> {
if args.len() < 3 {
return Err("not enough arguments");
}
let query = args[1].clone();
let file_path = args[2].clone();
let ignore_case = env::var("IGNORE_CASE").is_ok();
Ok(Config {
query,
file_path,
ignore_case,
})
}
}
we create a new variable
ignore_case
. To set its value, we call theenv::var
function and pass it the name of theIGNORE_CASE
environment variable.
Now, let’s run the program with IGNORE_CASE
set to 1
but with the same query
to
IGNORE_CASE=1 cargo run -- to poem.txt
In most terminals, there are two kinds of output: standard output (stdout
) for general information and standard error (stderr
) for error messages.
Command line programs are expected to send error messages to the standard error stream so we can still see error messages on the screen even if we redirect the standard output stream to a file.
cargo run > output.txt
The standard library provides the eprintln!
macro that prints to the standard error stream
fn main() {
let args: Vec<String> = env::args().collect();
let config = Config::build(&args).unwrap_or_else(|err| {
eprintln!("Problem parsing arguments: {err}");
process::exit(1);
});
if let Err(e) = minigrep::run(config) {
eprintln!("Application error: {e}");
process::exit(1);
}
}
now using standard output for successful output and standard error for error output as appropriate.
Programming in a functional style often includes using functions as values by passing them in arguments, returning them from other functions, assigning them to variables for later execution, and so forth.
Rust’s closures are anonymous functions you can save in a variable or pass as arguments to other functions.
For this example, we’re going to use an enum called ShirtColor that has the variants Red
and Blue
(limiting the number of colors available for simplicity). We represent the company’s inventory
with an Inventory struct that has a field named shirts
that contains a Vec<ShirtColor>
representing the shirt colors currently in stock. The method giveaway
defined on Inventory
gets the optional shirt color preference of the free shirt winner, and returns the shirt color the person will get.
#[derive(Debug, PartialEq, Copy, Clone)]
enum ShirtColor {
Red,
Blue,
}
struct Inventory {
shirts: Vec<ShirtColor>,
}
impl Inventory {
fn giveaway(&self, user_preference: Option<ShirtColor>) -> ShirtColor {
// passed a closure that calls self.most_stocked()
// on the current Inventory instance
// The standard library didn’t need to know anything
// about the Inventory or ShirtColor types we defined,
// Functions, on the other hand,
// are not able to capture their environment in this way.
user_preference.unwrap_or_else(|| self.most_stocked())
}
fn most_stocked(&self) -> ShirtColor {
let mut num_red = 0;
let mut num_blue = 0;
for color in &self.shirts {
match color {
ShirtColor::Red => num_red += 1,
ShirtColor::Blue => num_blue += 1,
}
}
if num_red > num_blue {
ShirtColor::Red
} else {
ShirtColor::Blue
}
}
}
Closures don’t usually require you to annotate the types of the parameters or the return value like fn
functions do.
Closures are typically short and relevant only within a narrow context rather than in any arbitrary scenario.
As with variables, we can add type annotations if we want to increase explicitness and clarity at the cost of being more verbose than is strictly necessary.
let expensive_closure = |num: u32| -> u32 {
println!("calculating slowly...");
thread::sleep(Duration::from_secs(2));
num
};
This illustrates how closure syntax is similar to function syntax except for the use of pipes and the amount of syntax that is optional:
fn add_one_v1 (x: u32) -> u32 { x + 1 }
let add_one_v2 = |x: u32| -> u32 { x + 1 };
let add_one_v3 = |x| { x + 1 };
let add_one_v4 = |x| x + 1 ;
Because there are no type annotations, we can call the closure with any type, which we’ve done here with String
the first time. If we then try to call example_closure with an integer
, we’ll get an error.
let example_closure = |x| x;
let s = example_closure(String::from("hello"));
let n = example_closure(5); // error here, cant use second time
Closures can capture values from their environment in three ways, which directly map to the three ways a function can take a parameter:
- borrowing immutably
- borrowing mutably
- taking ownership
A closure that captures an immutable reference to the vector named list
because it only needs an immutable reference to print the value:
fn main() {
let list = vec![1, 2, 3];
println!("Before defining closure: {:?}", list);
let only_borrows = || println!("From closure: {:?}", list); // bind closure
println!("Before calling closure: {:?}", list);
only_borrows(); // using closure
println!("After calling closure: {:?}", list);
}
Because we can have multiple immutable references to list at the same time, list is still accessible from the code before the closure definition, after the closure definition but before the closure is called, and after the closure is called.
we change the closure body so that it adds an element to the list vector. The closure now captures a mutable reference
fn main() {
let mut list = vec![1, 2, 3];
println!("Before defining closure: {:?}", list);
let mut borrows_mutably = || list.push(7); // borrow bind with mut
// no printing here because borrowing
borrows_mutably();
println!("After calling closure: {:?}", list);
}
Note that there’s no longer a
println!
between the definition and the call of theborrows_mutably
closure: whenborrows_mutably
is defined, it captures a mutable reference tolist
If you want to force the closure to take ownership of the values it uses in the environment even though the body of the closure doesn’t strictly need ownership, you can use the move
keyword before the parameter list.
use std::thread;
fn main() {
let list = vec![1, 2, 3];
println!("Before defining closure: {:?}", list);
thread::spawn(move || println!("From thread: {:?}", list))
.join()
.unwrap();
}
This technique is mostly useful when passing a closure to a new
thread
to move the data so that it’s owned by the new thread. If the main thread maintained ownership of list but ended before the new thread did and dropped list, the immutable reference in the thread would be invalid.
A closure body can do any of the following: move a captured value out of the closure, mutate the captured value, neither move nor mutate the value, or capture nothing from the environment to begin with. The way a closure captures and handles values from the environment affects which traits the closure implements:
FnOnce
applies to closures that can be called onceFnMut
applies to closures that don’t move captured values out of their body, but that might mutate the captured values.Fn
applies to closures that don’t move captured values out of their body and that don’t mutate captured values, as well as closures that capture nothing from their environment.
Let’s look at the definition of the unwrap_or_else method on Option
impl<T> Option<T> {
pub fn unwrap_or_else<F>(self, f: F) -> T
where
F: FnOnce() -> T
{
match self {
Some(x) => x,
None => f(),
}
}
}
The trait bound specified on the generic type
F
isFnOnce() -> T
, which meansF
must be able to be called once, take no arguments, and return aT
. Because all closures implementFnOnce
,unwrap_or_else
accepts the most different kinds of closures and is as flexible as it can be.
Note: Functions can implement all three of the Fn
traits too. If what we want to do doesn’t require capturing a value from the environment, we can use the name of a function rather than a closure where we need something that implements one of the Fn
traits. For example, on an Option<Vec<T>>
value, we could call unwrap_or_else(Vec::new)
to get a new, empty vector if the value is None
.
Now let’s look at the standard library method sort_by_key
defined on slices, to see how that differs from unwrap_or_else
and why sort_by_key
uses FnMut
instead of FnOnce
for the trait bound.
#[derive(Debug)]
struct Rectangle {
width: u32,
height: u32,
}
fn main() {
let mut list = [
Rectangle { width: 10, height: 1 },
Rectangle { width: 3, height: 5 },
Rectangle { width: 7, height: 12 },
];
list.sort_by_key(|r| r.width);
println!("{:#?}", list);
}
The iterator pattern allows you to perform some task on a sequence of items in turn. In Rust, iterators are lazy
, meaning they have no effect until you call methods that consume the iterator to use it up.
let v1 = vec![1, 2, 3];
let v1_iter = v1.iter();
When the for loop is called using the iterator in v1_iter, each element in the iterator is used in one iteration of the loop, which prints out each value.
let v1 = vec![1, 2, 3];
let v1_iter = v1.iter();
for val in v1_iter {
println!("Got: {}", val);
}
All iterators implement a trait named Iterator that is defined in the standard library.
pub trait Iterator {
type Item;
fn next(&mut self) -> Option<Self::Item>;
// methods with default implementations elided
}
type Item: this code says implementing the Iterator trait requires that you also define an Item type, and this Item type is used in the return type of the next method. In other words, the Item type will be the type returned from the iterator.
We can call the next method on iterators directly
#[test]
fn iterator_demonstration() {
let v1 = vec![1, 2, 3];
let mut v1_iter = v1.iter();
assert_eq!(v1_iter.next(), Some(&1));
assert_eq!(v1_iter.next(), Some(&2));
assert_eq!(v1_iter.next(), Some(&3));
assert_eq!(v1_iter.next(), None);
}
Note that we needed to make
v1_iter
mutable: calling thenext
method on an iterator changes internal state that the iterator uses to keep track of where it is in the sequence. In other words, this code consumes, or uses up, the iterator. Each call tonext
eats up an item from the iterator. We didn’t need to makev1_iter
mutable when we used afor
loop because the loop took ownership ofv1_iter
and made it mutable behind the scenes.
Also note that the values we get from the calls to next
are immutable references to the values in the vector. The iter
method produces an iterator over immutable references. If we want to create an iterator that takes ownership of v1
and returns owned values, we can call into_iter
instead of iter. Similarly, if we want to iterate over mutable references, we can call iter_mut
instead of iter
.
Methods that call next
are called consuming adaptors
, because calling them uses up the iterator. One example is the sum
method
#[test]
fn iterator_sum() {
let v1 = vec![1, 2, 3];
let v1_iter = v1.iter();
let total: i32 = v1_iter.sum();
assert_eq!(total, 6);
}
Iterator adaptors
are methods defined on the Iterator
trait that don’t consume the iterator. The map
method returns a new iterator that produces the modified items.
let v1: Vec<i32> = vec![1, 2, 3];
// map will produce new iterator
// we need to collect items from it
let v2: Vec<_> = v1.iter().map(|x| x + 1).collect();
assert_eq!(v2, vec![2, 3, 4]);
Many iterator adapters take closures as arguments, and commonly the closures we’ll specify as arguments to iterator adapters will be closures that capture their environment.
#[derive(PartialEq, Debug)]
struct Shoe {
size: u32,
style: String,
}
fn shoes_in_size(shoes: Vec<Shoe>, shoe_size: u32) -> Vec<Shoe> {
// takes ownership of a vector of shoes and a shoe size as parameters
shoes.into_iter().filter(|s| s.size == shoe_size).collect()
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn filters_by_size() {
let shoes = vec![
Shoe {
size: 10,
style: String::from("sneaker"),
},
Shoe {
size: 13,
style: String::from("sandal"),
},
Shoe {
size: 10,
style: String::from("boot"),
},
];
let in_my_size = shoes_in_size(shoes, 10);
assert_eq!(
in_my_size,
vec![
Shoe {
size: 10,
style: String::from("sneaker")
},
Shoe {
size: 10,
style: String::from("boot")
},
]
);
}
}
The closure captures the shoe_size parameter from the environment and compares the value with each shoe’s size, keeping only shoes of the size specified. Finally, calling collect gathers the values returned by the adapted iterator into a vector that’s returned by the function.
Let’s look at how iterators can improve our implementation of the Config::build
function and the search
function.
With our new knowledge about iterators, we can change the build
function to take ownership of an iterator as its argument instead of borrowing a slice.
impl Config {
pub fn build(args: &[String]) -> Result<Config, &'static str> {
if args.len() < 3 {
return Err("not enough arguments");
}
let query = args[1].clone();
let file_path = args[2].clone();
let ignore_case = env::var("IGNORE_CASE").is_ok();
Ok(Config {
query,
file_path,
ignore_case,
})
}
}
Once
Config::build
takes ownership of the iterator and stops using indexing operations that borrow, we can move theString
values from the iterator intoConfig
rather than callingclone
and making a new allocation.
Change main function
fn main() {
let config = Config::build(env::args()).unwrap_or_else(|err| {
eprintln!("Problem parsing arguments: {err}");
process::exit(1);
});
// --snip--
}
The
env::args
function returns an iterator! Rather than collecting the iterator values into a vector and then passing a slice toConfig::build
, now we’re passing ownership of the iterator returned fromenv::args
toConfig::build
directly.
Change lib
impl Config {
pub fn build(
mut args: impl Iterator<Item = String>,
) -> Result<Config, &'static str> {
// --snip--
The standard library documentation for the
env::args
function shows that the type of the iterator it returns isstd::env::Args
, and that type implements theIterator
trait and returnsString
values.
Next, we’ll fix the body of Config::build. Because args implements the Iterator trait, we know we can call the next method on it!
impl Config {
pub fn build(
mut args: impl Iterator<Item = String>,
) -> Result<Config, &'static str> {
args.next();
let query = match args.next() {
Some(arg) => arg,
None => return Err("Didn't get a query string"),
};
let file_path = match args.next() {
Some(arg) => arg,
None => return Err("Didn't get a file path"),
};
let ignore_case = env::var("IGNORE_CASE").is_ok();
Ok(Config {
query,
file_path,
ignore_case,
})
}
}
We can also take advantage of iterators in the search
function in our I/O project
pub fn search<'a>(query: &str, contents: &'a str) -> Vec<&'a str> {
let mut results = Vec::new();
for line in contents.lines() {
if line.contains(query) {
results.push(line);
}
}
results
}
We can write this code in a more concise way using iterator adaptor methods. Doing so also lets us avoid having a mutable intermediate results
vector. The functional programming style prefers to minimize the amount of mutable state to make code clearer.
pub fn search<'a>(query: &str, contents: &'a str) -> Vec<&'a str> {
contents
.lines()
.filter(|line| line.contains(query))
.collect()
}
uses the filter
adaptor
to keep only the lines thatline.contains(query)
returnstrue
for. We then collect the matching lines into another vector withcollect
.
Most Rust programmers prefer to use the iterator style
. It’s a bit tougher to get the hang of at first, but once you get a feel for the various iterator adaptors and what they do, iterators can be easier to understand.
To determine whether to use loops or iterators, you need to know which implementation is faster: the version of the search function with an explicit for loop or the version with iterators.
test bench_search_for ... bench: 19,620,300 ns/iter (+/- 915,700)
test bench_search_iter ... bench: 19,234,900 ns/iter (+/- 657,200)
The iterator version was slightly faster!
In Rust, release profiles are predefined and customizable profiles with different configurations that allow a programmer to have more control over various options for compiling code
$ cargo build
Finished dev [unoptimized + debuginfo] target(s) in 0.0s
$ cargo build --release
Finished release [optimized] target(s) in 0.0s
Cargo has default settings for each of the profiles that apply when you haven't explicitly added any [profile.*]
sections in the project’s Cargo.toml file
.
[profile.dev]
opt-level = 0
[profile.release]
opt-level = 3
We’ve used packages from crates.io as dependencies of our project, but you can also share your code with other people by publishing your own packages.
Documentation comments use three slashes, ///, instead of two and support Markdown notation for formatting the text.
/// Adds one to the number given.
///
/// # Examples
///
/// ```
/// let arg = 5;
/// let answer = my_crate::add_one(arg);
///
/// assert_eq!(6, answer);
/// ```
pub fn add_one(x: i32) -> i32 {
x + 1
}
For convenience, running
cargo doc --open
will build the HTML for your current crate’s documentation
Here are some other sections that crate authors commonly use in their documentation:
# Examples
# Panics
# Errors
# Safety
Adding example code blocks in your documentation comments can help demonstrate how to use your library, and doing so has an additional bonus: running cargo test
will run the code examples in your documentation as tests!
The style of doc comment //!
adds documentation to the item that contains the comments rather than to the items following the comments.
//! # My Crate
//!
//! `my_crate` is a collection of utilities to make performing certain
//! calculations more convenient.
/// Adds one to the number given.
// --snip--
To remove the internal organization from the public API, we can add pub use
statements to re-export the items at the top level
//! # Art
//!
//! A library for modeling artistic concepts.
pub use self::kinds::PrimaryColor;
pub use self::kinds::SecondaryColor;
pub use self::utils::mix;
pub mod kinds {
// --snip--
}
pub mod utils {
// --snip--
}
Before you can publish any crates, you need to create an account on crates.io and get an API token.
cargo login abcdefghijklmnopqrstuvwxyz012345
Let’s say you have a crate you want to publish. Before publishing, you’ll need to add some metadata in the [package] section of the crate’s Cargo.toml file.
[package]
name = "guessing_game"
version = "0.1.0"
edition = "2021"
description = "A fun game where you guess what number the computer has chosen."
license = "MIT OR Apache-2.0"
[dependencies]
Be careful, because a publish is permanent. The version can never be overwritten, and the code cannot be deleted.
cargo publish
When you’ve made changes to your crate and are ready to release a new version, you change the version value specified in your Cargo.toml file and republish.
Although you can’t remove previous versions of a crate, you can prevent any future projects from adding them as a new dependency.
$ cargo yank --vers 1.0.1
Updating crates.io index
Yank guessing_game@1.0.1
Cargo offers a feature called workspaces that can help manage multiple related packages that are developed in tandem.
A workspace is a set of packages that share the same Cargo.lock and output directory.
[workspace]
members = [
"adder",
]
$ cargo new adder
Created binary (application) `adder` package
├── Cargo.lock
├── Cargo.toml
├── adder
│ ├── Cargo.toml
│ └── src
│ └── main.rs
└── target
Next, let’s create another member package in the workspace and call it add_one. Change the top-level Cargo.toml to specify the add_one path in the members list:
[workspace]
members = [
"adder",
"add_one",
]
$ cargo new add_one --lib
Created library `add_one` package
├── Cargo.lock
├── Cargo.toml
├── add_one
│ ├── Cargo.toml
│ └── src
│ └── lib.rs
├── adder
│ ├── Cargo.toml
│ └── src
│ └── main.rs
└── target
In the add_one/src/lib.rs file, let’s add an add_one function:
pub fn add_one(x: i32) -> i32 {
x + 1
}
Now we can have the adder package with our binary depend on the add_one package that has our library.
[dependencies]
add_one = { path = "../add_one" }
Next, let’s use the add_one function (from the add_one crate) in the adder crate.
use add_one;
fn main() {
let num = 10;
println!("Hello, world! {num} plus one is {}!", add_one::add_one(num));
}
To run the binary crate from the add directory, we can specify which package in the workspace we want to run by using the -p argument and the package name with cargo run
$ cargo run -p adder
Finished dev [unoptimized + debuginfo] target(s) in 0.0s
Running `target/debug/adder`
Hello, world! 10 plus one is 11!
Notice that the workspace has only one Cargo.lock file at the top level, rather than having a Cargo.lock in each crate’s directory.
For another enhancement, let’s add a test of the add_one::add_one function within the add_one crate:
pub fn add_one(x: i32) -> i32 {
x + 1
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn it_works() {
assert_eq!(3, add_one(2));
}
}
Now run cargo test in the top-level add directory. Running cargo test in a workspace structured like this one will run the tests for all the crates in the workspace
We can also run tests for one particular crate in a workspace from the top-level directory by using the -p flag and specifying the name of the crate we want to test:
cargo test -p add_one
The cargo install command allows you to install and use binary crates locally.
cargo install ripgrep
Cargo is designed so you can extend it with new subcommands without having to modify Cargo. If a binary in your $PATH is named cargo-something, you can run it as if it was a Cargo subcommand by running cargo something. Custom commands like this are also listed when you run cargo --list. Being able to use cargo install to install extensions and then run them just like the built-in Cargo tools is a super convenient benefit of Cargo’s design!
A pointer is a general concept for a variable that contains an address in memory. This address refers to, or “points at,” some other data.
Box<T>
for allocating values on the heapRc<T>
a reference counting type that enables multiple ownershipRef<T> and RefMut<T>
accessed throughRefCell<T>
, a type that enforces the borrowing rules at runtime instead of compile time
Boxes don’t have performance overhead, other than storing their data on the heap instead of on the stack. But they don’t have many extra capabilities either.
use a box to store an i32 value on the heap:
fn main() {
let b = Box::new(5);
println!("b = {}", b);
}
A value of recursive type can have another value of the same type as part of itself.
More Information About the Cons List
// representing recursive list from Lisp language
(1, (2, (3, Nil)))
// recreate in rust
enum List {
Cons(i32, List),
Nil,
}
// using recursive list
use crate::List::{Cons, Nil};
fn main() {
let list = Cons(1, Cons(2, Cons(3, Nil)));
}
Because a Box is a pointer, Rust always knows how much space a Box needs: a pointer’s size doesn’t change based on the amount of data it’s pointing to. This means we can put a Box inside the Cons variant instead of another List value directly.
enum List {
Cons(i32, Box<List>),
Nil,
}
use crate::List::{Cons, Nil};
// using Box<t> to store recursive list
fn main() {
let list = Cons(1, Box::new(Cons(2, Box::new(Cons(3, Box::new(Nil))))));
}
Implementing the Deref trait allows you to customize the behavior of the dereference operator *
we create a reference to an i32 value and then use the dereference operator to follow the reference to the value
fn main() {
let x = 5;
let y = &x; // use ref
assert_eq!(5, x);
assert_eq!(5, *y); // deref pointer
}
We can rewrite the code in Listing 15-6 to use a Box instead of a reference;
fn main() {
let x = 5;
let y = Box::new(x); // use Box
assert_eq!(5, x);
assert_eq!(5, *y);
}
Let’s build a smart pointer similar to the Box type provided by the standard library to experience how smart pointers behave differently from references by default
struct MyBox<T>(T);
impl<T> MyBox<T> {
fn new(x: T) -> MyBox<T> {
MyBox(x)
}
}
Our
MyBox<T>
type can’t be dereferenced because we haven’t implemented that ability on our type. To enable dereferencing with the*
operator, we implement theDeref
trait.
The Deref trait, provided by the standard library, requires us to implement one method named deref that borrows self and returns a reference to the inner data.
use std::ops::Deref;
impl<T> Deref for MyBox<T> {
type Target = T;
fn deref(&self) -> &Self::Target {
&self.0
}
}
When we entered *y in Listing 15-9, behind the scenes Rust actually ran this code:
*(y.deref())
Deref coercion converts a reference to a type that implements the Deref
trait into a reference to another type.
fn hello(name: &str) {
println!("Hello, {name}!");
}
fn main() {
let m = MyBox::new(String::from("Rust"));
hello(&m);
}
Here we’re calling the
hello
function with the argument&m
, which is a reference to aMyBox<String>
value. Because we implemented theDeref
trait onMyBox<T>
, Rust can turn&MyBox<String>
into&String
by callingderef
. The standard library provides an implementation ofDeref
onString
that returns a string slice, and this is in the API documentation for Deref. Rust calls deref again to turn the&String
into&str
, which matches the hello function’s definition.
If Rust didn’t implement deref coercion
fn main() {
let m = MyBox::new(String::from("Rust"));
hello(&(*m)[..]);
}
Rust does deref coercion when it finds types and trait implementations in three cases:
- From
&T
to &U when T: Deref<Target=U> - From
&mut T
to &mut U when T: DerefMut<Target=U> - From
&mut T
to &U when T: Deref<Target=U>
The second trait important to the smart pointer pattern is Drop
, which lets you customize what happens when a value is about to go out of scope.
Rust automatically called drop
for us when our instances went out of scope, calling the code we specified. Variables are dropped in the reverse order
Unfortunately, it’s not straightforward to disable the automatic drop
functionality. Disabling drop isn’t usually necessary
So, if we need to force a value to be cleaned up early, we use the std::mem::drop
function.
In the majority of cases, ownership is clear: you know exactly which variable owns a given value. However, there are cases when a single value might have multiple owners
We’ll create list a that contains 5 and then 10. Then we’ll make two more lists: b that starts with 3 and c that starts with 4. Both b and c lists will then continue on to the first a list containing 5 and 10.
enum List {
Cons(i32, Box<List>),
Nil,
}
use crate::List::{Cons, Nil};
fn main() {
let a = Cons(5, Box::new(Cons(10, Box::new(Nil))));
let b = Cons(3, Box::new(a));
let c = Cons(4, Box::new(a));
}
The Cons variants own the data they hold, so when we create the b list, a is moved into b and b owns a. Then, when we try to use a again when creating c, we’re not allowed to because a has been moved.
we’ll change our definition of List to use Rc in place of Box
enum List {
Cons(i32, Rc<List>), // change to Rc<T>
Nil,
}
use crate::List::{Cons, Nil};
use std::rc::Rc;
fn main() {
let a = Rc::new(Cons(5, Rc::new(Cons(10, Rc::new(Nil)))));
let b = Cons(3, Rc::clone(&a));
let c = Cons(4, Rc::clone(&a));
}
When looking for performance problems in the code, we only need to consider the deep-copy clones and can disregard calls to Rc::clone
Via immutable references, Rc<T>
allows you to share data between multiple parts of your program for reading only. If Rc<T>
allowed you to have multiple mutable references too, you might violate one of the borrowing rules discussed in Chapter 4: multiple mutable borrows to the same place can cause data races and inconsistencies.
Interior mutability is a design pattern in Rust that allows you to mutate data even when there are immutable references to that data; normally, this action is disallowed by the borrowing rules.
With references and Box<T>
, the borrowing rules’ invariants are enforced at compile time. With RefCell<T>
, these invariants are enforced at runtime. With references, if you break these rules, you’ll get a compiler error. With RefCell<T>
, if you break these rules, your program will panic and exit.
Here is a recap of the reasons to choose Box<T>
, Rc<T>
, or RefCell<T>
:
- Rc enables multiple owners of the same data; Box and RefCell have single owners
- Box allows immutable or mutable borrows checked at compile time; Rc allows only immutable borrows checked at compile time; RefCell allows immutable or mutable borrows checked at runtime.
- Because RefCell allows mutable borrows checked at runtime, you can mutate the value inside the RefCell even when the RefCell is immutable.
A consequence of the borrowing rules is that when you have an immutable value, you can’t borrow it mutably.
fn main() {
let x = 5;
let y = &mut x; // don't work
}
Sometimes during testing a programmer will use a type in place of another type, in order to observe particular behavior and assert it’s implemented correctly. This placeholder type is called a test double. - stunt double
from filmmaking
Here’s the scenario we’ll test: we’ll create a library that tracks a value against a maximum value and sends messages based on how close to the maximum value the current value is.
pub trait Messenger {
fn send(&self, msg: &str);
}
pub struct LimitTracker<'a, T: Messenger> {
messenger: &'a T,
value: usize,
max: usize,
}
impl<'a, T> LimitTracker<'a, T>
where
T: Messenger,
{
pub fn new(messenger: &'a T, max: usize) -> LimitTracker<'a, T> {
LimitTracker {
messenger,
value: 0,
max,
}
}
pub fn set_value(&mut self, value: usize) {
self.value = value;
let percentage_of_max = self.value as f64 / self.max as f64;
if percentage_of_max >= 1.0 {
self.messenger.send("Error: You are over your quota!");
} else if percentage_of_max >= 0.9 {
self.messenger
.send("Urgent warning: You've used up over 90% of your quota!");
} else if percentage_of_max >= 0.75 {
self.messenger
.send("Warning: You've used up over 75% of your quota!");
}
}
}
One important part of this code is that the Messenger trait has one method called send that takes an immutable reference to self and the text of the message. This trait is the interface our mock object needs to implement so that the mock can be used in the same way a real object is.
We need a mock object that, instead of sending an email or text message when we call send, will only keep track of the messages it’s told to send.
#[cfg(test)]
mod tests {
use super::*;
struct MockMessenger {
sent_messages: Vec<String>,
}
impl MockMessenger {
fn new() -> MockMessenger {
MockMessenger {
sent_messages: vec![],
}
}
}
impl Messenger for MockMessenger {
fn send(&self, message: &str) {
self.sent_messages.push(String::from(message));
}
}
#[test]
fn it_sends_an_over_75_percent_warning_message() {
let mock_messenger = MockMessenger::new();
let mut limit_tracker = LimitTracker::new(&mock_messenger, 100);
limit_tracker.set_value(80);
assert_eq!(mock_messenger.sent_messages.len(), 1);
}
}
This test code defines a MockMessenger struct that has a sent_messages field with a Vec of String values to keep track of the messages it’s told to send.
This is a situation in which interior mutability can help! We’ll store the sent_messages within a RefCell, and then the send method will be able to modify sent_messages to store the messages we’ve seen.
#[cfg(test)]
mod tests {
use super::*;
use std::cell::RefCell;
struct MockMessenger {
sent_messages: RefCell<Vec<String>>,
}
impl MockMessenger {
fn new() -> MockMessenger {
MockMessenger {
sent_messages: RefCell::new(vec![]),
}
}
}
impl Messenger for MockMessenger {
fn send(&self, message: &str) {
self.sent_messages.borrow_mut().push(String::from(message));
}
}
#[test]
fn it_sends_an_over_75_percent_warning_message() {
// --snip--
assert_eq!(mock_messenger.sent_messages.borrow().len(), 1);
}
}
The sent_messages field is now of type
RefCell<Vec<String>>
instead ofVec<String>
. In the new function, we create a newRefCell<Vec<String>>
instance around the empty vector.
When creating immutable and mutable references, we use the &
and &mut
syntax, respectively. With RefCell<T>
, we use the borrow and borrow_mut methods, which are part of the safe API that belongs to RefCell<T>
.
If we try to violate these rules, rather than getting a compiler error as we would with references, the implementation of RefCell will panic at runtime.
impl Messenger for MockMessenger {
fn send(&self, message: &str) {
let mut one_borrow = self.sent_messages.borrow_mut();
let mut two_borrow = self.sent_messages.borrow_mut();
one_borrow.push(String::from(message));
two_borrow.push(String::from(message));
}
}
We create a variable one_borrow for the RefMut smart pointer returned from borrow_mut. Then we create another mutable borrow in the same way in the variable two_borrow. This makes two mutable references in the same scope, which isn’t allowed.
Choosing to catch borrowing errors at runtime rather than compile time, as we’ve done here, means you’d potentially be finding mistakes in your code later in the development process: possibly not until your code was deployed to production. Also, your code would incur a small runtime performance penalty as a result of keeping track of the borrows at runtime rather than compile time. However, using RefCell makes it possible to write a mock object that can modify itself to keep track of the messages it has seen while you’re using it in a context where only immutable values are allowed.
A common way to use RefCell<T>
is in combination with Rc<T>
.
#[derive(Debug)]
enum List {
Cons(Rc<RefCell<i32>>, Rc<List>),
Nil,
}
use crate::List::{Cons, Nil};
use std::cell::RefCell;
use std::rc::Rc;
fn main() {
let value = Rc::new(RefCell::new(5));
let a = Rc::new(Cons(Rc::clone(&value), Rc::new(Nil)));
let b = Cons(Rc::new(RefCell::new(3)), Rc::clone(&a));
let c = Cons(Rc::new(RefCell::new(4)), Rc::clone(&a));
*value.borrow_mut() += 10; // change inner value
println!("a after = {:?}", a); // 15 not 5
println!("b after = {:?}", b); // 15
println!("c after = {:?}", c); // 15
}
We create a value that is an instance of
Rc<RefCell<i32>>
and store it in a variable named value so we can access it directly later.
Rust’s memory safety guarantees make it difficult, but not impossible, to accidentally create memory that is never cleaned up (known as a memory leak).
Let’s look at how a reference cycle might happen and how to prevent it, starting with the definition of the List enum
and a tail
method
use crate::List::{Cons, Nil};
use std::cell::RefCell;
use std::rc::Rc;
#[derive(Debug)]
enum List {
Cons(i32, RefCell<Rc<List>>),
Nil,
}
impl List {
fn tail(&self) -> Option<&RefCell<Rc<List>>> {
match self {
Cons(_, item) => Some(item),
Nil => None,
}
}
}
fn main() {}
adding a main function that uses the definitions
fn main() {
let a = Rc::new(Cons(5, RefCell::new(Rc::new(Nil)))); // instance with 5
println!("a initial rc count = {}", Rc::strong_count(&a));
println!("a next item = {:?}", a.tail());
let b = Rc::new(Cons(10, RefCell::new(Rc::clone(&a)))); // points to a
println!("a rc count after b creation = {}", Rc::strong_count(&a));
println!("b initial rc count = {}", Rc::strong_count(&b));
println!("b next item = {:?}", b.tail());
if let Some(link) = a.tail() { // change a to point to b -> cycle
*link.borrow_mut() = Rc::clone(&b);
}
println!("b rc count after changing a = {}", Rc::strong_count(&b));
println!("a rc count after changing a = {}", Rc::strong_count(&a));
// Uncomment the next line to see that we have a cycle;
// it will overflow the stack
// println!("a next item = {:?}", a.tail());
}
The reference count of the
Rc<List>
instances in both a and b are 2 after we change the list ina
to point tob
. At the end of main, Rust drops the variableb
, which decreases the reference count of theb
Rc<List>
instance from 2 to 1. The memory thatRc<List>
has on the heap won’t be dropped at this point, because its reference count is 1, not 0.
You can also create a weak reference to the value within an Rc<T>
instance by calling Rc::downgrade
and passing a reference to the Rc<T>
.
Creating a Tree Data Structure: a Node with Child Nodes
use std::cell::RefCell;
use std::rc::Rc;
#[derive(Debug)]
struct Node {
value: i32,
children: RefCell<Vec<Rc<Node>>>,
}
We want a Node to own its children, and we want to share that ownership with variables so we can access each Node in the tree directly.
We’ll use our struct definition and create one Node instance named leaf with the value 3 and no children, and another instance named branch with the value 5 and leaf as one of its children,
fn main() {
let leaf = Rc::new(Node {
value: 3,
children: RefCell::new(vec![]),
});
let branch = Rc::new(Node {
value: 5,
children: RefCell::new(vec![Rc::clone(&leaf)]),
});
}
We clone the Rc in leaf and store that in branch, meaning the Node in leaf now has two owners: leaf and branch.
Adding a Reference from a Child to Its Parent
use std::cell::RefCell;
use std::rc::{Rc, Weak};
#[derive(Debug)]
struct Node {
value: i32,
parent: RefCell<Weak<Node>>, // use weak instead of RC<T>
children: RefCell<Vec<Rc<Node>>>,
}
To make the child node aware of its parent, we need to add a parent field to our Node struct definition.
A node will be able to refer to its parent node but doesn’t own its parent.
fn main() {
let leaf = Rc::new(Node {
value: 3,
parent: RefCell::new(Weak::new()), // empty ref
children: RefCell::new(vec![]),
});
println!("leaf parent = {:?}", leaf.parent.borrow().upgrade());
let branch = Rc::new(Node {
value: 5,
parent: RefCell::new(Weak::new()),
children: RefCell::new(vec![Rc::clone(&leaf)]),
});
*leaf.parent.borrow_mut() = Rc::downgrade(&branch);
println!("leaf parent = {:?}", leaf.parent.borrow().upgrade());
}
When we create the branch node, it will also have a new Weak reference in the parent field, because branch doesn’t have a parent node. We still have leaf as one of the children of branch. Once we have the Node instance in branch, we can modify leaf to give it a Weak reference to its parent. We use the borrow_mut method on the RefCell<Weak> in the parent field of leaf, and then we use the Rc::downgrade function to create a Weak reference to branch from the Rc in branch.
When we print the parent of leaf again, this time we’ll get a Some variant holding branch: now leaf can access its parent! When we print leaf, we also avoid the cycle that eventually ended in a stack overflow
leaf parent = Some(Node { value: 5, parent: RefCell { value: (Weak) },
children: RefCell { value: [Node { value: 3, parent: RefCell { value: (Weak) },
children: RefCell { value: [] } }] } })
Let’s look at how the strong_count and weak_count values of the Rc instances change by creating a new inner scope and moving the creation of branch into that scope.
use std::cell::RefCell;
use std::rc::{Rc, Weak};
#[derive(Debug)]
struct Node {
value: i32,
parent: RefCell<Weak<Node>>,
children: RefCell<Vec<Rc<Node>>>,
}
fn main() {
let leaf = Rc::new(Node {
value: 3,
parent: RefCell::new(Weak::new()),
children: RefCell::new(vec![]),
});
println!(
"leaf strong = {}, weak = {}",
Rc::strong_count(&leaf),
Rc::weak_count(&leaf),
);
{
let branch = Rc::new(Node {
value: 5,
parent: RefCell::new(Weak::new()),
children: RefCell::new(vec![Rc::clone(&leaf)]),
});
*leaf.parent.borrow_mut() = Rc::downgrade(&branch);
println!(
"branch strong = {}, weak = {}",
Rc::strong_count(&branch),
Rc::weak_count(&branch),
);
println!(
"leaf strong = {}, weak = {}",
Rc::strong_count(&leaf),
Rc::weak_count(&leaf),
);
}
println!("leaf parent = {:?}", leaf.parent.borrow().upgrade());
println!(
"leaf strong = {}, weak = {}",
Rc::strong_count(&leaf),
Rc::weak_count(&leaf),
);
}
After leaf is created, its Rc has a strong count of 1 and a weak count of 0. In the inner scope, we create branch and associate it with leaf, at which point when we print the counts, the Rc in branch will have a strong count of 1 and a weak count of 1 (for leaf.parent pointing to branch with a Weak). When we print the counts in leaf, we’ll see it will have a strong count of 2, because branch now has a clone of the Rc of leaf stored in branch.children, but will still have a weak count of 0. When the inner scope ends, branch goes out of scope and the strong count of the Rc decreases to 0, so its Node is dropped. The weak count of 1 from leaf.parent has no bearing on whether or not Node is dropped, so we don’t get any memory leaks!
All of the logic that manages the counts and value dropping is built into Rc and Weak and their implementations of the Drop trait. By specifying that the relationship from a child to its parent should be a Weak reference in the definition of Node, you’re able to have parent nodes point to child nodes and vice versa without creating a reference cycle and memory leaks.
This chapter covered how to use smart pointers to make different guarantees and trade-offs from those Rust makes by default with regular references. The Box<T>
type has a known size and points to data allocated on the heap. The Rc<T>
type keeps track of the number of references to data on the heap so that data can have multiple owners. The RefCell<T>
type with its interior mutability gives us a type that we can use when we need an immutable type but need to change an inner value of that type; it also enforces the borrowing rules at runtime instead of at compile time.