modify

Helius

2020-11-12 58aa9ff8da187d28afc6934bb06d1fe7cc78c987

 src/main/java/com/xcong/excoin/modules/coin/controller/Trc20Controller.java

@@ -1,7 +1,158 @@
package com.xcong.excoin.modules.coin.controller;/**
*
* @author wzy
* @date 2020-11-10
**/
package com.xcong.excoin.modules.coin.controller;

import cn.hutool.core.collection.CollUtil;
import cn.hutool.crypto.SecureUtil;
import com.xcong.excoin.common.enumerates.CoinTypeEnum;
import com.xcong.excoin.common.response.Result;
import com.xcong.excoin.modules.coin.dao.MemberAccountMoneyChangeDao;
import com.xcong.excoin.modules.coin.dao.TrcAddressDao;
import com.xcong.excoin.modules.coin.entity.MemberAccountMoneyChange;
import com.xcong.excoin.modules.coin.entity.TrcAddressEntity;
import com.xcong.excoin.modules.coin.parameter.dto.TrcRechargeDto;
import com.xcong.excoin.modules.coin.parameter.dto.TrcWithdrawDto;
import com.xcong.excoin.modules.member.dao.MemberCoinChargeDao;
import com.xcong.excoin.modules.member.dao.MemberCoinWithdrawDao;
import com.xcong.excoin.modules.member.dao.MemberWalletCoinDao;
import com.xcong.excoin.modules.member.entity.MemberCoinChargeEntity;
import com.xcong.excoin.modules.member.entity.MemberCoinWithdrawEntity;
import com.xcong.excoin.modules.member.entity.MemberWalletCoinEntity;
import com.xcong.excoin.utils.TRC20ApiUtils;
import lombok.extern.slf4j.Slf4j;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import javax.annotation.Resource;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * @author wzy
 * @date 2020-11-10
 **/
@Slf4j
@RestController
@RequestMapping(value = "/trc")
public class Trc20Controller {

    @Resource
    private MemberWalletCoinDao memberWalletCoinDao;
    @Resource
    private MemberAccountMoneyChangeDao memberAccountMoneyChangeDao;
    @Resource
    private MemberCoinChargeDao memberCoinChargeDao;
    @Resource
    private TrcAddressDao trcAddressDao;
    @Resource
    private MemberCoinWithdrawDao memberCoinWithdrawDao;

    @PostMapping(value = "/rechargeTrcAmount")
    @Transactional(rollbackFor = Exception.class)
    public Result rechargeTrcAmount(@RequestBody TrcRechargeDto trcRechargeDto) {
        log.info("传入参数 : {}", trcRechargeDto);
        String sign = SecureUtil.md5(trcRechargeDto.getNum() + trcRechargeDto.getNum().toPlainString() + CoinTypeEnum.USDT.name() + trcRechargeDto.getType() + TRC20ApiUtils.SIGN_STR);
        if (!sign.equals(trcRechargeDto.getKey())) {
            return Result.fail("参数错误");
        }
        Long memberId = Long.parseLong(trcRechargeDto.getUid());

        MemberWalletCoinEntity wallet = memberWalletCoinDao.selectWalletCoinBymIdAndCode(memberId, CoinTypeEnum.USDT.name());
        memberWalletCoinDao.updateWalletBalance(wallet.getId(), trcRechargeDto.getNum(), null, null);

        TrcAddressEntity trcAddress = trcAddressDao.selectSrcAddressByMemberId(memberId);
        MemberCoinChargeEntity charge = new MemberCoinChargeEntity();
        charge.setAddress(trcAddress.getAddress());
        charge.setAmount(trcRechargeDto.getNum());
        charge.setMemberId(memberId);
        charge.setSymbol(CoinTypeEnum.USDT.name());
        charge.setStatus(1);
        charge.setTag("TRC20");
        charge.setLastAmount(trcRechargeDto.getNum());
        String orderNo = generateNo();
        charge.setOrderCode(orderNo);
        memberCoinChargeDao.insert(charge);

        MemberAccountMoneyChange change = new MemberAccountMoneyChange();
        change.setAmount(trcRechargeDto.getNum());
        change.setContent("转入");
        change.setSymbol(CoinTypeEnum.USDT.name());
        change.setStatus(1);
        change.setType(1);
        change.setMemberId(memberId);

        memberAccountMoneyChangeDao.insert(change);
        return Result.ok("充值成功");
    }

