package com.pollex.pam.service; import com.pollex.pam.domain.AppointmentCustomerView; import com.pollex.pam.domain.Consultant; import com.pollex.pam.domain.CustomerFavoriteConsultant; import com.pollex.pam.repository.AppointmentCustomerViewRepository; import com.pollex.pam.repository.ConsultantRepository; 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.ConsultantMapper; import com.pollex.pam.web.rest.errors.ConsultantNotFoundException; 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.List; import java.util.stream.Collectors; @Service public class ConsultantService { private static final Logger log = LoggerFactory.getLogger(ConsultantService.class); @Autowired ConsultantRepository consultantRepository; @Autowired CustomerFavoriteConsultantRepository customerFavoriteConsultantRepository; @Autowired ConsultantMapper consultantMapper; @Autowired AppointmentService appointmentService; @Autowired AppointmentCustomerViewRepository appointmentCustomerViewRepository; @Autowired AppointmentCustomerViewMapper appointmentCustomerViewMapper; public List getMyConsultantList() { Long userId = SecurityUtils.getCustomerId(); return customerFavoriteConsultantRepository.findAllByCustomerId(userId) .stream() .map(consultantMapper::toDto) .collect(Collectors.toList()); } public List getRecommendConsultantList() { return consultantRepository.findAllByRecommendIsTrue() .stream() .map(consultantMapper::toDto) .collect(Collectors.toList()); } public List strictQueryConsultant(StrictQueryConsultantParam param) { return consultantRepository.findAll(ConsultantQuerySpec.getStrictQuerySpec(param)) .stream() .map(consultantMapper::toDto) .collect(Collectors.toList()); } public List fastQueryConsultant(FastQueryConsultantParam param) { return consultantRepository.findAll(ConsultantQuerySpec.getFastQuerySpec(param)) .stream() .map(consultantMapper::toDto) .collect(Collectors.toList()); } public ConsultantDetailDTO getConsultantDetail(String agentNo) { Consultant consultant = consultantRepository.findOneByAgentNo(agentNo).orElseThrow(ConsultantNotFoundException::new); return consultantMapper.toDetailDto(consultant); } @Transactional public void addConsultantToCustomList(AddConsultantParam param) { List agentNoList = param.getAgentNoList(); List consultants = consultantRepository.findAllByAgentNoIn(agentNoList); Long userId = SecurityUtils.getCustomerId(); consultants.forEach(consultant -> { boolean isConsultantInList = customerFavoriteConsultantRepository.findOneByCustomerIdAndConsultant(userId, consultant).isPresent(); if(!isConsultantInList) { CustomerFavoriteConsultant customerFavoriteConsultant = new CustomerFavoriteConsultant(); customerFavoriteConsultant.setConsultant(consultant); customerFavoriteConsultant.setCustomerId(userId); customerFavoriteConsultantRepository.save(customerFavoriteConsultant); } else { log.info("The consultant is in customer favorite list! customId = {}, consultant AgentNo = {}", userId, consultant.getAgentNo()); } }); } public List getMyAppointment() { String agentNo = SecurityUtils.getAgentNo(); List appointmentList = appointmentCustomerViewRepository.findByAgentNo(agentNo); return appointmentCustomerViewMapper.toAppointmentCustomerViewDTO(appointmentList); } public void removeConsultantFromCustomList(String agentNo) { Long customId = SecurityUtils.getCustomerId(); 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); } } }