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 全部都是奇数

小结

1. 密封类（更好的枚举类），密封类的单例子类
2. 省略参数的 when 表达式
3. 扩展函数
4. 泛型【协变】
5. lambda 表达式，最后一行表达式的结果就是 lambda 的返回值