본문 바로가기

Effective Java

[Effective Java] 아이템 58 - 전통적인 for 문보다는 for-each문을 사용하라

개요

이전 아이템에서 for문이 while문 보다 더 적격이다 라는 내용을 다루었지만

같은 for문 내에서도 for-each 문을 사용하는 것이 더 바람직하다

 

반복자와 인덱스 변수는 모두 코드를 지저분하게 할 뿐 우리에게 진짜 필요한 건 원소들 뿐이다.

쓰이는 요소 종류가 늘어나면 오류가 발생할 가능성이 높아진다.

중첩되어 반복문이 사용되는 경우 변수를 잘못 사용할 틈새가 넓어진다.

 

혹시라도 잘못된 변수를 사용했을 때 컴파일러가 잘못 사용한 변수를 잡아준다는 보장도 없다.

컬렉션인지 배열인지에 따라 코드 형태도 상당히 달라지므로 주의해야 한다.


해결법 : for-each

위에서 언급한 내용들은 for-each문을 사용하면 모두 해결이 가능하다.

반복자와 인덱스 변수를 사용하지 않으니 코드가 깔끔해지고 오류가 날 일도 없다.

하나의 관용구로 컬렉션과 배열을 모두 처리할 수 있어 어떤 컨테이너를 다루는지는 신경 쓰지 않아도 된다.

for(Element e: elements){
	doSomething(e);
}

 

 

Error Catch: NoSuchElementException

 

enum Suit{CLUB,DIAMOND, HEART, SPADE}
    enum Rank{ACE, DEUCE, THREE, FOUR, FIVE, SIX, SEVEN, EIGHT, NINE,
        TEN, JACK, QUEEN, KING}

static Collection<Suit> suits = new Arrays.asList(Suit.values());
static Collection<Suit> ranks = new Arrays.asList(Rank.values());

List<Card> deck = new ArrayList<>();
for(Iterator<Suit> i = suits.iterator(); i.hasNext(); )
	for(Iterator<Rank> j = ranks.iterator(); j.hasNext(); )
    	deck.add(new Card(i.next(), j.next());

 

위와 같은 코드에서 i에 대한 순회가 지속적으로 반복되는 과정으로 인하여 NoSuchElementException이 발생할 수 있다.

이러한 코드 구성을 for-each 문을 사용하면 다음과 같은 코드로 더욱 간결하고 safe한 코드를 작성할 수 있다.

 

for(Suit suit : suits){
	for( Rank rank : ranks){
    	deck.add(new Card(suit, rank);
     }
 }

for-each 문을 사용할 수 없는 세 가지 상황

1. 파괴적인 필터링(destructive filtering)

컬렉션을 순회하면서 선택된 원소를 제거해야 한다면 반복자의 remove 메소드를 호출해야 한다. 자바 8부터는 Collection의 removeIf 메소드를 사용해 컬렉션을 명시적으로 순회하는 일을 피할 수 있다.

예를 들어 1부터 10까지 들어있는 리스트를 순회하면서 조건문에 해당하는 원소를 지우는 경우 위와 같은 예외가 발생.....

한다고 개인적인 예를 가져와봤습니다.

 

2. 변형(transforming)

리스트나 배열을 순회하면서 그 원소의 값 일부 혹은 전체를 교체해야한다면 리스트의 반복자나 배열의 인덱스를 사용해야 한다.

 

3. 병렬 반복(parallel iteration)

여러 컬렉션을 병렬로 순회해야 한다면 각각의 반복자와 인덱스 변수를 사용해 엄격하고 명시적으로 제어해야 한다.


핵심 정리

위에서 말한 세 가지 상황이 아니라면 되도록 for-each문을 사용하여 구현을 하도록 하자. 

for-each문은 컬렉션과 배열은 물론 Iterator 인터페이스를 구현한 객체라면 무엇이든 순회할 수 있다.

 

전통적인 for문과 비교했을 때 for-each 문은 명료하고, 유연하고, 버그를 예방해준다.
성능 저하도 없다. 가능한 모든 곳에서 for문이 아닌 for-each 문을 사용하도록 하자.