xiaoyong931011
2022-12-31 c2b0b1cc50e7401a085c2b86602ef26073da6eeb
src/test/java/cc/mrbird/febs/ProfitTest.java
@@ -6,6 +6,12 @@
import cc.mrbird.febs.common.utils.AppContants;
import cc.mrbird.febs.common.utils.LoginUserUtil;
import cc.mrbird.febs.common.utils.MallUtils;
import cc.mrbird.febs.common.utils.RedisUtils;
import cc.mrbird.febs.mall.chain.enums.ChainEnum;
import cc.mrbird.febs.mall.chain.service.BaseCoinService;
import cc.mrbird.febs.mall.chain.service.ChainService;
import cc.mrbird.febs.mall.chain.service.ContractEventService;
import cc.mrbird.febs.mall.chain.service.UsdtErc20UpdateService;
import cc.mrbird.febs.mall.entity.*;
import cc.mrbird.febs.mall.mapper.*;
import cc.mrbird.febs.mall.quartz.OrderSettlementJob;
@@ -26,7 +32,9 @@
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
import javax.annotation.Resource;
import java.math.BigDecimal;
import java.math.BigInteger;
import java.text.DecimalFormat;
import java.text.SimpleDateFormat;
import java.util.*;
@@ -337,11 +345,63 @@
    @Autowired
    private IMallAchieveService mallAchieveService;
    @Autowired
    private MallMemberStarBuyMapper mallMemberStarBuyMapper;
    @Test
    public void achieveTest() {
        List<MallOrderItem> items = mallOrderItemMapper.selectList(null);
        for (MallOrderItem item : items) {
            mallAchieveService.add(item.getId());
        List<MallMemberStarBuy> mallMemberStarBuys = mallMemberStarBuyMapper.selectByStatus(1);
        if(CollUtil.isEmpty(mallMemberStarBuys)){
            return;
        }
        for(MallMemberStarBuy mallMemberStarBuy : mallMemberStarBuys){
            Long memberId = mallMemberStarBuy.getMemberId();
            MallMember mallMember = memberMapper.selectById(memberId);
            //用户不存在跳出本次循环
            if(ObjectUtil.isEmpty(mallMember)){
                continue;
            }
            //返还总金额
            BigDecimal amount = mallMemberStarBuy.getAmount();
            //剩余返还总金额不足,跳出本次循环
            BigDecimal avaAmount = mallMemberStarBuy.getAvaAmount();
            if(avaAmount.compareTo(BigDecimal.ZERO) <= 0){
                continue;
            }
            //每次返还比例
            DataDictionaryCustom returnPercentDic = dataDictionaryCustomMapper.selectDicDataByTypeAndCode(
                    DataDictionaryEnum.RETURN_PERCENT.getType(),
                    DataDictionaryEnum.RETURN_PERCENT.getCode()
            );
            String returnPercentStr = returnPercentDic.getValue();
            BigDecimal returnPercent = new BigDecimal(returnPercentStr).multiply(new BigDecimal(0.01)).setScale(2,BigDecimal.ROUND_DOWN);
            //本次返还金额
            BigDecimal returnAmount = amount.multiply(returnPercent).setScale(2,BigDecimal.ROUND_DOWN);
            /**
             * 如果本次返还金额大于剩余返还总金额
             */
            if(returnAmount.compareTo(avaAmount) >= 0){
                returnAmount = avaAmount;
                avaAmount = BigDecimal.ZERO;
            }else{
                avaAmount = avaAmount.subtract(returnAmount).setScale(2,BigDecimal.ROUND_DOWN);
            }
            //增加用户金额
            memberWalletService.addBalance(returnAmount,mallMember.getId());
            //生成流水记录
            Long subsidyAmountFlowId = mallMoneyFlowService.addMoneyFlow(
                    mallMember.getId(),
                    mallMemberStarBuy.getBuyCode(),
                    returnAmount,
                    MallMoneyFlowTypeEnum.RETURN_MARKET_STAR.getCode(),
                    MallMoneyFlow.STATUS_SUCCESS,
                    MallMoneyFlow.IS_RETURN_Y,
                    mallMember.getId(),
                    FlowTypeEnum.BALANCE.getValue(),
                    MallMoneyFlowTypeEnum.RETURN_MARKET_STAR.getName()
            );
            //更新记录
            mallMemberStarBuyMapper.updateAvaAmountById(avaAmount,mallMemberStarBuy.getId());
        }
    }
//
@@ -358,10 +418,32 @@
//        amount.put("amount", 2);
//    }
//
//    @Test
//    public void scorePool(){
//        memberProfitService.scorePool();
//    }
    @Autowired
    private ContractEventService bscUsdtContractEvent;
    @Autowired
    private BaseCoinService baseCoinService;
    @Autowired
    private RedisUtils redisUtils;
    @Resource
    private UsdtErc20UpdateService usdtErc20UpdateService;
    @Test
    public void scorePool(){
        long start = System.currentTimeMillis();
        Object incrementObj = redisUtils.get(AppContants.REDIS_KEY_BLOCK_ETH_INCREMENT_NUM);
        BigInteger newest = ChainService.getInstance(ChainEnum.BSC_USDT.name()).blockNumber();
        BigInteger block;
        if (incrementObj == null) {
            block = newest;
        } else {
            block = (BigInteger) incrementObj;
        }
//        ChainService.wssBaseCoinEventListener(BigInteger.valueOf(24317595), baseCoinService);
        ChainService.wssContractEventListener(BigInteger.valueOf(24317595), bscUsdtContractEvent, ChainEnum.BSC_USDT.name());
    }
//
//    @Test
//    public void staticProfit(){