    @PostMapping(value = "/withdrawTrcAmount")
    public Result withdrawTrcAmount(@RequestBody TrcWithdrawDto trcWithdrawDto) {
        if (!CoinTypeEnum.USDT.name().equals(trcWithdrawDto.getSymbol())) {
            return Result.fail("币种错误");
        }

        if (!TrcWithdrawDto.STATUS_SUCCESS.equals(trcWithdrawDto.getStatus()) && !TrcWithdrawDto.STATUS_FAIL.equals(trcWithdrawDto.getStatus())) {
            return Result.fail("状态错误");
        }

        String sign = SecureUtil.md5(trcWithdrawDto.getMemberId() + trcWithdrawDto.getOrderNo() + trcWithdrawDto.getStatus() + trcWithdrawDto.getSymbol());
        if (!sign.equals(trcWithdrawDto.getKey())) {
            return Result.fail("参数错误");
        }

        Map<String, Object> columnMaps = new HashMap<>();
        columnMaps.put("tag", trcWithdrawDto.getOrderNo());
        columnMaps.put("member_id", trcWithdrawDto.getMemberId());
        columnMaps.put("status", 1);
        List<MemberCoinWithdrawEntity> withdrawEntities = memberCoinWithdrawDao.selectByMap(columnMaps);
        if (CollUtil.isEmpty(withdrawEntities)) {
            return Result.fail("该订单不存在");
        }

        MemberCoinWithdrawEntity withdrawEntity = withdrawEntities.get(0);

        MemberWalletCoinEntity wallet = memberWalletCoinDao.selectWalletCoinBymIdAndCode(withdrawEntity.getMemberId(), CoinTypeEnum.USDT.name());
        Map<String, Object> changeParam = new HashMap<>();
        changeParam.put("withdraw_id", withdrawEntity.getId());
        MemberAccountMoneyChange accountMoneyChanges = memberAccountMoneyChangeDao.selectByMap(changeParam).get(0);

        // 审核成功
        if (TrcWithdrawDto.STATUS_SUCCESS.equals(trcWithdrawDto.getStatus())) {
            memberWalletCoinDao.updateWalletBalance(wallet.getId(), null, null, withdrawEntity.getAmount().negate());

            accountMoneyChanges.setStatus(MemberAccountMoneyChange.STATUS_SUCCESS_INTEGER);
            accountMoneyChanges.setAmount(withdrawEntity.getAmount().negate());
            memberAccountMoneyChangeDao.updateById(accountMoneyChanges);

            withdrawEntity.setStatus(MemberCoinWithdrawEntity.IS_STATUS_Y);
            memberCoinWithdrawDao.updateById(withdrawEntity);

            return Result.ok("提币成功");
        }

        if (TrcWithdrawDto.STATUS_FAIL.equals(trcWithdrawDto.getStatus())) {
            memberWalletCoinDao.updateWalletBalance(wallet.getId(), withdrawEntity.getAmount(), null, withdrawEntity.getAmount().negate());

            accountMoneyChanges.setStatus(MemberAccountMoneyChange.STATUS_FAIL_INTEGER);
            accountMoneyChanges.setAmount(withdrawEntity.getAmount());
            memberAccountMoneyChangeDao.updateById(accountMoneyChanges);

            withdrawEntity.setStatus(MemberCoinWithdrawEntity.IS_STATUS_N);
            memberCoinWithdrawDao.updateById(withdrawEntity);

            return Result.ok("提币申请驳回成功");
        }

        return Result.fail("提币失败");
    }

    private String generateNo() {
        // 生成订单号
        Long timestamp = System.currentTimeMillis();
        // 随机数
        int random = (int) (Math.random() * 10);
        return String.valueOf(timestamp).substring(2) + random;
    }
}