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

   1 package org.openecomp.core.tools.Commands.exportdata;
   2
   3 import com.amdocs.zusammen.datatypes.Id;
   4 import com.amdocs.zusammen.datatypes.SessionContext;
   5 import com.amdocs.zusammen.datatypes.item.Item;
   6 import com.amdocs.zusammen.plugin.statestore.cassandra.dao.ItemDao;
   7 import com.amdocs.zusammen.plugin.statestore.cassandra.dao.ItemDaoFactory;
   8 import com.google.common.base.Strings;
   9 import org.openecomp.core.tools.Commands.ExportDataCommand;
  10 import org.openecomp.core.utilities.json.JsonUtil;
  11 import org.openecomp.sdc.logging.api.Logger;
  12 import org.openecomp.sdc.logging.api.LoggerFactory;
  13
  14 import java.io.IOException;
  15 import java.net.URISyntaxException;
  16 import java.nio.file.Path;
  17 import java.nio.file.Paths;
  18 import java.util.ArrayList;
  19 import java.util.List;
  20 import java.util.Optional;
  21 import java.util.Set;
  22
  23 import static java.io.File.separator;
  24 import static java.nio.file.Files.*;
  25 import static org.openecomp.core.tools.Commands.exportdata.ImportProperties.JSON_POSTFIX;
  26
  27 public class ItemHandler {
  28     private static final Logger logger = LoggerFactory.getLogger(ExportDataCommand.class);
  29
  30
  31     public ItemHandler() {
  32         ImportProperties.initParams();
  33     }
  34
  35     public void createItemsData(SessionContext context, Set<String> filteredItem) throws URISyntaxException, IOException {
  36         final List<Item> items = new ArrayList<>();
  37         if (filteredItem.isEmpty()) {
  38             items.addAll(getItemDao(context).list(context));
  39         } else {
  40             Optional<Item> item = addItem(context, filteredItem.iterator().next(), items);
  41             item.ifPresent(foundedItem -> {
  42                 Object vendorId =  foundedItem.getInfo().getProperty("vendorId");
  43                 if (vendorId != null) {
  44                     String vendorIdStr = vendorId.toString();
  45                     addItem(context, vendorIdStr, items);
  46                     filteredItem.add(vendorIdStr);
  47                 }
  48             });
  49         }
  50
  51         items.parallelStream().forEach(item -> createItemDirectoryAndFiles(item, filteredItem));
  52
  53     }
  54
  55
  56
  57     private Optional<Item> addItem(SessionContext context, String filteredItem, List<Item> items) {
  58         Optional<Item> item = getItemDao(context).get(context, new Id(filteredItem));
  59         item.ifPresent(itemData -> items.add(itemData));
  60         return item;
  61     }
  62
  63     private final void createItemDirectoryAndFiles(Item item, Set<String> filteredItem) {
  64         try {
  65             String itemId = item.getId().getValue();
  66             if (!filteredItem.isEmpty() && !filteredItem.contains(itemId)) {
  67                 return;
  68             }
  69             Path itemPath = Paths.get(ImportProperties.ROOT_DIRECTORY + separator + itemId);
  70             Path itemFilePath = Paths.get(ImportProperties.ROOT_DIRECTORY + separator +
  71                     itemId + separator + itemId + JSON_POSTFIX);
  72             if (notExists(itemPath)) {
  73                 createDirectories(itemPath);
  74             }
  75             String itemJson = JsonUtil.object2Json(item);
  76             write(itemFilePath, itemJson.getBytes());
  77         } catch (Exception ex) {
  78             logger.error(ex.getMessage(), ex);
  79         }
  80
  81     }
  82
  83
  84     private ItemDao getItemDao(SessionContext context) {
  85         return ItemDaoFactory.getInstance().createInterface(context);
  86     }
  87 }