org.web3j.spring.actuate.Web3jHealthIndicator Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of web3j-spring-boot-starter Show documentation
web3j Spring Boot Starter
The newest version!
package org.web3j.spring.actuate;

import org.springframework.boot.actuate.health.AbstractHealthIndicator;
import org.springframework.boot.actuate.health.Health;
import org.springframework.util.Assert;
import org.web3j.protocol.Web3j;

import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.CompletableFuture;

/**
 * Health check indicator for Web3j
 */
public class Web3jHealthIndicator extends AbstractHealthIndicator {

    private Web3j web3j;

    public Web3jHealthIndicator(Web3j web3j) {
        Assert.notNull(web3j, "Web3j must not be null");
        this.web3j = web3j;
    }

    @Override
    protected void doHealthCheck(Health.Builder builder) throws Exception {
        try {
            boolean listening = web3j.netListening().send().isListening();
            if (!listening) {
                builder.down();
            } else {
                builder.up();
                List futures = new ArrayList<>();

                futures.add(web3j.netVersion()
                        .sendAsync()
                        .thenApply(netVersion ->
                                builder.withDetail("netVersion", netVersion.getNetVersion())));

                futures.add(web3j.web3ClientVersion()
                        .sendAsync()
                        .thenApply(web3ClientVersion ->
                                builder.withDetail("clientVersion", web3ClientVersion.getWeb3ClientVersion())));

                futures.add(web3j.ethBlockNumber()
                        .sendAsync()
                        .thenApply(ethBlockNumber ->
                                builder.withDetail("blockNumber", ethBlockNumber.getBlockNumber())));

                futures.add(web3j.ethProtocolVersion()
                        .sendAsync()
                        .thenApply(ethProtocolVersion ->
                                builder.withDetail("protocolVersion", ethProtocolVersion.getProtocolVersion())));

                futures.add(web3j.netPeerCount()
                        .sendAsync()
                        .thenApply(netPeerCount ->
                                builder.withDetail("netPeerCount", netPeerCount.getQuantity())));

                CompletableFuture.allOf(futures.toArray(new CompletableFuture[futures.size()])).get();
            }

        } catch (Exception ex) {
            builder.down(ex);
        }
    }
}