Skip to content

Commit

Permalink
[#38]Refactor: Change TownMapService for new infraPlace api response …
Browse files Browse the repository at this point in the history
…format
  • Loading branch information
gmkim20713 committed Feb 15, 2023
1 parent 3f8bb84 commit d5b1f0b
Showing 1 changed file with 14 additions and 17 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,7 @@
import java.util.Collections;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Optional;
import java.util.stream.Collectors;
import lombok.RequiredArgsConstructor;
Expand All @@ -15,11 +16,12 @@
import yapp.domain.member.entity.MemberWishTown;
import yapp.domain.member.repository.MemberWishTownRepository;
import yapp.domain.townMap.converter.LocationConverter;
import yapp.domain.townMap.dto.response.InfraPlaceDto;
import yapp.domain.townMap.converter.PlaceConverter;
import yapp.domain.townMap.dto.InfraPlaceDto;
import yapp.domain.townMap.dto.response.LocationInfoResponse;
import yapp.domain.townMap.dto.response.ThemePlaceDto;
import yapp.domain.townMap.dto.ThemePlaceDto;
import yapp.domain.townMap.dto.response.ThemePlaceResponse;
import yapp.domain.townMap.entity.Infra;
import yapp.domain.townMap.entity.Theme;
import yapp.domain.townMap.repository.InfraRepository;
import yapp.domain.townMap.repository.PlaceRepository;
import yapp.domain.townMap.repository.ThemeRepository;
Expand All @@ -35,6 +37,7 @@ public class TownMapService {
private final ThemeRepository themeRepository;
private final PlaceRepository placeRepository;
private final LocationConverter locationConverter;
private final PlaceConverter placeConverter;

public LocationInfoResponse getLocationInfo(
String memberId,
Expand Down Expand Up @@ -91,27 +94,21 @@ public HashMap<String, Object> getInfraPlaceInfo(
return infraPlaceListHashMap;
}

public HashMap<String, Object> getThemePlaceInfo(
public HashMap<String, List<ThemePlaceResponse>> getThemePlaceInfo(
Long object_id,
String category
) {

HashMap<String, Object> listHashMap = new HashMap<>();
HashMap<String, List<ThemePlaceResponse>> themePlaceListHashMap = new HashMap<>();

List<Theme> themeList = themeRepository.findThemeByCategory(category);
List<ThemePlaceResponse> themePlaceList = placeRepository.findByTheme(object_id, category)
.stream()
.map(placeConverter::toThemePlace)
.collect(Collectors.toList());

for (int i = 0; i < themeList.size(); i++) {
List<ThemePlaceDto> themePlaceList = placeRepository.findByTheme(
object_id, category, themeList.get(i).getSubCategory())
.stream()
.map(ThemePlaceDto::new)
.collect(Collectors.toList());
;

listHashMap.put(themeList.get(i).getSubCategoryName(), themePlaceList);
}
themePlaceListHashMap.put("placeList", themePlaceList);

return listHashMap;
return themePlaceListHashMap;
}

}

0 comments on commit d5b1f0b

Please sign in to comment.