loop문에서 continue와 break를 사용해야 했다. Java에서 작성하던 방식과 비슷하게 작성햇는데 내가 생각한 대로 동작이 일어나지 않아 레퍼런스 문서를 찾게 되었고, 잘못 사용했다는 것을 깨달아 블로그에 정리한다.
1. break / continue
우리가 흔히 사용하는 break, continue는 for문 안에서 사용할 수 있다.
fun main() {
for (i in 1..5) {
if (i==3) break
print("$i ")
}
}// 결과
1 2
문제는 Iterable<T>.forEach() 에서는 break, continue를 사용할 수 없다는 점이다. 아래와 같이 작성하면 break 때문에 compile이 불가능하다.
fun main() {
(1..5).forEach { i ->
if (i==3) break
print("$it ")
}
}'break' and 'continue' are only allowed inside a loop
collection.forEach()는 for를 Iterable로 한번 감싼 형태이기 때문에 loop라고 인식을 하지 않는 듯하다.
나는 collection을 chaining하던 중이었기 때문에 위의 for()문은 사용할 수 없었고, 다른 방법을 찾아 보았다.
2. Label 사용
fun main() {
listOf(1, 2, 3, 4, 5).forEach {
if (it == 3) return
print("$it ")
}
println("다른 로직 처리")
}// 결과
1 2
break 자리에 return을 사용하면 될줄 알았는데 아예 함수가 끝나버린다..ㅎㅎ
2-1. label을 사용한 continue
fun main() {
listOf(1, 2, 3, 4, 5).forEach {
if (it == 3) return@forEach
print("$it ")
}
println("다른 로직 처리")
}
// 결과
1 2 4 5 다른 로직 처리
return이지만 현재의 loop를 태깅하면 continue 동작이 실행된다...
2-2. label을 사용한 break
fun main() {
run {
listOf(1, 2, 3, 4, 5).forEach {
if (it == 3) return@run
print("$it ")
}
}
println("다른 로직 처리")
}// 결과
1 2 다른 로직 처리
그리고 만약 코드에 run이나 forEach가 많다면 태그를 붙여 확실하게 return 하는 expression을 지정할 수도 있다.
fun main() {
run startForeach@ { // 이름 태깅
listOf(1, 2, 3, 4, 5).forEach loop@ {
if (it == 3) return@startForeach // 태깅한 이름을 지정
print("$it ")
}
}
println("다른 로직 처리")
}
이제서야 내가 원하는 break 방식을 찾았다. 그렇지만 불필요한 run을 한번 더 감싸는 방식이 마음에 들지 않았다.
다른 방식을 찾아보기로 했다.
3. Collection 함수 사용
3-1. filter()를 사용한 continue
만약 중간에 continue가 들어가야 한다면 사용할 수 없는 방법이지만, 처음부터 조건문을 사용하여 continue 문을 거는 경우라면 filter()를 사용해볼 수 있다. 조건에 따라 filterNot()을 사용할 수도 있다.
fun main() {
run startForeach@ {
listOf(1, 2, 3, 4, 5)
.filter { it != 3 }
// .filterNot { it == 3 }
.forEach {
print("$it ")
}
}
println("다른 로직 처리")
}// 결과
1 2 4 5 다른 로직 처리
3-2. takeWhile()을 사용한 break
위와 마찬가지로 처음부터 조건문을 사용하여 break를 거는 경우라면 takeWhile()을 사용할 수 있다. takeWhile()은 자주 쓰지 않는 함수라 찾아보았다.
public inline fun <T> Iterable<T>.takeWhile(predicate: (T) -> Boolean): List<T> {
val list = ArrayList<T>()
for (item in this) {
if (!predicate(item))
break
list.add(item)
}
return list
}조건문이 실행되기 전까지의 list를 반환하는 함수이다. filter와 다른 점은 filter는 전체 중 조건문에 해당하는 전체 리스트 반환이지만, takeWhile()은 조건문에 해당하지 않는 순간 loop가 종료되고, 그 순간까지의 리스트만 반환한다.
fun main() {
run startForeach@ {
listOf(1, 2, 3, 4, 5)
.takeWhile { it != 3 }
.forEach {
print("$it ")
}
}
println("다른 로직 처리")
}// 결과
1 2 다른 로직 처리
결론
나는 개인적으로 2-2의 방법은 불필요한 run()을 작성하는게 별로라고 생각되어 3-2의 takeWhile()문을 사용하여 해결했다. 그러나 로직을 실행하다 break를 해야하는 순간에는 2-2를 선택할 수 밖에 없고, 어떻게 보면 그게 더 직관적인 코드일 수도 있다. 이것은 개인이 취향이기 때문에 여러가지 방법을 알아두고 사용하는 것이 좋겠다.
참고1: https://kotlinlang.org/docs/returns.html#break-and-continue-labels
참고2: https://www.baeldung.com/kotlin/break-continue-functional-loop
