ฟังก์ชันส่วนขยาย: ไวยากรณ์และกฎการเลือกใช้

ฟังก์ชันส่วนขยายคืออะไร

ไวยากรณ์พื้นฐานของฟังก์ชันส่วนขยาย

fun String.shout(): String = uppercase() + "!"
println("hello".shout())  // HELLO!
fun Int.isEven() = this % 2 == 0
println(4.isEven())  // true

this ในส่วนขยาย

fun List<Int>.secondOrNull(): Int? {
    return if (size >= 2) this[1] else null
}
println(listOf(10, 20, 30).secondOrNull())  // 20
println(listOf(5).secondOrNull())           // null

การจัดส่งแบบสแตติก: ส่วนขยายไม่เขียนทับ

open class Base
class Child : Base()
fun Base.greet() = "Hello from Base"
fun Child.greet() = "Hello from Child"

val obj: Base = Child()
println(obj.greet())  // "Hello from Base" (static dispatch!)

ส่วนขยายเทียบกับฟังก์ชันสมาชิก

class Greeter {
    fun hello() = "member hello"
}
fun Greeter.hello() = "extension hello"  // shadowed!
println(Greeter().hello())  // "member hello"

ส่วนขยายของผู้รับที่เป็นค่าว่างได้

fun String?.orEmpty() = this ?: ""
fun Any?.toStringSafe() = this?.toString() ?: "null"
println(null.orEmpty())  // ""
println(42.toStringSafe())  // "42"

การมองเห็นและขอบเขต

// Top-level extension:
fun List<String>.joinUppercase() = joinToString { it.uppercase() }
// Private to file:
private fun String.clean() = trim().lowercase()

ส่วนขยายของอ็อบเจ็กต์คู่หู

class User(val name: String) {
    companion object
}
fun User.Companion.guest() = User("Guest")
val guest = User.guest()

ส่วนขยายบนชนิดของไลบรารีมาตรฐาน

fun Double.roundTo(decimals: Int): Double {
    val factor = Math.pow(10.0, decimals.toDouble())
    return Math.round(this * factor) / factor
}
println(3.14159.roundTo(2))  // 3.14

การจัดระเบียบส่วนขยาย

// StringExtensions.kt
fun String.titleCase() = split(" ")
    .joinToString(" ") {
        it.replaceFirstChar { c -> c.uppercase() }
    }
println("hello world".titleCase())  // Hello World

ส่วนขยายเข้าถึงสมาชิกส่วนตัวไม่ได้

class Secret {
    private val key = "hidden"
    val name = "public"
}
fun Secret.reveal() {
    println(name)  // OK
    // println(key)  // Error: key is private
}

ตรวจสอบอย่างรวดเร็ว

ทบทวน