From 43e9c46308e72856c710bb726fc831ab8db9e769 Mon Sep 17 00:00:00 2001 From: KKSU <15274802129@163.com> Date: Fri, 12 Jul 2024 14:27:02 +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 66a17f4..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.startsWith(ETH_PREFIX)) { - 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.startsWith(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