gerrit.onap Code Review - sdc.git/blob

   1 package org.openecomp.core.tools.commands;
   2
   3 import static org.openecomp.core.tools.commands.CommandName.RESET_OLD_VERSION;
   4
   5 import com.datastax.driver.core.ResultSet;
   6 import java.util.ArrayList;
   7 import org.apache.commons.cli.CommandLine;
   8 import org.apache.commons.cli.Option;
   9 import org.openecomp.core.tools.store.HealingHandler;
  10 import org.openecomp.core.tools.store.VersionCassandraLoader;
  11 import org.openecomp.core.tools.store.zusammen.datatypes.HealingEntity;
  12
  13
  14 /**
  15  * Created by ayalaben on 10/15/2017
  16  */
  17 public class SetHealingFlag extends Command {
  18
  19     private static final String VERSION_OPTION = "v";
  20
  21     SetHealingFlag() {
  22         options.addOption(Option.builder(VERSION_OPTION).hasArg().argName("version").desc("release version").build());
  23     }
  24
  25     @Override
  26     public boolean execute(String[] args) {
  27         CommandLine cmd = parseArgs(args);
  28         String oldVersion = cmd.hasOption(VERSION_OPTION) ? cmd.getOptionValue(VERSION_OPTION) : null;
  29
  30         VersionCassandraLoader versionCassandraLoader = new VersionCassandraLoader();
  31         ResultSet listItemVersion = versionCassandraLoader.listItemVersion();
  32
  33         ArrayList<HealingEntity> healingEntities = new ArrayList<>();
  34
  35         listItemVersion.iterator().forEachRemaining(entry -> healingEntities.add(new HealingEntity(entry.getString(0),
  36                 entry.getString(1), entry.getString(2), true, oldVersion)));
  37
  38         HealingHandler healingHandler = new HealingHandler();
  39         healingHandler.populateHealingTable(healingEntities);
  40
  41         return true;
  42     }
  43
  44     @Override
  45     public CommandName getCommandName() {
  46         return RESET_OLD_VERSION;
  47     }
  48 }