本章我们将学习怎么样在Kotlin中通过枚举和when实现表示与选择。
我们来看一下java中的用法:
public enum Color
{
RED, ORANGE, YELLOW, GREEN, BLUE, INDIGO, VIOLET
}
kotlin中的用法:
enum class Color {
RED, ORANGE, YELLOW, GREEN, BLUE, INDIGO, VIOLET
}
你会发现kotlin中使用enum
同时还要使用class
这与java
中不同,思考下这是为什么?
知识点
与java相同,枚举中可以声明属性和方法
先看看java中的代码:
public enum Color
{
RED(255, 0, 0), //当每个变量创建的时候,指定属性值
ORANGE(255, 165, 0), //逗号是必须的
YELLOW(255, 255, 0),
GREEN(0, 255, 0),
BLUE(0, 0, 255),
INDIGO(75, 0, 130),
VIOLET(238, 130, 238);
private int r;
private int g;
private int b;
Color(int r, int g, int b)
{
this.r = r;
this.g = g;
this.b = b;
}
public int getRGB(){
return (r * 256 + g * 256 + b*256);
}
public int getR()
{
return r;
}
public int getG()
{
return g;
}
public int getB()
{
return b;
}
}
是不是超级多,再来看看kotlin,是不是少了好多,瞬间清爽了很多
enum class Color(
val r: Int, val g: Int, val b: Int //声明枚举常量的属性
) {
RED(255, 0, 0), //当每个变量创建的时候,指定属性值
ORANGE(255, 165, 0), //逗号是必须的
YELLOW(255, 255, 0),
GREEN(0, 255, 0),
BLUE(0, 0, 255),
INDIGO(75, 0, 130),
VIOLET(238, 130, 238);
fun rgb() = (r * 256 + g * 256 + b*256)//定义枚举的方法
}
从两个例子中可以看出,当你定义一个枚举常量,你需要为它提供属性值。
这个例子中展示了Kotlin语法唯一需要分号的地方:在枚举类中如果你定义任何方法,分号区分了枚举常量列表和方法声明。
接下来我们看一下枚举的处理:
接下来我们使用when选择枚举,我们输入一个颜色,返回对应的一个字符串
fun getMnemonic(color: Color) = //直接返回一个“when”的表达式
when (color) { //如果颜色等于枚举常量,返回响应的字符串
Color.RED -> "Richard"
Color.ORANGE -> "Of"
Color.YELLOW -> "York"
Color.GREEN -> "Gave"
Color.BLUE -> "Battle"
Color.INDIGO -> "In"
Color.VIOLET -> "Vain"
}
println(getMnemonic(Color.BLUE)) // Battle
上面代码我们可以处,你不需要为每个分支写break语句(在java中遗漏break通常会导致bug)。
可以吧多个值合并到同一个分支,只需要逗号隔开。
如下:
fun getWarmth(color: Color) = when(color) {
Color.RED, Color.ORANGE, Color.YELLOW -> "warm"
Color.GREEN -> "neutral"
Color.BLUE, Color.INDIGO, Color.VIOLET -> "cold"
}
println(getWarmth(Color.ORANGE)) //warm
上面的例子用全名来使用枚举常量,指定Color枚举类名。你可以用导入常量来简化代码:
import ch02.colors.Color //导入声明在另外一个包的Color类
import ch02.colors.Color.*//用名字显示导入枚举常量
fun getWarmth(color: Color) = when(color) {
RED, ORANGE, YELLOW -> "warm" //用名字导入常量
GREEN -> "neutral"
BLUE, INDIGO, VIOLET -> "cold"
}
kotlin中的when结构比java中的switch强大得多。switch必须要求使用常量(枚举常量、字符串或者数字字面值)作为分支条件,而when可以使用任何对象。
下面我们写一个混合两者颜色的函数:
fun mix(c1: Color, c2: Color) =
when (setOf(c1, c2)) {//when表达式的参数可以是任何实例,用来被分支条件检查
setOf(RED, YELLOW) -> ORANGE//枚举可以混合的颜色对
setOf(YELLOW, BLUE) -> GREEN
setOf(BLUE, VIOLET) -> INDIGO
else -> throw Exception("Dirty color")//执行这个,如果没有分支可以匹配
}
println(mix(BLUE, YELLOW))//GREEN
Kotlin标准库中含有一个setOf的函数,用来创建Set,包含参数指定的实例;一个set是一个集合,它的项的次序并不重要。所以,如果setOf(c1, c2)和setOf(RED, YELLOW)是相等的,那么意味着要不然c1是RED和c2是YELLOW,或者相反。
上面的例子有点效率低下,因为每次你调用这个函数,它都会创建几个Set实例,仅仅是用在检查两个颜色是否匹配另外两个颜色。正常情况下,通常不是个问题。但是如果这个函数经常被调用,那么为了避免GC,值得用另外一种方式来重写这个代码。你可以用不带参数的when表达式完成。代码虽然可读性差一点,但是这是为了达到更好性能付出的代价。
fun mixOptimized(c1: Color, c2: Color) =
//没有实参传给when
when {
(c1 == RED && c2 == YELLOW) || (c1 == YELLOW && c2 == RED) -> ORANGE
(c1 == YELLOW && c2 == BLUE) || (c1 == BLUE && c2 == YELLOW) -> GREEN
(c1 == BLUE && c2 == VIOLET) || (c1 == VIOLET && c2 == BLUE) -> INDIGO
else -> throw Exception("Dirty color")
}
println(mixOptimized(BLUE, YELLOW)) //GREEN
if和when都可以用代码块作为分支。
在这个例子中,代码块中最后最后一个表达式作为结果。如果你想在例子函数中加日志,你可以在代码块中完成,并用最后一个值返回。
fun evalWithLogging(e: Expr): Int =
when (e) {
is Num -> {
println("num: ${e.value}")
e.value //如果e是Num类型,这是代码块最后一个表达式,并被返回
}
is Sum -> {
val left = evalWithLogging(e.left)
val right = evalWithLogging(e.right)
println("sum: $left + $right")
left + right//如果表达式被返回当e是Sum类型
}
else -> throw IllegalArgumentException("Unknown expression")
}
println(evalWithLogging(Sum(Sum(Num(1), Num(2)), Num(4))))
//num: 1
//num: 2
//sum: 1 + 2
//num: 4
//sum: 3 + 4
//7
规则 :“代码块中最后一个表达式是返回值”,在所有使用代码块并期望得到一个结果的地方成立。
现在已经学会了如何正确从众多选项中作出选择,下一章我们一起去看下如何迭代事物。