[Swift] 열거형

📌 열거형

Swift의 열거형은 다른 언어에서의 열거형과는 다름

• 유사한 종류의 여러 값을 한 곳에 모아서 정의한 것 예) 요일, 월, 계절 등
• 대문자 카멜케이스를 사용하여 이름을 정의한다.
• 각 case는 소문자 카멜케이스로 정의한다.
• 각 case는 그 자체가 고유의 값이다. (각 case에 자동으로 정수값이 할당되지 않음)
• 각 case는 한 줄에 개별로도, 한 줄에 여러개도 정의할 수 있다.

enum Weekday {
	case mon
    case tue
    case wed
    case thu, fri, sat, sun
}

열거형 사용

// 열거형 타입과 케이스를 모두 사용 가능
var day: Weekday = Weekday.mon

// 타입이 명확하다면 .케이스 처럼 표현해도 무방
day = .tue

print(day) // tue

// switch의 비교값에 열거형 타입이 위치할 때
// 모든 열거형 케이스를 포함한다면 default를 작성할 필요 X
switch day {
case .mon, .tue, .wed, .thu:
    print("평일입니다")
case Weekday.fri:
    print("불금 파티!!")
case .sat, .sun:
    print("신나는 주말!!")
}

원시값

다른 언어에서처럼 case에 값을 할당할 수도 있다.

Int 값을 할당하는 경우

• rawValue는 case별로 각각 다른 값을 가져야 한다.
• 자동으로 1이 증가된 값이 할당된다. (아래 코드의 경우 peach에 자동으로 2가 할당된다. 또한 apple, grape에 값을 대입하지 않아도 자동으로 0, 1이 할당된다.)
• 단, 이 상태에서는 case mango = 0 을 추가할 수 없다. mango와 apple의 원시값이 같을 수 없기 때문이다.
• rawValue 값을 꺼내서 쓸 때는 case에서 꺼내서 쓰면 된다. Fruit.peach.rawValue 요렇게!

enum Fruit: Int {
    case apple = 0
    case grape = 1
    case peach
    // case mango = 0
}

print("Fruit.peach.rawValue == \(Fruit.peach.rawValue)")
// Fruit.peach.rawValue == 2

String 값을 할당하는 경우

정수 타입 뿐만 아니라 Hashable 프로토콜을 따르는 모든 타입이 원시값의 타입으로 지정될 수 있다.

• 마찬가지로 rawValue 값을 꺼내서 쓸 때는 case에서 꺼내면 된다.
• 자동으로 1이 증가된 값이 할당되던 Int타입 열거형과는 달리 String타입은 1 증가시킬 수가 없다. 따라서 값을 대입하지 않은 case의 경우 rawValue를 출력하면 그냥 case 이름이 출력된다.

enum School: String {
    case elementary = "초등"
    case middle = "중등"
    case high = "고등"
    case university
}

print("School.middle.rawValue == \(School.middle.rawValue)")
// School.middle.rawValue == 중등

print("School.university.rawValue == \(School.university.rawValue)")
// School.university.rawValue == university

원시값 정보를 안다면 원시값을 통해 열거형 변수/상수를 생성할 수 있다.

• 현재 Fruit 열거형의 case에 할당된 정수값들은 0, 1, 2이지만 3 이상의 값으로 초기화를 시도할 수도 있다. 때문에 원시값을 통해 초기화한 인스턴스는 옵셔널 타입으로 지정해주어야 한다.

// let apple: Fruit = Fruit(rawValue: 0) → 에러
let apple: Fruit? = Fruit(rawValue: 0)

• if-let 구문을 사용하면 rawValue에 해당하는 case를 곧바로 사용할 수 있다.

if let name: School = School(rawValue: "고등") {
    print("\(name.rawValue)학생입니다")
} else {
    print("학생이 아닙니다")
}

• 열거형에 method를 추가할 수 있다.

enum Month {
    case dec, jan, feb
    case mar, apr, may
    case jun, jul, aug
    case sep, oct, nov
    
    func printMessage() {
        switch self {
        case .mar, .apr, .may:
            print("따스한 봄~")
        case .jun, .jul, .aug:
            print("여름 더워요~")
        case .sep, .oct, .nov:
            print("가을은 독서의 계절!")
        case .dec, .jan, .feb:
            print("추운 겨울입니다")
        }
    }
}

Month.mar.printMessage()

야곰의 스위프트 기본 문법 강좌를 수강하며 작성한 내용입니다.