openecomp-be/tools/zusammen-tools/src/main/java/org/openecomp/core/tools/loaders/zusammen/VspGeneralLoader.java

   1 package org.openecomp.core.tools.loaders.zusammen;
   2
   3 import com.amdocs.zusammen.datatypes.Id;
   4 import com.amdocs.zusammen.datatypes.SessionContext;
   5 import com.amdocs.zusammen.plugin.statestore.cassandra.dao.types.ElementEntityContext;
   6 import org.openecomp.core.zusammen.plugin.dao.impl.CassandraElementRepository;
   7 import org.openecomp.core.zusammen.plugin.dao.types.ElementEntity;
   8 import org.openecomp.sdc.vendorsoftwareproduct.dao.impl.zusammen.StructureElement;
   9
  10 import java.util.HashMap;
  11 import java.util.List;
  12 import java.util.Map;
  13 import java.util.Optional;
  14
  15 public class VspGeneralLoader {
  16   public static Map<String, ElementEntity> load(SessionContext context,
  17                                                 Map<String, List<String>> vspItemVersionsMap) {
  18     Map<String, ElementEntity> elementEntityMap = new HashMap<>();
  19     System.setProperty("cassandra.dox.keystore", "zusammen_dox");
  20     CassandraElementRepository cassandraElementRepository = new CassandraElementRepository();
  21     for (Map.Entry<String, List<String>> entry : vspItemVersionsMap.entrySet()) {
  22
  23       for (String version : entry.getValue()) {
  24
  25         Optional<ElementEntity> result =
  26             cassandraElementRepository.get(context, new ElementEntityContext(
  27                     context.getUser().getUserName(),
  28                     new Id(entry.getKey()),
  29                     new Id(version)),
  30                 new ElementEntity(new Id(StructureElement.General.name())));
  31         if (result.isPresent()) {
  32           elementEntityMap.put(context.getUser().getUserName() + "_" + entry.getKey()
  33               + "_" + version, result.get());
  34         }
  35       }
  36     }
  37
  38     return elementEntityMap;
  39   }
  40 }