Enums with data in Rust

Introduction

Enums with data let you group different types of related information together in one place. This helps keep your code organized and easy to understand.

When you want to represent different kinds of messages that carry extra information.

When you need to handle different shapes, each with its own measurements.

When you want to store different types of errors with details.

When you want to group related but different data types under one name.

Syntax

Rust

enum Name {
    Variant1(Type1),
    Variant2(Type2, Type3),
    Variant3 { field1: Type4, field2: Type5 },
}

Each variant can hold different types and amounts of data.

You can use tuple-like variants or struct-like variants with named fields.

Examples

This enum has variants with no data, tuple data, and named fields.

Rust

enum Message {
    Quit,
    Move(i32, i32),
    Write(String),
    ChangeColor { r: u8, g: u8, b: u8 },
}

Create a Message variant with two numbers.

Rust

let msg = Message::Move(10, 20);

Create a Message variant with named color fields.

Rust

let color = Message::ChangeColor { r: 255, g: 0, b: 0 };

Sample Program

This program defines a Shape enum with data. It calculates area differently based on the shape variant.

Rust

enum Shape {
    Circle(f64),
    Rectangle { width: f64, height: f64 },
}

fn area(shape: &Shape) -> f64 {
    match shape {
        Shape::Circle(radius) => 3.14159 * radius * radius,
        Shape::Rectangle { width, height } => width * height,
    }
}

fn main() {
    let c = Shape::Circle(2.0);
    let r = Shape::Rectangle { width: 3.0, height: 4.0 };

    println!("Circle area: {}", area(&c));
    println!("Rectangle area: {}", area(&r));
}

Important Notes

Use match to check which variant you have and get its data.

Enums with data help you write clear and safe code by grouping related data.

Summary

Enums can hold different types of data in each variant.

This helps organize related but different information together.

You use match to work with enum data safely.

Practice

1. What is the main purpose of enums with data in Rust?

easy

A. To group related values that can be different types under one name

B. To create multiple copies of the same data

C. To perform arithmetic operations on numbers

D. To store only strings in a list

Solution

1. Step 1: Understand what enums with data do

   Enums with data allow each variant to hold different types or amounts of data, grouping related but different information.

2. Step 2: Compare options with this understanding

   Only To group related values that can be different types under one name correctly describes grouping related values of different types under one name.

3. Final Answer:

   To group related values that can be different types under one name -> Option A

4. Quick Check:

   Enums with data = group different types [OK]

Hint: Enums group different data types under one name [OK]

Common Mistakes:

• Thinking enums only store one data type
• Confusing enums with arrays or vectors
• Assuming enums perform calculations

2. Which of the following is the correct syntax to define an enum variant with data in Rust?

easy

A. enum Color { Red(u8, u8, u8), Blue }

B. enum Color { Red = (u8, u8, u8), Blue }

C. enum Color { Red: (u8, u8, u8), Blue }

D. enum Color { Red -> (u8, u8, u8), Blue }

Solution

1. Step 1: Recall enum variant syntax with data

   In Rust, enum variants with data are defined by listing the variant name followed by parentheses containing the data types.

2. Step 2: Check each option

   enum Color { Red(u8, u8, u8), Blue } uses correct syntax: Red(u8, u8, u8). Options A, B, and C use invalid syntax with ->, =, or : which are incorrect for enum variant data.

3. Final Answer:

   enum Color { Red(u8, u8, u8), Blue } -> Option A

4. Quick Check:

   Variant with data uses parentheses () [OK]

Hint: Use parentheses () after variant name for data [OK]

Common Mistakes:

• Using = instead of () for variant data
• Using : or -> which are invalid here
• Forgetting parentheses for data

3. What will be the output of this Rust code?

enum Message {
    Quit,
    Move { x: i32, y: i32 },
    Write(String),
}

fn main() {
    let msg = Message::Move { x: 10, y: 20 };
    match msg {
        Message::Quit => println!("Quit"),
        Message::Move { x, y } => println!("Move to {}, {}", x, y),
        Message::Write(text) => println!("Write {}", text),
    }
}

medium

A. Compilation error

B. Quit

C. Write 10, 20

D. Move to 10, 20

Solution

1. Step 1: Identify the enum variant used

   The variable msg is assigned Message::Move with x=10 and y=20.

2. Step 2: Match arms and output

   The match statement checks msg. For Message::Move, it prints "Move to {}, {}" with x and y values, so output is "Move to 10, 20".

3. Final Answer:

   Move to 10, 20 -> Option D

4. Quick Check:

   Match Message::Move prints coordinates [OK]

Hint: Match variant and print its data fields [OK]

Common Mistakes:

• Confusing variant names
• Expecting Quit or Write output
• Thinking code won't compile

4. The following Rust code has an error. What is the problem?

enum Shape {
    Circle(f64),
    Rectangle { width: f64, height: f64 },
}

fn area(shape: Shape) -> f64 {
    match shape {
        Shape::Circle(radius) => 3.14 * radius * radius,
        Shape::Rectangle(width, height) => width * height,
    }
}

medium

A. Missing semicolon after match arms

B. Circle variant should have named fields

C. Incorrect pattern for Rectangle variant in match

D. Function return type should be i32

Solution

1. Step 1: Check enum variant definitions

   Circle has tuple data (f64), Rectangle has named fields { width, height }.

2. Step 2: Check match patterns

   Circle pattern is correct. Rectangle pattern uses (width, height) which is wrong; it should use { width, height } to match named fields.

3. Final Answer:

   Incorrect pattern for Rectangle variant in match -> Option C

4. Quick Check:

   Match named fields with braces {} [OK]

Hint: Match named fields with braces {}, tuple fields with () [OK]

Common Mistakes:

• Using parentheses for named fields
• Ignoring pattern mismatch errors
• Changing return type unnecessarily

5. Given this enum and function, what is the output?

enum Event {
    Click { x: i32, y: i32 },
    KeyPress(char),
    Resize(u32, u32),
}

fn describe(event: Event) -> String {
    match event {
        Event::Click { x, y } if x > 0 && y > 0 => format!("Clicked at positive coords: {}, {}", x, y),
        Event::Click { .. } => "Clicked at non-positive coords".to_string(),
        Event::KeyPress(c) if c.is_alphabetic() => format!("Key pressed: {}", c),
        Event::KeyPress(_) => "Non-alphabetic key pressed".to_string(),
        Event::Resize(w, h) if w == h => "Resized to square".to_string(),
        Event::Resize(_, _) => "Resized to rectangle".to_string(),
    }
}

fn main() {
    let e = Event::Resize(100, 100);
    println!("{}", describe(e));
}

hard

A. Clicked at positive coords: 100, 100

B. Resized to square

C. Resized to rectangle

D. Non-alphabetic key pressed

Solution

1. Step 1: Identify the enum variant and values

   The event is Event::Resize with width=100 and height=100.

2. Step 2: Match arms with conditions

   The Resize arm with condition w == h matches because 100 == 100, so it returns "Resized to square".

3. Final Answer:

   Resized to square -> Option B

4. Quick Check:

   Resize with equal sides = square [OK]

Hint: Check match guards (if conditions) carefully [OK]

Common Mistakes:

• Ignoring if conditions in match arms
• Confusing Resize with Click variant
• Assuming default arm without conditions