Add collaboration feature
[sdc.git] / openecomp-be / lib / openecomp-sdc-vendor-license-lib / openecomp-sdc-vendor-license-core / src / main / java / org / openecomp / sdc / vendorlicense / dao / impl / zusammen / LimitZusammenDaoImpl.java
index 78c26c8..7b19236 100644 (file)
@@ -9,26 +9,30 @@ import com.amdocs.zusammen.datatypes.item.Action;
 import com.amdocs.zusammen.datatypes.item.ElementContext;
 import com.amdocs.zusammen.datatypes.item.Info;
 import org.openecomp.core.zusammen.api.ZusammenAdaptor;
-import org.openecomp.core.zusammen.api.ZusammenUtil;
+import org.openecomp.sdc.datatypes.model.ElementType;
 import org.openecomp.sdc.vendorlicense.dao.LimitDao;
 import org.openecomp.sdc.vendorlicense.dao.types.AggregationFunction;
 import org.openecomp.sdc.vendorlicense.dao.types.LimitEntity;
 import org.openecomp.sdc.vendorlicense.dao.types.LimitType;
 import org.openecomp.sdc.versioning.dao.types.Version;
+import org.openecomp.types.ElementPropertyName;
 
 import java.util.Collection;
 import java.util.Collections;
-import java.util.Optional;
 import java.util.stream.Collectors;
 
