Toán tử dấu hỏi '?'

Thuộc phần Logic và luồng chương trình trong hành trình Rust của Coddy. Bài học 46 trên 66.

Khi làm việc với các hàm trả về Result, bạn thường cần xử lý lỗi tại chỗ hoặc chuyển lỗi lên hàm gọi. Rust cung cấp một toán tử tinh gọn cho mẫu phổ biến này: toán tử dấu hỏi ?.

Toán tử ? giúp truyền lỗi một cách gọn gàng mà không cần viết các câu lệnh match dài dòng. Khi đặt ? sau một giá trị Result, toán tử này tự động lấy giá trị bên trong Ok hoặc trả về ngay từ hàm hiện tại cùng với Err:

fn parse_and_double(s: &str) -> Result<i32, std::num::ParseIntError> {
    let number = s.parse::<i32>()?; // Nếu phân tích cú pháp thất bại, trả về lỗi
    Ok(number * 2) // Nếu thành công, nhân đôi số đó
}

Cách này tương đương với việc viết đầy đủ câu lệnh match, nhưng ngắn gọn hơn nhiều. Nếu s.parse::<i32>() trả về Ok, ? sẽ trích xuất giá trị và gán cho number. Nếu nó trả về Err, ? sẽ lập tức trả về lỗi đó từ toàn bộ hàm.

Hãy chú ý đến cú pháp ::<i32>. Cách viết này được gọi là turbofish, và cho Rust biết cần phân tích chuỗi thành kiểu dữ liệu nào. Hàm trả về Result<i32, std::num::ParseIntError>, trong đó i32 là kiểu dữ liệu khi thành công, còn std::num::ParseIntError là kiểu lỗi được trả về khi phân tích thất bại.

Toán tử ? chỉ có thể dùng trong các hàm trả về Result (hoặc Option), vì nó cần có khả năng trả về lỗi.

Thử thách

Dễ

Bạn sẽ nhận hai đầu vào. Đầu vào thứ nhất là một chuỗi cần được phân tích thành số nguyên, còn đầu vào thứ hai là một chuỗi khác cũng cần được phân tích thành số nguyên. Tạo một hàm có tên add_numbers, nhận hai lát cắt chuỗi làm tham số và trả về Result<i32, std::num::ParseIntError>. Bên trong hàm, dùng toán tử ? để phân tích cả hai chuỗi và trả về tổng của chúng được bọc trong Ok. Gọi hàm với hai đầu vào và xử lý kết quả bằng match.

Yêu cầu:

Đọc đầu vào thứ nhất (số thứ nhất dưới dạng chuỗi) và loại bỏ khoảng trắng
Đọc đầu vào thứ hai (số thứ hai dưới dạng chuỗi) và loại bỏ khoảng trắng
Tạo một hàm add_numbers nhận hai tham số &str và trả về Result<i32, std::num::ParseIntError>
Bên trong hàm, dùng toán tử ? để phân tích chuỗi thứ nhất thành i32
Dùng toán tử ? để phân tích chuỗi thứ hai thành i32
Trả về Ok(sum), trong đó sum là tổng của hai số đã phân tích
Gọi hàm với hai chuỗi đầu vào
Dùng biểu thức match để xử lý Result
Trong nhánh Ok(value), in ra: Sum: [value]
Trong nhánh Err(_), in ra: Parsing error

Đầu vào:

Dòng thứ nhất: Một chuỗi biểu thị một số (ví dụ: 15)
Dòng thứ hai: Một chuỗi biểu thị một số khác (ví dụ: 27)

Đầu ra:

Nếu cả hai chuỗi đều là số hợp lệ: Sum: [result]
Nếu không thể phân tích một trong hai chuỗi: Parsing error

Tự làm thử

use std::io;

// TODO: Tạo hàm (function) add_numbers tại đây
// Hàm nhận hai tham số &str và trả về Result<i32, std::num::ParseIntError>


fn main() {
    // Đọc đầu vào thứ nhất
    let mut input1 = String::new();
    io::stdin().read_line(&mut input1).expect("Failed to read line");
    let input1 = input1.trim();
    
    // Đọc đầu vào thứ hai
    let mut input2 = String::new();
    io::stdin().read_line(&mut input2).expect("Failed to read line");
    let input2 = input2.trim();
    
    // TODO: Gọi add_numbers với input1 và input2
    // TODO: Dùng match để xử lý Result
    // In ra "Sum: [value]" khi nhận Ok(value)
    // In ra "Parsing error" khi nhận Err(_)
}