PaymentServiceImpl.java 3.57 KB
package vion.service.impl;

import cn.dev33.satoken.stp.StpUtil;
import cn.hutool.core.lang.Opt;
import cn.hutool.core.util.ObjUtil;
import cn.hutool.core.util.StrUtil;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.github.yulichang.base.MPJBaseServiceImpl;
import com.github.yulichang.wrapper.MPJLambdaWrapper;
import io.github.linpeilie.Converter;
import lombok.RequiredArgsConstructor;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import vion.dto.PaymentDTO;
import vion.mapper.PaymentMapper;
import vion.model.Contract;
import vion.model.Payment;
import vion.service.IContractPaymentService;
import vion.service.IContractService;
import vion.service.IPaymentService;
import vion.vo.PaymentVO;
import vion.vo.RoleVO;
import vion.vo.UserVO;

import java.math.BigDecimal;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

/**
 * @author HlQ
 * @date 2023/12/5
 */
@Service
@RequiredArgsConstructor
public class PaymentServiceImpl extends MPJBaseServiceImpl<PaymentMapper, Payment> implements IPaymentService {

    private final IContractService contractService;
    private final IContractPaymentService contractPaymentService;
    private final Converter converter;

    @Override
    public Page<PaymentVO> list(PaymentDTO dto) {
        UserVO userVO = (UserVO) StpUtil.getTokenSession().get("curLoginUser");
        String saleName = Opt.ofEmptyAble(userVO.getRoleVOList())
                .map(l -> l.stream().map(RoleVO::getCode).collect(Collectors.toList()))
                .map(roleCodeList -> {
                    if (!roleCodeList.contains("admin") && !roleCodeList.contains("shangwu") && !roleCodeList.contains("caiwu")) {
                        return userVO.getUsername();
                    } else {
                        return "";
                    }
                }).orElse("未知");

        MPJLambdaWrapper<Payment> wrapper = new MPJLambdaWrapper<>(converter.convert(dto, Payment.class))
                .selectAll(Payment.class)
                .leftJoin(Contract.class, Contract::getContractNo, Payment::getContractNo)
                .eq(StrUtil.isNotBlank(saleName), Contract::getSaleName, saleName);
        return this.selectJoinListPage(Page.of(dto.getPageNum(), dto.getPageSize()), PaymentVO.class, wrapper);
    }

    @Override
    @Transactional(rollbackFor = Exception.class)
    public String save(List<PaymentDTO> dto) {
        List<Payment> paymentList = converter.convert(dto, Payment.class);
        paymentList.forEach(p -> {
            Payment payment = this.lambdaQuery().eq(Payment::getSerialNo, p.getSerialNo()).one();
            if (ObjUtil.isNull(payment)) {
                this.save(p);
            } else {
                p.setId(payment.getId());
                this.updateById(p);
            }
        });
        Map<String, List<Payment>> no2PaymentMap = paymentList.stream().collect(Collectors.groupingBy(Payment::getContractNo));
        no2PaymentMap.forEach((no, list) -> {
            BigDecimal sumAmount = list.stream().map(Payment::getPaymentAmount).reduce(BigDecimal.ZERO, BigDecimal::add);

            Contract existContract = contractService.lambdaQuery().eq(Contract::getContractNo, no).one();
            // 最新的应收款去计算
            Contract updDto = new Contract();
            updDto.setId(existContract.getId());
            updDto.setPaidAmount(sumAmount);
            contractPaymentService.calMoney(existContract, updDto);
            contractService.updateById(updDto);
        });
        return "成功";
    }
}