gerrit.onap Code Review - sdc.git/blob

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