Mercurial > hg > release > thermostat-0.4
view agent/cli/src/main/java/com/redhat/thermostat/agent/cli/StorageCommand.java @ 651:205565370bab
Move command usage to properties file
Reviewed-by: sgehwolf
Review-thread: http://icedtea.classpath.org/pipermail/thermostat/2012-September/003395.html
author | Jon VanAlten <vanaltj@gmail.com> |
---|---|
date | Tue, 02 Oct 2012 17:43:38 -0400 |
parents | 486a299ab905 |
children | e88742733601 |
line wrap: on
line source
/* * Copyright 2012 Red Hat, Inc. * * This file is part of Thermostat. * * Thermostat is free software; you can redistribute it and/or modify * it under the terms of the GNU General Public License as published * by the Free Software Foundation; either version 2, or (at your * option) any later version. * * Thermostat 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 for more details. * * You should have received a copy of the GNU General Public License * along with Thermostat; see the file COPYING. If not see * <http://www.gnu.org/licenses/>. * * Linking this code with other modules is making a combined work * based on this code. Thus, the terms and conditions of the GNU * General Public License cover the whole combination. * * As a special exception, the copyright holders of this code give * you permission to link this code with independent modules to * produce an executable, regardless of the license terms of these * independent modules, and to copy and distribute the resulting * executable under terms of your choice, provided that you also * meet, for each linked independent module, the terms and conditions * of the license of that module. An independent module is a module * which is not derived from or based on this code. If you modify * this code, you may extend this exception to your version of the * library, but you are not obligated to do so. If you do not wish * to do so, delete this exception statement from your version. */ package com.redhat.thermostat.agent.cli; import java.io.File; import java.io.FileInputStream; import java.io.IOException; import java.util.Properties; import org.apache.commons.cli.Options; import com.redhat.thermostat.agent.cli.db.DBConfig; import com.redhat.thermostat.agent.cli.db.DBOptionParser; import com.redhat.thermostat.agent.cli.db.DBStartupConfiguration; import com.redhat.thermostat.agent.cli.db.MongoProcessRunner; import com.redhat.thermostat.common.cli.Arguments; import com.redhat.thermostat.common.cli.CommandContext; import com.redhat.thermostat.common.cli.CommandException; import com.redhat.thermostat.common.config.ConfigUtils; import com.redhat.thermostat.common.config.InvalidConfigurationException; import com.redhat.thermostat.common.tools.ApplicationException; import com.redhat.thermostat.common.tools.ApplicationState; import com.redhat.thermostat.common.tools.BasicCommand; public class StorageCommand extends BasicCommand { private static final String NAME = "storage"; private DBStartupConfiguration configuration; private DBOptionParser parser; private MongoProcessRunner runner; private void parseArguments(Arguments args) throws InvalidConfigurationException { this.configuration = new DBStartupConfiguration(); // configs, read everything that is in the configs File propertyFile = ConfigUtils.getStorageConfigurationFile(); if (!propertyFile.exists()) { throw new InvalidConfigurationException("can't access database configuration file " + propertyFile); } readAndSetProperties(propertyFile); parser = new DBOptionParser(configuration, args); parser.parse(); } @Override public void run(CommandContext ctx) throws CommandException { try { parseArgsAndRun(ctx); } catch (InvalidConfigurationException e) { throw new CommandException(e); } } private void parseArgsAndRun(CommandContext ctx) throws InvalidConfigurationException { parseArguments(ctx.getArguments()); // dry run means we don't do anything at all if (parser.isDryRun()) return; runner = createRunner(); try { switch (parser.getAction()) { case START: startService(); break; case STOP: stopService(); break; default: break; } getNotifier().fireAction(ApplicationState.SUCCESS); } catch (Exception e) { getNotifier().fireAction(ApplicationState.FAIL, e); } } private void readAndSetProperties(File propertyFile) throws InvalidConfigurationException { Properties properties = new Properties(); try { properties.load(new FileInputStream(propertyFile)); } catch (IOException e) { throw new InvalidConfigurationException(e); } if (properties.containsKey(DBConfig.PORT.name())) { String port = (String) properties.get(DBConfig.PORT.name()); int localPort = Integer.parseInt(port); configuration.setPort(localPort); } else { throw new InvalidConfigurationException(DBConfig.PORT + " property missing"); } if (properties.containsKey(DBConfig.PROTOCOL.name())) { String url = (String) properties.get(DBConfig.PROTOCOL.name()); configuration.setProtocol(url); } else { throw new InvalidConfigurationException(DBConfig.PROTOCOL + " property missing"); } if (properties.containsKey(DBConfig.BIND.name())) { String ip = (String) properties.get(DBConfig.BIND.name()); configuration.setBindIP(ip); } else { throw new InvalidConfigurationException(DBConfig.BIND + " property missing"); } } private void startService() throws IOException, InterruptedException, InvalidConfigurationException, ApplicationException { runner.startService(); getNotifier().fireAction(ApplicationState.START); } private void stopService() throws IOException, InterruptedException, InvalidConfigurationException, ApplicationException { check(); runner.stopService(); getNotifier().fireAction(ApplicationState.STOP); } MongoProcessRunner createRunner() { return new MongoProcessRunner(configuration, parser.isQuiet()); } private void check() throws InvalidConfigurationException { if (!configuration.getDBPath().exists() || !configuration.getLogFile().getParentFile().exists() || !configuration.getPidFile().getParentFile().exists()) { throw new InvalidConfigurationException("database directories do not exist..."); } } @Override public DBStartupConfiguration getConfiguration() { return configuration; } @Override public String getName() { return NAME; } @Override public Options getOptions() { return DBOptionParser.getOptions(); } }