LogSummarizer.java

package org.ggp.base.apps.logging;

import java.io.IOException;
import java.net.ServerSocket;
import java.net.Socket;

import org.ggp.base.util.http.HttpReader;
import org.ggp.base.util.http.HttpWriter;
import org.ggp.base.util.logging.LogSummaryGenerator;

import external.JSON.JSONException;

/**
 * The "Exponent" Log Summarizer Server is a multi-threaded web server that makes
 * log summaries and sends them back to remote clients. These log summaries should
 * not contain any sensitive data; the summarizer can be queried by anyone and its
 * summaries are made publicly available on the GGP.org viewer alongside the other
 * information about each match.
 *
 * SAMPLE INVOCATION (when running locally):
 *
 * ResourceLoader.load_raw('http://127.0.0.1:9199/matchABC');
 *
 * The Log Summarizer Server replies with a JSON summary of the logs for "matchABC".
 *
 * @author Sam Schreiber
 */
public class LogSummarizer
{
    public final LogSummaryGenerator theGenerator;
    public static final int SERVER_PORT = 9199;

    public LogSummarizer(LogSummaryGenerator theGenerator) {
        this.theGenerator = theGenerator;
    }

    class SummarizeLogThread extends Thread {
        private Socket connection;

        public SummarizeLogThread(Socket connection) throws IOException, JSONException {
            this.connection = connection;
        }

        @Override
        public void run() {
            try {
                String matchId = HttpReader.readAsServer(connection);
                System.out.println(System.currentTimeMillis() + ": Got request for logs for " + matchId);
                String theResponse = theGenerator.getLogSummary(matchId);
                System.out.println(System.currentTimeMillis() + ": Got " + theResponse.length() + " bytes of logs for " + matchId);
                HttpWriter.writeAsServer(connection, theResponse);
                connection.close();
                System.out.println(System.currentTimeMillis() + ": Successfully replied to log request for " + matchId);
            } catch (IOException e) {
                e.printStackTrace();
                throw new RuntimeException(e);
            }
        }
    }

    private boolean shouldStop = false;
    private ServerSocket listener = null;

    public void stopAbruptly() throws IOException {
        shouldStop = true;
        if (listener != null) {
            listener.close();
            listener = null;
        }
    }

    public void runSummarizer() {
        try {
            listener = new ServerSocket(SERVER_PORT);
        } catch (IOException e) {
            System.err.println("Could not open server on port " + SERVER_PORT + ": " + e);
            e.printStackTrace();
            return;
        }

        while (!shouldStop) {
            try {
                Socket connection = listener.accept();
                Thread handlerThread = new SummarizeLogThread(connection);
                handlerThread.start();
            } catch (Exception e) {
                System.err.println(e);
            }
        }
    }

    public static void main(String[] args) {
        new LogSummarizer(null).runSummarizer();
    }
}