Swift의 Protocol Extension: 개념과 사용 방법

Protocol Extension이란?

Protocol Extension은 기존 프로토콜에 메서드와 계산된 속성의 구현을 추가하는 방법입니다. 확장이 없으면 프로토콜은 요구 사항만 정의하고 각 타입이 개별적으로 구현합니다.

Protocol Extension은 코드 중복 문제를 해결합니다. 5개의 구조체가 동일한 프로토콜을 채택하고 동일한 메서드를 구현하는 경우, 확장이 기본 구현을 한 번 제공합니다.

Swift Evolution proposal SE-0186에 따르면 protocol extensions는 POP의 성공을 예고한 주요 기능 중 하나입니다. 기본 클래스를 만들지 않고 단일 책임 원칙을 위반하지 않으면서 공통 동작을 추가할 수 있습니다.

Protocol extension 구문

Protocol Extension은 일반 확장과 같이 선언되지만 타입 대신 프로토콜 이름을 사용합니다.

protocol Greetable {
    var name: String { get }
    func greet() -> String
}

extension Greetable {
    func greet() -> String {
        return "Hello, \(name)!"
    }
}

이제 Greetable을 채택하는 모든 타입이 자동으로 greet 구현을 얻습니다.

struct Person: Greetable {
    let name: String
}
// Person이 자동으로 greet()를 가짐

let user = Person(name: "Alice")
print(user.greet()) // "안녕하세요, Alice!"

Protocol Extension에는 계산된 속성을 포함할 수 있지만 저장 속성은 포함할 수 없습니다(프로토콜은 저장소를 정의할 수 없음). 확장을 통해 서브스크립트와 중첩 타입도 추가할 수 있습니다.

기본 메서드 구현

기본 구현은 protocol extension의 주요 용도입니다. 타입은 자체 버전을 제공하여 메서드를 재정의할 수 있습니다.

protocol Loggable {
    func log(message: String)
}

extension Loggable {
    func log(message: String) {
        print("[Default] \(message)")
    }
}

struct ConsoleLogger: Loggable {}
// 기본 구현 사용

struct FileLogger: Loggable {
    func log(message: String) {
        // 사용자 정의 구현이 기본값을 재정의
        writeToFile(message)
    }
}

클래스 상속과의 중요한 차이점: 타입 자체가 프로토콜 메서드를 구현하면 해당 구현이 호출됩니다. 그렇지 않으면 확장의 기본값이 사용됩니다. 이는 정적 디스패치이며 컴파일 시간에 결정됩니다.

Protocol extension의 where 제약 조건

Where 절을 사용하면 추가 조건을 충족하는 타입에만 protocol extension을 제한할 수 있습니다. 이는 특화된 동작을 추가하기 위한 강력한 메커니즘입니다.

protocol Printable {
    var content: String { get }
}

extension Printable where Self: CustomStringConvertible {
    func debugPrint() -> String {
        return "[Printable] \(content)"
    }
}

여기서 debugPrint는 Printable과 CustomStringConvertible을 동시에 구현하는 타입에만 사용할 수 있습니다. Swift 표준 라이브러리는 이 패턴을 광범위하게 사용합니다. 예를 들어, Collection의 확장 where Element가 있습니다.

타입 동등성 제약 조건이 있는 where 절은 특히 유용합니다.

extension Collection where Element == String {
    func commaJoined() -> String {
        return self.joined(separator: ", ")
    }
}

let words = ["Swift", "Kotlin", "Java"]
print(words.commaJoined()) // "Swift, Kotlin, Java"

이 메커니즘은 protocol extension을 선택적으로 만듭니다. commaJoined 메서드는 문자열 컬렉션에만 사용할 수 있고 숫자 컬렉션에는 사용할 수 없습니다. 컴파일러는 제약 조건을 정적으로 확인합니다.

Where 제약 조건은 다음을 확인할 수 있습니다.

• 프로토콜 준수: where Self: Equatable
• 타입 제약: where Element == String
• 조합: where Element: Numeric, Element: Comparable

이것은 protocol extension을 일반적인 프로토콜 구현을 오염시키지 않고 특화된 동작을 추가하기 위한 강력한 메커니즘으로 만듭니다.

Protocol Extensions vs 상속

많은 개발자가 궁금해합니다. protocol extensions은 언제 사용하고 클래스 상속은 언제 사용해야 할까요? 답은 아키텍처 패러다임에 따라 다릅니다.

Apple은 protocol + extension으로 시작하고 공유 상태나 정체성(참조 의미론)이 필요한 경우에만 클래스로 전환할 것을 권장합니다. Protocol Extensions는 상속 대신 구성을 제공하여 더 유연하고 테스트 가능한 접근 방식을 제공합니다.

실제로 protocol extensions는 프로토콜 요구 사항 위에 편리한 래퍼 메서드를 추가하는 데 자주 사용됩니다. 예를 들어, 프로토콜이 자세한 보고서와 함께 validate 메서드를 요구하는 경우 확장은 전체 버전을 기반으로 부울 값을 반환하는 isValid 메서드를 추가할 수 있습니다. 이는 프로토콜 계약을 변경하지 않고 클라이언트 코드를 단순화합니다. 이 패턴을 “파생 API가 있는 기본 구현”이라고 하며 Swift 표준 라이브러리 및 인기 있는 타사 프레임워크에서 널리 사용됩니다. 이는 실제 프로토콜 지향 프로그래밍의 핵심 기술 중 하나이며 유연한 아키텍처의 기초입니다.