com.yelstream.topp.standard.net.tool.HttpScanner2 Maven / Gradle / Ivy

Go to download
package com.yelstream.topp.standard.net.tool;

import com.fasterxml.jackson.annotation.JsonInclude;
import com.fasterxml.jackson.annotation.JsonProperty;
import com.fasterxml.jackson.databind.ObjectMapper;

import java.net.URI;
import java.net.http.HttpClient;
import java.net.http.HttpRequest;
import java.net.http.HttpResponse;
import java.util.List;
import java.util.concurrent.CompletableFuture;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;

public class HttpScanner2 {

    private static final String HOST = "localhost";
    private static final int PORT = 8080;
    private static final List PATHS =
            List.of(
                    "/q/arc",
                    "/q/arc/beans",
                    "/q/arc/observers",
                    "/q/arc/removed-beans",
                    "/q/dev-ui",
                    "/q/health",
                    "/q/health-ui",
                    "/q/health/group",
                    "/q/health/group/*",
                    "/q/health/live",
                    "/q/health/ready",
                    "/q/health/started",
                    "/q/health/well",
                    "/q/metrics",
                    "/q/openapi",
                    "/q/swagger-ui"
            );

    private static final int SIMULTANEOUS_REQUESTS = 5;

    static class HalLink {
        public String href;
        public HalLink(String href) { this.href = href; }
    }

    static class HalResource {
        @JsonProperty("_links")
        public Links links;
        public String path;
        public int statusCode;
        public String body;

        public HalResource(String path, int statusCode, String body) {
            this.links = new Links(path);
            this.path = path;
            this.statusCode = statusCode;
            this.body = body;
        }
    }

    static class Links {
        public HalLink self;

        public Links(String path) {
            this.self = new HalLink("http://" + HOST + ":" + PORT + path);
        }
    }

    public static void main(String[] args) throws Exception {
        ExecutorService executorService = Executors.newFixedThreadPool(SIMULTANEOUS_REQUESTS);
        HttpClient client = HttpClient.newBuilder()
                .executor(executorService)
                .build();  //TO-DO: Auto-closable!

        List> futures = PATHS.stream()
                .map(path -> {
                    URI uri = URI.create("http://" + HOST + ":" + PORT + path);
                    HttpRequest request = HttpRequest.newBuilder().uri(uri).build();
                    return client.sendAsync(request, HttpResponse.BodyHandlers.ofString())
                            .thenApply(response -> new HalResource(
                                    path,
                                    response.statusCode(),
                                    response.body().substring(0, Math.min(response.body().length(), 100))
                            )).exceptionally(e -> {
                                e.printStackTrace();
                                return new HalResource(path, 500, "Request failed: " + e.getMessage());
                            });
                })
                .toList();

        @SuppressWarnings({"unchecked","rawtypes"})
        CompletableFuture allFutures = CompletableFuture.allOf(futures.toArray(new CompletableFuture[0]));

        allFutures.thenRun(() -> {
            List results = futures.stream()
                    .map(CompletableFuture::join)
                    .toList();
            try {
                ObjectMapper mapper = new ObjectMapper();
                mapper.setSerializationInclusion(JsonInclude.Include.NON_NULL);
                String jsonOutput = mapper.writerWithDefaultPrettyPrinter().writeValueAsString(results);
                System.out.println(jsonOutput);
            } catch (Exception e) {
                e.printStackTrace();
            }
        }).join();

        executorService.shutdown();
    }
}