자바 iterators란?

🐣 Java Iteration 🐣

 

Java에서 Map 순회하는 방법을 보면 for문을 사용한 방법과 iterator를 사용한 방법 2가지가 있다. 

그런데 iterator로 Map을 순회하는 코드를 작성하면 인텔리제이에서는 향상된 for문으로 변경할 수 있다고 한다. 

인텔리제이에서 밑줄을 확인하면 'while' loop can be replaced with enhanced 'for'라고 나온다

 

인텔리제이에서 추천하는대로 enhanced 'for' loop으로 바꿔보면 기존에 작성한 for문을 활용한 Map 순회방법과 똑같이 변경되는데 향상된 for문은 사실 iterator object를 사용해서 순회하는 코드를 사용하기 쉽고 예쁘게 바꿔놓은 것으로 볼 수 있다. (물론, iterator를 사용하면 순회하는 도중에 삭제가 가능하지만 향상된 for문은 ConcurrentModificationException이 발생하기 때문에 순회 도중 삭제가 불가능하다는 차이점이 있다.)

iterator를 활용한 코드와 enhanced for문 모두 123을 출력해준다

 

 

그래서 향상된 for문을 사용하고자 한다면 컴파일러는 

1. 해당 클래스에 iterator 객체를 반환하는  iterator() 메소드가 있는지

2. iterator 인터페이스에 hasNext(), next() 메소드가 있는지 확인한다. 

 

Set/List Collection에서 향상된 for문을  사용할 수 있는 이유는 Set/List인터페이스는 Collection을 상속하고 Collection은 Iterable을 상속하기 때문이다. 따라서 만약 사용자 생성 클래스에 향상된 for문을 사용하고자 한다면, Set/List와 같이 Iterable을 상속받아야 한다. 

Set 인터페이스는 Collection을 상속받고

 

Collection인터페이스는 Iterable을 상속받는다.

 

 

Iterables는 iterators를 생성할 수 있는 객체를 말하며, iterable interface를 상속한 클래스는 iterator를 반환하는 iterator() 메소드를 정의해야 한다. 

public interface Iterable<T> {
	Iterator<T> iterator();
}

 

Iterators는 순회될 수 있는 객체를 말한다. Iterator를 상속하면 hasNext()와 next() 메소드를 제공해야 한다. 

public interface Iterator<T> {
	boolean hasNext();
	T next();
}

 

 

예제 1

Iterable을 상속 받는 Books 클래스 예제를 만들어보았다. Books 클래스는 Book 객체를 list로 가지고 있는데 List가 이미 iterator 메소드를 정의하고 있기 때문에 books.iterator()로 list가 가지고 있는 iterator를 반환할 수 있도록 했다.  

Books 클래스

Book 클래스

 

향상된 for문을 iterable을 상속한 Books 클래스에 적용할 수 있다.

 

 

 

하지만 Books 클래스는 List로  Book을 가지고 있어서 구현이 쉬웠지만 iterator() 메소드를 직접 구현해야 될 때는 hasNext()와 next() 메소드를 만들어 주어야 한다. 

 

 

 

예제 2

Book 클래스를 조금 수정했는데 다음 Book 객체의 reference를 가지고 있을 next 필드를 추가했다.

Book 클래스

 

Books 클래스는 기존과 동일하지만 생성자로 Book 객체를 받고 새로 만든 BookIterator를 반환한다. 

 

 

Iterator를 상속받은 custom BookIterator를 만들어서 직접 hasNext()와 next() 메소드를 오버라이드 해주고 나면

BookIterator 클래스

 

 

향상된 for 문을 예제 1과 동일하게 사용할 수 있다. 

 

 

 

 

 

 

 

참고자료 

1. https://cs61b-2.gitbook.io/cs61b-textbook/12.-inheritance-iv-iterators-object-methods/12.3-iteration