[TIL] Kotlin 표현식 활용하여 리팩토링 하기

 

 

PR 리뷰에서 Kotlin 표현식 관련 피드백을 받았다.

`when`, `ifBlank`, `let`을 활용하여 리팩토링을 진행해봤다. 🫵🏻

 

 

 

`when` 표현식

 

Java 에서는 `switch~case`문,  Kotlin에서는 `when` 키워드를 사용한다.

 

when(값) {
    조건부 -> 어떠한 구문
    조건부 -> 어떠한 구문
    else -> 어떠한 구문
}

 

Java 에서는 `switch~case`문,  Kotlin에서는 `when` 키워드를 사용한다.

 

 

when {
    score >= 90 -> "A"
    score >= 80 -> "B"
    else -> "C"
}

 

`when()` 에 값 없이 `ealry return`처럼 활용 가능하다.

`if-else`문 대신 사용하면 가독성이 좋아진다.

 

 

val grade = when {
    score >= 90 -> "A"
    score >= 80 -> "B"
    else -> "C"
}

 

`when`은 값을 반환하는 표현식이라서 변수에 바로 할당할 수 있다.

 

 

 

🚧 수정 전 코드 1 — `if`문

        if (deleteImage) {
            this.imageId = null
        }
        imageId?.let { this.imageId = it }

 

 

이미지 파일은 기본값이 없기 때문에, `deleteImage`라는 flag 값을 기준으로 수정을 했다.

`deleteImage = true`이면서 `imageId`가 null이 아니라면, 먼저 기존 값을 삭제한 후에 (null 처리) 바로 새 값으로 덮어쓴다.

기능상 문제는 없지만 가독성이 떨어진다는 피드백이 있었다.

 

 

👷🏻 수정 후 코드 1 — `when` 표현식

    when {
        imageId != null -> this.imageId = imageId
        deleteImage -> this.imageId = null
    }

 

수정 전 코드에서는 무조건 삭제(null 처리) 처리 후에 새 값을 추가했지만,

수정 후에는 새 이미지 업로드 하는 경우에는 요청 데이터를 그대로 덮어쓰기 하고(삭제 처리 필요 없음)

새 이미지 업로드를 하지 않고 삭제 처리만 하는 요청된 경우에만 null처리가 된다.

 

 

 

 

🚧 수정 전 코드 2  —  `if-else`문

        content?.let {
            if (it.isNotBlank()) {
                this.content = it
            } else {
                this.content = null
            }
        }

 

수정 관련 api는 아래와 같은 기준으로 작업을 진행했다.

1) content key 값 자체가 넘어오지 않은 경우 (null) -> 내용 수정 없음

2) content 값이 빈 문자열("")인 경우 -> 내용 삭제

3) content 값이 빈 문자열("")이 아닌 경우 -> 새 값으로 덮어쓰기

 

 

위 코드와 마찬가지로 가독성이 떨어진다는 피드백이 있었다.

 

 

👷🏻 수정 후 코드 2 — `ifBlank` 표현식

content?.let { this.content = it.ifBlank { null } }

 

`ifBlank { null }`은 문자열이 blank이면 `null`을, 아니면 원래 값을 반환하는 표현식이다.

 

 

 

`let`은 마지막 표현식을 반환한다

`let`은 스코프 함수다.  블록 `{ }` 안의 마지막 표현식이 반환된다.

 

val result = "hello".let {
    println(it)       // 실행만 하고
    it.uppercase()    // ⬅️ 마지막 표현식 = 반환값
}

// result = "HELLO"

 

 

사이드 프로젝트에서는 아래와 같이 활용을 했다.

        val activityCategoryList =
            activityCategoryIdList?.let {
                val activityCategoryList = activityCategoryRepository.findAllById(it)
                if (it.size != activityCategoryList.size) {
                    throw ActivityCategoryNotFoundException()
                }

                activityCategoryList ⬅️ 해당 변수가 반환됨
            }