자바 스트림(Java Stream) API - 2 (스트림 API 활용)

걸러내기

• Filter(Predicate)

• 예) 이름이 3글자 이상인 데이터만 새로운 스트림으로

 

변경하기

• Map(Function)

• 예) 각각의 Post 인스턴스에서 String title만 새로운 스트림으로

• FlatMap(Function)

• 예) List<Stream<String>>을 String의 스트림으로

 

생성하기

• generate(Supplier) 또는 Iterate(T seed, UnaryOperator)

• 예) 10부터 1씩 증가하는 무제한 숫자 스트림

• 예) 랜덤 int 무제한 스트림

 

제한하기

• limit(long)

• 예) 최대 5개의 요소가 담긴 스트림을 리턴한다.

• skip(long)

• 예) 앞에서 3개를 뺀 나머지 스트림을 리턴한다.

 

스트림에 있는 데이터가 특정 조건을 만족하는지 확인

• anyMatch(), allMatch(), nonMatch()

• 예) k로 시작하는 문자열이 있는지 확인한다. (true 또는 false를 리턴한다.)

• 예) 스트림에 있는 모든 값이 10보다 작은지 확인한다.

 

개수 세기

• count()

• 예) 10보다 큰 수의 개수를 센다.

 

스트림을 데이터 하나로 뭉치기

• reduce(identity, BiFunction), collect(), sum(), max()

• 예) 모든 숫자 합 구하기

• 예) 모든 데이터를 하나의 List 또는 Set에 옮겨 담기

 

스트림 API 활용 예제

public class OnlineClass {

    private Integer id;

    private String title;

    private boolean closed;

    public OnlineClass(Integer id, String title, boolean closed) {
        this.id = id;
        this.title = title;
        this.closed = closed;
    }

    public Integer getId() {
        return id;
    }

    public void setId(Integer id) {
        this.id = id;
    }

    public String getTitle() {
        return title;
    }

    public void setTitle(String title) {
        this.title = title;
    }

    public boolean isClosed() {
        return closed;
    }

    public void setClosed(boolean closed) {
        this.closed = closed;
    }
}

 

public class Main {
    public static void main(String[] args) {
        List<OnlineClass> springClasses = new ArrayList<>();
        springClasses.add(new OnlineClass(1, "spring boot", true));
        springClasses.add(new OnlineClass(2, "spring data jpa", true));
        springClasses.add(new OnlineClass(3, "spring mvc", false));
        springClasses.add(new OnlineClass(4, "spring core", false));
        springClasses.add(new OnlineClass(5, "rest api development", false));

        System.out.println("spring 으로 시작하는 수업");
        springClasses.stream()
                .filter((c) -> c.getTitle().startsWith("spring"))
                .forEach((c) -> System.out.println("\t" + c.getId()));
        System.out.println("#######################");

        System.out.println("close 되지 않은 수업");
        springClasses.stream()
                .filter(Predicate.not(OnlineClass::isClosed))
                .forEach((c) -> System.out.println("\t" + c.getId()));
        System.out.println("#######################");

        System.out.println("수업 이름만 모아서 스트림 만들기");
        springClasses.stream()
                .map(OnlineClass::getTitle)
                .forEach(System.out::println);
        System.out.println("#######################");

        List<OnlineClass> javaClasses = new ArrayList<>();
        javaClasses.add(new OnlineClass(1, "The Java, Test", true));
        javaClasses.add(new OnlineClass(2, "The Java, Code manipulation", true));
        javaClasses.add(new OnlineClass(3, "The Java, 8 to 11", false));

        List<List<OnlineClass>> devHistoryEvent = new ArrayList<>();
        devHistoryEvent.add(springClasses);
        devHistoryEvent.add(javaClasses);

        System.out.println("두 수업 목록에 들어있는 모든 수업 아이디 출력");
        devHistoryEvent.stream()
                .flatMap(Collection::stream) //1. 리스트를 납작하게 만들어 온라인 클래스로만 스트림을 구성
                .forEach(c -> System.out.println(c.getId())); //2. 온라인 클래스에서 수업 아이디 출력
        System.out.println("#######################");

        System.out.println("10부터 1씩 증가하는 무제한 스트림 중에서 앞에 10개 빼고 최대 10개 까지만");
        Stream.iterate(10, i -> i + 1)
                .skip(10) //1. 처음의 10개 (제외)
                .limit(10) //2. 이후 10개 까지 (제한)
                .forEach(System.out::println); // 20~ 29
        System.out.println("#######################");

        System.out.println("자바 수업 중에 Test가 들어있는 수업이 있는지 확인");
        boolean test = javaClasses.stream().anyMatch(c -> c.getTitle().contains("Test"));
        System.out.println(test);
        System.out.println("#######################");

        System.out.println("스프링 수업 중에 제목에 spring이 들어간 제목만 모아서 List로 만들기");
        List<String> spring = springClasses.stream()
                .filter(c -> c.getTitle().contains("spring"))
                .map(OnlineClass::getTitle)
                .collect(Collectors.toList());
        spring.forEach(System.out::println);
        System.out.println("=================");
        List<String> spring1 = springClasses.stream()
                .map(OnlineClass::getTitle)
                .filter(c -> c.contains("spring"))
                .collect(Collectors.toList());
        spring1.forEach(System.out::println);

    }
}

---

[참고자료]

더 자바, Java 8, 백기선