From fa5e63f49f1e5c891312d9cd659b91c09107ded5 Mon Sep 17 00:00:00 2001
From: KKSU <15274802129@163.com>
Date: Sat, 06 Jul 2024 13:06:20 +0800
Subject: [PATCH] 逻辑

---
 src/main/java/cc/mrbird/febs/dapp/chain/ChainService.java |  292 +++++++++++++++++++++++++++++++++++++++++++++++----------
 1 files changed, 239 insertions(+), 53 deletions(-)

diff --git a/src/main/java/cc/mrbird/febs/dapp/chain/ChainService.java b/src/main/java/cc/mrbird/febs/dapp/chain/ChainService.java
index b81b2b5..0ba0c6d 100644
--- a/src/main/java/cc/mrbird/febs/dapp/chain/ChainService.java
+++ b/src/main/java/cc/mrbird/febs/dapp/chain/ChainService.java
@@ -1,89 +1,275 @@
 package cc.mrbird.febs.dapp.chain;
 
 import cc.mrbird.febs.common.exception.FebsException;
-import cn.hutool.core.util.StrUtil;
-import cn.hutool.http.HttpUtil;
-import com.alibaba.fastjson.JSONObject;
+import io.reactivex.Flowable;
+import io.reactivex.disposables.Disposable;
+import io.reactivex.schedulers.Schedulers;
+import lombok.extern.slf4j.Slf4j;
+import org.web3j.crypto.Credentials;
+import org.web3j.protocol.Web3j;
+import org.web3j.protocol.core.DefaultBlockParameter;
+import org.web3j.protocol.core.DefaultBlockParameterName;
+import org.web3j.protocol.core.DefaultBlockParameterNumber;
+import org.web3j.protocol.core.methods.request.EthFilter;
+import org.web3j.protocol.http.HttpService;
+import org.web3j.protocol.websocket.WebSocketClient;
+import org.web3j.protocol.websocket.WebSocketService;
+import org.web3j.tx.gas.StaticGasProvider;
 
-import java.math.BigDecimal;
 import java.math.BigInteger;
+import java.net.URI;
+import java.util.HashMap;
+import java.util.Map;
+import java.util.concurrent.TimeUnit;
+import java.util.concurrent.atomic.AtomicInteger;
 
 /**
- * @author wzy
+ * @author
  * @date 2022-03-23
  **/
