Тимофеев Кирилл, ИТМО DWS, Stage 5

src/main/java/ru/vk/itmo/test/timofeevkirill/reference/ExceptionUtils.java

@@ -0,0 +1,26 @@
+package ru.vk.itmo.test.timofeevkirill.reference;
+
+import one.nio.http.HttpSession;
+import one.nio.http.Response;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import java.io.IOException;
+
+public final class ExceptionUtils {
+    private static final Logger log = LoggerFactory.getLogger(ExceptionUtils.class);
+
+    public static void handleErrorFromHandleRequest(Exception e, HttpSession session) {
+        log.error("Exception during handleRequest", e);
+        try {
+            session.sendResponse(new Response(Response.INTERNAL_ERROR, Response.EMPTY));
+        } catch (IOException ex) {
+            log.error("Exception while sending close connection", ex);
+            session.scheduleClose();
+        }
+    }
+
+    private ExceptionUtils() {
+
+    }
+}

src/main/java/ru/vk/itmo/test/timofeevkirill/reference/HandleResult.java

@@ -0,0 +1,14 @@
+package ru.vk.itmo.test.timofeevkirill.reference;
+
+public record HandleResult(int status, byte[] data, long timestamp) {
+
+    public HandleResult(int status, byte[] data, long timestamp) {
+        this.status = status;
+        this.data = data;
+        this.timestamp = timestamp;
+    }
+
+    public HandleResult(int status, byte[] data) {
+        this(status, data, 0);
+    }
+}

src/main/java/ru/vk/itmo/test/timofeevkirill/reference/MergeHandleResult.java

@@ -0,0 +1,83 @@
+package ru.vk.itmo.test.timofeevkirill.reference;
+
+import one.nio.http.HttpSession;
+import one.nio.http.Response;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import java.net.HttpURLConnection;
+import java.util.Queue;
+import java.util.concurrent.CompletableFuture;
+import java.util.concurrent.ConcurrentLinkedQueue;
+import java.util.concurrent.atomic.AtomicInteger;
+
+public class MergeHandleResult {
+    private static final Logger log = LoggerFactory.getLogger(MergeHandleResult.class);
+    private final Queue<CompletableFuture<HandleResult>> resultFutures;
+    private final Queue<HandleResult> results;
+    private final AtomicInteger count;
+    private final int ack;
+    private final int from;
+    private final HttpSession session;
+
+    public MergeHandleResult(HttpSession session, int size, int ack) {
+        this.session = session;
+        this.resultFutures = new ConcurrentLinkedQueue<>();
+        this.results = new ConcurrentLinkedQueue<>();
+        this.count = new AtomicInteger();
+        this.ack = ack;
+        this.from = size;
+    }
+
+    @SuppressWarnings("FutureReturnValueIgnored")
+    public void add(CompletableFuture<HandleResult> resultFuture) {
+        resultFutures.add(resultFuture);
+        resultFuture.whenComplete((result, e) -> {
+            if (e == null) {
+                results.add(result);
+                int get = count.incrementAndGet();
+                if (get == from) {
+                    sendResult();
+                    cancelRemainingFutures();
+                }
+            } else {
+                log.error("Error remote handle result: " + e);
+            }
+        });
+    }
+
+    private void cancelRemainingFutures() {
+        CompletableFuture<HandleResult> future;
+        while ((future = resultFutures.poll()) != null) {
+            future.cancel(true);
+        }
+    }
+
+    private void sendResult() {
+        HandleResult mergedResult = new HandleResult(HttpURLConnection.HTTP_GATEWAY_TIMEOUT, null);
+
+        int positiveCount = 0;
+        for (HandleResult handleResult : results) {
+            if (handleResult.status() == HttpURLConnection.HTTP_OK
+                    || handleResult.status() == HttpURLConnection.HTTP_CREATED
+                    || handleResult.status() == HttpURLConnection.HTTP_ACCEPTED
+                    || handleResult.status() == HttpURLConnection.HTTP_NOT_FOUND) {
+                positiveCount++;
+                if (mergedResult.timestamp() <= handleResult.timestamp()) {
+                    mergedResult = handleResult;
+                }
+            }
+        }
+
+        try {
+            if (positiveCount < ack) {
+                session.sendResponse(new Response(Response.GATEWAY_TIMEOUT, Response.EMPTY));
+            } else {
+                session.sendResponse(new Response(String.valueOf(mergedResult.status()), mergedResult.data()));
+            }
+        } catch (Exception e) {
+            ExceptionUtils.handleErrorFromHandleRequest(e, session);
+        }
+
+    }
+}