[Swift] CaseIterable

코드를 짜다보면 enum 타입으로 어떤 종류의 케이스를 나열하는 경우가 많죠~?

예를 들면 이런식으로요!

enum Country {
    case korea
    case swiss
    case france
    case england
}

근데 이렇게 케이스는 작성해놨는데, 우리는 저 케이스에 뭐가 뭐가 있는지는 몰라서

enum Country {
    case korea
    case swiss
    case france
    case england
    
    static var allCountry: [Country] {
        return [.korea, .swiss, .france, .england]
    }
}

이렇게 property를 임의로 작성해줘야 했죠 😭

근데 Swift 4.2부터는 CaseIterable이라는 프로토콜을 제공해주면서, 저런 짓(?)을 할 필요가 없습니다!! 

그럼 CaseIterable 이게 뭔지~~~??

 

"어떤 컬렉션의 모든 값들을 제공하는 타입"의 프로토콜 이라고 하네요!! 

즉, 위에서 예를 든 Country라는 컬렉션에서 모든 값(나라들)을 제공해주는 프로토콜 이래요.

그럼 아까 만들었던 Country가 이 프로토콜을 채택하도록 할게요!

enum Country: CaseIterable {
    case korea
    case swiss
    case france
    case england
}

이렇게 CaseIterable 프로토콜을 채택하면, 우리는 allCases라는 프로퍼티를 통해 모든 케이스의 값을 가져올 수 있어요!

모든 케이스의 값을 가져올 수 있게 되서, 코드를 짜다가 행여나 케이스를 누락할 일이 없어졌어요 👍

오호~ Country타입의 배열이네요!!

let countries = Country.allCases
print(countries)

한번 출력해보니, 썩 보기좋은 포맷은 아니지만 Country가 가지고 있는 모든 케이스를 가지고 있네요!

 

for country in  countries {
    if country == .korea {
        print(country) // "korea"
    }
}

Array타입이니까 이렇게 바로 for문을 통해서 뭔가 로직을 넣어서 활용도 가능해요!

 

조금 이쁘게 String타입의 배열로 변환도 가능하구요~~

let list = countries.map { "\($0)" }
print(list) // ["korea", "swiss", "france", "england"]

 

extension을 통해 다른 property를 깔끔하게 정의해줄 수도 있어요~~

extension Country {
    var continent: String {
        switch self {
        case .korea:
            return "Asia"
        case .swiss, .england, .france:
            return "Europe"
        }
    }
}

 

추가로 정의해준 것도 for문을 통해 한번 탐색도 해보고요!

for country in  countries {
    if country.continent == "Europe" {
        print(country)
    }
}
/*
swiss
france
england
*/

 

 

사용은 이렇게 간단하지만 효과는 좋은 👍 CaseIterable에 대해서 간략하게 알아보았습니다~!!

오늘도 누군가에게 도움이 되었길 바래요 🙏