openecomp-be/tools/zusammen-tools/src/main/java/org/openecomp/core/tools/Commands/ExportDataCommand.java

   1 package org.openecomp.core.tools.Commands;
   2
   3 import static java.nio.file.Files.createDirectories;
   4
   5 import com.amdocs.zusammen.datatypes.SessionContext;
   6 import org.apache.commons.io.FileUtils;
   7 import org.openecomp.core.tools.Commands.exportdata.ElementHandler;
   8 import org.openecomp.core.tools.Commands.exportdata.ImportProperties;
   9 import org.openecomp.core.tools.Commands.exportdata.ItemHandler;
  10 import org.openecomp.core.tools.Commands.exportdata.VersionHandler;
  11 import org.openecomp.core.tools.util.ZipUtils;
  12 import org.openecomp.core.zusammen.impl.CassandraConnectionInitializer;
  13 import org.openecomp.sdc.logging.api.Logger;
  14 import org.openecomp.sdc.logging.api.LoggerFactory;
  15
  16 import java.io.File;
  17 import java.io.IOException;
  18 import java.nio.file.Files;
  19 import java.nio.file.Path;
  20 import java.nio.file.Paths;
  21 import java.time.LocalDateTime;
  22 import java.time.format.DateTimeFormatter;
  23 import java.util.HashSet;
  24 import java.util.Set;
  25 public class ExportDataCommand {
  26     private static final Logger logger = LoggerFactory.getLogger(ExportDataCommand.class);
  27
  28     public static void exportData(SessionContext context, String filterItem) {
  29         try {
  30             Set<String> filteredItem = new HashSet<>();
  31             filteredItem.add(filterItem);
  32             ImportProperties.initParams();
  33             CassandraConnectionInitializer.setCassandraConnectionPropertiesToSystem();
  34             Path rootDir = Paths.get(ImportProperties.ROOT_DIRECTORY);
  35             initDir(rootDir);
  36             if (filterItem != null) {
  37                 filterItem = filterItem.replaceAll("\\r", "");
  38             }
  39             new ItemHandler().createItemsData(context, filteredItem);
  40             new VersionHandler().loadVersions(filteredItem);
  41             new ElementHandler().loadElements(filteredItem);
  42             zipPath(rootDir,filteredItem);
  43             FileUtils.forceDelete(rootDir.toFile());
  44         } catch (Exception ex) {
  45             logger.error(ex.getMessage(), ex);
  46         }
  47
  48     }
  49     private static void zipPath(Path rootDir,Set<String> filterItem ) throws Exception{
  50         LocalDateTime date = LocalDateTime.now();
  51         DateTimeFormatter formatter = DateTimeFormatter.ISO_LOCAL_DATE_TIME;
  52         String dateStr = date.format(formatter);
  53         String zipFile = System.getProperty("user.home")+ File.separatorChar+"onboarding_import"+ dateStr + ".zip";
  54         ZipUtils.createZip(zipFile, rootDir,filterItem);
  55         logger.info("Exported file :" + zipFile);
  56         System.out.println("Exported file :" + zipFile);
  57     }
  58
  59
  60     public static void initDir(Path rootDir ) throws IOException{
  61         if (Files.exists(rootDir)) {
  62             FileUtils.forceDelete(rootDir.toFile());
  63         }
  64         createDirectories(rootDir);
  65     }
  66
  67 }