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 |   17 +++++++++++++----
 1 files changed, 13 insertions(+), 4 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 af9c247..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)
@@ -74,14 +78,19 @@
 
     public ConsultantDetailDTO getConsultantDetail(String agentNo) {
         Consultant consultant = consultantRepository.findOneByAgentNo(agentNo).orElseThrow(ConsultantNotFoundException::new);
-        return consultantMapper.toDetailDto(consultant);
+        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,7 +116,7 @@
 	}
 
     public void removeConsultantFromCustomList(String agentNo) {
-        Long customId = SecurityUtils.getCustomerId();
+        Long customId = SecurityUtils.getCustomerDBId();
         Consultant consultant = consultantRepository.findOneByAgentNo(agentNo).orElseThrow(ConsultantNotFoundException::new);
         CustomerFavoriteConsultant target = customerFavoriteConsultantRepository.findOneByCustomerIdAndConsultant(customId, consultant).orElse(null);
 

--
Gitblit v1.8.0