2.3 enum, when, is, as - ericbykim/kotlin-in-action GitHub Wiki

enum

• enum은 soft keyword이다. class앞에 있을 때는 특별한 의미를 지니지만 다른 곳에서는 이름을 사용할 수 있다.

enum class Color (
	val r: Int, val g: Int, val b: Int
) {
	RED(255,0,0), ORANGE(255,165,0),YELLO(255,255,0), ...
	VIOLET(238,130,238); // <- 반드시 세미콜론을 사용해야 한다.

	fun rgb() = (r * 256 + g) * 256 + b
}

when

• 자바의 switch 문에 해당하는 코틀린 요소

fun getMnmonic(color: Color) = 
	when (color) {
		Color.RED -> "Richard"
		Color.ORANGE -> "Of"
		Color.YELLOW -> "York"
		...
		Color.VIOLET -> "Vain"
	}

• 여러 개 매칭

fun getWarmth(color: Color) = when(color) {
	Color.RED, Color.ORANGE, Color.YELLOW -> "Warm"
	Color.BLUE, Color.INDIGO, Color.VIOLET -> "Cold"
	else -> "i don't know"
}

• 위 코드를 더 간단하게?

fun getWarmth(color: Color) = when(color) {
	RED, ORANGE, YELLOW -> "Warm"
	BLUE, INDIGO, VIOLET -> "Cold"
	else -> "i don't know"
}

• 혼합 조건으로 매칭

fun mix(c1: Color, c2: Color) =
	when (setOf(c1, c2)) {
		setOf(RED, YELLOW) -> ORANGE
		setOf(YELLOW, BLUE) -> GREEN
		setOf(BLUE, VIOLET) -> INDIGO
		else -> throw Exception("dirty color")
	}

println(mix(BLUE, YELLOW)) // GREEN

• 위처럼 setOf를 사용하면 set인스턴스를 생성해서 가비지 객체가 늘어난다. 코드의 가독성 보다 성능이 더 중요할 때도 있다.

// 가독성은 떨어지지만 추가 객체를 만들지 않는 장점이 있다.
fun mixOptimized(c1: Color, c2: Color) = 
	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")
	}

• Kotlin 1.3 부터 가능: When의 대상을 변수에 대입

fun Request.getBody() =
	when(val response = executeRequest()) {
		is Success -> response.body
		is HttpError -> throw HttpException(response.status)
	}

Smart Cast: 타입 검사 + 타입 캐스트를 조합 (is, as)

interface Expr
class Num(val value: Int) : Expr
class Sum(val left: Expr, val right: Expr) : Expr

// println ( eval(Sum(Sum(Num(1), Num(2)), Num(4))) )
// -> 7

• 자바 스타일 구현 (if 사용)

fun eval(e: Expr) : Int {
	if (e is Num) {
		val n = e as Num // 불필요한 코드 (스마트 캐스트로 이미 Num으로 캐스트 됨)
		return n.value
	}
	if (e is Sum) {
		return eval(e.right) + eval(e.left) // 스마트 캐스트 됨
	}
	throw IllegalArgumentException("Unknown expression")
}

• 스마트 캐스트: is 를 통해 타입 체크를 하면 컴파일러가 원하는 타입으로 캐스트 해준다.
• is 는 자바의 instanceof 와 비슷하다.
• 스마트 캐스트 안하는 경우: val이 아니거나 val이지만 커스텀 접근자를 사용하는 경우에는 해당 프로퍼티에 대한 접근이 항상 같은 값을 내놓는다고 확신할 수 없기 때문에 스마트 캐스트 하지 않는다.
• 명시적 타입 캐스팅 as
• 코틀린 방식으로 구현

fun eval(e: Expr) : Int = 
	if (e is Num) {
		e.value // 스마트 캐스트 됨
	} else if (e is Sum) {
		eval(e.right) + eval(e.left) // 스마트 캐스트 됨
	} else {
		throw IllegalArgumentException("Unknown expression")
	}

fun eval(e: Expr) : Int = 
	when (e) {
		is Num -> e.value // 스마트 캐스트 됨
		is Sum -> eval(e.right) + eval(e.left) // 스마트 캐스트 됨
		else -> throw IllegalArgumentException("Unknown expression")
	}

• 블록 사용

fun eval(e: Expr) : Int = 
	when (e) {
		is Num -> {
			println("num: ${e.value}")
			e.value
		}

		is Sum -> {
			val left = eval(e.left)
			val right = eval(e.right)
			println("sum: $left + $right")
			left + right
		}

		else -> throw IllegalArgumentException("Unknown expression")
	}