package org.example;

import com.sun.net.httpserver.HttpServer;

import java.net.InetSocketAddress;
import java.net.URI;
import java.net.http.HttpClient;
import java.net.http.HttpRequest;
import java.net.http.HttpResponse;
import java.net.http.HttpResponse.BodyHandlers;
import java.util.concurrent.CompletableFuture;
import java.util.concurrent.Executors;

public class ReqCloseIssue {

    static synchronized void log(String format, Object... args) {
        System.out.printf(format + "%n", args);
    }

    public static void main(String[] args) throws Exception {

        HttpServer slowServer = HttpServer.create(new InetSocketAddress(8080), 0);
        slowServer.createContext("/", exchange -> {
            try {
                Thread.sleep(1000);
            } catch (InterruptedException ignored) {
            }
            String response = "** message body **";
            exchange.sendResponseHeaders(200, response.length());
            exchange.getResponseBody().write(response.getBytes());
            exchange.close();
        });
        slowServer.setExecutor(Executors.newCachedThreadPool(r -> {
            Thread t = new Thread(r);
            t.setDaemon(true);
            return t;
        }));

        slowServer.start();

        try (HttpClient client = HttpClient.newHttpClient()) {
            HttpRequest request = HttpRequest.newBuilder(new URI("http://localhost:8080/")).build();
            CompletableFuture<String> future =
                    client.sendAsync(request, BodyHandlers.ofString())
                            .thenApply(HttpResponse::body)
                            .whenComplete((res, ex) -> {
                                if (ex == null) {
                                    log("Request was not cancelled; got response %s", res);
                                } else {
                                    log("Request was cancelled with %s", ex.getCause());
                                }
                            })
                            .exceptionally(ex -> "** cancelled **");
            Thread.sleep(200); // request will still be in progress at end of sleep
            future.cancel(true);
            log("Got response: %s", future.get()); // throws exception when agent present
        } finally {
            slowServer.stop(0);
        }
    }
}