1. 新建 BooleanExt.kt 文件
/**
* @author 刘建波
* 2022/6/1
* 描述：实现带【泛化】返回值的Boolean链式扩展
*/
/**
* 密封类基类，作为中间类型，实现链式调用
*/
sealed class BooleanExt<out T> // 作为生产者 + 子类型关系 = 使用协变
/**
* 密封类子类，负责执行正向逻辑
*【需要携带数据给中间件，完成中介二次调用】
* 普通类，支持不同泛化
*/
class TransferData<T>(val data: T) : BooleanExt<T>()
/**
* 密封类子类，负责执行反向逻辑
*【不需要携带数据给中间件，用Nothing完成协变语法即可，不必泛化】
* 单例节约内存
*/
object Otherwise : BooleanExt<Nothing>()
/**
* 调用以扩展函数[yes]开头，以密封类为中介返回，延续链式调用
*/
inline fun <T> Boolean.yes(action: () -> T) = when {
this -> {
// 执行正向 lambda, 携带传递 lambda 结果，完成链式调用
TransferData(action.invoke())
}
else -> {
Otherwise
}
}
/**
* 中介扩展函数，处理[Boolean.yes]的另外一种情况
*/
inline fun <T> BooleanExt<T>.otherwise(action: () -> T) = when (this) {
is Otherwise -> {
// 执行反向的 lambda
action.invoke()
}
is TransferData -> {
this.data
}
}
2. 不带返回值的时候
测试代码
val booleanA = true
val aResult = booleanA
.yes { Logger.d("liujianbo", "booleanA 执行 yes") }
.otherwise { Logger.d("liujianbo", "booleanA 执行 otherwise") }
Logger.d("liujianbo", "aResult is $aResult")
结果
2022-06-01 14:27:58.434 19085-19085/com.jhj.cloudman D/liujianbo: booleanA 执行 yes
2022-06-01 14:27:58.435 19085-19085/com.jhj.cloudman D/liujianbo: aResult is kotlin.Unit
3. 带返回值的时候
测试代码
val booleanB = false
val bResult = booleanB
.yes {
Logger.d("liujianbo", "booleanB 执行 yes")
1 //返回值
}
.otherwise {
Logger.d("liujianbo", "booleanB 执行 otherwise")
0 // 返回值
}
Logger.d("liujianbo", "bResult is $bResult")
结果
2022-06-01 14:27:58.435 19085-19085/com.jhj.cloudman D/liujianbo: booleanB 执行 otherwise
2022-06-01 14:27:58.435 19085-19085/com.jhj.cloudman D/liujianbo: bResult is 0
4. 和别的表达式联合链式调用
测试代码
val listC = listOf(1, 3, 5)
val cResult = (listC
.filter { it % 2 == 1 }
.count() == listC.size)
.yes { "全部都是奇数" }
.otherwise { "非全部都是奇数" }
Logger.d("liujianbo", "cResult is $cResult")
结果
2022-06-01 14:27:58.436 19085-19085/com.jhj.cloudman D/liujianbo: cResult is 全部都是奇数
小结
- 密封类（更好的枚举类），密封类的单例子类
- 省略参数的 when 表达式
- 扩展函数
- 泛型【协变】
- lambda 表达式，最后一行表达式的结果就是 lambda 的返回值
