Java - forEach 사용 방법

JS · 05 Dec 2020

forEach는 Java8에서 추가된 메소드이며, List, Map 등을 순회(Iterate)하는데 사용됩니다.

List, Map, Set, Array에서 forEach()를 사용하는 방법을 알아보겠습니다.

List forEach

List와 같은 Collection에서 forEach()는 다음과 같이 Consumer라는 함수형 인터페이스를 인자로 받습니다.

default void forEach(Consumer<? super T> action) {
    for (T t : this) {
        action.accept(t);
    }
}

Consumer는 1개의 인자를 받고 리턴 값이 없는 함수형 인터페이스입니다.

@FunctionalInterface
public interface Consumer<T> {
    void accept(T t);
}

다음 예제와 같이 Cosumer 객체를 forEach에 전달하여 List의 모든 아이템을 순회할 수 있습니다.

List<String> items = new ArrayList<>();
items.add("Paris");
items.add("Seoul");
items.add("Tokyo");
items.add("Washington");

items.forEach(new Consumer<String>() {
    @Override
    public void accept(String name) {
        System.out.println(name);
    }
});

Output:

Paris
Seoul
Tokyo
Washington

익명 클래스 대신에 Lambda를 사용하면 더 간단한 코드로 구현할 수 있습니다.

List<String> items = new ArrayList<>();
items.add("Paris");
items.add("Seoul");
items.add("Tokyo");
items.add("Washington");

items.forEach(name -> System.out.println(name));

메소드 레퍼런스를 사용하면 다음과 같이 더 간단히 표현할 수도 있습니다.

items.forEach(System.out::println);

Set forEach

Set도 List와 동일합니다.

다음과 같이 모든 아이템을 순회할 수 있습니다.

Set<String> set = new HashSet<>();
set.add("Paris");
set.add("Seoul");
set.add("Tokyo");
set.add("Washington");

set.forEach(item -> System.out.println(item));

Output:

Seoul
Tokyo
Paris
Washington

Map forEach

Collection이 아닌 Map은 List와 조금 다릅니다. Map은 Key와 Value라는 두개의 값을 Pair로 저장합니다.

Map.forEach()는 다음과 같이 BiConsumer라는 함수형 인터페이스를 인자로 받습니다.

default void forEach(BiConsumer<? super K, ? super V> action) {
    Objects.requireNonNull(action);
    for (Map.Entry<K, V> entry : entrySet()) {
        K k = entry.getKey();
        V v = entry.getValue();
        action.accept(k, v);
    }
}

BiConsumer는 두개의 인자를 받고 리턴 값은 없는 함수형 인터페이스입니다.

@FunctionalInterface
public interface BiConsumer<T, U> {
    void accept(T t, U u);
}

다음과 같이 BiConsumer 객체를 인자로 전달하여 Map의 모든 아이템을 순회할 수 있습니다.

Map<String, String> map = new HashMap<>();
map.put("Paris", "France");
map.put("Seoul", "Korea");
map.put("Tokyo", "Japan");
map.put("Washington", "USA");

map.forEach((new BiConsumer<String, String>() {
    @Override
    public void accept(String k, String v) {
        System.out.println("key: " + k + ", value: " + v);
    }
}));

Output:

key: Seoul, value: Korea
key: Tokyo, value: Japan
key: Paris, value: France
key: Washington, value: USA

익명 클래스 대신에 Lambda를 사용하면 다음과 같이 더 간단히 표현할 수 있습니다.

map.forEach((k, v) -> {
    System.out.println("key: " + k + ", value: " + v);
});

Map.entrySet()에 대한 forEach

다음과 같이 EntrySet을 얻고, Map을 순회할 수 있습니다.

Map<String, String> map = new HashMap<>();
map.put("Paris", "France");
map.put("Seoul", "Korea");
map.put("Tokyo", "Japan");
map.put("Washington", "USA");

map.entrySet().forEach(entry -> {
    System.out.println("key: " + entry.getKey()
            + ", value: " + entry.getValue());
});

Output:

key: Seoul, value: Korea
key: Tokyo, value: Japan
key: Paris, value: France
key: Washington, value: USA

Array forEach

Array는 다음과 같이 Stream으로 변경 후 forEach()를 사용할 수 있습니다.

String[] array = {"Paris", "Seoul", "Tokyo", "Washington"};
Arrays.stream(array).forEach(item -> System.out.println(item));

Output:

Paris
Seoul
Tokyo
Washington
댓글을 보거나 쓰려면 이 버튼을 눌러주세요.
codechachaCopyright ©2019 codechacha