📁 Extension

extension은 기존 타입에 기능을 추가한다. 여기서 기존의 타입은 구조체, 열거형, 클래스를 말한다. 확장은 타입에 새로운 기능을 추가할 수는 있지만, 기존에 존재하는 기능을 재정의할 수는 없다! 그렇다고 기능을 추가하기 위해 타입의 구현된 모든 소스코드를 알 수는 없다. extension은 타입만 알고 있다면 그 타입의 기능을 확장할 수 있도록 해준다.

Extension 구문

extension 키워드를 사용하여 확장을 선언한다.

extension SomeType {

}

struct Person {
	let name: String
    var country: String
    let isMale: Bool
}

extension Person {
	let age: Int
    let bloodType: String
    var height: Float
}

확장으로 기존에 존재하는 타입이 프로토콜을 채택할 수 있다. 이때 프로토콜을 다중 채택할 수 있다.

extension SomeType: SomeProtocol, AnotherProtocol {
    
}

extension 구문으로는 다양한 것들이 가능하다. 기존 타입에 새로운 연산 프로퍼티(computed property)를 추가, 새로운 인스턴스/타입 메서드 추가, 초기화 구문 추가, 프로토콜 채택, 서브스크립트 추가, 중첩타입 추가가 가능하다.

📝 연산 프로퍼티 추가

확장은 기존 타입에 타입 연산 프로퍼티와 인스턴스 연산 프로퍼티를 추가할 수 있다. 연산 프로퍼티라고 특별히 지정한것 처럼 저장 프로퍼티는 추가가 불가능하다. 또한 기존 타입의 저장된 프로퍼티에 기본값 설정 시 컴파일 에러가 발생한다.

extension String {
    var lastTwoWord: String {
        let suffixString = self.suffix(2)
        return String(suffixString)
    }
}

let str = "Hello"
print(str.lastTwoWord)	//"lo"

위 코드는 String 타입에 대해 연산 프로퍼티를 추가하였다. String 타입의 인스턴스의 마지막 두 글자를 반환해주는 연산 프로퍼티이다. 이렇게 확장해주면 모든 String 타입에서 해당 프로퍼티를 사용 가능하다. 이 프로퍼티는 읽기전용 연산프로퍼티이므로 get키워드 없이 표현이 가능하다.

🧐 인스턴스 연산 프로퍼티와 타입 연산 프로퍼티 : swift에서 프로퍼티를 정의하는 두가지 방법이다.

• 타입 연산 프로퍼티는 타입 자체와 관련이 있는 값을 나타내며 static키워드를 사용하여 타입 내부에 선언한다. 그럼 해당 타입의 모든 인스턴스가 공유하게 된다.
• 인스턴스 연산 프로퍼티는 특정 인스턴스와 관련이 있는 값을 나타내며 var키워드를 사용하여 타입 내부에 선언된다. 인스턴스가 생성될 때 마다 고유한 값을 가진다.

📝 초기화 구문 추가

extension으로 기존 타입에 새로운 편의 초기화 구문을 추가할 수 있다. 그러나 새로운 지정 초기화 구문이나 초기화 해제 구문(디이니셜라이저)을 추가할 수는 없다. 이들은 항상 기존 클래스의 구현부에 작성되어야 한다. 값 타입(구조체,열거형)은 상관없다.

extension Double {
    init(fromString str: String) {
        self = Double(str) ?? 0.0
    }
}

let stringValue = "3.14"
let doubleValue = Double(fromString: stringValue) //3.14

이 코드는 Double 타입에 대한 확장을 정의하고 있다. 만약 초기화 구문으로 받아들인 문자열이 Double로 변환할 수 있다면 해당 값으로 Double 인스턴스를 초기화하고 아니면 0.0을 사용한다.

🧐 지정초기화와 편의초기화 : 인스턴스를 생성하고 그 인스턴스의 프로퍼티를 초기화하는 두가지 유형

• 지정초기화 : 클래스나 구조체의 모든 저장 프로퍼티를 초기화하는 구문으로 반드시 하나 이상의 지정 초기화 구문이 있어야 한다! (흔히 초기화 구문이라고 부르는 블록)
• 편의초기화 : 초기화 작업을 더 편리하게 만들기 위해 지정 초기화를 간단하게 호출할 수 잇는 보조 초기화 작업을 담당하는 구문으로 작성하지 않아도 된다. convenience키워드를 사용한다. 보통 매개변수의 일부 만을 받아 지정 초기화를 호출하는 방식으로 동작한다.

convenience init(name: String) {
        self.init(name: name, age: 0)
    }

📝 메서드 추가

extension으로 기존 타입에 새로운 인스턴스 메서드와 타입 메서드를 추가할 수 있다.

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

3.repetitions{
	print("hello")
}
//hello
//hello
//hello

위의 코드는 Int타입으로 repetition이라는 새로운 인스턴스 메서드를 추가한 예시이다. 이 메서드는 파라미터가 없고 반환값이 없는 함수타입을 인수로 가진다.
self 또는 프로퍼티를 변경하는 구조체와 열거형 메서드는 기존의 변경 메서드와 같이 mutating 으로 인스턴스 메서드를 표시해야 한다. 아래의 예시는 원래 제공되는 메서드인 square라는 메서드를 추가로 변경하는 코드이다.

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

📝 서브스크립트 추가

extension으로 기존 타입에 새로운 서브스크립트를 추가할 수 있다.

struct Matrix {
    private var data: [[Int]]
    
    init(rows: Int, columns: Int) {
        data = Array(repeating: Array(repeating: 0, count: columns), count: rows)
    }
}

extension Matrix {
    subscript(row: Int, column: Int) -> Int {
        get {
            return data[row][column]
        }
        set {
            data[row][column] = newValue
        }
    }
}

var matrix = Matrix(rows: 5, columns: 5)
matrix[0, 0] = 1
matrix[1, 1] = 3
print(matrix)
//Matrix(data: [[1, 0, 0, 0, 0], [0, 3, 0, 0, 0], [0, 0, 0, 0, 0], [0, 0, 0, 0, 0], [0, 0, 0, 0, 0]])

📝 프로토콜 채택과 구현

extension을 이용해서 프로토콜을 채택하여 프로토콜의 기본값을 설정해줄 수 있다. 프로토콜은 틀을 만드는 코드로 실제로 기능이 구현되지 않은 프로퍼티와 메서드를 작성하였다. extension을 이용해서 프로토콜에 기능을 구현할 수 있다.

protocol Printable {
    func printDescription()
}

extension Printable {
	func printDescription(){
    	print("printing...")
     }
}

struct MyPrint: Printable{}

let myprinter = MyPrint()
myprinter.printDescription() //printing...