Java tutorial
/* * * * RHQ Management Platform * * Copyright (C) 2005-2012 Red Hat, Inc. * * All rights reserved. * * * * This program is free software; you can redistribute it and/or modify * * it under the terms of the GNU General Public License, version 2, as * * published by the Free Software Foundation, and/or the GNU Lesser * * General Public License, version 2.1, also as published by the Free * * Software Foundation. * * * * 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 General Public License and the GNU Lesser General Public License * * for more details. * * * * You should have received a copy of the GNU General Public License * * and 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.rhq.server.control.command; import java.io.File; import org.apache.commons.cli.CommandLine; import org.apache.commons.cli.Options; import org.rhq.server.control.ControlCommand; import org.rhq.server.control.RHQControl; import org.rhq.server.control.RHQControlException; import org.rhq.server.control.util.ExecutorAssist; /** * This command is registered on Windows only. It performs Windows service removal. * <p/> * If we decide to enable this for Linux it is coded to by a synonym for stop. * * @author Jay Shaughnessy */ public class Remove extends ControlCommand { private Options options; public Remove() { options = new Options().addOption(null, "storage", false, "Remove RHQ storage node service") .addOption(null, "server", false, "Remove RHQ server service") .addOption(null, "agent", false, "Remove RHQ agent service"); } @Override public String getName() { return "remove"; } @Override public String getDescription() { return "Removes RHQ services on Windows."; } @Override public Options getOptions() { return options; } @Override protected String getReadmeFilename() { return "REMOVE_README.txt"; } @Override protected int exec(CommandLine commandLine) { int rValue = RHQControl.EXIT_CODE_OK; try { // if no options specified, then stop whatever is installed if (commandLine.getOptions().length == 0) { if (isAgentInstalled()) { rValue = Math.max(rValue, removeAgentService()); } // the server service may be installed even if the full server install fails. The files to execute // the remove are there after the initial unzip, so just go ahead and try to remove the service. This // may help clean up a failed install. rValue = Math.max(rValue, removeServerService()); if (isStorageInstalled()) { rValue = Math.max(rValue, removeStorageService()); } } else { if (commandLine.hasOption(AGENT_OPTION)) { if (isAgentInstalled()) { rValue = Math.max(rValue, removeAgentService()); } else { log.warn("It appears that the agent is not installed. The --" + AGENT_OPTION + " option will be ignored."); rValue = RHQControl.EXIT_CODE_INVALID_ARGUMENT; } } if (commandLine.hasOption(SERVER_OPTION)) { // the server service may be installed even if the full server install fails. The files to execute // the remove are there after the initial unzip, so just go ahead and try to remove the service. This // may help clean up a failed install. rValue = Math.max(rValue, removeServerService()); } if (commandLine.hasOption(STORAGE_OPTION)) { if (isStorageInstalled()) { rValue = Math.max(rValue, removeStorageService()); } else { log.warn("It appears that the storage node is not installed. The --" + STORAGE_OPTION + " option will be ignored."); rValue = RHQControl.EXIT_CODE_INVALID_ARGUMENT; } } } } catch (Exception e) { throw new RHQControlException("Failed to stop services", e); } return rValue; } private int removeStorageService() throws Exception { log.debug("Stopping RHQ storage node"); org.apache.commons.exec.CommandLine commandLine; int rValue; if (isWindows()) { commandLine = getCommandLine("rhq-storage", "remove"); try { rValue = ExecutorAssist.execute(getBinDir(), commandLine); } catch (Exception e) { log.debug("Failed to remove storage service", e); rValue = RHQControl.EXIT_CODE_OPERATION_FAILED; } } else { String pid = getStoragePid(); if (pid != null) { System.out.println("Stopping RHQ storage node..."); System.out.println("RHQ storage node (pid=" + pid + ") is stopping..."); commandLine = new org.apache.commons.exec.CommandLine("kill").addArgument(pid); rValue = ExecutorAssist.execute(getBinDir(), commandLine); System.out.println("RHQ storage node has stopped"); } else { rValue = RHQControl.EXIT_CODE_OK; } } return rValue; } private int removeServerService() throws Exception { log.debug("Stopping RHQ server"); org.apache.commons.exec.CommandLine commandLine; int rValue; if (isWindows()) { try { commandLine = getCommandLine("rhq-server", "remove"); rValue = ExecutorAssist.execute(getBinDir(), commandLine); } catch (Exception e) { // Ignore, service may not exist or be running, , script returns 1 log.debug("Failed to remove server service", e); rValue = RHQControl.EXIT_CODE_OPERATION_FAILED; } } else { commandLine = getCommandLine("rhq-server", "stop"); rValue = ExecutorAssist.execute(getBinDir(), commandLine); } return rValue; } private int removeAgentService() throws Exception { log.debug("Stopping RHQ agent"); File agentBinDir = new File(getAgentBasedir(), "bin"); org.apache.commons.exec.CommandLine commandLine; int rValue; if (isWindows()) { try { commandLine = getCommandLine("rhq-agent-wrapper", "remove"); rValue = ExecutorAssist.execute(agentBinDir, commandLine); } catch (Exception e) { // Ignore, service may not exist, script returns 1 log.debug("Failed to remove agent service", e); rValue = RHQControl.EXIT_CODE_OPERATION_FAILED; } } else { commandLine = getCommandLine("rhq-agent-wrapper", "stop"); rValue = ExecutorAssist.execute(agentBinDir, commandLine); } return rValue; } }