From 25c3288859d660507dd78be62dadecfa3c543180 Mon Sep 17 00:00:00 2001 From: Helius <wangdoubleone@gmail.com> Date: Wed, 11 Nov 2020 10:57:11 +0800 Subject: [PATCH] trc20 modify --- src/main/java/com/xcong/excoin/modules/coin/parameter/dto/TrcRechargeDto.java | 30 ++++++++- src/main/java/com/xcong/excoin/modules/coin/controller/Trc20Controller.java | 91 ++++++++++++++++++++++++++++- src/main/java/com/xcong/excoin/quartz/job/TRC20OrderJob.java | 41 +++++++++++++ src/main/java/com/xcong/excoin/utils/TRC20ApiUtils.java | 2 src/main/java/com/xcong/excoin/modules/member/entity/MemberCoinWithdrawEntity.java | 4 + src/main/java/com/xcong/excoin/configurations/security/WebSecurityConfig.java | 1 6 files changed, 158 insertions(+), 11 deletions(-) diff --git a/src/main/java/com/xcong/excoin/configurations/security/WebSecurityConfig.java b/src/main/java/com/xcong/excoin/configurations/security/WebSecurityConfig.java index 004f851..be1e85a 100644 --- a/src/main/java/com/xcong/excoin/configurations/security/WebSecurityConfig.java +++ b/src/main/java/com/xcong/excoin/configurations/security/WebSecurityConfig.java @@ -45,6 +45,7 @@ .antMatchers("/v2/**").permitAll() .antMatchers("/api/symbols/**").permitAll() .antMatchers("/common/**").permitAll() + .antMatchers("/trc/**").permitAll() .antMatchers("/api/exchange/**").permitAll() .antMatchers("/api/member/getMemberAccountInfo").permitAll() .antMatchers("/api/member/memberForgetPwd").permitAll() diff --git a/src/main/java/com/xcong/excoin/modules/coin/controller/Trc20Controller.java b/src/main/java/com/xcong/excoin/modules/coin/controller/Trc20Controller.java index fc731c0..4dc7faf 100644 --- a/src/main/java/com/xcong/excoin/modules/coin/controller/Trc20Controller.java +++ b/src/main/java/com/xcong/excoin/modules/coin/controller/Trc20Controller.java @@ -1,7 +1,88 @@ -package com.xcong.excoin.modules.coin.controller;/** -* -* @author wzy -* @date 2020-11-10 -**/ +package com.xcong.excoin.modules.coin.controller; + +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.member.dao.MemberCoinChargeDao; +import com.xcong.excoin.modules.member.dao.MemberWalletCoinDao; +import com.xcong.excoin.modules.member.entity.MemberCoinChargeEntity; +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; + +/** + * @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; + + @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("充值成功"); + } + + private String generateNo() { + // 生成订单号 + Long timestamp = System.currentTimeMillis(); + // 随机数 + int random = (int) (Math.random() * 10); + return String.valueOf(timestamp).substring(2) + random; + } } diff --git a/src/main/java/com/xcong/excoin/modules/coin/parameter/dto/TrcRechargeDto.java b/src/main/java/com/xcong/excoin/modules/coin/parameter/dto/TrcRechargeDto.java index 727a8db..a03a198 100644 --- a/src/main/java/com/xcong/excoin/modules/coin/parameter/dto/TrcRechargeDto.java +++ b/src/main/java/com/xcong/excoin/modules/coin/parameter/dto/TrcRechargeDto.java @@ -1,7 +1,27 @@ -package com.xcong.excoin.modules.coin.parameter.dto;/** -* -* @author wzy -* @date 2020-11-10 -**/ +package com.xcong.excoin.modules.coin.parameter.dto; + +import lombok.Data; + +import javax.validation.constraints.NotBlank; +import javax.validation.constraints.NotNull; +import java.math.BigDecimal; + +/** + * @author wzy + * @date 2020-11-10 + **/ +@Data public class TrcRechargeDto { + + @NotBlank(message = "uid不能为空") + private String uid; + + @NotNull(message = "充币数量不能为空") + private BigDecimal num; + + @NotNull(message = "类型不能为空") + private Integer type; + + @NotBlank(message = "参数错误") + private String key; } diff --git a/src/main/java/com/xcong/excoin/modules/member/entity/MemberCoinWithdrawEntity.java b/src/main/java/com/xcong/excoin/modules/member/entity/MemberCoinWithdrawEntity.java index cea3765..c19c6c9 100644 --- a/src/main/java/com/xcong/excoin/modules/member/entity/MemberCoinWithdrawEntity.java +++ b/src/main/java/com/xcong/excoin/modules/member/entity/MemberCoinWithdrawEntity.java @@ -42,6 +42,10 @@ */ private int status; public static final int STATUS_DOING = 1; + //同意 + public static final int IS_STATUS_Y = 2; + //拒绝 + public static final int IS_STATUS_N = 3; /** * 是否内部转账 Y-是N-不是 */ diff --git a/src/main/java/com/xcong/excoin/quartz/job/TRC20OrderJob.java b/src/main/java/com/xcong/excoin/quartz/job/TRC20OrderJob.java index ab29562..934373e 100644 --- a/src/main/java/com/xcong/excoin/quartz/job/TRC20OrderJob.java +++ b/src/main/java/com/xcong/excoin/quartz/job/TRC20OrderJob.java @@ -1,11 +1,18 @@ package com.xcong.excoin.quartz.job; import cn.hutool.core.collection.CollUtil; +import com.alibaba.fastjson.JSONObject; +import com.xcong.excoin.modules.coin.dao.MemberAccountMoneyChangeDao; +import com.xcong.excoin.modules.coin.entity.MemberAccountMoneyChange; import com.xcong.excoin.modules.member.dao.MemberCoinWithdrawDao; +import com.xcong.excoin.modules.member.dao.MemberWalletCoinDao; 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.boot.autoconfigure.condition.ConditionalOnProperty; import org.springframework.stereotype.Component; +import org.springframework.transaction.annotation.Transactional; import javax.annotation.Resource; import java.util.HashMap; @@ -18,19 +25,53 @@ **/ @Slf4j @Component +@ConditionalOnProperty(prefix = "app", name = "block-job", havingValue = "true") public class TRC20OrderJob { @Resource private MemberCoinWithdrawDao memberCoinWithdrawDao; + @Resource + private MemberWalletCoinDao memberWalletCoinDao; + @Resource + private MemberAccountMoneyChangeDao memberAccountMoneyChangeDao; + + @Transactional(rollbackFor = Exception.class) public void trc20WithdrawOrder() { + log.info("trc20订单查询"); Map<String, Object> param = new HashMap<>(); param.put("label", "TRC20"); param.put("status", 1); List<MemberCoinWithdrawEntity> withdrawEntities = memberCoinWithdrawDao.selectByMap(param); if (CollUtil.isNotEmpty(withdrawEntities)) { for (MemberCoinWithdrawEntity withdrawEntity : withdrawEntities) { + MemberWalletCoinEntity wallet = memberWalletCoinDao.selectWalletCoinBymIdAndCode(withdrawEntity.getMemberId(), withdrawEntity.getSymbol()); String applyOrderInfo = TRC20ApiUtils.getApplyOrderInfo(withdrawEntity.getTag()); + Integer transStatus = JSONObject.parseObject(applyOrderInfo).getInteger("transStatus"); + + Map<String, Object> columnMaps = new HashMap<>(); + columnMaps.put("withdraw_id", withdrawEntity.getId()); + MemberAccountMoneyChange accountMoneyChanges = memberAccountMoneyChangeDao.selectByMap(columnMaps).get(0); + // 状态:2待审核3提币中(审核成功)4提币成功5审核失败 + if (transStatus == 4) { + 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); + } else if (transStatus == 5) { + 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); + } } } } diff --git a/src/main/java/com/xcong/excoin/utils/TRC20ApiUtils.java b/src/main/java/com/xcong/excoin/utils/TRC20ApiUtils.java index 6a02891..d60d55b 100644 --- a/src/main/java/com/xcong/excoin/utils/TRC20ApiUtils.java +++ b/src/main/java/com/xcong/excoin/utils/TRC20ApiUtils.java @@ -17,7 +17,7 @@ @Slf4j public class TRC20ApiUtils { private static final String TRC20_API = "http://27.50.59.35:5002/"; - private static final String SIGN_STR = "w@a!llokmet"; + public static final String SIGN_STR = "w@a!llokmet"; /** * 提币申请 -- Gitblit v1.9.1