Lambda & Stream 활용

1. 활용 가이드

가. Stream<Integer> 보다 IntStream을 적극 활용

나. stream만으로 배열의 값 생성하기

다. Map::values() → List

람다를 통한 중복 제거

2. 스트림 생성, 중개연산, 최종연산

1. 활용 가이드

가. Stream<Integer> 보다 IntStream을 적극 활용

•

Stream<Integer>를 int 형 등의 기본형으로 unboxing하는 것은 비효율적인 작업

•

IntSream 등 기본 자료형의 Stream으로 처리하는데 유용한 메소드들이 많이 있음

나. stream만으로 배열의 값 생성하기

•

첫번째 생성한 stream으로 배열 구성

→ flatMap 사용하여 Integer → Stream<int[]>로 변환

IntStream.rangeClosed(START_NUM, END_NUM)
    .boxed() 
		// flatMap은 IntStream을 인자로 받지 않음, Stream<T[]> -> Stream<T>로 변환
    // integer -> Stream.of(1, 2, 3, 4, 5, 6) : 1 ~ 6까지를 6번 생성
		// .map(integer1 -> new int[]{integer, integer1}) : {1, 1}, {1, 2} ~ {6, 6} 생성
    .flatMap(integer -> Stream.of(1, 2, 3, 4, 5, 6).map(integer1 -> new int[]{integer, integer1}))
    .forEach(arr -> System.out.println(arr[0] + " " + arr[1]));
Java
복사

•

만약 flatMap 대신 map을 사용했다면 Integer → Stream<Stream<int[]>>으로 변환됨

→ 내부에 map이 쓰이기 때문에 flatMap으로 끊어줘야함

다. Map::values() → List

•

다음과 같이 ArrayList 생성자의 인자로 tasks.values()를 전달하여 List 타입으로 반환 받을 수 있음

Map<Integer, Task> tasks = new HashMap<>();

@Override
public List<Task> findAllByCompletedFalse() {
    return new ArrayList<>(tasks.values());
}
Java
복사

•

단, Map의 value 각각에 조건을 걸기 위해서는 stream 활용

Map<Integer, Task> tasks = new HashMap<>();

@Override
public List<Task> findAllByCompletedFalse() {
    return tasks.values().stream()
            .filter(task -> !task.isCompleted())
            .collect(Collectors.toCollection(ArrayList::new));
}
Java
복사

람다를 통한 중복 제거

•

적용 절차

1) 공통부분과 변경부분을 구분

2) 변경부분에 대해 인터페이스의 메소드로 선언

3) 인터페이스의 메소드를 본문의 메소드에 인자로 전달함

→ 하나의 인터페이스에는 하나의 메소드만 선언됨

→ 단, 인터페이스의 메소드를 overriding해서 사용할 수 있음

4) 인터페이스의 메소드의 구현부는 본문의 메소드를 사용하는 곳(테스트 코드 등)에 익명 클래스로 구현

5) 익명 클래스를 람다로 변경

6) 스트림을 활용해서 본문의 유사 메소드 여러 개를 하나의 메소드로 통합

•

실습(위의 적용 절차에 따라)

1) Lambda.java의 변경부분은 내부 조건(노란색 배경 부분), 공통부분은 그 외 전부

// Lambda
public static int sumAll(List<Integer> numbers) {
	    int total = 0;
	    for (int number : numbers) {
	        total += number;
	    }
	    return total;
	}
	
	public static int sumAllEven(List<Integer> numbers) {
	    int total = 0;
	    for (int number : numbers) {
	        if (number % 2 == 0) {
	            total += number;
	        }
	    }
	    return total;
	}

	public static int sumAllOverThree(List<Integer> numbers) {
	    int total = 0;
	    for (int number : numbers) {
	    	if (number > 3) {
	    		total += number;
	    	}
	    }
	    return total;
	}
Java
복사

2) 변경부분에 대해 인터페이스의 메소드로 선언

public interface Conditional {
    boolean condition(Integer number);
}
Java
복사

3) 인터페이스의 메소드를 본문의 관련 메소드에 인자로 전달

public static int sumAll(List<Integer> numbers, Conditional conditional) {
    int total = 0;
    for (int number : numbers) {
        if (conditional.condition(number)) {
            total += number;
        }
    }
    return total;
}

public static int sumAllEven(List<Integer> numbers, Conditional conditional) {
    int total = 0;
    for (int number : numbers) {
        if (conditional.condition(number)) {
            total += number;
        }
    }
    return total;
}

public static int sumAllOverThree(List<Integer> numbers, Conditional conditional) {
    int total = 0;
    for (int number : numbers) {
        if (conditional.condition(number)) {
            total += number;
        }
    }
    return total;
}
Java
복사

4) 본문의 메소드를 사용하는 곳에서 인터페이스의 메소드의 구현부를 익명클래스로 작성

// LambdaTest

@Test
public void sumAll() throws Exception {
	int sum = Lambda.sumAll(numbers, new Lambda.Conditional() {
		@Override
		public boolean condition(Integer number) {
			return true;
		}
	});
	assertThat(sum).isEqualTo(21);
}

@Test
public void sumAllEven() throws Exception {
	int sum = Lambda.sumAllEven(numbers, new Lambda.Conditional() {
		@Override
		public boolean condition(Integer number) {
			return number % 2 == 0;
		}
	});
	assertThat(sum).isEqualTo(12);
}

@Test
public void sumAllOverThree() throws Exception {
	int sum = Lambda.sumAllOverThree(numbers, new Lambda.Conditional() {
		@Override
		public boolean condition(Integer number) {
			return number > 3;
		}
	});
	assertThat(sum).isEqualTo(15);
}
Java
복사

5) 익명 클래스를 람다로 대체

// LambdaTest
@Test
public void sumAll() throws Exception {
	int sum = Lambda.sumAll(numbers, number -> true);
	assertThat(sum).isEqualTo(21);
}

@Test
public void sumAllEven() throws Exception {
	int sum = Lambda.sumAllEven(numbers, number -> number % 2 == 0);
	assertThat(sum).isEqualTo(12);
}

@Test
public void sumAllOverThree() throws Exception {
	int sum = Lambda.sumAllOverThree(numbers, number -> number > 3);
	assertThat(sum).isEqualTo(15);
}
Java
복사

6) 스트림을 활용해서 본문의 유사 메소드 다수를 하나의 메소드로 통합

// Lambda.java
public static int sumAllByCondition(List<Integer> numbers, Conditional conditional) {
    return numbers.stream()
            .filter(number -> conditional.condition(number))
            .reduce(0, (frontNum, nextNum) -> Integer.sum(frontNum, nextNum));
}

// to(메소드 참조 활용하여 간략하게 정리)

public static int sumAllByCondition(List<Integer> numbers, Conditional conditional) {
    return numbers.stream()
            .filter(conditional::condition)
            .reduce(0, Integer::sum);
}
Java
복사