[JAVA] Map, Stream - Boolean값이 true인 value의 key 반환받기

 

😲 자바 Boolean값이 true인 value의 key 반환받기

 

 

 

코드를 짜다보면 boolean값을 여러 개 가진 객체에서 true값을 가진 변수만 추출해야 하는 경우가 종종 있다.

 

나는 이번에 참가한 서울시 IoT 해커톤에서 위와 같은 기능을 구현해야 했었다.

 

하지만 이를 효율적인 코드로 구현하는 방법을 몰랐기에

 

if문을 boolean변수의 수만큼 사용하여 true인 데이터의 변수명을 List<String>에 저장하도록 구현했다.

 

 

 

문제의 DTO

@Getter
@Setter
public class AmenityRequestDto {
    private boolean elevator;   // 엘리베이터
    private boolean wideExit;   //넓은 출입구
    private boolean ramp;   // 경사로
    private boolean accessRoads;    // 접근로
    private boolean wheelchairLift;    // 휠체어 리프트
    private boolean brailleBlock;    // 점자 블록
    private boolean exGuidance;    // 시각장애인 유도 안내
    private boolean exTicketOffice;    // 장애인 전용 매표소
    private boolean wideRestroom;   // 장애인 전용 화장실
    private Double longitude;
    private Double latitude;

    public Map<String, Boolean> toMap(){
        ObjectMapper objectMapper = new ObjectMapper();
        HashMap result = objectMapper.convertValue(this, HashMap.class);
        result.remove("longitude");
        result.remove("latitude");
        return result;
    }
}

 

 

Before Refactoring

private List<String> getExistAmenityList(AmenityRequestDto data) {
        List<String> trueList = new ArrayList<>();
        if (data.isElevator()) trueList.add("a.elevator");
        if (data.isWideExit()) trueList.add("a.wideExit");
        if (data.isRamp()) trueList.add("a.ramp");
        if (data.isAccessRoads()) trueList.add("a.accessRoads");
        if (data.isWheelchairLift()) trueList.add("a.wheelchairLift");
        if (data.isBrailleBlock()) trueList.add("a.brailleBlock");
        if (data.isExGuidance()) trueList.add("a.exGuidance");
        if (data.isExTicketOffice()) trueList.add("a.exTicketOffice");
        return trueList;
    }

 

아무리 봐도 하드코딩이 아닌가?라는 생각에 리팩터링을 결심했다.

 

어떻게 코드를 짜야할지 고민 중 지인의 도움으로 Map을 통해 true인 값들만 List로 저장하는 것이 가능하다는 것을 알게 되었다.

 

After Refactoring

private List<String> getExistAmenityList(AmenityRequestDto data) {
    List<String> collect = data.toMap()
            .entrySet() // Set<Map<K,V>.Entry<String,Boolean>> 리턴
            .stream() // Stream<Map<K,V>.Entry<String,Boolean>> 리턴
            .filter(Map.Entry::getValue) // m -> m.getValue().equals(true) 와 같음
            .map(Map.Entry::getKey) 
            .collect(Collectors.toList());
    collect.forEach(c -> System.out.println("c = " + c));

    return collect;
}

그렇게 해서 구현한 것이 위의 코드이다. 

 

훨씬 깔끔해 보이지 않은가?

 

---

👨‍💻 코드 설명

 

 

1. toMap()

Dto객체에 toMap() 메서드를 구현하여 Map으로 반환받는다.

 

 

2.entrySet()

Map의 entrySet() 메서드를 사용하여 Set객체로 묶인 Map<Key, Value>를 가져온다.

 

map에는 keySet과 entrySet이 있는데

keySet은 key만 필요한 경우 사용하고, entrySet은 key와 value가 모두 필요할 때 사용한다.

Set객체로 값을 리턴해준다.

 

 

3. Set객체를 stream() 메서드를 사용하여 Stream객체로 변환한다.

.

 

4. filter(Map.Entry::getValue)

filter를 사용하여 value가 True인 데이터만 필터링

[ Stream에서 사용하는 ::는 왼쪽 객체(Map.Entry)에서 오른쪽 메서드(getValue)를 사용한다는 뜻이다. ]

 

filter() 메서드는 파라미터가 true인 데이터만 필터링하여 리턴해준다.

그래서 m -> m.getValue().equals(true)로 사용하는 것과 똑같다.

 

 

5. map(Map.Entry::getKey)

Value값이 True인 데이터만으로 필터링된 Map의 Key를 Stream으로 묶은 객체로 반환

 

6. collect(Collectors.toList())

반환받은 Stream을 Collectors 클래스의 toList() 메서드를 통해 List형태로 반환

 

 

---

🧹 정리

 

.filter(Map.Entry::getValue) -> Map에서 getValue가 true인 데이터만 필터링

.map(Map.Entry::getKey) -> value가 true인 데이터의 key들을 리턴

collect(Collectors.toList()) -> 리턴 받은 key들을 Collectors 클래스를 사용해 List형태로 변환

 

 

 

 

앞으로도 계속 좋은 코드를 위해 고민하고,
이미 끝난 프로젝트더라도 이것저것 실험해보는 도전 정신을 불태우자.