youl

youl

푸하하-자아자 ˘◡˘

[Kotlin] Extensions

2 분 소요


🙋‍♀️Extensions?

  • extension(확장)을 통해서 간단하게 객체의 함수, 프로퍼티를 확장 정의할 수 있다.


🙋‍♀️Extension functions

  • index1, index2인 두 값을 교환하는 프로그램을 작성하고자 한다.
fun MutableList<Int>.swap(index1: Int, index2: Int) {
    val tmp = this[index1] // 'this' corresponds to the list
    this[index1] = this[index2]
    this[index2] = tmp
}

  • 확장 함수의 this는 MutableLis(receiver object)와 대응하게 된다.

  • swap()은 모든 MutableList에서 사용할 수 있다.

  • generic으로도 사용 가능하다.

    fun <T> MutableList<T>.swap(index1: Int, index2: Int) {
  val tmp = this[index1] // 'this' corresponds to the list
  this[index1] = this[index2]
  this[index2] = tmp
}

🙄사용의 예를 한 번 살펴보자.

val list = mutableListOf(1, 2, 3)
list.swap(0, 2) // 'this' inside 'swap()' will hold the value of 'list'

print(list) // [3, 2, 1]

🙄 위의 예에서 확인했듯이 마치 MutableList의 함수인 것처럼 swap()함수를 정의할 수 있다.


🙋‍♀️Extension properties

extension 함수와 같이 extension properties도 제공한다.

val <T> List<T>.lastIndex: Int
    get() = size - 1


🙇‍♀️Nullable receiver 사용!

  • extension은 nullable receiver 타입과 함께 정의될 수 있다.
  • 즉, null에 의해서도 extension이 호출될 수 있다는 의미이다.
fun Any?.toString(): String {
    if (this == null) return "null"
    // after the null check, 'this' is autocast to a non-null type, so the toString() below
    // resolves to the member function of the Any class
    return toString()
}
  • extension 함수 안에서 null 체크가 이루어진다.
  • 따라서 별도의 null 체크없이 extension 함수인 toString()을 사용할 수 있다.


🙋‍♀️Extension 사용 규칙

🙇‍♀️Extension은 정적으로 처리된다.

open class Shape
class Rectangle: Shape()

fun Shape.getName() = "Shape"
fun Rectangle.getName() = "Rectangle"

fun printClassName(s: Shape) {
    println(s.getName())
}

printClassName(Rectangle())
  • 위 예제를 확인하면 최종적으로 “Rectangle”이 출력될 것이라 생각할 것이다.
  • 하지만 Extension의 경우 정적으로 처리되기 때문에 실제로 참조되고 있는 타입인 Rectangle이 아니라 PrintClassName에 선언된 타입인 Shape의 getName이 호출되게 된다.


🙇‍♀️멤버가 Extension보다 우선이다.

class Person {
    fun getName() { println("yuri") }
}
fun Person.getName() { println("Hello, I'm yuri!") }

fun main() {
    Person().getName() //yuri
}

🙄 위 코드를 통해서 쉽게 이해할 수 있을 것이다.

  • getName 함수는 멤버로도 선언되어 있으며, extension 함수로도 선언되어 있다.
  • 이럴경우, 멤버가 우선이 된다.


🙇‍♀️ extesions의 범위

  • extensions을 선택적으로 import해 사용할 수 있다.
package org.example.declarations

fun List<String>.getLongestString() { /*...*/}

package org.example.usage

import org.example.declarations.getLongestString

fun main() {
    val list = listOf("red", "green", "blue")
    list.getLongestString()
}
  • extensions을 클래스 멤버로 선언할 경우에는 해당 클래스 내로 범위가 결정된다.


🙄연산자 오버로딩

이번 시간에 배운 extension을 이용해서 연산자 오버로딩도 진행할 수 있다. 간단한 내용이기에 이번 게시글에서 함께 짚고 넘어가고자 한다.

우선 오버로딩이 가능한 산술연산자를 다음의 표를 통해서 살표보자.

expression translated to
+a a.unaryPlus()
-a a.unaryMinus()
!a a.not()
a++ a.inc()
a– a.dec()
a+b a.plus(b)
a-b a.minus(b)
a*b a.times(b)
a/b a.div(b)
a%b a.rem(b)
a..b a.rangeTo(b)
a in b b.contains(a)
a !in b !b.contains(a)
a[i] a.get(i)
a() a.invoke()
a += b a.plusAssign(b)
a -= b a.minusAssign(b)
a *= b a.timesAssign(b)
a /= b a.divAssign(b)
a %= b a.remAssign(b)
a > b a.compareTo(b) > 0
a < b a.compareTo(b) < 0
a >= b a.compareTo(b) >= 0
a <= b a.compareTo(b) <= 0

🙄 많아 보여도 비슷한 부분이 많아 쉽게 익힐 수 있을 것이다!


그렇다면 위 표에 존재하는 산술연산자를 어떻게 오버로딩하는지 한 번 살펴보자! 예로 Point 객체를 -Point 연산이 가능하도록 하고자 한다.

data class Point(val x: Int, val y: Int)
  • -Point 연산은 unaryMinus()를 오버로딩 함으로써 구현할 수 있다. 
    operator fun Point.unaryMinus() = Point(-x, -y)
    • 연산자 오버로딩을 진행할 경우, operator 키워드를 사용한다.
    • Point 객체의 extension function인 unaryMinus()를 정의한다.

🙄간단하지 않은가? 그렇다면 사용법을 한 번 살펴보자!

val point = Point(10, 20)

fun main() {
   println(-point)// "Point(x=-10, y=-20)"
}


🙇‍♀️ 부족한 부분이 있다면 말씀해주세요! 감사합니다!

📃참고

댓글남기기

참고