LoggingBox.java

package com.loggingbox;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.PrintStream;
import java.net.URL;

import javax.servlet.ServletContextEvent;

import org.apache.log4j.Logger;
import org.eclipse.jetty.server.Handler;
import org.eclipse.jetty.server.Server;
import org.eclipse.jetty.server.handler.DefaultHandler;
import org.eclipse.jetty.server.handler.GzipHandler;
import org.eclipse.jetty.server.handler.HandlerList;
import org.eclipse.jetty.server.handler.ResourceHandler;
import org.eclipse.jetty.servlet.ServletContextHandler;
import org.eclipse.jetty.servlet.ServletHolder;
import org.springframework.web.context.ContextLoader;
import org.springframework.web.context.ContextLoaderListener;
import org.springframework.web.context.WebApplicationContext;

import com.loggingbox.service.ServerProperties;
import com.loggingbox.servlet.LogGetServlet;
import com.loggingbox.servlet.LogInsertServlet;
import com.loggingbox.servlet.LogsInsertServlet;

public class LoggingBox {

	private final static String DEFAULT_PROPERTIES_FILE = "default.properties";
	private final static String SPECIFIC_PROPERTIES_FILE = "loggingbox.properties";

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

		String homePath = "";

		homePath = args[0];
		System.setProperty("loggingbox.home", homePath);
		
		if (args.length > 1 && args[1].equals("1")) {
			PrintStream out = new PrintStream(new FileOutputStream(homePath
					+ "/logs/loggingbox.out", true));
			System.setOut(out);
			System.setErr(out);
		}

		Logger LOGGER = Logger.getLogger(LoggingBox.class);

		ContextLoaderListener contextLoaderListener = new ContextLoaderListener();
		Server server = new Server(8080);

		ResourceHandler resource_handler = new ResourceHandler();
		resource_handler.setDirectoriesListed(true);
		resource_handler.setWelcomeFiles(new String[] { "index.html" });
		resource_handler.setResourceBase(homePath + "/war");

		ServletContextHandler servletContext = new ServletContextHandler();
		servletContext.setContextPath("/api");
		servletContext.setInitParameter(ContextLoader.CONFIG_LOCATION_PARAM,
				"classpath*:applicationContext.xml");

		HandlerList handlers = new HandlerList();
		handlers.setHandlers(new Handler[] { resource_handler, servletContext,
				new DefaultHandler() });

		GzipHandler gzipHandler = new GzipHandler();
		gzipHandler.setHandler(handlers);

		// GZipRequestHandler gZipRequestHandler = new GZipRequestHandler();
		// gZipRequestHandler.setHandler(gzipHandler);
		server.setHandler(gzipHandler);

		server.start();

		contextLoaderListener.contextInitialized(new ServletContextEvent(
				servletContext.getServletContext()));

		WebApplicationContext context = ContextLoaderListener
				.getCurrentWebApplicationContext();

		ServerProperties serverProperties = context
				.getBean(ServerProperties.class);

		// load the default config file
		InputStream is = Thread.currentThread().getContextClassLoader()
				.getResourceAsStream(DEFAULT_PROPERTIES_FILE);
		if (is != null) {
			LOGGER.info(String.format("Found default file conf {%s}",
					DEFAULT_PROPERTIES_FILE));
			try {
				serverProperties.load(is);
			} catch (FileNotFoundException e) {
				LOGGER.error(String.format(
						"Failed to load default configuration file {%s}",
						DEFAULT_PROPERTIES_FILE), e);
			} catch (IOException e) {
				LOGGER.error(String.format(
						"Failed to load default configuration file {%s}",
						DEFAULT_PROPERTIES_FILE), e);
			}
		} else {
			URL url = Thread.currentThread().getContextClassLoader()
					.getResource(DEFAULT_PROPERTIES_FILE);
			if (url != null) {
				LOGGER.info(String.format("Found default file conf {%s}",
						url.toString()));
				try {
					serverProperties.load(new FileInputStream(new File(url
							.getFile())));
					LOGGER.info(url.getFile() + " has been loaded. ");
				} catch (FileNotFoundException e) {
					LOGGER.error(String.format(
							"Failed to load default configuration file {%s}",
							DEFAULT_PROPERTIES_FILE), e);
				} catch (IOException e) {
					LOGGER.error(String.format(
							"Failed to load default configuration file {%s}",
							DEFAULT_PROPERTIES_FILE), e);
				}
			} else {
				LOGGER.error(DEFAULT_PROPERTIES_FILE
						+ " file has not been found in your classpath. ");
			}
		}

		// check a specific config file to override some properties
		File specificFile = new File(homePath + "/etc/"
				+ SPECIFIC_PROPERTIES_FILE);
		if (specificFile != null && specificFile.exists()) {
			LOGGER.info(String.format("Found file conf {%s}",
					specificFile.toString()));
			try {
				serverProperties.load(new FileInputStream(specificFile));
			} catch (FileNotFoundException e) {
				LOGGER.error(
						String.format("Failed to load configuration file {%s}",
								specificFile), e);
			} catch (IOException e) {
				LOGGER.error(
						String.format("Failed to load configuration file {%s}",
								specificFile), e);
			}
		}

		servletContext.addServlet(
				new ServletHolder(context.getBean(LogGetServlet.class)),
				"/log/get");
		servletContext.addServlet(
				new ServletHolder(context.getBean(LogInsertServlet.class)),
				"/log/insert");
		servletContext.addServlet(
				new ServletHolder(context.getBean(LogsInsertServlet.class)),
				"/log/insert_batch");

//		 SystemHarvester systemHarvester = new SystemHarvester(InetAddress
//		 .getLocalHost().getHostName(), "8080", "test");
//		 systemHarvester.setDelayInMs(10000);
//		 systemHarvester.start();

		server.join();
	}

}