STUDY/JAVA | SPRING
[Java8] stream()을 활용하여 객체의 특정 필드로 구성된 List 생성하기
nnh
2023. 6. 28. 14:59
728x90
api로 가져온 List<Object>에서 각 Object의 특정 필드값만 따로 추출할 필요가 있었다.
for문을 사용하여 각 객체에서 get을 통해 필드 값을 추출하는 방법도 있었지만,
자바8에서 새로 추가된 기능인 strem()을 활용하여 코드를 간결하게 작성해보았다.
간단한 예제로 정리를 해보자면
@Getter
public class PersonDto {
private String name;
private Integer age;
}
위와 같은 객체로 이루어진 List<PersonDto> personList에서 stream()을 이용하여 name 값만 추출하여 List<String>으로 만들어보자.
List<String> nameList = personList.stream().map(PersonDto::getName).collect(Collectors.toList();
먼저 stream()의 map 메서드를 사용하여 name만 추출한 다음 collect 메서드를 통해 List 타입으로 변환하면 된다.
+)
추출한 필드값을 구분자로 나열하여 하나의 문자열로 만드는 방법도 있다.
구분자를 ','로 지정했을 때의 코드는 아래와 같다.
String names = personList.map(PersonDto::getName).collect(Collectors.joining(","));
728x90