[Swift] Extension에 대해서 알아보자.

안녕하세요~

 

오늘은 Extension에 대해서 작성하려고 합니다 : )

 

Extension 이란?

Swift의 extension은 기존의 클래스, 구조체, 열거형 등의 타입에 새로운 기능을 추가할 수 있는 기능입니다.

이를 통해 기존의 타입을 확장하거나, 프로토콜 적용 등을 통해 새로운 기능을 추가할 수 있습니다. 

 

기본 문법

extension을 사용하여 타입을 확장하는 문법은 아래와 같습니다.

 

extension 확장할 타입 {
   // 기능 구현
}

 

위 문법에서 확장할 타입은

기존에 정의된 클래스, 구조체, 열거형 등의 타입을 의미합니다.

 

기능 구현에는 기존 타입에 추가될 프로퍼티, 메서드, 서브스크립트 등의 멤버를 구현할 수 있습니다.

예를 들어, Int 타입에 isEven이라는 프로퍼티를 추가하고 싶다면 아래와 같이 extension을 사용할 수 있습니다.

 

extension Int {
  var isEven: Bool {
    return self % 2 == 0
  }
}

 

위 코드에서

앞서 문법에서 설명했던

확장할 타입은 Int가 되는 것이고 isEven이라는 새로운 프로퍼티가 기능구현이라고 생각하시면 됩니다.

 

Int 타입의 인스턴스에서 isEven 프로퍼티를 사용하는 방법은 아래와 같습니다.

 

let number = 10
print(number.isEven) // true

 

프로토콜 적용

extension을 사용하여 기존 타입에 새로운 프로토콜을 적용할 수 있습니다.

이를 통해 기존 타입이 새로운 프로토콜을 준수하도록 만들 수 있습니다.

 

예를 들어, CustomStringConvertible 프로토콜을 적용하여

Person이라는 구조체의 인스턴스를 문자열로 변환할 수 있도록 만들 수 있습니다.

 

struct Person {
  let name: String
  let age: Int
}

extension Person: CustomStringConvertible {

   var description: String {
     return "\(name)(은/는) \(age)살입니다."
   }
}

 

Person 구조체에 CustomStringConvertible 프로토콜을 적용하고

description 프로퍼티를 구현하여 문자열로 변환할 수 있도록 만들었습니다.

 

이제 Person 구조체의 인스턴스를 print 함수에 전달하여 다음과 같이 출력되는 걸 확인할 수 있습니다. 

 

let person = Person(name: "경푸", age: 20)
print(person) // 경푸(은/는) 20살입니다.

 

제네릭 타입 확장

extension을 사용하여 제네릭 타입에 대해서도 확장할 수 있습니다.

예를 들어, Array 타입에 lastIndex(of:) 메서드를 추가하고 싶다면 아래와 같이 extension을 사용할 수 있습니다.

 

extension Array where Element: Equatable {

  func lastIndex(of element: Element) -> Int? {
    for (index, item) in self.enumerated().reversed() {
      if item == element {
        return index
      }
    }
    return nil
  }
  
}

 

위 코드에서는 Array 타입에 where 절을 사용하여 Element가 Equatable 프로토콜을 준수하는 경우에만

lastIndex(of:) 메서드를 사용할 수 있도록 제한했습니다.

 

Array 타입의 인스턴스에서 lastIndex(of:) 메서드를 사용하면 다음과 같이 동작합니다.

 

let numbers = [1, 3, 5, 3, 7, 9, 3]
print(numbers.lastIndex(of: 3)) // Optional(6)

 

 

초기화 구문 추가

extension을 사용하여 초기화 구문을 추가할 수도 있습니다.

이를 통해 기존 타입에 다양한 초기화 구문을 추가하여 사용할 수 있습니다.

 

예를 들어, Date 타입에 String 값을 입력받아

Date 인스턴스를 생성하는 초기화 구문을 만들어 보면 다음과 같이 작성할 수 있습니다.

extension Date {

  init?(fromString dateString: String, format: String = "yyyy-MM-dd") {
  
    let dateFormatter = DateFormatter()
    dateFormatter.dateFormat = format
    
    if let date = dateFormatter.date(from: dateString) {
      self = date
    } else {
      return nil
    }
  }
}

 

extension을 사용하여 Date 타입에 초기화 구문을 추가하였고

String 값과 format 값을 입력받아 Date 인스턴스를 생성하며, 초기화에 실패한 경우 nil을 반환게 만들었습니다.

 

Date 타입의 인스턴스를 생성할 때 fromString 메서드를 사용하면 다음과 같이 사용될 수 있습니다.

 

let dateString = "2022-04-01"
if let date = Date(fromString: dateString) {
  print(date) // 2022-04-01 00:00:00 +0000
}

if let date = Date(fromString: dateString, format: "yyyy/MM/dd") {
  print(date) // 2022-04-01 00:00:00 +0000
}

 

---

 

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

그럼 이만  👋🏻 👋🏻 👋🏻