[iOS - swift] enum에 공통 변수가 필요한 경우 리펙토링 방법 (wrapping)

enum에 공통 변수가 필요한 경우

• ColorType이라는 enum이 있고 color타입들을 분류하는 case들을 나타내는 경우
  • 여기에 defaultColor도 다 갖고 있게 하고싶은 경우?

enum ColorType {
    case blue
    case black
    case green
}

ex) 단순히 associated type으로 설정

• 선언하는 곳, 사용하는 곳 모두 일일이 다 써줘야 하므로 복잡한 코드가 됨

enum ColorType {
    case blue(defaultColor: UIColor)
    case black(defaultColor: UIColor)
    case green(defaultColor: UIColor)
}

(사용하는쪽)

let colorType = ColorType.blue(defaultColor: .white)

switch colorType {
case .blue(let defaultColor):
    <#code#>
case .black(let defaultColor):
    <#code#>
case .green(let defaultColor):
    <#code#>
}

struct로 wrapping하여 리펙토링하기

• associated type없이 enum 타입을 가지고 있는 프로퍼티 하나와, 공통적으로 가지고 있는 프로퍼티를 선언

struct ColorModel {
    let defaultColor: UIColor
    let color: ColorType
}

enum ColorType {
    case blue
    case black
    case green
}

• 이렇게 리펙토링하면 사용하는쪽에서도 일일이 defaultColor를 선언해주지 않아도 되므로 더욱 깔끔한 코드 유지가 가능

사용하는쪽)

let colorModel = ColorModel(defaultColor: .white, color: .blue)

switch colorModel.color {
case .blue:
    <#code#>
case .black:
    <#code#>
case .green:
    <#code#>
}