SolrMeterMain.java

/**
 * Copyright Plugtree LLC
 *
 * 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.plugtree.solrmeter;

import com.google.inject.Guice;
import com.google.inject.Injector;
import com.google.inject.Module;
import com.plugtree.solrmeter.controller.StatisticsRepository;
import com.plugtree.solrmeter.runMode.SolrMeterRunMode;
import com.plugtree.solrmeter.model.*;
import com.plugtree.solrmeter.view.*;
import com.plugtree.stressTestScope.StressTestScopeModule;
import org.apache.log4j.Logger;

import java.io.File;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;

/**
 * 
 * @author tflobbe
 *
 */
public class SolrMeterMain {
	
	public static ConsoleFrame mainFrame;
	
	private static Injector injector;
    private static SolrMeterRunMode runMode;

	public static void main(String[] args) throws Exception {
		addPlugins(new ExpectedParameter(args, "statisticsLocation", "./plugins").getValue());
		createInjector();
        runMode = injector.getInstance(SolrMeterRunMode.class);
        runMode.main(injector);
        mainFrame = runMode.getMainFrame();
	}
	
	private static void addPlugins(String statisticsPath) {
		try {
			Logger.getLogger("boot").info("Adding plugins from " + statisticsPath);
			File pluginsDir = new File(statisticsPath);
			if(!pluginsDir.exists() || pluginsDir.list().length == 0) {
				Logger.getLogger("boot").warn("No plugins directory found. No pluggin added");
				return;
			}
			for(String jarName:pluginsDir.list()) {
				if(jarName.endsWith(".jar")) {
					Logger.getLogger("boot").info("Adding file " + jarName + " to classpath.");
					ClassPathHacker.addFile(new File(pluginsDir, jarName));
				}
			}
			SolrMeterConfiguration.setTransientProperty(StatisticsRepository.PLUGIN_STATISTICS_CONF_FILE_PROPERTY, statisticsPath + "/statistics-config.xml");
		} catch (IOException e) {
			Logger.getLogger("boot").error("Error while adding plugins to classpath", e);
			throw new RuntimeException(e);
		}
	}

	private static void createInjector() {
        List<Module> modules = new ArrayList<Module>();
        modules.add(createModule("guice.statisticsModule"));
        modules.add(createModule("guice.modelModule"));
        if (SolrMeterConfiguration.isHeadless()) {
            modules.add(createModule("guice.headlessModule"));
        }
        else {
            modules.add(createModule("guice.standalonePresentationModule"));
        }
        modules.add(createModule("guice.solrMeterRunModeModule"));
        modules.add(new StressTestScopeModule());
		injector = Guice.createInjector(modules);
	}

	private static Module createModule(String moduleKey) {
		String moduleClassName = SolrMeterConfiguration.getProperty(moduleKey);
		Logger.getLogger(SolrMeterMain.class).info("Using module: " + moduleClassName);
		Class<?> moduleClass;
		try {
			moduleClass = Class.forName(moduleClassName);
		} catch (ClassNotFoundException e) {
			Logger.getLogger(SolrMeterMain.class).error("Module for name " + moduleClassName + " can't be found! Make sure it is in classpath.", e);
			throw new RuntimeException("Could not start application, module for name " + moduleClassName + " was not found.", e);
		}
		Module moduleInstance;
		try {
			moduleInstance = (Module) moduleClass.newInstance();
		} catch (Exception e) {
			Logger.getLogger(SolrMeterMain.class).error("Module for name " + moduleClassName + " could not be instantiated.", e);
			throw new RuntimeException("Module for name " + moduleClassName + " could not be instantiated.", e);
		}
		return moduleInstance;
	}

	public static void restartApplication() {
        runMode.restartApplication();
	}

    public static SolrMeterRunMode getRunMode() {
        return runMode;
    }

}