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 | 85 +++++++++++++++++++++++------------------- 1 files changed, 46 insertions(+), 39 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 d26676d..76af3d3 100644 --- a/pamapi/src/main/java/com/pollex/pam/service/ConsultantService.java +++ b/pamapi/src/main/java/com/pollex/pam/service/ConsultantService.java @@ -1,28 +1,23 @@ package com.pollex.pam.service; -import com.pollex.pam.domain.Appointment; import com.pollex.pam.domain.AppointmentCustomerView; import com.pollex.pam.domain.Consultant; -import com.pollex.pam.domain.CustomFavoriteConsultant; -import com.pollex.pam.domain.User; -import com.pollex.pam.enums.ContactStatusEnum; +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.CustomFavoriteConsultantRepository; +import com.pollex.pam.repository.CustomerFavoriteConsultantRepository; import com.pollex.pam.security.SecurityUtils; import com.pollex.pam.service.dto.*; import com.pollex.pam.service.mapper.AppointmentCustomerViewMapper; -import com.pollex.pam.service.mapper.AppointmentMapper; import com.pollex.pam.service.mapper.ConsultantMapper; import com.pollex.pam.web.rest.errors.ConsultantNotFoundException; -import com.pollex.pam.web.rest.errors.NotLoginException; import org.slf4j.Logger; import org.slf4j.LoggerFactory; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.stereotype.Service; import org.springframework.transaction.annotation.Transactional; -import java.util.Collections; import java.util.List; import java.util.stream.Collectors; @@ -31,35 +26,30 @@ private static final Logger log = LoggerFactory.getLogger(ConsultantService.class); - private final ConsultantRepository consultantRepository; - private final CustomFavoriteConsultantRepository customFavoriteConsultantRepository; - private final ConsultantMapper consultantMapper; - private final UserService userService; - + @Autowired + ConsultantRepository consultantRepository; + + @Autowired + CustomerFavoriteConsultantRepository customerFavoriteConsultantRepository; + + @Autowired + ConsultantMapper consultantMapper; + @Autowired AppointmentService appointmentService; - + @Autowired AppointmentCustomerViewRepository appointmentCustomerViewRepository; - + @Autowired AppointmentCustomerViewMapper appointmentCustomerViewMapper; - - public ConsultantService( - ConsultantRepository consultantRepository, - CustomFavoriteConsultantRepository customFavoriteConsultantRepository, - ConsultantMapper consultantMapper, - UserService userService - ) { - this.consultantRepository = consultantRepository; - this.customFavoriteConsultantRepository = customFavoriteConsultantRepository; - this.consultantMapper = consultantMapper; - this.userService = userService; - } + + @Autowired + LoginRecordService loginRecordService; public List<ConsultantDTO> getMyConsultantList() { - User user = userService.getUserWithAuthorities().orElseThrow(NotLoginException::new); - return customFavoriteConsultantRepository.findAllByCustomId(user.getId()) + Long userId = SecurityUtils.getCustomerDBId(); + return customerFavoriteConsultantRepository.findAllByCustomerId(userId) .stream() .map(consultantMapper::toDto) .collect(Collectors.toList()); @@ -87,29 +77,33 @@ } 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); - User user = userService.getUserWithAuthorities().orElseThrow(NotLoginException::new); + Long userId = SecurityUtils.getCustomerDBId(); consultants.forEach(consultant -> { - boolean isConsultantInList = customFavoriteConsultantRepository.findOneByCustomIdAndConsultant(user.getId(), consultant).isPresent(); + boolean isConsultantInList = customerFavoriteConsultantRepository.findOneByCustomerIdAndConsultant(userId, consultant).isPresent(); if(!isConsultantInList) { - CustomFavoriteConsultant customFavoriteConsultant = new CustomFavoriteConsultant(); - customFavoriteConsultant.setConsultant(consultant); - customFavoriteConsultant.setCustomId(user.getId()); - customFavoriteConsultant.setContactStatus(ContactStatusEnum.PICKED); + CustomerFavoriteConsultant customerFavoriteConsultant = new CustomerFavoriteConsultant(); + customerFavoriteConsultant.setConsultant(consultant); + customerFavoriteConsultant.setCustomerId(userId); - customFavoriteConsultantRepository.save(customFavoriteConsultant); + customerFavoriteConsultantRepository.save(customerFavoriteConsultant); } else { - log.info("The consultant is in custom favorite list! customId = {}, consultant AgentNo = {}", user.getId(), consultant.getAgentNo()); + log.info("The consultant is in customer favorite list! customId = {}, consultant AgentNo = {}", userId, consultant.getAgentNo()); } }); @@ -120,4 +114,17 @@ List<AppointmentCustomerView> appointmentList = appointmentCustomerViewRepository.findByAgentNo(agentNo); return appointmentCustomerViewMapper.toAppointmentCustomerViewDTO(appointmentList); } + + public void removeConsultantFromCustomList(String agentNo) { + Long customId = SecurityUtils.getCustomerDBId(); + Consultant consultant = consultantRepository.findOneByAgentNo(agentNo).orElseThrow(ConsultantNotFoundException::new); + CustomerFavoriteConsultant target = customerFavoriteConsultantRepository.findOneByCustomerIdAndConsultant(customId, consultant).orElse(null); + + if(target != null) { + customerFavoriteConsultantRepository.delete(target); + } + else { + log.info("this consultant is not in customer list! agentNo = {}, customId = {}", agentNo, customId); + } + } } -- Gitblit v1.8.0