Java tutorial
/* * Licensed to the Apache Software Foundation (ASF) under one * or more contributor license agreements. See the NOTICE file * distributed with this work for additional information * regarding copyright ownership. The ASF licenses this file * to you 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 org.apache.usergrid.chop.webapp; import com.google.inject.Guice; import com.google.inject.Injector; import com.google.inject.servlet.GuiceServletContextListener; import com.netflix.config.ConcurrentCompositeConfiguration; import com.netflix.config.ConfigurationManager; import org.apache.commons.cli.CommandLine; import org.apache.shiro.guice.aop.ShiroAopModule; import org.apache.usergrid.chop.webapp.dao.SetupDao; import org.apache.usergrid.chop.webapp.elasticsearch.ElasticSearchFig; import org.apache.usergrid.chop.webapp.elasticsearch.EsEmbedded; import org.apache.usergrid.chop.webapp.elasticsearch.IElasticSearchClient; import org.apache.usergrid.chop.webapp.service.InjectorFactory; import org.apache.usergrid.chop.webapp.service.shiro.CustomShiroWebModule; import org.apache.usergrid.chop.webapp.service.util.TimeUtil; import org.safehaus.guicyfig.Env; import org.slf4j.Logger; import org.slf4j.LoggerFactory; import javax.servlet.ServletContext; import javax.servlet.ServletContextEvent; import java.io.IOException; import java.util.Enumeration; import static org.apache.usergrid.chop.webapp.ChopUiFig.CONTEXT_TEMPDIR_KEY; /** * ... */ @SuppressWarnings("UnusedDeclaration") public class ChopUiConfig extends GuiceServletContextListener { private final static Logger LOG = LoggerFactory.getLogger(ChopUiConfig.class); private EsEmbedded esEmbedded; private Injector injector; private ServletContext context; @Override protected Injector getInjector() { if (injector != null) { return injector; } injector = Guice.createInjector(new CustomShiroWebModule(context), new ShiroAopModule(), new ChopUiModule()); InjectorFactory.setInjector(injector); return injector; } @Override public void contextInitialized(ServletContextEvent servletContextEvent) { context = servletContextEvent.getServletContext(); context.setAttribute(Injector.class.getName(), getInjector()); Injector injector = getInjector(); ElasticSearchFig elasticSearchFig = injector.getInstance(ElasticSearchFig.class); ChopUiFig chopUiFig = injector.getInstance(ChopUiFig.class); /* * -------------------------------------------------------------------- * Archaius Configuration Settings * -------------------------------------------------------------------- */ ConcurrentCompositeConfiguration ccc = new ConcurrentCompositeConfiguration(); Env env = Env.getEnvironment(); boolean embedded = false; if (env == Env.ALL) { ConfigurationManager.getDeploymentContext().setDeploymentEnvironment("PROD"); LOG.info("Setting environment to: PROD"); /* * -------------------------------------------------------------------- * Extract Configuration Settings from CommandLine * -------------------------------------------------------------------- */ if (ChopUiJettyRunner.getCommandLine() != null) { CommandLine cl = ChopUiJettyRunner.getCommandLine(); if (cl.hasOption('d')) { String dataDir = cl.getOptionValue('d'); LOG.info("The -d option is given, replacing data directory with {}", dataDir); elasticSearchFig.bypass(ElasticSearchFig.DATA_DIR_KEY, dataDir); } if (cl.hasOption('e')) { startEmbeddedES(elasticSearchFig); } if (cl.hasOption('c')) { String serverHostPort = cl.getOptionValue('c'); int seperatorIndex = serverHostPort.indexOf(":"); String serverHost = serverHostPort.substring(0, seperatorIndex); String serverPort = serverHostPort.substring(seperatorIndex + 1, serverHostPort.length()); LOG.info("The -c option is given, replacing host with {} and port with {}", serverHost, serverPort); elasticSearchFig.bypass(ElasticSearchFig.SERVERS_KEY, serverHost); elasticSearchFig.bypass(ElasticSearchFig.PORT_KEY, serverPort); } if (cl.hasOption('n')) { String clusterName = cl.getOptionValue('n'); elasticSearchFig.bypass(ElasticSearchFig.CLUSTER_NAME_KEY, clusterName); } } else { LOG.warn("ChopUi not started via Launcher - no command line argument processing will take place."); } } else if (env == Env.UNIT) { LOG.info("Operating in UNIT environment"); } try { ConfigurationManager.loadCascadedPropertiesFromResources("chop-ui"); } catch (IOException e) { LOG.warn("Failed to cascade load configuration properties: ", e); } /* * -------------------------------------------------------------------- * Environment Based Configuration Property Adjustments * -------------------------------------------------------------------- */ if (LOG.isDebugEnabled()) { Enumeration<String> names = context.getAttributeNames(); LOG.debug("Dumping attribute names: "); while (names.hasMoreElements()) { String name = names.nextElement(); LOG.debug("attribute {} = {}", name, context.getAttribute(name)); } } // Checking if a temp directory is defined - usually null String contextTempDir = (String) context.getAttribute(CONTEXT_TEMPDIR_KEY); LOG.info("From servlet context: {} = {}", CONTEXT_TEMPDIR_KEY, contextTempDir); if (contextTempDir == null) { LOG.info("From ChopUiFig {} = {}", CONTEXT_TEMPDIR_KEY, chopUiFig.getContextTempDir()); } setupStorage(); } private static EsEmbedded startEmbeddedES(ElasticSearchFig elasticSearchFig) { LOG.info("The -e option has been provided: launching embedded elasticsearch instance."); // This will set the parameters needed in the fig to attach to the embedded instance EsEmbedded es = new EsEmbedded(elasticSearchFig); es.start(); long pause = 5000; LOG.info("Pausing for {} ms so embedded elasticsearch can complete initialization.", pause); TimeUtil.sleep(pause); return es; } private void setupStorage() { LOG.info("Setting up the storage..."); IElasticSearchClient esClient = getInjector().getInstance(IElasticSearchClient.class); esClient.start(); SetupDao setupDao = getInjector().getInstance(SetupDao.class); LOG.info("esClient: {}", esClient); LOG.info("setupDao: {}", setupDao); try { setupDao.setup(); } catch (Exception e) { LOG.error("Failed to setup the storage!", e); } } @Override public void contextDestroyed(final ServletContextEvent servletContextEvent) { super.contextDestroyed(servletContextEvent); if (esEmbedded != null) { esEmbedded.stop(); } } }