com.aitusoftware.flute.server.http.HistogramQueryServer Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of flute-server Show documentation
Flute
The newest version!
/*
 * Copyright 2016 Aitu Software Limited.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.aitusoftware.flute.server.http;

import com.aitusoftware.flute.config.ConnectionFactory;
import com.aitusoftware.flute.config.DatabaseConfig;
import com.aitusoftware.flute.config.HistogramConfig;
import com.aitusoftware.flute.server.config.ServerConfig;
import com.aitusoftware.flute.server.dao.jdbc.HistogramRetrievalDao;
import com.aitusoftware.flute.server.dao.jdbc.MetricIdentifierDao;
import com.aitusoftware.flute.server.reporting.dao.ReportDao;
import com.aitusoftware.flute.server.reporting.http.DeleteReportHandler;
import com.aitusoftware.flute.server.reporting.http.GetReportHandler;
import com.aitusoftware.flute.server.reporting.http.ListReportsHandler;
import com.aitusoftware.flute.server.reporting.http.ModifyReportHandler;
import com.aitusoftware.flute.server.reporting.http.ViewReportConfigHandler;
import org.eclipse.jetty.server.Handler;
import org.eclipse.jetty.server.Request;
import org.eclipse.jetty.server.Server;
import org.eclipse.jetty.server.handler.ContextHandler;
import org.eclipse.jetty.server.handler.ContextHandlerCollection;
import org.eclipse.jetty.server.handler.DefaultHandler;
import org.eclipse.jetty.server.handler.ResourceHandler;

import javax.servlet.ServletException;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.io.PrintWriter;
import java.net.InetSocketAddress;

import static java.net.InetSocketAddress.createUnresolved;

public final class HistogramQueryServer
{
    private final ServerConfig serverConfig;
    private final DatabaseConfig databaseConfig;
    private final HistogramConfig histogramConfig;
    private final DatabaseConfig reportDatabaseConfig;

    public HistogramQueryServer(
            final ServerConfig serverConfig,
            final DatabaseConfig metricsDatabaseConfig,
            final HistogramConfig histogramConfig,
            final DatabaseConfig reportDatabaseConfig)
    {
        this.serverConfig = serverConfig;
        this.databaseConfig = metricsDatabaseConfig;
        this.histogramConfig = histogramConfig;
        this.reportDatabaseConfig = reportDatabaseConfig;
    }

    public void run() throws Exception
    {
        final InetSocketAddress socketAddress = createUnresolved("0.0.0.0", serverConfig.getHttpPort());
        final Server server = new Server(socketAddress);
        final ContextHandlerCollection handlers = new ContextHandlerCollection();

        try
        {
            final ConnectionFactory connectionFactory = new ConnectionFactory(databaseConfig);
            final HistogramRetrievalDao histogramRetrievalDao =
                    new HistogramRetrievalDao(connectionFactory, histogramConfig, 1024 * 1024);
            final MetricIdentifierDao metricIdentifierDao = new MetricIdentifierDao(connectionFactory);
            final ResourceHandler resourceHandler = new ResourceHandler();
            resourceHandler.setDirectoriesListed(false);
            // TODO needs to be parameterised
            if(System.getProperty("flute.resource.base") != null)
            {
                resourceHandler.setResourceBase(System.getProperty("flute.resource.base"));
            }
            else
            {
                resourceHandler.setResourceBase("/work/deploy/resources");
            }
            final ReportDao reportDao = new ReportDao(new ConnectionFactory(reportDatabaseConfig));
            addContextHandler(handlers, "/flute/report/create", ModifyReportHandler.createMode(reportDao));
            addContextHandler(handlers, "/flute/report/amend", ModifyReportHandler.amendMode(reportDao));
            addContextHandler(handlers, "/flute/report/delete", new DeleteReportHandler(reportDao));
            addContextHandler(handlers, "/flute/report/get", new GetReportHandler(reportDao));
            addContextHandler(handlers, "/flute/report/list", new ListReportsHandler(reportDao));
            addContextHandler(handlers, "/flute/report/spec", new ViewReportConfigHandler(reportDao, metricIdentifierDao));

            addContextHandler(handlers, "/flute/query/slaReport", new SlaReportHandler(histogramRetrievalDao, metricIdentifierDao, false, histogramConfig.asSupplier()));
            addContextHandler(handlers, "/flute/query/slaPercentiles", new SlaPercentileChartHandler(histogramRetrievalDao, metricIdentifierDao, false, histogramConfig.asSupplier()));
            addContextHandler(handlers, "/flute/query/timeSeries", new TimeSeriesChartHandler(histogramRetrievalDao, metricIdentifierDao, false));
            addContextHandler(handlers, "/flute/query/aggregator", new SlaReportHandler(histogramRetrievalDao, metricIdentifierDao, true, histogramConfig.asSupplier()));
            addContextHandler(handlers, "/flute/query/metricSearch", new MetricIdentifierSearchHandler(metricIdentifierDao));

            addContextHandler(handlers, "/flute/resources", resourceHandler);
        }
        catch(final Exception e)
        {
            addContextHandler(handlers, "/flute", new DefaultHandler()
            {
                @Override
                public void handle(final String target, final Request baseRequest, final HttpServletRequest request, final HttpServletResponse response) throws IOException, ServletException
                {
                    response.setContentType("text/plain; charset=utf-8");
                    response.setStatus(HttpServletResponse.SC_OK);
                    baseRequest.setHandled(true);
                    final PrintWriter writer = response.getWriter();
                    writer.append("Failed to configure: ").append(e.getMessage());
                    writer.flush();
                }
            });
        }

        server.setHandler(handlers);
        server.start();
        server.join();
    }

    private void addContextHandler(final ContextHandlerCollection handlers,
                                   final String contextPath,
                                   final Handler childHandler)
    {
        final ContextHandler handler = new ContextHandler();
        handler.setContextPath(contextPath);
        handler.setHandler(childHandler);
        handlers.addHandler(handler);
    }
}