/** SonarQube* Copyright (C) 2009-2016 SonarSource SA* mailto:contact AT sonarsource DOT com** This program is free software; you can redistribute it and/or* modify it under the terms of the GNU Lesser General Public* License as published by the Free Software Foundation; either* version 3 of the License, or (at your option) any later version.** This program is distributed in the hope that it will be useful,* but WITHOUT ANY WARRANTY; without even the implied warranty of* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU* Lesser General Public License for more details.** You should have received a copy of the GNU Lesser General Public License* along with this program; if not, write to the Free Software Foundation,* Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.*/
package org.sonar.search;import org.apache.lucene.util.StringHelper;
import org.elasticsearch.cluster.health.ClusterHealthStatus;
import org.elasticsearch.common.unit.TimeValue;
import org.elasticsearch.node.Node;
import org.elasticsearch.node.NodeBuilder;
import org.sonar.process.Jmx;
import org.sonar.process.MinimumViableSystem;
import org.sonar.process.Monitored;
import org.sonar.process.ProcessEntryPoint;
import org.sonar.process.Props;public class SearchServer implements Monitored {private final EsSettings settings;private Node node;public SearchServer(Props props) {this.settings = new EsSettings(props);new MinimumViableSystem().checkWritableTempDir();}@Overridepublic void start() {Jmx.register(EsSettingsMBean.OBJECT_NAME, settings);initBootstrap();node = NodeBuilder.nodeBuilder().settings(settings.build()).build();node.start();}// copied from https://github.com/elastic/elasticsearch/blob/v2.3.3/core/src/main/java/org/elasticsearch/bootstrap/Bootstrap.javaprivate static void initBootstrap() {// init lucene random seed. it will use /dev/urandom where available:StringHelper.randomId();}@Overridepublic boolean isUp() {return node != null && node.client().admin().cluster().prepareHealth().setWaitForYellowStatus().setTimeout(TimeValue.timeValueSeconds(30L)).get().getStatus() != ClusterHealthStatus.RED;}@Overridepublic void awaitStop() {try {while (node != null && !node.isClosed()) {Thread.sleep(200L);}} catch (InterruptedException e) {// Restore the interrupted statusThread.currentThread().interrupt();}}@Overridepublic void stop() {if (node != null && !node.isClosed()) {node.close();}Jmx.unregister(EsSettingsMBean.OBJECT_NAME);}public static void main(String... args) {ProcessEntryPoint entryPoint = ProcessEntryPoint.createForArguments(args);new SearchLogging().configure(entryPoint.getProps());SearchServer searchServer = new SearchServer(entryPoint.getProps());entryPoint.launch(searchServer);}
}
/** SonarQube* Copyright (C) 2009-2016 SonarSource SA* mailto:contact AT sonarsource DOT com** This program is free software; you can redistribute it and/or* modify it under the terms of the GNU Lesser General Public* License as published by the Free Software Foundation; either* version 3 of the License, or (at your option) any later version.** This program is distributed in the hope that it will be useful,* but WITHOUT ANY WARRANTY; without even the implied warranty of* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU* Lesser General Public License for more details.** You should have received a copy of the GNU Lesser General Public License* along with this program; if not, write to the Free Software Foundation,* Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.*/
package org.sonar.process;import java.io.File;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;public class ProcessEntryPoint implements Stoppable {public static final String PROPERTY_PROCESS_KEY = "process.key";public static final String PROPERTY_PROCESS_INDEX = "process.index";public static final String PROPERTY_TERMINATION_TIMEOUT = "process.terminationTimeout";public static final String PROPERTY_SHARED_PATH = "process.sharedDir";private final Props props;private final String processKey;private final int processNumber;private final File sharedDir;private final Lifecycle lifecycle = new Lifecycle();private final ProcessCommands commands;private final SystemExit exit;private volatile Monitored monitored;private volatile StopperThread stopperThread;private final StopWatcher stopWatcher;// new Runnable() is important to avoid conflict of call to ProcessEntryPoint#stop() with Thread#stop()private Thread shutdownHook = new Thread(new Runnable() {@Overridepublic void run() {exit.setInShutdownHook();stop();}});ProcessEntryPoint(Props props, SystemExit exit, ProcessCommands commands) {this(props, getProcessNumber(props), getSharedDir(props), exit, commands);}private ProcessEntryPoint(Props props, int processNumber, File sharedDir, SystemExit exit, ProcessCommands commands) {this.props = props;this.processKey = props.nonNullValue(PROPERTY_PROCESS_KEY);this.processNumber = processNumber;this.sharedDir = sharedDir;this.exit = exit;this.commands = commands;this.stopWatcher = new StopWatcher(commands, this);}public ProcessCommands getCommands() {return commands;}public Props getProps() {return props;}public String getKey() {return processKey;}public int getProcessNumber() {return processNumber;}public File getSharedDir() {return sharedDir;}/*** Launch process and waits until it's down*/public void launch(Monitored mp) {if (!lifecycle.tryToMoveTo(Lifecycle.State.STARTING)) {throw new IllegalStateException("Already started");}monitored = mp;Logger logger = LoggerFactory.getLogger(getClass());try {logger.info("Starting " + getKey());Runtime.getRuntime().addShutdownHook(shutdownHook);stopWatcher.start();monitored.start();boolean up = false;while (!up) {up = monitored.isUp();Thread.sleep(20L);}// notify monitor that process is readycommands.setUp();if (lifecycle.tryToMoveTo(Lifecycle.State.STARTED)) {monitored.awaitStop();}} catch (Exception e) {logger.warn("Fail to start " + getKey(), e);} finally {stop();}}boolean isStarted() {return lifecycle.getState() == Lifecycle.State.STARTED;}/*** Blocks until stopped in a timely fashion (see {@link org.sonar.process.StopperThread})*/void stop() {stopAsync();try {// stopperThread is not null for sure// join() does nothing if thread already finishedstopperThread.join();lifecycle.tryToMoveTo(Lifecycle.State.STOPPED);} catch (InterruptedException e) {Thread.currentThread().interrupt();}exit.exit(0);}@Overridepublic void stopAsync() {if (lifecycle.tryToMoveTo(Lifecycle.State.STOPPING)) {stopperThread = new StopperThread(monitored, commands, Long.parseLong(props.nonNullValue(PROPERTY_TERMINATION_TIMEOUT)));stopperThread.start();stopWatcher.stopWatching();}}Lifecycle.State getState() {return lifecycle.getState();}Thread getShutdownHook() {return shutdownHook;}public static ProcessEntryPoint createForArguments(String[] args) {Props props = ConfigurationUtils.loadPropsFromCommandLineArgs(args);File sharedDir = getSharedDir(props);int processNumber = getProcessNumber(props);ProcessCommands commands = DefaultProcessCommands.main(sharedDir, processNumber);return new ProcessEntryPoint(props, processNumber, sharedDir, new SystemExit(), commands);}private static int getProcessNumber(Props props) {return Integer.parseInt(props.nonNullValue(PROPERTY_PROCESS_INDEX));}private static File getSharedDir(Props props) {return props.nonNullValueAsFile(PROPERTY_SHARED_PATH);}
}
