[Swift] Extension (익스텐션)

Extension은 클래스, 구조체, 열거형, 프로토콜에 새로운 기능을 추가하여 형식을 확장하는 것입니다. 

 

- 계산 인스턴스 속성과 계산 형식 속성을 추가할 수 있습니다. 

- 인스턴스 메소드와 형식 메소드를 정의할 수 있습니다. 

- 새로운 이니셜라이저를 제공할 수 있습니다. 

- 서브스크립트를 정의할 수 있습니다. 

- 새로운 nested types를 정의하고 사용할 수 있습니다. 

- 이미 정의된 형식을 프토토콜로 만들 수 있습니다.

 

extension SomeType {
	//new functionality to add to SomeType goes here
}
extension SomeType: SomeProtocol, AnotherProtocol {
	// implementation of protocol requirements goes here
}

 

Computed Properties

Extension은 존재하고 있는 형식에 계산 속성을 추가할 수 있습니다.

Swift의 Double은 구조체로 구현되어 있기 때문에 extension을 통해 새로운 계산 속성을 추가할 수 있습니다. 

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.4mm
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"

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

Extension은 계산 속성은 추가할 수 있지만 저장 속성은 추가할 수 없고, 존재하는 속성에 속성감시자를 추가할 수는 없습니다. 

 

 

Initializers

존재하는 형식의 이니셜라이저를 추가할 수 있습니다. 

이니셜라이저를 추가하면 기본으로 제공되는 이니셜라이저와 함께 사용할 수 있습니다. 

클래스에 Convenience 이니셜라이저는 추가할 수 있지만, 새로운 designated 이니셜라이저나 deinitializers는 추가할 수 없습니다. 

 

[예제1]

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)

 

Methods

존재하는 형식에 새로운 인스턴스 메소드와 형식 메소드를 추가할 수 있습니다. 

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

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

 

[Mutating Instance Methods]

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

 

Subscripts

- 123456789[0]    returns 9
- 123456789[1]     returns 8

extension Int {
	subscript(digitIndex: Int) -> Int {
    	var decimalBase = 1
        for _ in 0..<digitIndex {
        	decimalBase *= 10
        }
        return (self / decimalBase) % 10
    }
}
746381295[0]
// returns 5
746381295[1]
// returns 9

746381295[9]
// returns 0, as if you had requested:
0746381295[9]

 

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
        }
    }
}
                

이제 어떠한 Int 형식에서도 nested enumeration을 사용할 수 있습니다. 

func printIntegerKinds(_ numbers: [Int]) {
	for number in numbers {
    	switch number.kind {
        case .negative:
        	print("- ", terminator: "")
        case .zero:
        	print("0 ", terminator: "")
        case .positive:
        	print("+ ", terminator: "")
        }
    }
    print("")
}
printIntegerKinds([3, 19, -27, 0, -6, 0, 7])
// Prints "+ + - 0 - 0 + "
//print(terminator: "")는 new line 으로 출력 안하는 법

 

 

 

코드 출처 : Swift 공식 홈페이지 - Extensions

https://docs.swift.org/swift-book/LanguageGuide/Extensions.html