1 files changed, 28 insertions, 13 deletions

diff --git a/openecomp-be/tools/zusammen-tools/src/main/java/org/openecomp/core/tools/commands/SetHealingFlag.java b/openecomp-be/tools/zusammen-tools/src/main/java/org/openecomp/core/tools/commands/SetHealingFlag.java
index fc20b66e3b..c8aee891d0 100644
--- a/openecomp-be/tools/zusammen-tools/src/main/java/org/openecomp/core/tools/commands/SetHealingFlag.java
+++ b/openecomp-be/tools/zusammen-tools/src/main/java/org/openecomp/core/tools/commands/SetHealingFlag.java
@@ -1,33 +1,48 @@
 package org.openecomp.core.tools.commands;
 
+import static org.openecomp.core.tools.commands.CommandName.RESET_OLD_VERSION;
+
 import com.datastax.driver.core.ResultSet;
+import java.util.ArrayList;
+import org.apache.commons.cli.CommandLine;
+import org.apache.commons.cli.Option;
 import org.openecomp.core.tools.store.HealingHandler;
 import org.openecomp.core.tools.store.VersionCassandraLoader;
 import org.openecomp.core.tools.store.zusammen.datatypes.HealingEntity;
 
-import java.util.ArrayList;
-
 
 /**
  * Created by ayalaben on 10/15/2017
  */
-public class SetHealingFlag {
+public class SetHealingFlag extends Command {
+
+    private static final String VERSION_OPTION = "v";
 
+    SetHealingFlag() {
+        options.addOption(Option.builder(VERSION_OPTION).hasArg().argName("version").desc("release version").build());
+    }
 
-  private SetHealingFlag(){}
+    @Override
+    public boolean execute(String[] args) {
+        CommandLine cmd = parseArgs(args);
+        String oldVersion = cmd.hasOption(VERSION_OPTION) ? cmd.getOptionValue(VERSION_OPTION) : null;
 
-  public static void populateHealingTable(String oldVersion) {
+        VersionCassandraLoader versionCassandraLoader = new VersionCassandraLoader();
+        ResultSet listItemVersion = versionCassandraLoader.listItemVersion();
 
-    VersionCassandraLoader versionCassandraLoader = new VersionCassandraLoader();
-    ResultSet listItemVersion = versionCassandraLoader.listItemVersion();
+        ArrayList<HealingEntity> healingEntities = new ArrayList<>();
 
-    ArrayList<HealingEntity> healingEntities = new ArrayList<>();
+        listItemVersion.iterator().forEachRemaining(entry -> healingEntities.add(new HealingEntity(entry.getString(0),
+                entry.getString(1), entry.getString(2), true, oldVersion)));
 
-    listItemVersion.iterator().forEachRemaining(entry -> healingEntities.add(new HealingEntity
-        (entry.getString(0),entry.getString(1),entry.getString(2),true,oldVersion)));
+        HealingHandler healingHandler = new HealingHandler();
+        healingHandler.populateHealingTable(healingEntities);
 
-    HealingHandler healingHandler = new HealingHandler();
-    healingHandler.populateHealingTable(healingEntities);
+        return true;
+    }
 
-  }
+    @Override
+    public CommandName getCommandName() {
+        return RESET_OLD_VERSION;
+    }
 }