diff --git a/src/main/java/com/ruoyi/airline/service/impl/AirlineMarkerServiceImpl.java b/src/main/java/com/ruoyi/airline/service/impl/AirlineMarkerServiceImpl.java index 91e95b2..a57bb07 100644 --- a/src/main/java/com/ruoyi/airline/service/impl/AirlineMarkerServiceImpl.java +++ b/src/main/java/com/ruoyi/airline/service/impl/AirlineMarkerServiceImpl.java @@ -13,6 +13,7 @@ import org.springframework.beans.factory.annotation.Autowired; import org.springframework.stereotype.Service; import com.ruoyi.common.security.utils.SecurityUtils; +import java.util.ArrayList; import java.util.List; /** @@ -87,22 +88,22 @@ public class AirlineMarkerServiceImpl implements IAirlineMarkerService { @Override public List selectMarkerList(AirlineMarkerDTO dto) { - AirlineMarker model = AirlineMarkerServiceConvert.to(dto); - List markers = iAirlineMarkerDomain.selectMarkerList(model); - List dtos = AirlineMarkerServiceConvert.fromList(markers); - + + AirlineMarkerGroupInfoDTO groupInfoDTO = new AirlineMarkerGroupInfoDTO(); + groupInfoDTO.setMarkerId(dto.getGroupId()); + List groupInfos = iAirlineMarkerGroupInfoService.selectMarkerGroupInfoList(groupInfoDTO); + + List result = new ArrayList<>(); // 为每个标注设置分组ID - for (AirlineMarkerDTO markerDto : dtos) { + for (AirlineMarkerGroupInfoDTO markerDto : groupInfos) { // 查询分组关系 - AirlineMarkerGroupInfoDTO groupInfoDTO = new AirlineMarkerGroupInfoDTO(); - groupInfoDTO.setMarkerId(markerDto.getId()); - List groupInfos = iAirlineMarkerGroupInfoService.selectMarkerGroupInfoList(groupInfoDTO); - if (!groupInfos.isEmpty()) { - markerDto.setGroupId(groupInfos.get(0).getGroupId()); - } + AirlineMarker airlineMarker = iAirlineMarkerDomain.selectMarkerById(markerDto.getMarkerId()); + AirlineMarkerDTO airlineMarkerDTO = AirlineMarkerServiceConvert.from(airlineMarker); + airlineMarkerDTO.setGroupId(groupInfos.get(0).getGroupId()); + result.add(airlineMarkerDTO); } - return dtos; + return result; } @Override