From bb9499339d13dcc8a9ed6d75e6b040ecd8a57acb Mon Sep 17 00:00:00 2001 From: wayne <wayne8692wayne8692@gmail.com> Date: 星期二, 30 十一月 2021 13:06:17 +0800 Subject: [PATCH] [ref] 重構OTP response code --- pamapi/src/main/java/com/pollex/pam/service/ConsultantService.java | 21 +++++++++++++++------ 1 files changed, 15 insertions(+), 6 deletions(-) diff --git a/pamapi/src/main/java/com/pollex/pam/service/ConsultantService.java b/pamapi/src/main/java/com/pollex/pam/service/ConsultantService.java index 3dbcbd9..76af3d3 100644 --- a/pamapi/src/main/java/com/pollex/pam/service/ConsultantService.java +++ b/pamapi/src/main/java/com/pollex/pam/service/ConsultantService.java @@ -3,6 +3,7 @@ import com.pollex.pam.domain.AppointmentCustomerView; import com.pollex.pam.domain.Consultant; import com.pollex.pam.domain.CustomerFavoriteConsultant; +import com.pollex.pam.enums.LoginResult; import com.pollex.pam.repository.AppointmentCustomerViewRepository; import com.pollex.pam.repository.ConsultantRepository; import com.pollex.pam.repository.CustomerFavoriteConsultantRepository; @@ -43,8 +44,11 @@ @Autowired AppointmentCustomerViewMapper appointmentCustomerViewMapper; + @Autowired + LoginRecordService loginRecordService; + public List<ConsultantDTO> getMyConsultantList() { - Long userId = SecurityUtils.getCustomerId(); + Long userId = SecurityUtils.getCustomerDBId(); return customerFavoriteConsultantRepository.findAllByCustomerId(userId) .stream() .map(consultantMapper::toDto) @@ -73,15 +77,20 @@ } public ConsultantDetailDTO getConsultantDetail(String agentNo) { - Consultant consultant = consultantRepository.findFirstByAgentNo(agentNo).orElseThrow(ConsultantNotFoundException::new); - return consultantMapper.toDetailDto(consultant); + Consultant consultant = consultantRepository.findOneByAgentNo(agentNo).orElseThrow(ConsultantNotFoundException::new); + ConsultantDetailDTO consultantDetailDTO = consultantMapper.toDetailDto(consultant); + + loginRecordService.findLatestLoginRecord(agentNo, LoginResult.SUCCESS) + .ifPresent(loginRecord -> consultantDetailDTO.setLatestLoginTime(loginRecord.getLoginDate())); + + return consultantDetailDTO; } @Transactional public void addConsultantToCustomList(AddConsultantParam param) { List<String> agentNoList = param.getAgentNoList(); List<Consultant> consultants = consultantRepository.findAllByAgentNoIn(agentNoList); - Long userId = SecurityUtils.getCustomerId(); + Long userId = SecurityUtils.getCustomerDBId(); consultants.forEach(consultant -> { boolean isConsultantInList = customerFavoriteConsultantRepository.findOneByCustomerIdAndConsultant(userId, consultant).isPresent(); @@ -107,8 +116,8 @@ } public void removeConsultantFromCustomList(String agentNo) { - Long customId = SecurityUtils.getCustomerId(); - Consultant consultant = consultantRepository.findFirstByAgentNo(agentNo).orElseThrow(ConsultantNotFoundException::new); + Long customId = SecurityUtils.getCustomerDBId(); + Consultant consultant = consultantRepository.findOneByAgentNo(agentNo).orElseThrow(ConsultantNotFoundException::new); CustomerFavoriteConsultant target = customerFavoriteConsultantRepository.findOneByCustomerIdAndConsultant(customId, consultant).orElse(null); if(target != null) { -- Gitblit v1.8.0