From cd4a64b39c429f2f5b45cb042d1b66f1cd3904cd Mon Sep 17 00:00:00 2001
From: 劉鈞霖 <benson@gmail.com>
Date: 星期一, 15 十一月 2021 18:06:40 +0800
Subject: [PATCH] [ Update ] 更新 navbar 清單 根據 角色 顯示清單
---
pamapi/src/main/java/com/pollex/pam/service/ConsultantService.java | 86 ++++++++++++++++++++++++++++++------------
1 files changed, 61 insertions(+), 25 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 e71667b..e36e0cf 100644
--- a/pamapi/src/main/java/com/pollex/pam/service/ConsultantService.java
+++ b/pamapi/src/main/java/com/pollex/pam/service/ConsultantService.java
@@ -1,45 +1,55 @@
package com.pollex.pam.service;
+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.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.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.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
-import java.util.Collections;
import java.util.List;
import java.util.stream.Collectors;
@Service
public class ConsultantService {
- private final ConsultantRepository consultantRepository;
- private final CustomFavoriteConsultantRepository customFavoriteConsultantRepository;
- private final ConsultantMapper consultantMapper;
- private final UserService userService;
+ private static final Logger log = LoggerFactory.getLogger(ConsultantService.class);
+
+ final ConsultantRepository consultantRepository;
+ final CustomerFavoriteConsultantRepository customerFavoriteConsultantRepository;
+ final ConsultantMapper consultantMapper;
+ final AppointmentService appointmentService;
+ final AppointmentCustomerViewRepository appointmentCustomerViewRepository;
+ final AppointmentCustomerViewMapper appointmentCustomerViewMapper;
public ConsultantService(
ConsultantRepository consultantRepository,
- CustomFavoriteConsultantRepository customFavoriteConsultantRepository,
+ CustomerFavoriteConsultantRepository customerFavoriteConsultantRepository,
ConsultantMapper consultantMapper,
- UserService userService
+ AppointmentService appointmentService,
+ AppointmentCustomerViewRepository appointmentCustomerViewRepository,
+ AppointmentCustomerViewMapper appointmentCustomerViewMapper
) {
this.consultantRepository = consultantRepository;
- this.customFavoriteConsultantRepository = customFavoriteConsultantRepository;
+ this.customerFavoriteConsultantRepository = customerFavoriteConsultantRepository;
this.consultantMapper = consultantMapper;
- this.userService = userService;
+ this.appointmentService = appointmentService;
+ this.appointmentCustomerViewRepository = appointmentCustomerViewRepository;
+ this.appointmentCustomerViewMapper = appointmentCustomerViewMapper;
}
public List<ConsultantDTO> getMyConsultantList() {
- User user = userService.getUserWithAuthorities().orElseThrow(NotLoginException::new);
- return customFavoriteConsultantRepository.findAllByCustomId(user.getId())
+ Long userId = SecurityUtils.getCustomerId();
+ return customerFavoriteConsultantRepository.findAllByCustomerId(userId)
.stream()
.map(consultantMapper::toDto)
.collect(Collectors.toList());
@@ -53,9 +63,10 @@
}
public List<ConsultantDTO> strictQueryConsultant(StrictQueryConsultantParam param) {
-
-
- return Collections.emptyList();
+ return consultantRepository.findAll(ConsultantQuerySpec.getStrictQuerySpec(param))
+ .stream()
+ .map(consultantMapper::toDto)
+ .collect(Collectors.toList());
}
public List<ConsultantDTO> fastQueryConsultant(FastQueryConsultantParam param) {
@@ -74,16 +85,41 @@
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.getCustomerId();
consultants.forEach(consultant -> {
- CustomFavoriteConsultant customFavoriteConsultant = new CustomFavoriteConsultant();
- customFavoriteConsultant.setConsultant(consultant);
- customFavoriteConsultant.setCustomId(user.getId());
- customFavoriteConsultant.setContactStatus(ContactStatusEnum.PICKED);
+ boolean isConsultantInList = customerFavoriteConsultantRepository.findOneByCustomerIdAndConsultant(userId, consultant).isPresent();
- customFavoriteConsultantRepository.save(customFavoriteConsultant);
+ 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<AppointmentCustomerViewDTO> getMyAppointment() {
+ String agentNo = SecurityUtils.getAgentNo();
+ List<AppointmentCustomerView> appointmentList = appointmentCustomerViewRepository.findByAgentNo(agentNo);
+ return appointmentCustomerViewMapper.toAppointmentCustomerViewDTO(appointmentList);
+ }
+
+ public void removeConsultantFromCustomList(String agentNo) {
+ Long customId = SecurityUtils.getCustomerId();
+ Consultant consultant = consultantRepository.findFirstByAgentNo(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