【Kotlin】Kotlin 领域特定语言 DSL 原理 二 ( 中缀表达式 )

//类中定义支持中缀表达式方法
infix fun 函数名( 参数列表 ) : 返回值类型 {
  //方法体
}
//为现有类扩展函数
infix fun 类名.函数名( 参数列表 ) : 返回值类型 {
  //方法体
}

类对象名称 函数名称 参数列表

package dsl
data class Student(var name : String, var age : Int, var height : Int)
/*
    这里的  listOf 方法 , 返回一个集合 , 原型如下
    public fun <T> listOf(vararg elements: T): List<T> 
      = if (elements.size > 0) elements.asList() else emptyList()
 */
var students : List<Student> = listOf<Student>(
    Student("Tom", 12, 190),
    Student("Jerry", 18, 165),
    Student("Trump", 18, 178),
    Student("Jack", 29, 165)
)
/*
    DSL 扩展函数
    中缀表达式 : 使用空格代替 . 和 ()
    为 List<Student> 类型变量扩展函数 printMaxAgeStudent
        扩展内容是 查找身高高于 180 的人
 */
infix fun List<Student>.printHeightUpStudent(height : Int) : Unit{
    //调用 集合的高阶函数 filter , 筛选出身高高于 180 的学生
    filter {
        it.height > height
    }.forEach(::println)
}
fun main() {
    //调用 DSL 扩展函数 , 打印身高高于 180 的学生
    students.printHeightUpStudent(180)
    //使用中缀表达式 , 也可以实现上述功能 , 前提是方法前需要使用 infix 修饰
    students printHeightUpStudent 180
}

Student(name=Tom, age=12, height=190)
Student(name=Tom, age=12, height=190)