+import static org.openecomp.core.zusammen.api.ZusammenUtil.buildElement;
+import static org.openecomp.core.zusammen.api.ZusammenUtil.buildStructuralElement;
+import static org.openecomp.core.zusammen.api.ZusammenUtil.createSessionContext;
+
 public class LimitZusammenDaoImpl implements LimitDao {
 
-  public static final String LIMT_TYPE = "type";
-  public static final String METRIC = "metric";
-  public static final String AGGREGATIONFUNCTION = "aggregationfunction";
-  public static final String TIME = "time";
-  public static final String UNIT = "unit";
-  public static final String VALUE = "value";
+  private static final String LIMT_TYPE = "type";
+  private static final String METRIC = "metric";
+  private static final String AGGREGATIONFUNCTION = "aggregationfunction";
+  private static final String TIME = "time";
+  private static final String UNIT = "unit";
+  private static final String VALUE = "value";
   private ZusammenAdaptor zusammenAdaptor;
 
   public LimitZusammenDaoImpl(ZusammenAdaptor zusammenAdaptor) {
@@ -39,36 +43,30 @@ public class LimitZusammenDaoImpl implements LimitDao {
   public void create(LimitEntity limitEntity) {
     ZusammenElement limitElement = limitToZusammen(limitEntity, Action.CREATE);
 
-    ZusammenElement limitsElement =
-        VlmZusammenUtil.buildStructuralElement(StructureElement.Limits, null);
+    ZusammenElement limitsElement = buildStructuralElement(ElementType.Limits, null);
     limitsElement.setSubElements(Collections.singletonList(limitElement));
 
-    ZusammenElement epLkgElement =
-        buildZusammenElement(new Id(limitEntity.getEpLkgId()), Action.IGNORE);
+    ZusammenElement epLkgElement = buildElement(new Id(limitEntity.getEpLkgId()), Action.IGNORE);
     epLkgElement.setSubElements(Collections.singletonList(limitsElement));
 
-    SessionContext context = ZusammenUtil.createSessionContext();
-    Id itemId = new Id(limitEntity.getVendorLicenseModelId());
-    ElementContext elementContext = new ElementContext(itemId,
-        VlmZusammenUtil.getFirstVersionId(context, itemId, zusammenAdaptor));
+    SessionContext context = createSessionContext();
+    ElementContext elementContext =
+        new ElementContext(limitEntity.getVendorLicenseModelId(), limitEntity.getVersion().getId());
 
-    Optional<Element> savedElement =
+    Element savedElement =
         zusammenAdaptor.saveElement(context, elementContext, epLkgElement, "Create limit");
-    savedElement.ifPresent(element ->
-        limitEntity.setId(element.getSubElements().iterator().next()
-            .getSubElements().iterator().next().getElementId().getValue()));
+    limitEntity.setId(savedElement.getSubElements().iterator().next()
+        .getSubElements().iterator().next().getElementId().getValue());
   }
 
   @Override
   public boolean isLimitPresent(LimitEntity limitEntity) {
-    SessionContext context = ZusammenUtil.createSessionContext();
-    Id itemId = new Id(limitEntity.getVendorLicenseModelId());
-    ElementContext elementContext = new ElementContext(itemId,
-        VlmZusammenUtil.getFirstVersionId(context, itemId, zusammenAdaptor),
-        VlmZusammenUtil.getVersionTag(limitEntity.getVersion()));
+    SessionContext context = createSessionContext();
+    ElementContext elementContext =
+        new ElementContext(limitEntity.getVendorLicenseModelId(), limitEntity.getVersion().getId());
 
     Collection<ElementInfo> elementInfos = zusammenAdaptor.listElementsByName(context,
-        elementContext, new Id(limitEntity.getEpLkgId()),StructureElement.Limits.name());
+        elementContext, new Id(limitEntity.getEpLkgId()), ElementType.Limits.name());
 
     for (ElementInfo elementInfo : elementInfos) {
       if (elementInfo.getId().getValue().equals(limitEntity.getId())) {
@@ -81,20 +79,18 @@ public class LimitZusammenDaoImpl implements LimitDao {
 
   @Override
   public Collection<LimitEntity> list(LimitEntity limitEntity) {
-    SessionContext context = ZusammenUtil.createSessionContext();
-    Id itemId = new Id(limitEntity.getVendorLicenseModelId());
-    ElementContext elementContext = new ElementContext(itemId,
-        VlmZusammenUtil.getFirstVersionId(context, itemId, zusammenAdaptor),
-        VlmZusammenUtil.getVersionTag(limitEntity.getVersion()));
+    SessionContext context = createSessionContext();
+    ElementContext elementContext =
+        new ElementContext(limitEntity.getVendorLicenseModelId(), limitEntity.getVersion().getId());
 
     return listLimits(context, elementContext, limitEntity);
   }
 
   private Collection<LimitEntity> listLimits(SessionContext context, ElementContext elementContext,
-                                           LimitEntity limitEntity) {
+                                             LimitEntity limitEntity) {
     return zusammenAdaptor
         .listElementsByName(context, elementContext, new Id(limitEntity.getEpLkgId()),
-            StructureElement.Limits.name())
+            ElementType.Limits.name())
         .stream().map(elementInfo -> mapElementInfoToLimit(
             limitEntity.getVendorLicenseModelId(), limitEntity.getVersion(),
             limitEntity.getEpLkgId(), elementInfo))
@@ -102,18 +98,18 @@ public class LimitZusammenDaoImpl implements LimitDao {
   }
 
   private LimitEntity mapElementInfoToLimit(String vlmId, Version version,
-                                        String epLkgId, ElementInfo elementInfo) {
+                                            String epLkgId, ElementInfo elementInfo) {
     LimitEntity limitEntity =
         new LimitEntity(vlmId, version, epLkgId, elementInfo.getId().getValue());
 
     limitEntity.setName(elementInfo.getInfo().getName());
     limitEntity.setDescription(elementInfo.getInfo().getDescription());
-    limitEntity.setType( elementInfo.getInfo().getProperties().get(LIMT_TYPE) != null ?
+    limitEntity.setType(elementInfo.getInfo().getProperties().get(LIMT_TYPE) != null ?
         LimitType.valueOf((String) elementInfo.getInfo().getProperties().get(LIMT_TYPE)) :
         null);
-    limitEntity.setTime((String) elementInfo.getInfo().getProperties().get(TIME) );
-    limitEntity.setMetric( (String) elementInfo.getInfo().getProperties().get(METRIC));
-    limitEntity.setAggregationFunction( elementInfo.getInfo().getProperties().get
+    limitEntity.setTime((String) elementInfo.getInfo().getProperties().get(TIME));
+    limitEntity.setMetric((String) elementInfo.getInfo().getProperties().get(METRIC));
+    limitEntity.setAggregationFunction(elementInfo.getInfo().getProperties().get
         (AGGREGATIONFUNCTION) != null ?
         AggregationFunction.valueOf((String) elementInfo.getInfo().getProperties()
             .get(AGGREGATIONFUNCTION)) : null);
@@ -126,25 +122,22 @@ public class LimitZusammenDaoImpl implements LimitDao {
   }
 
   @Override
-  public void update(LimitEntity entity) {
-    ZusammenElement limitElement = limitToZusammen(entity, Action.UPDATE);
+  public void update(LimitEntity limitEntity) {
+    ZusammenElement limitElement = limitToZusammen(limitEntity, Action.UPDATE);
 
-    SessionContext context = ZusammenUtil.createSessionContext();
-    Id itemId = new Id(entity.getVendorLicenseModelId());
-    ElementContext elementContext =  new ElementContext(itemId,
-        VlmZusammenUtil.getFirstVersionId(context, itemId, zusammenAdaptor));
+    SessionContext context = createSessionContext();
+    ElementContext elementContext =
+        new ElementContext(limitEntity.getVendorLicenseModelId(), limitEntity.getVersion().getId());
 
-    zusammenAdaptor.saveElement(context,elementContext, limitElement,
-        String.format("Update limit with id %s", entity.getId()));
+    zusammenAdaptor.saveElement(context, elementContext, limitElement,
+        String.format("Update limit with id %s", limitEntity.getId()));
   }
 
   @Override
   public LimitEntity get(LimitEntity limitEntity) {
-    SessionContext context = ZusammenUtil.createSessionContext();
-    Id itemId = new Id(limitEntity.getVendorLicenseModelId());
-    ElementContext elementContext = new ElementContext(itemId,
-        VlmZusammenUtil.getFirstVersionId(context, itemId, zusammenAdaptor),
-        VlmZusammenUtil.getVersionTag(limitEntity.getVersion()));
+    SessionContext context = createSessionContext();
+    ElementContext elementContext =
+        new ElementContext(limitEntity.getVendorLicenseModelId(), limitEntity.getVersion().getId());
 
     return zusammenAdaptor.getElementInfo(context, elementContext, new Id(limitEntity.getId()))
         .map(elementInfo -> mapElementInfoToLimit(
@@ -154,18 +147,14 @@ public class LimitZusammenDaoImpl implements LimitDao {
   }
 
   @Override
-  public void delete(LimitEntity entity) {
-    SessionContext context = ZusammenUtil.createSessionContext();
-    ZusammenElement zusammenElement = new ZusammenElement();
-    zusammenElement.setAction(Action.DELETE);
-    zusammenElement.setElementId(new Id(entity.getId()));
+  public void delete(LimitEntity limitEntity) {
+    ZusammenElement zusammenElement = buildElement(new Id(limitEntity.getId()), Action.DELETE);
 
-    Id itemId = new Id(entity.getVendorLicenseModelId());
+    SessionContext context = createSessionContext();
     ElementContext elementContext =
-            new ElementContext(itemId,
-                    VlmZusammenUtil.getFirstVersionId(context, itemId, zusammenAdaptor));
+        new ElementContext(limitEntity.getVendorLicenseModelId(), limitEntity.getVersion().getId());
     zusammenAdaptor.saveElement(context, elementContext, zusammenElement,
-            "delete limit Id:" + entity.getId() + ".");
+        "delete limit Id:" + limitEntity.getId() + ".");
   }
 
   @Override
@@ -173,23 +162,13 @@ public class LimitZusammenDaoImpl implements LimitDao {
 
   }
 
-  private ZusammenElement limitToZusammen(LimitEntity limit,
-                                                     Action action) {
-    ZusammenElement limitElement = buildLimitElement(limit, action);
-    return limitElement;
-  }
-
-  private ZusammenElement buildLimitElement(LimitEntity limit,
-                                                       Action action) {
-    ZusammenElement limitElement = new ZusammenElement();
-    limitElement.setAction(action);
-    if (limit.getId() != null) {
-      limitElement.setElementId(new Id(limit.getId()));
-    }
-
+  private ZusammenElement limitToZusammen(LimitEntity limit, Action action) {
+    ZusammenElement limitElement =
+        buildElement(limit.getId() == null ? null : new Id(limit.getId()), action);
     Info info = new Info();
     info.setName(limit.getName());
     info.setDescription(limit.getDescription());
+    info.addProperty(ElementPropertyName.elementType.name(), ElementType.Limit);
     info.addProperty(LIMT_TYPE, limit.getType());
     info.addProperty(METRIC, limit.getMetric());
     info.addProperty(AGGREGATIONFUNCTION, limit.getAggregationFunction());
@@ -199,11 +178,4 @@ public class LimitZusammenDaoImpl implements LimitDao {
     limitElement.setInfo(info);
     return limitElement;
   }
-
-  private ZusammenElement buildZusammenElement(Id elementId, Action action) {
-    ZusammenElement element = new ZusammenElement();
-    element.setElementId(elementId);
-    element.setAction(action);
-    return element;
-  }
 }