100 Exercises To Learn Rust - 5.1 enum 풀기

문제

// TODO: `Status`를 `Ticket::status`의 타입으로 사용하세요.
// 다른 모든 메서드의 시그니처와 구현을 필요에 따라 조정하세요.

#[derive(Debug, PartialEq)]
// `derive`는 재귀적입니다: 모든 필드가 PartialEq를 구현해야만 PartialEq를 derive할 수 있습니다.
// `Debug`도 마찬가지입니다.
struct Ticket {
    title: String,
    description: String,
    status: String,
}

enum Status {
    // TODO: add the missing variants
}

impl Ticket {
    pub fn new(title: String, description: String, status: String) -> Ticket {
        if title.is_empty() {
            panic!("Title cannot be empty");
        }
        if title.len() > 50 {
            panic!("Title cannot be longer than 50 bytes");
        }
        if description.is_empty() {
            panic!("Description cannot be empty");
        }
        if description.len() > 500 {
            panic!("Description cannot be longer than 500 bytes");
        }
        if status != "To-Do" && status != "In Progress" && status != "Done" {
            panic!("Only `To-Do`, `In Progress`, and `Done` statuses are allowed");
        }

        Ticket {
            title,
            description,
            status,
        }
    }

    pub fn title(&self) -> &String {
        &self.title
    }

    pub fn description(&self) -> &String {
        &self.description
    }

    pub fn status(&self) -> &String {
        &self.status
    }
}

풀이

열거형(enum) 정의

struct Ticket {
    title: String,
    description: String,
    status: Status,
}

#[derive(Debug, PartialEq, Copy, Clone)]
enum Status {
    // TODO: add the missing variants
    ToDo,
    InProgress,
    Done,
}

 

Ticket 구조체의 status 필드를 String에서 Status 열거형으로 수정한다.

 

Status 열거형의 배리언트(variant)는 테스트를 통과할 수 있게 ToDo, InProgress, Done으로 설정한다.

 

테스트를 통과하기 위해 Debug, PartialEq, Copy, Clone 트레이트를 derive한다.

Ticket 구현 수정

impl Ticket {
    pub fn new(title: String, description: String, status: Status) -> Ticket {
        //...
        if status != Status::ToDo && status != Status::InProgress && status != Status::Done {
            panic!("Only `To-Do`, `In Progress`, and `Done` statuses are allowed");
        }

        Ticket {
            title,
            description,
            status,
        }
    }

    //...

    pub fn status(&self) -> &Status {
        &self.status
    }
}

 

status를 검사할 때 문자열 리터럴이 아닌 열거형의 배리언트를 비교하도록 수정한다.

 

또한 status() 메서드의 반환 타입도 &Status로 변경한다.