Swift 공식 문서 정리 - (21) Extensions

Extensions (익스텐션)

익스텐션을 이용해 클래스, 구조체, 열거형 혹은 프로토콜 타입에 기능을 추가할 수 있음

원본 코드를 몰라도 그 타입에 대한 기능을 확장할 수 있음 → 이게 매우 큰 장점인듯.

익스텐션을 이용해 할 수 있는 것들

• 계산된 인스턴스,타입 프로퍼티 추가
• 인스턴스,타입 메소드 추가
• 새로운 이니셜라이저 제공
• 서브스크립트 정의
• 중첩 타입의 선언과 사용
• 특정 프로토콜을 따르는 타입 만들기

익스텐션

은 타입에 새 기능을 추가할 수 있지만 오버라이드는 불가능

익스텐션 문법

// extension 키워드 사용 (기본)
extension SomeType {
	// new
}

// 프토토콜 채택
extension SomeType: SomeProtocol, AnotherProtocol {
    // implementation of protocol requirements goes here
}

하나의 익스텐션에서 현재 존재 하는 타입에 한개 이상의 프로토콜을 따르도록 확장할 수 있음

하나의 익스텐션은 제네릭(generic) 타입으로 확장 하는데 사용할 수 있음

구체적인 내용은 프로토콜과 제네릭 챕터에서 다룸

익스텐션을 정의하여 새 기능을 추가하면, 익스텐션을 정의하기 이전에 생성한 인스턴스를 포함한

존재하는 모든 해당 타입의 인스턴스에서 사용 가능 함.

계산된 프로퍼티

익스텐션을 이용해 존재하는 타입에 계산된 인스턴스,타입 프로퍼티를 추가할 수 있음

예제는 스위프트의 기본타입인 Double에 5개의 계산된 인스턴스 프로퍼티를 추가하는 예제임

extension Double {
    var km: Double { return self 1_000.0 }
    var m: Double { return self }
    var cm: Double { return self / 100.0 }
    var mm: Double { return self / 1_000.0 }
    var ft: Double { return self / 3.28084 }
}
let oneInch = 25.4.mm
print("One inch is \\(oneInch) meters")
// Prints "One inch is 0.0254 meters"
let threeFeet = 3.ft
print("Three feet is \\(threeFeet) meters")
// Prints "Three feet is 0.914399970739201 meters"

이 프로퍼티 들은 읽기 전용 계산된 프로퍼티이기 때문에 get 을 붙이지 않았음

단위 변환 값이 Double 타입이기 때문에 각각의 변환 값이 연산도 가능함

let aMarathon = 42.km + 195.m
print("A marathon is \\(aMarathon) meters long")
// Prints "A marathon is 42195.0 meters long"

익스텐션은 새 계산된 프로퍼티 값을 추가할 수 있지만, 새로운 저장된 프로퍼티나 프로퍼티 옵저버를 추가할 수는 없음

이니셜라이저

익스텐션을 이용해 새로운 이니셜라이저를 추가 가능함

원래 구현에서 포함하지 않는 초기화 정보를 추가할 수 있음

익스텐션은 클래스에 새로운 편리한 이니셜라이저를 추가할 수 있지만, 지정된 이니셜라이저나 디이니셜라이저를 추가할 수는 없음

지정된 이니셜라이저는 항상 반드시 오리지널 클래스의 구현에서 작성해야함

만약 익스텐션을 값타입(열거형, 구조체)에 이니셜라이저를 추가하는데 사용하고, 그 값타입이 모든 프로퍼티에 대해 기본 값을 제공하고 커스텀 이니셜라이저를 정의하지 않았다면, 익스텐션에서 기본 이니셜라이저와 멤버쪽 이니셜라이저를 익스텐션에서 호출할 수 있음

만약 값타입의 오리지널 구현의 부분으로 이니셜라이저 코드를 작성했다면 그것은 제외함

만약 다른 모듈에 선언돼 있는 구조체에 이니셜라이저를 추가하는 익스텐션을 사용하면

새로운 이니셜라이저는 모듈에 정의된 이니셜라이저를 호출하기 전까지 self에 접근할 수 없음

struct Size {
    var width = 0.0, height = 0.0
}
struct Point {
    var x = 0.0, y = 0.0
}
struct Rect {
    var origin = Point()
    var size = Size()
}

let defaultRect = Rect()
let memberwiseRect = Rect(origin: Point(x: 2.0, y: 2.0),
   size: Size(width: 5.0, height: 5.0))

// 기본 이니셜라이저와 멤버쪽 이니셜라이저를 이용해 초기화 한 예제

이제 익스텐션으로 추가 이니셜라이저를 정의한 예제를 보자

extension Rect {
    init(center: Point, size: Size) {
        let originX = center.x - (size.width / 2)
        let originY = center.y - (size.height / 2)
        self.init(origin: Point(x: originX, y: originY), size: size)
    }
}

let centerRect = Rect(center: Point(x: 4.0, y: 4.0),
                      size: Size(width: 3.0, height: 3.0))
// centerRect's origin is (2.5, 2.5) and its size is (3.0, 3.0)

익스텐션에서 이니셜라이저를 제공할 때 각 인스턴스가 이니셜라이저가 한번 완료되면 완전히 초기화 되도록 확실히 해야함

메소드

익스텐션을 이용해 존재하는 타입에 인스턴스 메소드나 타입 메소드를 추가할 수 있음

다음 예제는 Int 타입에 repetitions 라는 인스턴스 메소드를 추가한 예제임

extension Int {
    func repetitions(task: () -> Void) {
        for _ in 0..<self {
            task()
        }
    }
}

repetitions(task:) 메소드는 () → Void 타입의 하나의 인자를 받고 파라미터와 반환 값이 없는 함수임

3.repetitions {
    print("Hello!")
}
// Hello!
// Hello!
// Hello!

변경 가능한 인스턴스 메소드 (Mutating Instance Methods)

익스텐션에서 추가된 인스턴스 메소드는 인스턴스 자신을 변경할 수 있음

구조체와 열거형 메소드 중 자기자신을 변경하는 메소드는 원본 구현의 mutating 메소드와 같이

반드시 mutating 으로 선언 되어야 함

extension Int {
    mutating func square() {
        self = self * self
    }
}
var someInt = 3
someInt.square()
// someInt is now 9

서브스크립트

익스텐션을 이용해 존재하는 타입에 새로운 서브스크립트를 추가할 수 있음

아래 예제에서 서브스크립트 [n] 은 숫자의 오른쪽에서부터 n번째 위치하는 정수를 반환 함.

extension Int {
    subscript(digitIndex: Int) -> Int {
        var decimalBase = 1
        for _ in 0..<digitIndex {
            decimalBase *= 10
        }
        return (self / decimalBase) % 10
        // 10 * n번째 수로 현재 수를 나눈 것의 나머지
      // 1인 경우 746381295 % 10 -> 5가 나머지
      // 2인 경우 746381295 % 10 -> 9가 나머지
    }
}
746381295[0]
// returns 5
746381295[1]
// returns 9
746381295[2]
// returns 2
746381295[8]
// returns 7

만약 Int 값에서 요청한 값이 처리할 수 있는 자릿 수를 넘어가면 서브스크립트 구현에서 0을 반환함.

중첩 타입(Nested Types)

익스텐션을 이용해 존재하는 클래스, 구조체, 열거형에 중첩타입 추가 가능

extension Int {
    enum Kind {
        case negative, zero, positive
    }
    var kind: Kind {
        switch self {
        case 0:
            return .zero
        case let x where x > 0:
            return .positive
        default:
            return .negative
        }
    }
}