【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)