+@Slf4j
 public class ChainService {
+    private final static Map<String, ContractChainService> contractMap = new HashMap<>();
 
-    private final String ETH_PREFIX = "0x";
-    private final EthService ETH = new EthService();
-    private final TrxService TRX = TrxService.INSTANCE;
+    static {
+        for (ChainEnum chain : ChainEnum.values()) {
+            if ("TRX".equals(chain.getChain())) {
+                contractMap.put(chain.name(), new TrxService(chain.getAddress(), chain.getPrivateKey(), chain.getContractAddress(), chain.getApiKey()));
+            } else {
+                contractMap.put(chain.name(), new EthService(chain.getUrl(), chain.getAddress(), chain.getPrivateKey(), chain.getContractAddress()));
+            }
+        }
+    }
 
-    private ChainService() {}
+    private ChainService() {
+    }
 
     public final static ChainService INSTANCE = new ChainService();
 
-    /**
-     * 获取制定账号的USDT余额
-     *
-     * @param address
-     * @return
-     */
-    public BigDecimal balanceOf(String address) {
-        BigDecimal balance = BigDecimal.ZERO;
-        if (address.contains(ETH_PREFIX)) {
-            balance = ETH.tokenGetBalance(address);
-        } else {
-            balance = TRX.balanceOfDecimal(address);
+    public static ContractChainService getInstance(String chainType) {
+        ContractChainService contract = contractMap.get(chainType);
+        if (contract == null) {
+            throw new FebsException("参数错误");
         }
-        return balance;
+
+        return contract;
     }
 
     /**
-     * 判断地址是否授权给制定账户
+     * 监听合约事件
      *
-     * @param address
-     * @return
+     * @param startBlock 开始区块
      */
-    public boolean isAllowance(String address) {
-        BigInteger result;
-        if (address.indexOf(ETH_PREFIX) > 0) {
-            result = ETH.ethAllowance(address);
-        } else {
-            result = TRX.allowance(address);
+    public static void contractEventListener(BigInteger startBlock, ContractEventService event, String type) {
+        contractEventListener(startBlock, null, event, type);
+    }
+
+    public static void contractEventListener(BigInteger startBlock, BigInteger endBlock, ContractEventService event, String type) {
+        ChainEnum chain = ChainEnum.getValueByName(type);
+        assert chain != null;
+
+        EthUsdtContract contract = contract(chain.getPrivateKey(), chain.getContractAddress(), chain.getUrl());
+        EthFilter filter = getFilter(startBlock, endBlock, chain.getContractAddress());
+
+        Flowable<EthUsdtContract.TransferEventResponse> eventFlowable = contract.transferEventFlowable(filter);
+        eventFlowable.subscribe(e -> {
+            event.compile(e);
+        }, error -> {
+            log.error("合约监听启动报错", error);
+        });
+    }
+
+    public static void sdmUSDTEventListener(BigInteger startBlock, BigInteger endBlock, ContractEventService event, String type) {
+        ChainEnum chain = ChainEnum.getValueByName(type);
+        assert chain != null;
+
+        EthUsdtContract contract = contract(chain.getPrivateKey(), chain.getContractAddress(), chain.getUrl());
+        EthFilter filter = getFilter(startBlock, endBlock, chain.getContractAddress());
+
+        Flowable<EthUsdtContract.TransferEventResponse> eventFlowable = contract.transferEventFlowable(filter)
+                .doOnError(throwable ->
+                        log.error("合约事件监听发生错误: " + throwable.getMessage(), throwable)) // 更具体的错误日志记录
+                .retryWhen(errors -> {
+                    AtomicInteger counter = new AtomicInteger();
+                    return errors.takeWhile(e -> counter.getAndIncrement() != 3)
+                            .flatMap(e -> {
+                                System.out.println("delay retry by " + counter.get() + " second(s)");
+                                return Flowable.timer(counter.get(), TimeUnit.SECONDS);
+                            });
+                })
+                .subscribeOn(Schedulers.io()); // 指定subscribe操作在IO线程中执行,避免阻塞主线程
+
+        eventFlowable.subscribe(
+                e -> {
+                    try {
+                        event.sdmUSDT(e); // 处理事件
+                    } catch (Exception ex) {
+                        // 处理事件时可能出现的异常
+                        log.error("处理合约事件时出错", ex);
+                    }
+                },
+                Throwable::printStackTrace, // 打印错误堆栈,或者可以替换为更具体的错误处理逻辑
+                () -> log.info("合约事件监听已完成") // 在Flowable完成时执行的逻辑,如记录日志等
+        );
+
+    }
+
+    public static void receivePeopleEventListener(BigInteger startBlock, BigInteger endBlock, ContractEventService event, String type) {
+        ChainEnum chain = ChainEnum.getValueByName(type);
+        assert chain != null;
+
+        EthUsdtContract contract = contract(chain.getPrivateKey(), chain.getContractAddress(), chain.getUrl());
+        EthFilter filter = getFilter(startBlock, endBlock, chain.getContractAddress());
+
+        Flowable<EthUsdtContract.ReceivePeopleEventResponse> eventFlowable = contract.receivePeopleEventFlowable(filter)
+                .doOnError(throwable ->
+                        log.error("合约事件监听发生错误: " + throwable.getMessage(), throwable)) // 更具体的错误日志记录
+                .retryWhen(errors -> {
+                    AtomicInteger counter = new AtomicInteger();
+                    return errors.takeWhile(e -> counter.getAndIncrement() != 3)
+                            .flatMap(e -> {
+                                System.out.println("delay retry by " + counter.get() + " second(s)");
+                                return Flowable.timer(counter.get(), TimeUnit.SECONDS);
+                            });
+                })
+                .subscribeOn(Schedulers.io()); // 指定subscribe操作在IO线程中执行,避免阻塞主线程
+
+        eventFlowable.subscribe(
+                e -> {
+                    try {
+                        event.receivePeopleEvent(e); // 处理事件
+                    } catch (Exception ex) {
+                        // 处理事件时可能出现的异常
+                        log.error("处理合约事件时出错", ex);
+                    }
+                },
+                Throwable::printStackTrace, // 打印错误堆栈,或者可以替换为更具体的错误处理逻辑
+                () -> log.info("合约事件监听已完成") // 在Flowable完成时执行的逻辑,如记录日志等
+        );
+
+    }
+
+
+    public static void sdmChargeEventListener(BigInteger startBlock, BigInteger endBlock, ContractEventService event, String type) {
+        ChainEnum chain = ChainEnum.getValueByName(type);
+        assert chain != null;
+
+        EthUsdtContract contract = contract(chain.getPrivateKey(), chain.getContractAddress(), chain.getUrl());
+        EthFilter filter = getFilter(startBlock, endBlock, chain.getContractAddress());
+
+        Flowable<EthUsdtContract.TransferEventResponse> eventFlowable = contract.transferEventFlowable(filter)
+                .doOnError(throwable ->
+                        log.error("合约事件监听发生错误: " + throwable.getMessage(), throwable)) // 更具体的错误日志记录
+                .retryWhen(errors -> {
+                    AtomicInteger counter = new AtomicInteger();
+                    return errors.takeWhile(e -> counter.getAndIncrement() != 3)
+                            .flatMap(e -> {
+                                System.out.println("delay retry by " + counter.get() + " second(s)");
+                                return Flowable.timer(counter.get(), TimeUnit.SECONDS);
+                            });
+                })
+                .subscribeOn(Schedulers.io()); // 指定subscribe操作在IO线程中执行,避免阻塞主线程
+
+        eventFlowable.subscribe(
+                e -> {
+                    try {
+                        event.compile(e); // 处理事件
+                    } catch (Exception ex) {
+                        // 处理事件时可能出现的异常
+                        log.error("处理合约事件时出错", ex);
+                    }
+                },
+                Throwable::printStackTrace, // 打印错误堆栈,或者可以替换为更具体的错误处理逻辑
+                () -> log.info("合约事件监听已完成") // 在Flowable完成时执行的逻辑,如记录日志等
+        );
+
+    }
+
+    public static void wssContractEventListener(BigInteger startBlock, ContractEventService event, String type) {
+        WebSocketService ws = null;
+        WebSocketClient webSocketClient = null;
+        Web3j web3j = null;
+
+        try {
+            webSocketClient = new WebSocketClient(new URI("wss://bsc-mainnet.nodereal.io/ws/v1/78074065950e4915aef4f12b6f357d16"));
+            ws = new WebSocketService(webSocketClient, true);
+            ws.connect();
+            web3j = Web3j.build(ws);
+            ChainEnum chain = ChainEnum.getValueByName(type);
+            assert chain != null;
+
+            EthUsdtContract ethUsdtContract = wssContract(chain.getPrivateKey(), chain.getContractAddress(), web3j);
+            EthFilter filter = getFilter(startBlock, null, chain.getContractAddress());
+
+
+            Flowable<EthUsdtContract.TransferEventResponse> eventFlowable = ethUsdtContract.transferEventFlowable(filter);
+            Disposable subscribe = eventFlowable.subscribe(event::compile, error -> {
+                log.error("币安监听异常", error);
+            });
+        } catch (Exception e) {
+            e.printStackTrace();
         }
 
-        return result.intValue() != 0;
     }
 
-    /**
-     * 获取地址授权数量
-     *
-     * @param address
-     * @return
-     */
-    public int allowanceCnt(String address) {
-        String response = HttpUtil.get("https://apiasia.tronscan.io:5566/api/account/approve/list?address=" + address);
-        String total = JSONObject.parseObject(response).getString("total");
-        return Integer.parseInt(total);
+
+    private static EthUsdtContract contract(String privateKey, String contractAddress, String url) {
+        Credentials credentials = Credentials.create(privateKey);
+        HttpService httpService = new HttpService(url);
+//        httpService.addHeader("Authorization", "Bearer " + Base64.encode("tfc:tfc123".getBytes()));
+//        httpService.addHeader("Authorization", "Bearer eyJhbGciOiJSUzI1NiIsInR5cCI6IkpXVCJ9.eyJhdWQiOiJwdWJsaWMiLCJleHAiOjE2NTk5MzcxOTAsImp0aSI6IjRiMjNkYTVjLWRlZWEtNDYzNi04YjMwLWNmMmZmMjVkM2NlYyIsImlhdCI6MTY1OTkzMzU5MCwiaXNzIjoiQW5rciIsIm5iZiI6MTY1OTkzMzU5MCwic3ViIjoiZmNiNjY0YjItOGEwNC00N2E5LTg3ZjMtNTJhMjE2ODVlMzEzIn0.YfEwvDByU2MGHywsblZpEmKMIbjv4cWYkn5CaFglXY0TSANzd2pCSbIe40yU_R9_nV6xZeE8Uk74jJOdd_QvMpFyUgo-MMNWZP6uiEaYvK_K3tlpk5yzeZq9D4ruWaq8rFKggr-iaRGzu6coRSAOFv2prWll3a7NdEbmkM-y5Y85xYD6g1N-TPIpE_Y-_-WPf3JUavk744kG8YyHhGvAmk2IL0N2xePfC6CHesdJhwvmJJXzr_53dbPwit1y5KljS0iTZz3mGTML2bq4hGaEHbQxeY2fBpZOSm8sPMz-zB9IVJQKzH5-DXlPKz01mJ9XiBJlubfHsN72RdqFD-O2Tw");
+        return EthUsdtContract.load(contractAddress,
+                Web3j.build(httpService),
+                credentials,
+                new StaticGasProvider(BigInteger.valueOf(4500000L), BigInteger.valueOf(200000L)));
     }
 
-    public String transfer(String address) {
-        BigDecimal amount = balanceOf(address);
-
-        return transfer(address, amount);
+    private static EthUsdtContract wssContract(String privateKey, String contractAddress, Web3j web3j) {
+        Credentials credentials = Credentials.create(privateKey);
+//        httpService.addHeader("Authorization", "Bearer " + Base64.encode("tfc:tfc123".getBytes()));
+//        httpService.addHeader("Authorization", "Bearer eyJhbGciOiJSUzI1NiIsInR5cCI6IkpXVCJ9.eyJhdWQiOiJwdWJsaWMiLCJleHAiOjE2NTk5MzcxOTAsImp0aSI6IjRiMjNkYTVjLWRlZWEtNDYzNi04YjMwLWNmMmZmMjVkM2NlYyIsImlhdCI6MTY1OTkzMzU5MCwiaXNzIjoiQW5rciIsIm5iZiI6MTY1OTkzMzU5MCwic3ViIjoiZmNiNjY0YjItOGEwNC00N2E5LTg3ZjMtNTJhMjE2ODVlMzEzIn0.YfEwvDByU2MGHywsblZpEmKMIbjv4cWYkn5CaFglXY0TSANzd2pCSbIe40yU_R9_nV6xZeE8Uk74jJOdd_QvMpFyUgo-MMNWZP6uiEaYvK_K3tlpk5yzeZq9D4ruWaq8rFKggr-iaRGzu6coRSAOFv2prWll3a7NdEbmkM-y5Y85xYD6g1N-TPIpE_Y-_-WPf3JUavk744kG8YyHhGvAmk2IL0N2xePfC6CHesdJhwvmJJXzr_53dbPwit1y5KljS0iTZz3mGTML2bq4hGaEHbQxeY2fBpZOSm8sPMz-zB9IVJQKzH5-DXlPKz01mJ9XiBJlubfHsN72RdqFD-O2Tw");
+        return EthUsdtContract.load(contractAddress,
+                web3j,
+                credentials,
+                new StaticGasProvider(BigInteger.valueOf(4500000L), BigInteger.valueOf(200000L)));
     }
 
-    public String transfer(String address, BigDecimal amount) {
-        String hash;
-        if (address.contains(ETH_PREFIX)) {
-            hash = ETH.approveTransfer(address, amount, null);
+    private static EthFilter getFilter(BigInteger startBlock, String contractAddress) {
+        return getFilter(startBlock, null, contractAddress);
+    }
+
+    private static EthFilter getFilter(BigInteger startBlock, BigInteger endBlock, String contractAddress) {
+        DefaultBlockParameter startParameterName = null;
+        DefaultBlockParameter endParameterName = null;
+        if (startBlock != null) {
+            startParameterName = new DefaultBlockParameterNumber(startBlock);
         } else {
-            hash = TRX.transfer(address, amount);
+            startParameterName = DefaultBlockParameterName.EARLIEST;
         }
-        return hash;
+
+        if (endBlock != null) {
+            endParameterName = new DefaultBlockParameterNumber(endBlock);
+        } else {
+            endParameterName = DefaultBlockParameterName.LATEST;
+        }
+
+        return new EthFilter(startParameterName, endParameterName, contractAddress);
     }
 
     public static void main(String[] args) {
-        System.out.println(ChainService.INSTANCE.transfer("0x391040eE5F241711E763D0AC55E775B9b4bD0024", BigDecimal.valueOf(5)));
+//        ChainEnum chain = ChainEnum.getValueByName(ChainEnum.BSC_TFC.name());
+//        assert chain != null;
+//
+//        EthUsdtContract contract = contract(chain.getPrivateKey(), chain.getContractAddress(), chain.getUrl());
+//        EthFilter filter = getFilter(new BigInteger("18097238"), chain.getContractAddress());
+//
+//        contract.transferEventFlowable(filter).subscribe(e -> {
+//            System.out.println(1);
+//        }, error -> {
+//            log.error("--->", error);
+//        });
+
+        System.out.println(ChainService.getInstance(ChainEnum.BSC_TFC.name()).totalSupply());
     }
+
 }

--
Gitblit v1.9.1