Java 8 - Consumer 예제

javajava8

Java 8에서 도입된 Consumer의 기본적인 사용 방법 및 예제에 대해서 알아보겠습니다.

1. Consumer

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

@FunctionalInterface
public interface Consumer<T> {

    /**
     * Performs this operation on the given argument.
     *
     * @param t the input argument
     */
    void accept(T t);

}

2. Consumer 구현 방법 및 예제

Consumer는 Lambda 표현식으로 구현할 수 있습니다.

accept() 호출과 함께 인자를 전달하면 구현된 내용이 수행됩니다. 리턴 값은 없습니다.

import java.util.function.Consumer;

public class ConsumerExample1 {

    public static void main(String[] args) {
        Consumer<String> consumer = s -> System.out.println(s.toUpperCase());

        consumer.accept("hello world");
    }
}

Output:

HELLO WORLD

3. Primitive Consumer

Java는 Primitive type에 대한 Consumer 클래스를 제공합니다. IntConsumer 외에 DoubleConsumer, LongConsumer가 있습니다.

import java.util.function.IntConsumer;

public class ConsumerExample2 {

    public static void main(String[] args) {
        IntConsumer consumer = n -> System.out.println(n * 100);

        consumer.accept(5);

        consumer.accept(10);
    }
}

Output:

500
1000

4. List.forEach()와 Consumer로 리스트 순회

List.forEach()는 인자로 Consumer를 받습니다.

다음과 같이 Consumer를 이용하여 List의 모든 아이템들을 출력할 수 있습니다.

import java.util.Arrays;
import java.util.List;
import java.util.function.Consumer;

public class ConsumerExample3 {

    public static void main(String[] args) {
        Consumer<String> consumer = s -> System.out.println(s.toUpperCase());

        List<String> list = Arrays.asList("apple", "kiwi", "orange");

        list.forEach(consumer);
    }
}

Output:

APPLE
KIWI
ORANGE

5. andThen()으로 다수의 Consumer 연산

andThen()은 Consumer들을 연결해주는 Chaining 메소드입니다.

아래 코드에서 addNumber.andThen(printList).accept(fruits)를 호출하면 addNumber()가 수행된 이후에 printList()가 수행됩니다. 두개의 메소드로 전달되는 인자는 fruits로, 동일한 객체가 전달됩니다.

import java.util.Arrays;
import java.util.List;
import java.util.function.Consumer;

public class ConsumerExample4 {

    public static void main(String[] args) {
        List<String> fruits = Arrays.asList("apple", "kiwi", "orange");

        Consumer<List<String>> addNumber = list -> {
            for (int i = 0; i < list.size(); i++) {
                list.set(i, (i + 1) + ". " + list.get(i));
            }
        };

        Consumer<List<String>> printList = list -> {
            for (String fruit : list) {
                System.out.println(fruit);
            }
        };

        addNumber.andThen(printList).accept(fruits);
    }
}

Output:

1. apple
2. kiwi
3. orange

실제로 JDK에서 Consumer의 andThen()은 아래처럼 구현되어있습니다.

// Consumer.java
public interface Consumer<T> {
  ...
  default Consumer<T> andThen(Consumer<? super T> after) {
      Objects.requireNonNull(after);
      return (T t) -> { accept(t); after.accept(t); };
  }
}

6. 함수의 인자로 Consumer 전달 (Higher Order Function)

다음과 같이 Consumer를 인자로 받는 함수를 정의할 수 있습니다.

import java.util.Arrays;
import java.util.List;
import java.util.function.Consumer;

public class ConsumerExample5 {

    public static void main(String[] args) {
        List<String> fruits = Arrays.asList("apple", "kiwi", "orange");

        Consumer<String> printFruit = fruit -> System.out.println("I like " + fruit);

        forEach(fruits, printFruit);
    }

    static <T> void forEach(List<T> list, Consumer<T> consumer) {
        for (T t : list) {
            consumer.accept(t);
        }
    }

}

Output:

I like apple
I like kiwi
I like orange
Loading script...

Related Posts

codechachaCopyright ©2019 codechacha