[Swift] Enum(열거형)에 대해서

안녕하세요~

 

오늘은 Enum(열거형)에 대해서 정리하려고 합니다.

 

Enum이란?

Enum은 Enumeration(열거형)의 줄임말로, 서로 연관된 값들의 집합을 정의할 수 있는 자료형입니다.

Swift에서는 열거형이 일반적으로 Int, String 등의 원시값(Raw Value) 또는 연관값(Associated Value)을 가질 수 있습니다. 이를 통해 코드의 가독성을 높이고, 정확성을 유지하면서 간결한 코드를 작성할 수 있습니다.

 

Enum을 사용하지 않는 경우

Enum을 사용하지 않으면, 상태를 변수로 선언하고 이를 조건문 등을 통해 비교하여 사용할 수 있습니다. 예를 들어, 다음과 같은 코드를 생각해 볼 수 있습니다.

 

let stateA = "State A"
let stateB = "State B"
let currentState = stateA

if currentState == stateA {
  print("현재 상태는 A입니다.")
} else if currentState == stateB {
  print("현재 상태는 B입니다.")
} else {
  print("상태를 알 수 없습니다.")
}

 

이렇게 상태를 String으로 선언하고, 조건문으로 비교하여 사용할 수 있습니다.

하지만 이렇게 사용하면 코드가 길어지고 가독성이 떨어지는 등의 문제가 있을 수 있습니다.

 

Enum을 사용하는 경우

Enum을 사용하면, 상태를 Enum으로 정의하여 이를 변수로 선언하고 사용할 수 있습니다. 예를 들어, 다음과 같은 코드를 생각해 볼 수 있습니다.

 

enum State {
  case A
  case B
}

let currentState: State = .A

switch currentState {
  case .A:
    print("현재 상태는 A입니다.")
  case .B:
    print("현재 상태는 B입니다.")
}

 

이렇게 Enum을 사용하면, 변수의 값이 상태가 됩니다. 그리고 switch 문으로 변수의 값에 따라 다른 동작을 수행할 수 있습니다. 이는 가독성이 좋고, 코드가 간결해지는 등의 장점이 있습니다.

 

Enum을 활용하는 방법

Enum에서는 rawValue, caseIterable, 연관값 등의 기능을 사용할 수 있습니다.

1. rawValue(원시값)

Enum에서는 rawValue를 사용하여, Enum의 값을 다른 값으로 변환할 수 있습니다. rawValue는 Enum에서 각 case의 값으로, 정수형, 문자열 등을 사용할 수 있습니다. 예를 들어, 다음과 같이 Enum에서 rawValue를 사용할 수 있습니다.

 

enum State: Int {
  case A = 0
  case B = 1
}

let currentState = State(rawValue: 1)

switch currentState {
  case .A:
    print("현재 상태는 A입니다.")
  case .B:
    print("현재 상태는 B입니다.")
  case nil:
    print("상태를 알 수 없습니다.")
}

enum Color: String {
   case red = "red"
   case blue = "blue"
}

let redColor: Color = .red
print(redColor.rawValue) // "red"

 

2. caseIterable

caseIterable은 Enum에서 case를 모두 나열할 수 있는 기능입니다.

이를 사용하면, 다음과 같이 모든 case를 출력할 수 있습니다.

 

enum State: String, CaseIterable {
  case A = "A"
  case B = "B"
}

for state in State.allCases {
  print(state.rawValue)
}
// "A"
// "B"

 

3. 연관값

Enum의 각 case에는 연관값을 설정할 수 있습니다. 예를 들어, 다음과 같이 연관값을 설정할 수 있습니다.

 

enum State {
  case A(value: Int)
  case B(value: String)
}

let currentState = State.A(value: 10)

switch currentState {
  case .A(let value):
    print("현재 상태는 A이며, 값은 \(value)입니다.")
  case .B(let value):
    print("현재 상태는 B이며, 값은 \(value)입니다.")
}

 

4. switch를 이용하기

switch문을 이용하면 Enum의 다양한 기능을 활용할 수 있습니다.

예를 들어, Enum에서의 모든 case를 출력하거나, 연관값이 있는 case만 출력하는 등의 기능을 구현할 수 있습니다.

앞선 예시들에서도 지속적으로 switch 구문을 사용한 것을 알 수 있습니다.

 

enum State {
  case A(value: Int)
  case B(value: String)
  case C
}

let currentState = State.A(value: 10)

switch currentState {
  case .A(let value):
    print("현재 상태는 A이며, 값은 \(value)입니다.")
  case .B(let value):
    print("현재 상태는 B이며, 값은 \(value)입니다.")
  case .C:
    print("현재 상태는 C입니다.")
}

for state in State.allCases {
  print(state)
}

for case let .A(value) in State.allCases {
  print("A 상태의 값은 \(value)입니다.")
}

 

---

 

 

오늘은 여기까지 정리해 보겠습니다.

그럼 이만 👋🏻 👋🏻 👋🏻