保誠-保戶業務員媒合平台
wayne
2022-02-17 34b08e1c461f5e08675fcff95525956d7c4bef11
pamapi/src/main/java/com/pollex/pam/service/mapper/SatisfactionMapper.java
@@ -3,20 +3,36 @@
import static java.util.stream.Collectors.toList;
import java.util.List;
import java.util.Objects;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import com.pollex.pam.domain.Consultant;
import com.pollex.pam.domain.Customer;
import com.pollex.pam.domain.Satisfaction;
import com.pollex.pam.repository.ConsultantRepository;
import com.pollex.pam.repository.CustomerRepository;
import com.pollex.pam.service.dto.SatisfactionDTO;
@Service
public class SatisfactionMapper {
   @Autowired
   ConsultantRepository consultantRepository;
   @Autowired
   CustomerRepository customerRepository;
   public SatisfactionDTO toDTO(Satisfaction source) {
      SatisfactionDTO target = new SatisfactionDTO();
      target.setScore(source.getScore());
      BeanUtils.copyProperties(source, target);
      target.setScore(source.getScore());
      Consultant consultant= consultantRepository.findOneByAgentNo(source.getAgentNo()).orElse(null);
      target.setAgentName(Objects.isNull(consultant) ? null : consultant.getName());
      Customer customer = customerRepository.findById(source.getCustomerId()).get();
      target.setCustomerName(customer.getName());
      return target;
   }