代码生成

rk

昨天 ab9cd2c82bd64de8e33510db1d1e78a5b3b4de70

 server/services/src/main/java/com/doumee/service/business/impl/RevenueServiceImpl.java

@@ -11,13 +11,23 @@
import com.doumee.core.model.PageWrap;
import com.doumee.core.utils.Utils;
import com.doumee.dao.business.RevenueMapper;
import com.doumee.dao.business.WithdrawalOrdersMapper;
import com.doumee.dao.business.MemberMapper;
import com.doumee.dao.business.ShopInfoMapper;
import com.doumee.dao.business.model.Revenue;
import com.doumee.dao.business.model.ShopInfo;
import com.doumee.dao.business.model.Member;
import com.doumee.dao.business.model.WithdrawalOrders;
import com.doumee.dao.dto.RevenueQueryDTO;
import com.doumee.dao.vo.RevenueStatisticsVO;
import com.doumee.dao.vo.RevenueSummaryVO;
import com.doumee.service.business.RevenueService;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.util.List;
import java.util.Map;
import java.util.Objects;

/**
@@ -30,6 +40,15 @@

    @Autowired
    private RevenueMapper revenueMapper;

    @Autowired
    private WithdrawalOrdersMapper withdrawalOrdersMapper;

    @Autowired
    private ShopInfoMapper shopInfoMapper;

    @Autowired
    private MemberMapper memberMapper;

    @Override
    public Integer create(Revenue revenue) {
@@ -148,4 +167,188 @@
        return revenueMapper.selectCount(wrapper);
    }

    @Override
    public RevenueStatisticsVO getShopRevenueStatistics(Integer shopId) {
        ShopInfo shop = shopInfoMapper.selectById(shopId);
        if (shop == null || Constants.equalsInteger(shop.getDeleted(), Constants.ONE)) {
            throw new BusinessException(ResponseStatus.DATA_EMPTY.getCode(), "门店不存在");
        }
        Integer memberId = shop.getRegionMemberId();
        return buildRevenueStatistics(memberId, Constants.TWO, shop.getBalance());
    }

    @Override
    public RevenueStatisticsVO getDriverRevenueStatistics(Integer memberId) {
        Member member = memberMapper.selectById(memberId);
        if (member == null) {
            throw new BusinessException(ResponseStatus.DATA_EMPTY.getCode(), "用户不存在");
        }
        return buildRevenueStatistics(memberId, Constants.ONE, member.getAmount());
    }

    /**
     * 构建收益统计
     *
     * @param memberId   会员主键
     * @param memberType 会员类型：1=司机；2=门店
     * @param balance    当前余额（分）
     */
    private RevenueStatisticsVO buildRevenueStatistics(Integer memberId, Integer memberType, Long balance) {
        RevenueStatisticsVO vo = new RevenueStatisticsVO();
        vo.setBalance(balance != null ? balance : 0L);

        // 待结算金额：revenue中 vaildStatus=0 且 type=0（完成订单）的收入之和
        QueryWrapper<Revenue> pendingWrapper = new QueryWrapper<>();
        pendingWrapper.select("IFNULL(SUM(AMOUNT), 0) as amount")
                .eq("MEMBER_ID", memberId)
                .eq("MEMBER_TYPE", memberType)
                .eq("VAILD_STATUS", Constants.ZERO)
                .eq("TYPE", Constants.ZERO)
                .eq("OPT_TYPE", Constants.ONE)
                .eq("DELETED", Constants.ZERO);
        Map<String, Object> pendingResult = revenueMapper.selectMaps(pendingWrapper).stream().findFirst().orElse(null);
        vo.setPendingAmount(pendingResult != null && pendingResult.get("amount") != null
                ? Long.parseLong(pendingResult.get("amount").toString()) : 0L);

        // 累计提现金额：withdrawal_orders中 status=1（提现成功）且 type=0（提现）
        QueryWrapper<WithdrawalOrders> successWrapper = new QueryWrapper<>();
        successWrapper.select("IFNULL(SUM(AMOUNT), 0) as amount")
                .eq("MEMBER_ID", memberId)
                .eq("MEMBER_TYPE", memberType)
                .eq("STATUS", Constants.ONE)
                .eq("TYPE", Constants.ZERO)
                .eq("DELETED", Constants.ZERO);
        Map<String, Object> successResult = withdrawalOrdersMapper.selectMaps(successWrapper).stream().findFirst().orElse(null);
        vo.setTotalWithdrawn(successResult != null && successResult.get("amount") != null
                ? Long.parseLong(successResult.get("amount").toString()) : 0L);

        // 提现中金额：withdrawal_orders中 status=0（提现申请中）且 type=0（提现）
        QueryWrapper<WithdrawalOrders> progressWrapper = new QueryWrapper<>();
        progressWrapper.select("IFNULL(SUM(AMOUNT), 0) as amount")
                .eq("MEMBER_ID", memberId)
                .eq("MEMBER_TYPE", memberType)
                .eq("STATUS", Constants.ZERO)
                .eq("TYPE", Constants.ZERO)
                .eq("DELETED", Constants.ZERO);
        Map<String, Object> progressResult = withdrawalOrdersMapper.selectMaps(progressWrapper).stream().findFirst().orElse(null);
        vo.setWithdrawingAmount(progressResult != null && progressResult.get("amount") != null
                ? Long.parseLong(progressResult.get("amount").toString()) : 0L);

        return vo;
    }

    @Override
    public PageData<Revenue> findDriverRevenuePage(PageWrap<RevenueQueryDTO> pageWrap, Integer memberId) {
        IPage<Revenue> page = new Page<>(pageWrap.getPage(), pageWrap.getCapacity());
        QueryWrapper<Revenue> qw = buildRevenueQueryWrapper(pageWrap.getModel(), memberId);
        PageData<Revenue> result = PageData.from(revenueMapper.selectPage(page, qw));
        if (result != null && result.getRecords() != null) {
            for (Revenue model : result.getRecords()) {
                model.setAmountInfo(Constants.getFormatMoney(model.getAmount()));
            }
        }
        return result;
    }

    @Override
    public PageData<Revenue> findShopRevenuePage(PageWrap<RevenueQueryDTO> pageWrap, Integer shopId) {
        ShopInfo shop = shopInfoMapper.selectById(shopId);
        if (shop == null || Constants.equalsInteger(shop.getDeleted(), Constants.ONE)) {
            throw new BusinessException(ResponseStatus.DATA_EMPTY);
        }
        IPage<Revenue> page = new Page<>(pageWrap.getPage(), pageWrap.getCapacity());
        QueryWrapper<Revenue> qw = buildRevenueQueryWrapper(pageWrap.getModel(), shop.getRegionMemberId());
        PageData<Revenue> result = PageData.from(revenueMapper.selectPage(page, qw));
        if (result != null && result.getRecords() != null) {
            for (Revenue model : result.getRecords()) {
                model.setAmountInfo(Constants.getFormatMoney(model.getAmount()));
            }
        }
        return result;
    }

    @Override
    public RevenueSummaryVO getDriverRevenueSummary(RevenueQueryDTO queryDTO, Integer memberId) {
        return buildRevenueSummary(queryDTO, memberId);
    }

    @Override
    public RevenueSummaryVO getShopRevenueSummary(RevenueQueryDTO queryDTO, Integer shopId) {
        ShopInfo shop = shopInfoMapper.selectById(shopId);
        if (shop == null || Constants.equalsInteger(shop.getDeleted(), Constants.ONE)) {
            throw new BusinessException(ResponseStatus.DATA_EMPTY);
        }
        return buildRevenueSummary(queryDTO, shop.getRegionMemberId());
    }

    private QueryWrapper<Revenue> buildRevenueQueryWrapper(RevenueQueryDTO query, Integer memberId) {
        QueryWrapper<Revenue> qw = new QueryWrapper<>();
        qw.lambda().eq(Revenue::getDeleted, Constants.ZERO);
        qw.lambda().eq(Revenue::getMemberId, memberId);
        if (query != null) {
            if (query.getOptType() != null) {
                qw.lambda().eq(Revenue::getOptType, query.getOptType());
            }
            if (query.getType() != null) {
                qw.lambda().eq(Revenue::getType, query.getType());
            }
            if (query.getStartTime() != null) {
                qw.lambda().ge(Revenue::getCreateTime, query.getStartTime());
            }
            if (query.getEndTime() != null) {
                qw.lambda().le(Revenue::getCreateTime, Utils.Date.getEnd(query.getEndTime()));
            }
        }
        qw.lambda().orderByDesc(Revenue::getCreateTime);
        return qw;
    }

    private RevenueSummaryVO buildRevenueSummary(RevenueQueryDTO queryDTO, Integer memberId) {
        // 收入
        QueryWrapper<Revenue> incomeQw = new QueryWrapper<>();
        incomeQw.eq("DELETED", Constants.ZERO);
        incomeQw.eq("MEMBER_ID", memberId);
        incomeQw.eq("OPT_TYPE", Constants.ONE);
        if (queryDTO != null) {
            if (queryDTO.getType() != null) {
                incomeQw.eq("TYPE", queryDTO.getType());
            }
            if (queryDTO.getStartTime() != null) {
                incomeQw.ge("CREATE_TIME", queryDTO.getStartTime());
            }
            if (queryDTO.getEndTime() != null) {
                incomeQw.le("CREATE_TIME", Utils.Date.getEnd(queryDTO.getEndTime()));
            }
        }
        incomeQw.select("IFNULL(SUM(AMOUNT), 0) as amount");
        Map<String, Object> incomeResult = revenueMapper.selectMaps(incomeQw).stream().findFirst().orElse(null);
        long totalIncome = incomeResult != null && incomeResult.get("amount") != null
                ? Long.parseLong(incomeResult.get("amount").toString()) : 0L;
        // 支出
        QueryWrapper<Revenue> expenseQw = new QueryWrapper<>();
        expenseQw.eq("DELETED", Constants.ZERO);
        expenseQw.eq("MEMBER_ID", memberId);
        expenseQw.eq("OPT_TYPE", -Constants.ONE);
        if (queryDTO != null) {
            if (queryDTO.getType() != null) {
                expenseQw.eq("TYPE", queryDTO.getType());
            }
            if (queryDTO.getStartTime() != null) {
                expenseQw.ge("CREATE_TIME", queryDTO.getStartTime());
            }
            if (queryDTO.getEndTime() != null) {
                expenseQw.le("CREATE_TIME", Utils.Date.getEnd(queryDTO.getEndTime()));
            }
        }
        expenseQw.select("IFNULL(SUM(AMOUNT), 0) as amount");
        Map<String, Object> expenseResult = revenueMapper.selectMaps(expenseQw).stream().findFirst().orElse(null);
        long totalExpense = expenseResult != null && expenseResult.get("amount") != null
                ? Long.parseLong(expenseResult.get("amount").toString()) : 0L;

        RevenueSummaryVO vo = new RevenueSummaryVO();
        vo.setTotalIncome(totalIncome);
        vo.setTotalExpense(totalExpense);
        return vo;
    }

}