com.danielflower.apprunner.router.web.AppsCallAggregator Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of app-runner-router Show documentation

A reverse proxy for AppRunner that allows you horizontally scale AppRunner

There is a newer version: 2.0.2

package com.danielflower.apprunner.router.web;

import com.danielflower.apprunner.router.mgmt.Cluster;
import com.danielflower.apprunner.router.mgmt.MapManager;
import org.eclipse.jetty.server.Request;
import org.eclipse.jetty.server.handler.AbstractHandler;
import org.json.JSONArray;
import org.json.JSONObject;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import javax.servlet.ServletException;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.net.URI;
import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.TimeoutException;

class AppsCallAggregator extends AbstractHandler {
    private static final Logger log = LoggerFactory.getLogger(AppsCallAggregator.class);
    private final MapManager mapManager;
    private final Cluster cluster;

    public AppsCallAggregator(MapManager mapManager, Cluster cluster) {
        this.mapManager = mapManager;
        this.cluster = cluster;
    }

    public void handle(String target, Request baseRequest, HttpServletRequest request, HttpServletResponse response) throws IOException, ServletException {
        if (canHandle(target, request)) {
            try {
                List results = mapManager.loadAllApps(URI.create(request.getRequestURL().toString()), cluster.getRunners());
                JSONObject all = new JSONObject();
                List unsorted = new ArrayList<>();

                for (JSONObject result : results) {
                    JSONArray singleApps = result.getJSONArray("apps");
                    for (Object singleApp : singleApps) {
                        unsorted.add((JSONObject) singleApp);
                    }
                }
                unsorted.sort((o1, o2) -> o1.getString("name").compareTo(o2.getString("name")));
                JSONArray apps = new JSONArray();
                for (JSONObject jsonObject : unsorted) {
                    apps.put(jsonObject);
                }
                all.put("apps", apps);

                response.setStatus(200);
                response.setHeader("Content-Type", "application/json");
                response.getWriter().append(all.toString(4)).close();
            } catch (TimeoutException te) {
                response.sendError(504, "Timed out calling all the app runner instances with message " + te.getMessage());
            } catch (Exception e) {
                log.error("Error while aggregating the " + target + " call", e);
                response.sendError(502, "Error while aggregating the calls.");
            }
            baseRequest.setHandled(true);
        }
    }

    public static boolean canHandle(String target, HttpServletRequest request) {
        return "/api/v1/apps".equals(target) && request.getMethod().equals("GET");
    }
}