Kotlin 表达式与运算符

一、前言

Kotlin 中的类与接口Kotlin 空类型安全与智能类型转换 中我们已经将 Kotlin 中的类型初步讲完了,今天我们来讲 Kotlin中的表达式

二、分支表达式

2.1、if else 表达式

Kotlin:

if (a == 3) {
    c = 4
} else {
    c = 5
}

上面代码可以等价写成下面这种形式:

c = if (a == 3) 4 else 5 

2.2、when 表达式

这个 when 表达式就相当于 Java 里面的 switch case。

Java:

switch (a) {
    case 0:
        c = 5;
        break;
    case 1:
        c = 100;
        break;
    default:
        c = 20;
}

Kotlin:

when (a) {
    0 -> c = 5
    1 -> c = 100
    else -> c = 20
}
<==> 表达式的写法
c = when(a) {
    0 -> 5
    1 -> 100
    else -> 20
}

2.3、try catch 表达式

Java:

try {
    c = a / b;
} catch (Exception e) {
    e.printStackTrace();
    c = 0;
}

Kotlin:

try {
    c = a / b
} catch (e: Exception) {
    e.printStackTrace()
    c = 0
}
<==> 表达式的写法
c = try {
    c = a / b
} catch (e: Exception) {
    e.printStackTrace()
    0
}

三、运算符与中缀表达式

3.1、运算符

Kotlin 的运算符有以下特点:

  • Kotlin 支持运算符重载;
  • 运算符的范围仅限官方指定的符号;

3.1.1、== 运算符

在 Kotlin 中 == 与 equals 完全等价。

// 这两个写法完全等价
"Hello" == "World"
"Hello".equals("World")

3.1.2、+ 运算符

在 Kotlin 中 + 与 plus 完全等价。

// 这两个写法完全等价
2 + 3
2.plus(3)

3.1.3、in 运算符

val list = listOf(1, 2, 3, 4)
2 in list

3.1.4、[] 运算符

val map = mutableMapOf(
    "Hello" to 2,
    "World" to 3
)
val value = map["Hello"]
map["World"] = 4

...
这里只举几个运算符来讲,具体大家可以查看官网的运算符的用法。

3.2、运算符重载

运算符重载就是对已有的运算符赋予他们新的含义,重载的修饰符是 operator。

运算符重载我们可以看下面这个例子,这里我们是对复数 Complex 的运算符进行重载。

//复数
class Complex(var real: Double, var image: Double){
    override fun toString() = "$real + ${image}i"
}

operator fun Complex.plus(other: Complex): Complex {
    return Complex(this.real + other.real, this.image + other.image)
}

operator fun Complex.plus(other: Double): Complex {
    return Complex(this.real + other, this.image)
}

operator fun Complex.plus(other: Int): Complex {
    return Complex(this.real + other, this.image)
}

operator fun Complex.minus(real: Double): Complex {
    return Complex(this.real - real, this.image)
}

operator fun Complex.get(index: Int): Double = when(index){
    0 -> this.real
    1 -> this.image
    else -> throw IndexOutOfBoundsException()
}

fun main() {

    val c1 = Complex(3.0, 4.0)
    val c2 = Complex(2.0, 2.0)

    println(c1 + 2.0)
    println(c1 + c2)
    println(c1 + 3)
    println(c1 - 3.0)

    println(c1[0])
    println(c1[1])
    println(c1[2])
}

3.3、中缀表达式

只有一个参数,且用 infix 修饰的函数。

// 本没有 vs,自定义一个,不需要类名.方法即可调用
infix fun Int.vs(num: Int): String =
    when {
         this - num < 0 -> "$this 小于 $num"
         this - num > 0 -> "$this 大于 $num"
         else -> "$this 等于 $num"
     }

调用:

fun main() {
    println(5 vs 6)
    println(5 vs 5)
    println(7 vs 6)
}

运行结果:
Kotlin 表达式与运算符_第1张图片

四、Lambda 表达式

4.1、匿名函数

我们先来看一下一个普通函数:

// func 是函数名
fun func() {
    println("Hello")
}

我们把函数名 func 去掉就变成了匿名函数,如下:

// 去掉 func 变成匿名函数
fun() {
    println("Hello")
}

函数是可以传递的,所以匿名函数自然也能传递,如下:

// func 是变量名
var func = fun() {
    println("Hello")
}

我们知道一个变量是有类型的,那么 func 变量的类型是什么呢? func 变量的类型就是匿名函数的类型,如下:

// func/匿名函数的类型是 () -> Unit
var func: () -> Unit = fun() {
    println("Hello")
}

我们之所以说匿名函数的原因是 Lambda 表达式就是匿名函数的一个更具有表现力的写法,它本质上就是一个匿名函数。

4.2、Lambda 表达式的定义

Java:

Runnable lambda = () -> {
    System.out.println("Hello");
};

Kotlin:

val lambda = {
    println("Hello")
}

4.3、Lambda 表达式的类型

// 类型是 () -> Unit
val lambda: () -> Unit = {
    println("Hello")
}

// 类型是 (Int) -> Unit
val f1: (Int) -> Unit = { p: Int ->
    println(p)
}
<==> 等价写法
val f1: Function1 = { p ->
    println(p)
}
<==> 类型推倒
val f1 = { p: Int ->
    println(p)
}

Lambda 表达式的最后一行为返回值,下面这个 Lambda 表达式的意思就是接受一个整型参数,并且返回一个字符串 "Helll"。

val f1 = { p: Int ->
    println(p)
    "Hello"
}

4.4、Lambda 表达式的参数省略形式

val f1: Function1 = { p ->
    println(p)
}

如果 Lambda 表达式只有一个参数的话,可以省略参数,所以上面的 Lambda 表达式就可以改成下面这样:

val f1: Function1 = {
    println(it)
}

五、小结

本篇博客主要讲了 Kotlin 中的表达式运算符,下一节我们讲 Kotin 的函数进阶

六、源码

源码 已上传至 github,有需要可以直接下载。