AI产品交易

zyy

2025-07-14 d9d43aec7372dc579ced50eda32d7881265533b9

 src/main/java/com/nq/service/impl/UserAssetsServices.java

@@ -17,6 +17,7 @@
import javax.annotation.Resource;
import javax.validation.constraints.Email;
import java.math.BigDecimal;
import java.math.RoundingMode;
import java.util.Date;
import java.util.List;
import java.util.Objects;
@@ -291,4 +292,44 @@
        }
        extracted(userAssets);
    }


    public Boolean aiAvailableBalanceChange(UserAssets userAssets, EUserAssets eUserAssets, BigDecimal amount, BigDecimal buyAmount) {
        String before = userAssets.getAvailableBalance().toString();
        if (eUserAssets.getCode().equals(EUserAssets.BUY_AI.getCode())) {
            //冻结金额
            userAssets.setFreezeMoney(userAssets.getFreezeMoney().add(amount));
            //扣除可用金额
            userAssets.setAvailableBalance(userAssets.getAvailableBalance().add(amount.negate()));
        } else if (eUserAssets.getCode().equals(EUserAssets.BUY_AI_REJECT.getCode())) {
            //解除冻结金额
            userAssets.setFreezeMoney(userAssets.getFreezeMoney().add(amount.negate()));
            //归还可用金额
            userAssets.setAvailableBalance(userAssets.getAvailableBalance().add(amount));
        } else if (eUserAssets.getCode().equals(EUserAssets.AI_SETTLEMENT.getCode())) {
            //解除冻结金额
            userAssets.setFreezeMoney(userAssets.getFreezeMoney().add(buyAmount.negate()));
            //归还可用金额（买入金额+收益）
            userAssets.setAvailableBalance(userAssets.getAvailableBalance().add(amount));
        }
        String after = userAssets.getAvailableBalance().toString();
        MoneyLog moneyLog = new MoneyLog();
        moneyLog.setDescs(eUserAssets.getDesc());
        moneyLog.setBeFore(before);
        moneyLog.setAfter(after);
        moneyLog.setAmount(amount.toString());
        moneyLog.setAccectType(userAssets.getAccectType());
        moneyLog.setType(eUserAssets.getCode());
        moneyLog.setUserId(userAssets.getId()+"");
        moneyLog.setSymbol(EStockType.getEStockTypeByCode(userAssets.getAccectType()).getSymbol());
        moneyLog.setCreateTime(new Date());
        moneyLogMapper.insert(moneyLog);
        return userAssetsMapper.updateById(userAssets)>1;
    }

    @Override
    public BigDecimal exchangeAmountByRate(BigDecimal amount, BigDecimal rate) {
        //保留5位小数
        return amount.multiply(rate).setScale(5, RoundingMode.HALF_UP);
    }
}