[Java] Iterator()

먼저 Iterator를 설명하기 전에 제가 DB에서 조회한 목록 중 if 조건에 만족한 목록만 가져와야 하는 코드를 작성해야 했습니다. 그래서 list를 listTemp.addAll(list) 하고 listTemp를 for문 돌려서 if 조건에 만족하지 않은 데이터를 List에서 remove 했었습니다. 근데 혹시 다른 방법으로도 작성할 수 있지 않을까 해서 찾아보니 Iterator를 사용하는 방법도 있어서 글을 쓰게 되었습니다.

 

첫 번째 코드 블록은 처음에 작성한 코드와 같게 작성한 예제 코드이고, 두 번째 코드 블록은 Iterator를 사용한 예제 코드입니다.

 

1. Iterator를 사용하지 않은 코드 블록

List<Map<String, Object>> list = new ArrayList<Map<String, Object>>();

Map<String, Object> map0 = new HashMap<>();
map0.put("name", "맹구");

Map<String, Object> map1 = new HashMap<>();
map1.put("name", "짱구");

Map<String, Object> map2 = new HashMap<>();
map2.put("name", "철수");

Map<String, Object> map3 = new HashMap<>();
map3.put("name", "짱아");

list.add(map0);
list.add(map1);
list.add(map2);
list.add(map3);

String onlyFamily = "Y";

if ("Y".equals(onlyFamily)) {
    List<Map<String, Object>> listTemp = new ArrayList<Map<String, Object>>();
    listTemp.addAll(list);
    
    for (Map<String, Object> map : listTemp) {
        String name = map.get("name").toString();

        if (!"짱구".equals(name) && !"짱아".equals(name)) {
            list.remove(map);
        }
    }
}
System.out.println(list);
// 결과: [{name=짱구}, {name=짱아}]

2. Iterator를 사용한 코드 블록

List<Map<String, Object>> list = new ArrayList<Map<String, Object>>();

Map<String, Object> map0 = new HashMap<>();
map0.put("name", "맹구");

Map<String, Object> map1 = new HashMap<>();
map1.put("name", "짱구");

Map<String, Object> map2 = new HashMap<>();
map2.put("name", "철수");

Map<String, Object> map3 = new HashMap<>();
map3.put("name", "짱아");

list.add(map0);
list.add(map1);
list.add(map2);
list.add(map3);

String onlyFamily = "Y";

if ("Y".equals(onlyFamily)) {

    for (Iterator<Map<String, Object>> it = list.iterator(); it.hasNext();) {
        Map<String, Object> map = it.next();

        String name = String.valueOf(map.get("name"));
        if (!"짱구".equals(name) && !"짱아".equals(name)) {
            it.remove();
        }
    }
}

System.out.println(list);
// 결과: [{name=짱구}, {name=짱아}]

 

Iterator<E> iterator()를 설명하면 컬렉션의 반복자를 반환하는 메서드입니다.

자바의 컬렉션 프레임워크인 List, Set에 저장된 데이터를 Iterator를 통해서 데이터를 읽어오거나 삭제할 수 있습니다.

 

Iterator의 장점은 공통으로 사용이 가능하며, 사용법이 간단합니다. 단 처리속도가 느립니다.

 

저는 대용량의 데이터를 가지고 처리하는 게 아니어서 Iterator를 사용했습니다.

만약 대용량의 데이터를 가지고 처리한다면 사용하지 않을 거 같네요.

 

끝.