Kotlin中如何定义扩展函数

在 Kotlin 中，扩展函数允许你为一个已有的类添加新的功能，而无需修改其源代码或使用继承。扩展函数的语法相对简单，以下是定义扩展函数的基本步骤：

基本语法

fun <T> ClassName.extensionFunctionName(parameterList): ReturnType {
    // 函数体
}

• fun：关键字，表示定义一个函数。
• <T>：类型参数（可选），用于泛型扩展函数。
• ClassName：要扩展的类的名称。
• extensionFunctionName：扩展函数的名称。
• parameterList：函数的参数列表。
• ReturnType：函数的返回类型。

示例

假设我们有一个 String 类，想要为其添加一个名为 reverseAndCapitalize 的扩展函数，用于反转字符串并将其首字母大写。

fun String.reverseAndCapitalize(): String {
    return this.reversed().capitalize()
}

fun main() {
    val original = "hello world"
    val modified = original.reverseAndCapitalize()
    println(modified) // 输出: Dlrow Olleh
}

在这个例子中：

• String 是要扩展的类。
• reverseAndCapitalize 是扩展函数的名称。
• 函数体中使用了 this.reversed() 来反转字符串，并使用 capitalize() 将首字母大写。

扩展属性

除了函数，Kotlin 还支持扩展属性。以下是一个为 String 类添加扩展属性的示例：

val String.firstChar: Char
    get() = this.getOrNull(0) ?: throw IndexOutOfBoundsException("String is empty")

fun main() {
    val str = "Kotlin"
    println(str.firstChar) // 输出: K
}

在这个例子中：

• String.firstChar 是一个扩展属性。
• get() 方法用于获取字符串的第一个字符。

扩展函数的注意事项

1. 扩展函数不能有默认参数。
2. 扩展函数不能访问类的私有成员，除非它们是 internal 或 protected。
3. 扩展函数不能重载，即不能为同一个类定义多个同名的扩展函数，即使它们的参数列表不同。

使用场景

扩展函数在以下场景中非常有用：

• 为第三方库中的类添加功能，而无需修改其源代码。
• 组织代码，将相关的功能分组在一起。
• 提高代码的可读性和可维护性。

通过合理使用扩展函数，可以使代码更加简洁和易于理解。