From 3feee2892cad4decf78519a58aab4c0b12cbea45 Mon Sep 17 00:00:00 2001
From: Helius <wangdoubleone@gmail.com>
Date: Tue, 23 Feb 2021 10:24:02 +0800
Subject: [PATCH] merge trc20

---
 src/main/java/com/xcong/excoin/modules/blackchain/service/TrxUsdtUpdateService.java |  304 ++++++++++++++++++++++++++++++++++++++++++++++++++
 1 files changed, 304 insertions(+), 0 deletions(-)

diff --git a/src/main/java/com/xcong/excoin/modules/blackchain/service/TrxUsdtUpdateService.java b/src/main/java/com/xcong/excoin/modules/blackchain/service/TrxUsdtUpdateService.java
new file mode 100644
index 0000000..42367ee
--- /dev/null
+++ b/src/main/java/com/xcong/excoin/modules/blackchain/service/TrxUsdtUpdateService.java
@@ -0,0 +1,304 @@
+package com.xcong.excoin.modules.blackchain.service;
+
+import com.alibaba.fastjson.JSON;
+import com.alibaba.fastjson.JSONArray;
+import com.alibaba.fastjson.JSONObject;
+import com.xcong.excoin.common.enumerates.CoinTypeEnum;
+import com.xcong.excoin.modules.blackchain.model.EthUsdtChargeDto;
+import com.xcong.excoin.modules.member.dao.MemberCoinAddressDao;
+import com.xcong.excoin.modules.member.entity.MemberCoinAddressEntity;
+import com.xcong.excoin.rabbit.producer.UsdtUpdateProducer;
+import com.xcong.excoin.utils.RedisUtils;
+import org.apache.commons.collections.CollectionUtils;
+import org.apache.commons.lang3.StringUtils;
+import org.springframework.http.HttpEntity;
+import org.springframework.http.HttpHeaders;
+import org.springframework.http.MediaType;
+import org.springframework.http.ResponseEntity;
+import org.springframework.stereotype.Service;
+import org.springframework.web.client.RestTemplate;
+import org.tron.common.utils.ByteArray;
+import org.tron.walletserver.WalletApi;
+
+import javax.annotation.Resource;
+import java.math.BigDecimal;
+import java.math.BigInteger;
+import java.util.*;
+
+/**
+ *  TRX TRC20服务类
+ */
+@Service
+public class TrxUsdtUpdateService {
+
+    public static List<String> addressList = new ArrayList<>();
+    private static String http = "https://api.trongrid.io";
+
+    private static String TRC20_CONTRACT_ADDRESS = "TR7NHqjeKQxGTCi8q8ZY4pL8otSzgjLj6t";
+
+    /**
+     *  手续费
+     */
+    private final static BigDecimal TRX_FEE = new BigDecimal("10");
+
+    @Resource
+    private UsdtUpdateProducer usdtUpdateProducer;
+
+    @Resource
+    private MemberCoinAddressDao memberCoinAddressDao;
+
+    @Resource
+    RedisUtils redisUtils;
+
+    /**
+     *  扫块 同步充值USDT-TRC20和TRX
+     */
+    public void monitorCoinListener() {
+        if (CollectionUtils.isEmpty(addressList)) {
+            List<MemberCoinAddressEntity> coinAddressList = memberCoinAddressDao.selectAllBlockAddressBySymbolAndTag(CoinTypeEnum.USDT.name(), "TRC20");
+            if (CollectionUtils.isNotEmpty(coinAddressList)) {
+                coinAddressList.forEach(e -> {
+                    addressList.add(e.getAddress());
+                });
+            }
+        }
+        // 扫块区块
+        Object trc20BlockNum = redisUtils.get("USDT_TRC20_BLOCK_NUM");
+        if(trc20BlockNum==null){
+            trc20BlockNum = 27805917L;
+
+        }
+        Long blockNum = Long.valueOf(trc20BlockNum.toString());
+        redisUtils.set("USDT_TRC20_BLOCK_NUM",blockNum);
+        if (CollectionUtils.isEmpty(addressList)) {
+            return;
+        }
+
+
+        //  解析区块
+        httpTransactionInfo(addressList, blockNum);
+        redisUtils.set("USDT_TRC20_BLOCK_NUM", (blockNum + 1L));
+    }
+
+    /**
+     *  解析区块数据 同步用户充值
+     * @param addressList
+     * @param num
+     */
+    private void httpTransactionInfo(List<String> addressList, Long num) {
+        String transactionInfoByBlockNum = getTransactionInfoByBlockNum(BigInteger.valueOf(num));
+        if (StringUtils.isBlank(transactionInfoByBlockNum)) {
+            return;
+        }
+        JSONArray parseArray = JSON.parseArray(transactionInfoByBlockNum);
+        if (parseArray.size() > 0) {
+            for (Object e : parseArray) {
+                try {
+                    String txId = JSON.parseObject(e.toString()).getString("id");
+                    //判断 数据库 txId 有 就不用往下继续了
+                    JSONObject parseObject = JSON.parseObject(getTransactionById(txId));
+                    String contractRet = parseObject.getJSONArray("ret").getJSONObject(0).getString("contractRet");
+                    //交易成功
+                    if ("SUCCESS".equals(contractRet)) {
+                        String type = parseObject.getJSONObject("raw_data").getJSONArray("contract").getJSONObject(0).getString("type");
+                        if ("TriggerSmartContract".equals(type)) {
+                            //合约地址转账
+                            triggerSmartContract(addressList, txId, parseObject);
+
+                        } else if ("TransferContract".equals(type)) {
+                            //trx 转账
+                            transferContract(parseObject);
+                        }
+                    }
+                } catch (Exception exception) {
+                    exception.printStackTrace();
+                }
+            }
+        }
+    }
+
+
+    /**
+     *   比对本地地址 同步TRX充值
+     * @param parseObject
+     */
+    private void transferContract(JSONObject parseObject) {
+        //数量
+        BigDecimal amount = parseObject.getJSONObject("raw_data").getJSONArray("contract").getJSONObject(0).getJSONObject("parameter").getJSONObject("value").getBigDecimal("amount");
+
+        //调用者地址
+        String owner_address = parseObject.getJSONObject("raw_data").getJSONArray("contract").getJSONObject(0).getJSONObject("parameter").getJSONObject("value").getString("owner_address");
+        owner_address = WalletApi.encode58Check(ByteArray.fromHexString(owner_address));
+
+        //转入地址
+        String to_address = parseObject.getJSONObject("raw_data").getJSONArray("contract").getJSONObject(0).getJSONObject("parameter").getJSONObject("value").getString("to_address");
+        to_address = WalletApi.encode58Check(ByteArray.fromHexString(to_address));
+
+        amount = amount.divide(new BigDecimal(1 + TransformUtil.getSeqNumByLong(0L, 6)));
+
+    }
+
+    /**
+     * 获取特定区块的所有交易 Info 信息
+     *
+     * @param num 区块
+     * @return
+     */
+    public static String getTransactionInfoByBlockNum(BigInteger num) {
+        String url = http + "/wallet/gettransactioninfobyblocknum";
+        Map<String, Object> map = new HashMap<>();
+        map.put("num", num);
+        String param = JSON.toJSONString(map);
+        return postForEntity(url, param).getBody();
+    }
+
+
+    /**
+     * https://cn.developers.tron.network/docs/%E4%BA%A4%E6%98%9311#%E4%BA%A4%E6%98%93%E7%A1%AE%E8%AE%A4%E6%96%B9%E6%B3%95
+     * 按交易哈希查询交易
+     *
+     * @param txId 交易id
+     * @return
+     */
+    public static String getTransactionById(String txId) {
+        // String url = walletSolidityHttp + "/walletsolidity/gettransactionbyid";
+        String url = http + "/wallet/gettransactionbyid";
+        Map<String, Object> map = new HashMap<>();
+        map.put("value", txId);
+        String param = JSON.toJSONString(map);
+        return postForEntity(url, param).getBody();
+    }
+
+    /**
+     * 执行 post 请求
+     *
+     * @param url   url
+     * @param param 请求参数
+     * @return
+     */
+    private static ResponseEntity<String> postForEntity(String url, String param) {
+        RestTemplate restTemplate = new RestTemplate();
+        HttpHeaders headers = new HttpHeaders();
+        headers.setContentType(MediaType.APPLICATION_JSON);
+        HttpEntity<String> request = new HttpEntity<>(param);
+        ResponseEntity<String> result = restTemplate.postForEntity(url, request, String.class);
+//        System.out.println("url:" + url + ",param:" + param + ",result:" + result.getBody());
+        return result;
+    }
+
+    /**
+     *  比对本地地址 同步充值USDT-TRC20
+     * @param addressList
+     * @param txId
+     * @param parseObject
+     */
+    private void triggerSmartContract(List<String> addressList, String txId, JSONObject parseObject) {
+        //方法参数
+        String data = parseObject.getJSONObject("raw_data").getJSONArray("contract").getJSONObject(0).getJSONObject("parameter").getJSONObject("value").getString("data");
+        // 调用者地址
+        String owner_address = parseObject.getJSONObject("raw_data").getJSONArray("contract").getJSONObject(0).getJSONObject("parameter").getJSONObject("value").getString("owner_address");
+        owner_address = WalletApi.encode58Check(ByteArray.fromHexString(owner_address));
+        //System.out.println("owner_address:"+owner_address);
+        // 合约地址
+        String contract_address = parseObject.getJSONObject("raw_data").getJSONArray("contract").getJSONObject(0).getJSONObject("parameter").getJSONObject("value").getString("contract_address");
+        contract_address = WalletApi.encode58Check(ByteArray.fromHexString(contract_address));
+
+        String dataStr = data.substring(8);
+        List<String> strList = TransformUtil.getStrList(dataStr, 64);
+        //System.out.println(strList);
+        if (strList.size() != 2) {
+            return;
+        }
+
+        String to_address = TransformUtil.delZeroForNum(strList.get(0));
+
+        if (!to_address.startsWith("41")) {
+            to_address = "41" + to_address;
+        }
+
+        to_address = WalletApi.encode58Check(ByteArray.fromHexString(to_address));
+        //System.out.println("to_address:"+to_address);
+        String amountStr = TransformUtil.delZeroForNum(strList.get(1));
+
+        if (amountStr.length() > 0) {
+            amountStr = new BigInteger(amountStr, 16).toString(10);
+        }
+
+        BigDecimal amount = BigDecimal.ZERO;
+        //相匹配的合约地址
+        if (!TRC20_CONTRACT_ADDRESS.equals(contract_address)) {
+            return;
+        }
+
+        //币种
+        if (StringUtils.isNotEmpty(amountStr)) {
+            amount = new BigDecimal(amountStr).divide(new BigDecimal(1 + TransformUtil.getSeqNumByLong(0L, 6)));
+        }
+        for (String address : addressList) {
+            if (address.equals(to_address)) {
+                System.out.println("存在本地的地址:" + address);
+                // 金额
+                // 发送消息队列
+                EthUsdtChargeDto dto = new EthUsdtChargeDto(address, txId, amount);
+                dto.setSymbol(EthUsdtChargeDto.Symbol.USDT_TRC20);
+                usdtUpdateProducer.sendMsg(JSONObject.toJSONString(dto));
+                System.out.println("===to_address:" + to_address + "===amount:" + amount);
+            }
+        }
+
+    }
+
+    /**
+     *  根据地址归集USDT-TRC20
+     * @param address
+     * @return
+     */
+    public boolean poolByAddress(String address) {
+        // 首先查询trx余额
+        BigDecimal trxBalance = Trc20Service.getTrxBalance(address);
+        if (trxBalance == null) {
+            return false;
+        }
+        if (trxBalance.compareTo(TRX_FEE) >= 0) {
+            // 转
+            BigDecimal trc20Balance = Trc20Service.getTrc20Balance(address);
+            if (trc20Balance == null) {
+                return false;
+            }
+            MemberCoinAddressEntity coinAddressEntity = memberCoinAddressDao.selectCoinAddressByAddressAndSymbolTag(address, "USDT", "TRC20");
+            if (coinAddressEntity == null) {
+                return false;
+            }
+            Trc20Service.sendTrc20(coinAddressEntity.getPrivateKey(), Trc20Service.POOL_ADDRESS, trc20Balance);
+            // 需要将存在redis的待归集地址删除
+            Object trc20_pool = redisUtils.get("TRC20_POOL");
+            if(trc20_pool!=null){
+                List<String> poolList = (List) trc20_pool;
+                Iterator<String> iterator = poolList.iterator();
+                while (iterator.hasNext()){
+                    String next = iterator.next();
+                    if(address.equals(next)){
+                        iterator.remove();
+                    }
+                }
+               if(CollectionUtils.isEmpty(poolList)) {
+                   redisUtils.del("TRC20_POOL");
+               }else{
+                   redisUtils.set("TRC20_POOL", poolList);
+               }
+            }
+            return true;
+        } else {
+            Trc20Service.sendTrx(Trc20Service.TRX_PRIVATE_KEY, address, TRX_FEE);
+            // 将这个地址记录,后续同步
+            Object trc20_pool = redisUtils.get("TRC20_POOL");
+            List<String> poolList = new ArrayList<>();
+            if (trc20_pool != null) {
+                poolList = (List) trc20_pool;
+            }
+            poolList.add(address);
+            redisUtils.set("TRC20_POOL", poolList);
+            return true;
+        }
+    }
+}

--
Gitblit v1.9.1