- 확장 함수 정의하기
- 슈퍼 클래스에 확장 함수 정의하기
- 제네릭 확장 함수
- 확장 속성
- null 가능 타입의 확장함수
- 확장 함수의 바이트코드 구현
- 확장 함수로 추출하기
- 확장 파일 정의하기
- 확장 함수 이름 변경하기
확장이란, 기존 타입(클래스나 인터페이스)을 수정하지 않고 새로운 기능을 추가할 수 있는 방법.
ex. 예시코드1
fun String.addEnthusiasm(amount: Int = 1) = this + "!".repeat(amount)
fun main() {
println("마드리갈이 그 건물에서 나왔습니다".addEnthusiasm())
}String 은 수신자 타입이라고 하며 확장할 대상이 되는 클래스다.
위 코드는 String 클래스에 함수를 추가한 것과 같다.
위 코드처럼 String 클래스에 함수를 추가하려면 String 클래스를 상속받은
서브 클래스를 만들고 메소드를 만들어야 하지만,
코틀린의 String 클래스는 open 키워드가 없으므로 상속불가능하다.
Any 와 같이 최상위 클래스에 확장 함수를 정의하면 모든 타입에 대해서 호출될 수 있다.
예시코드1.
fun String.addEnthusiasm(amount: Int = 1) = this + "!".repeat(amount)
fun Any.easyPrint() = println(this)
fun main() {
"마드리갈이 그 건물에서 나왔습니다".addEnthusiasm().easyPrint()
42.easyPrint()
}확장함수는 이전에도 보았듯이 수신자 타입을 받아야 한다.
수신자 타입을 제네릭으로 변경하여 어떤 타입이든 받을 수 있도록 변경해보자.
제네릭을 사용하지 않은 예시코드.
fun String.addEnthusiasm(amount: Int = 1) = this + "!".repeat(amount)
fun Any.easyPrint() : Any {
println(this)
return this
}
fun main() {
// println("마드리갈이 그 건물에서 나왔습니다".addEnthusiasm())
"마드리갈이 그 건물에서 나왔습니다".easyPrint().addEnthusiasm().easyPrint()
42.easyPrint()
}위 코드에서 addEnthusiasm() 확장함수는 수신자 타입으로 String을 받아야하지만,
Any를 받고있으므로 컴파일 에러가 발생한다.
이런 경우 easyPrint() 확장함수의 수신자 타입을 제네릭으로 변경하여 오류를 제거할 수 있다.
제네릭을 사용한 예시코드.
fun String.addEnthusiasm(amount: Int = 1) = this + "!".repeat(amount)
fun <T> T.easyPrint() : T {
println(this)
return this
}
fun main() {
// println("마드리갈이 그 건물에서 나왔습니다".addEnthusiasm())
"마드리갈이 그 건물에서 나왔습니다".easyPrint().addEnthusiasm().easyPrint()
42.easyPrint()
}수신자 타입 Any를 T 라는 제네릭 타입으로 변경할 경우, 어떤 타입이든지 받을 수 있게되어 유연해지고
또한, 위 예제코드에서 String을 리턴하므로 addEnthusiasm() 확장함수를 호출할 수 있게된다.
제네릭 타입을 사용하는 코틀린 표준 라이브러리는 많다.
그중 let을 보면
public inline fun <T, R> T.let(block : (T) -> R) : R {
return block(this)
}let 은 제네릭 타입을 사용하여 어떤 타입이든지 받을 수 있고 다른 타입으로 실행결과를 반환해준다.
보통 let은 null check에서 사용한다. (기억 더듬기용..)
fun process(str: String?) {
str?.let { /*Do something*/ }
}또한, inline 키워드는 람다의 메모리 부담을 없애기 위해 만들어진 키워드이다. (이것도 기억 더듬기용..)
inline을 붙이지 않으면 jvm에서 람다는 새로운 객체로 생성하여 메모리에 할당하는데,,
inline을 붙일 경우 객체를 새로 생성하는 것이 아니라 해당 람다 코드를 통째로 복사하여
람다를 호출한 곳에 붙여넣기하여 사용한다고 한다.
함수 뿐만아니라 속성도 확장할 수 있다.
val String.numVowels
get() = count {"aeiouy".contains(it)}
fun main() {
"How many vowels?".numVowels.easyPrint()
}확장 속성은 산출 속성처럼 필드 값을 가지지 못하므로 반드시 get 메소드를 정의해야 한다.
(필드 값을 가지지 못하므로 setter 메소드와 var 키워드로 생성하지 못함)
확장함수에 null 인자를 받을 수 있도록 할 수 있다.
infix fun String?.printWithDefault(default: String) = print(this ?: default)
fun main() {
val nullableString: String? = null
nullableString printWithDefault "기본 문자열"
}확장할 대상이 되는 String 수신자 타입 뒤에 ?를 붙여주고 사용하려는 값이 null 일 경우 사용할
default 값으로 사용할 인자를 넘겨주고 null 복합연산자에서 사용하면 된다.
infix 키워드는 확장 함수나 클래스 함수에 인자가 하나라면 사용할 수 있으며,
main() 메소드에서와 같이 .(수신자 객채) 이나 (괄호)를 사용하지 않고도 함수호출 할 수 있도록 해준다.
참고 아래 코드의 to 도 키워드가 아닌 infix로 만들어진 함수이다.
var map0 = mapOf("one" to 1, "two" to 2)"one" 이라는 수신자 객체를 받아서 1 이라는 값과 함께 Pair 클래스의 속성으로 사용한다.
mapOf() 메소드는 결국 아래 코드처럼 for문을 돌리면서 map에 put 함.
public fun <K, V> MutableMap<in K, in V>.putAll(pairs: Array<out Pair<K, V>>): Unit {
for ((key, value) in pairs) {
put(key, value)
}
}그래서 아래 코드 처럼 사용하면 to 메소드 사용안하고 바로 Pair 클래스의 속성으로 사용하는 것.
(속도 차이는 거의 없을것 같고, 코틀린에서 친절히 to 메소드 만들어줬으니 쓰는게 좋을듯.)
var map1 = mapOf(Pair("one",1), Pair("two",2))확장 함수는 확장하려는 수신자 객체의 클래스에 직접 정의한 것이 아닌데 어떻게 구현되는지
바이트코드를 자바코드로 역컴파일한 코드를 보면 대충 이해할 수 있다.
(shift 2번 -> show kotlin bytecode -> Decompile 버튼 클릭)
kotlin
fun String?.addEnthusiasm(amount: Int = 1) = (this?: "1") + "!".repeat(amount)
fun main() {
println("마드리갈이 그 건물에서 나왔습니다".addEnthusiasm())
}
java 코드로 디컴파일
@NotNull
public static final String addEnthusiasm(@Nullable String $this$addEnthusiasm, int amount) {
StringBuilder var10000 = new StringBuilder();
String var10001 = $this$addEnthusiasm;
if ($this$addEnthusiasm == null) {
var10001 = "1";
}
return var10000.append(var10001).append(StringsKt.repeat((CharSequence)"!", amount)).toString();
}addEnthusiasm() 메소드를 static으로 선언하고 첫번째 인자로 수신자 타입의 값을 받는다. (위 예제 코드 "마드리갈~")
첫번째 인자로 받은 수신자 타입의 객체를 자바코드에서 코틀린에 addEnthusiasm() 메소드와 같은 기능을 하도록
변환 되어 있다.
코드내 중복적으로 발생하는 코드는 확장함수로 정의하여 제거할 수 있다.
기존 Tavern.kt 코드
val patronList: List<String> = listOf("Eli", "Mordoc", "Sopie")
val lastName = listOf("Ironfoot", "Fernsworh", "Baggins")
(0..9).forEach {index ->
val first = patronList.shuffled().first()
val last = lastName.shuffled().first()
val name = "$first $last"
uniquePatrons += name
}Tavern.kt 에는 shuffled().first() 라는 연쇄호출 함수를 여러번 사용되므로 확장 함수로 중복을 제거하여 사용할 수 있다.
private fun <T> Iterable<T>.random() : T = this.shuffled().first()
val patronList: List<String> = listOf("Eli", "Mordoc", "Sopie")
val lastName = listOf("Ironfoot", "Fernsworh", "Baggins")
(0..9).forEach {index ->
val first = patronList.random()
val last = lastName.random()
val name = "$first $last"
uniquePatrons += name
}수신자 타입을 Iterable로 한 이유는 shuffled() 와 first() 메소드는 Set과 List 타입에 연쇄 호출이 가능하도록 되어있으므로
두 타입의 슈퍼타입인 Iterable로 정의하였다.
(shuffled()는 Iterable 클래스를 수신자로 받아서 List 로 first()는 List를 받아서 값 리턴)
참고
List, Set은 Collection을 Collection은 Iterable을 슈퍼타입으로 정의되어 있다.
예시코드에서 random() 함수는 private으로 정의되어 있다.
private fun <T> Iterable<T>.random() : T = this.shuffled().first()
(0..9).forEach {index ->
val first = patronList.random()
val last = lastName.random()
val name = "$first $last"
uniquePatrons += name
}그래서 다른 파일에서는 random() 함수를 사용할 수 없으므로,
보통 코틀린에서는 확장함수나 확장속성은 별도 패키지로 관리한다고 한다.
패키지 이름은 관례상 extensions로
파일이름은 "수신자 타입 + Ext" 로 만든다
(ex. IterableExt.kt)
각 타입에 대한 확장 함수나 속성을 포함하는 코틀린의 표준 라이브러리 함수에는 파일 이름 끝에 s를 붙이고 있다.. (관례)
String 클래스의 확장함수 파일은 Strings.kt,
Sequence -> Sequence.kt
Range -> Ranges.kt
Map -> Maps.kt
코틀린에서는 서로 다른 패키지에 있는 동일한 이름의 클래스, 인터페이스, 함수들을 사용할 때
as 키워드를 사용하여 이름 충돌이 생기지 않도록 한다.
확장 함수도 똑같이 사용가능하다.
import extensions.random as randomizer
class Player {
private fun selectHometown() = File("data/towns.txt")
.readText()
.split("\r\n")
.randomizer()
}