java程序员的kotlin课

java程序员的kotlin课程(二): 高阶函数与泛型的几个套

2020-02-04  本文已影响0人  青_雉

函数在kotlin中是 一等公民,一个函数可以充当另外一个函数的入参或返回值,即所谓的高阶函数。
举个例子:

fun a (block: () -> Int) {
  println(block())
}

调用方式:

fun main() {
  a {
    123
  }
}

在使用kotlin的过程中发现kotlin标准库里的很多能力都会使用高阶函数来实现,比如 scope functionsrunBlockingcoroutineScopesequence。本文主要是基于标注库的几种玩法归纳整理了高阶函数与泛型配合使用的几个套路,大家在设计自己的工具库时可以致敬官方设计。
因为官方标准库里使用的模式比较多,本文关注重点在高阶函数和泛型的配合使用,所以下文中会对标准库的实现做简化说明,只剥离说明我们关注的部分。

apply函数的玩法

applyscope functions里的一个重要函数,其实现的套路大概是这样子的:

fun <T> T.b(block: T.() -> Int){
  println(block())
}

为任意类型T声明了b函数,b函数接受一个block函数,这个block函数也是类型T的一个函数,没有入参,返回值为Int类型,在block函数内部是类型T的scope的。
调用方式:

StringBuilder("hello").b {
    append(" ")
    append("xiaoming")
    toString().length
  }

with函数的玩法

with也是scope functions里的一个重要函数,其实现的套路大概是这样子的:

fun <T> c(receiver: T, block: T.() -> Unit){
  receiver.block()
}

apply函数不同, 并没有对c函数声明具体的receiver(Function Literal with Receiver),接受者是以入参的形式存在于c函数中的;同样block函数是具备T类型的scope的,调用方式如下:

fun main() {
  var sb = StringBuilder("hello")
  c (sb) {
    append(" ")
    append("xiaoming")
  }
  println(sb.toString())
}

coroutineScope函数玩法

coroutineScope函数是协程库里的重要函数,其玩法大概是这样的:

fun <R> d(block: String.() -> R) : R {
  return "abc".block()
}

返回值类型是由block的返回值类型决定的,block函数的receiver是固定的(本例中是String类型,coroutineScope函数里是CoroutineScope类型),调用方式如下:

 var r1 = d { 123 }
 var r2 = d { "hello" }
 println("r1 -> $r1, r2 -> $r2")

sequence 函数的玩法

sequence函数支持返回一个序列,类似一个管道的引用,管道内部后续可以不断的产生内容
这个函数的玩法特点是,block函数的receiver类型是固定的,但是这个类型还接受一个泛型, 泛型的类型是不固定的,会根据调用方的使用方式推断而来,这次我们先看使用方式:

fun main() {
  var r1 = e { doInfer("hello") }
  var r2 = e { doInfer(123) }
  println("r1 -> $r1, r2 -> $r2")
}

class Sth<T> {

  var sth : T? = null

  fun doInfer(t: T){
    sth = t
  }
}

可以看到,在e函数的入参,即 block函数里,是隶属于Sth类型的scope的,可以不通过this关键字直接低啊用doInfer函数;Sth对象里的T类型是根据调用方的使用姿势推断出来的:

@UseExperimental(ExperimentalTypeInference::class)
fun <T> e(@BuilderInference block: Sth<T>.() -> Unit): T? {
  val sth = Sth<T>()
  sth.block()
  return sth.sth
}

可以看到针对block函数是打了一个特殊的注解@BuilderInference的,而且这是一个实验性的注解,所以方法上还打了另外一个注解@UseExperimental(ExperimentalTypeInference::class)
@BuilderInference的解释如下:

Allows to infer generic type arguments of a function from the calls in the annotated function (笔者备注:annotated function就是我们的block函数)parameter of that function.
When this annotation is placed on a generic function parameter of a function,
it enables to infer the type arguments of that generic function from the lambda body passed to that parameter.
The calls that affect inference are either members of the receiver type of an annotated function parameter or
extensions for that type. The extensions must be themselves annotated with @BuilderInference.

上一篇下一篇

猜你喜欢

热点阅读