1 files changed, 158 insertions, 0 deletions
diff --git a/openecomp-be/lib/openecomp-sdc-vendor-software-product-lib/openecomp-sdc-vendor-software-product-core/src/main/java/org/openecomp/sdc/vendorsoftwareproduct/dao/impl/VspMergeDaoImpl.java b/openecomp-be/lib/openecomp-sdc-vendor-software-product-lib/openecomp-sdc-vendor-software-product-core/src/main/java/org/openecomp/sdc/vendorsoftwareproduct/dao/impl/VspMergeDaoImpl.java
new file mode 100644
index 0000000000..2ccc47a651
--- /dev/null
+++ b/openecomp-be/lib/openecomp-sdc-vendor-software-product-lib/openecomp-sdc-vendor-software-product-core/src/main/java/org/openecomp/sdc/vendorsoftwareproduct/dao/impl/VspMergeDaoImpl.java
@@ -0,0 +1,158 @@
+package org.openecomp.sdc.vendorsoftwareproduct.dao.impl;
+
+import com.amdocs.zusammen.adaptor.inbound.api.types.item.ElementInfo;
+import com.amdocs.zusammen.datatypes.Id;
+import com.amdocs.zusammen.datatypes.SessionContext;
+import com.amdocs.zusammen.datatypes.item.Action;
+import com.amdocs.zusammen.datatypes.item.ElementContext;
+import com.amdocs.zusammen.datatypes.item.Resolution;
+import com.datastax.driver.core.ResultSet;
+import com.datastax.driver.core.Row;
+import com.datastax.driver.mapping.annotations.Accessor;
+import com.datastax.driver.mapping.annotations.Query;
+import org.openecomp.core.nosqldb.factory.NoSqlDbFactory;
+import org.openecomp.core.zusammen.api.ZusammenAdaptor;
+import org.openecomp.sdc.common.session.SessionContextProviderFactory;
+import org.openecomp.sdc.datatypes.model.ElementType;
+import org.openecomp.sdc.vendorsoftwareproduct.dao.VspMergeDao;
+import org.openecomp.sdc.versioning.dao.types.Version;
+
+import java.util.List;
+import java.util.stream.Collectors;
+
+import static org.openecomp.core.zusammen.api.ZusammenUtil.buildElement;
+import static org.openecomp.core.zusammen.api.ZusammenUtil.createSessionContext;
+
+public class VspMergeDaoImpl implements VspMergeDao {
+  private static final String VSP_MODEL_NOT_EXIST =
+      "Vsp model does not exist for Vsp %s, version %s.";
+
+  private static VspMergeHintAccessor accessor =
+      NoSqlDbFactory.getInstance().createInterface()
+          .getMappingManager().createAccessor(VspMergeHintAccessor.class);
+
+  private ZusammenAdaptor zusammenAdaptor;
+
+  public VspMergeDaoImpl(ZusammenAdaptor zusammenAdaptor) {
+    this.zusammenAdaptor = zusammenAdaptor;
+  }
+
+  @Override
+  public boolean isVspModelConflicted(String vspId, Version version) {
+    SessionContext context = createSessionContext();
+    ElementContext elementContext = new ElementContext(new Id(vspId), new Id(version.getId()));
+
+    return listVspModels(context, elementContext).size() > 1;
+  }
+
+  @Override
+  public void updateVspModelId(String vspId, Version version) {
+    SessionContext context = createSessionContext();
+    ElementContext elementContext = new ElementContext(new Id(vspId), new Id(version.getId()));
+
+    String vspModelId = zusammenAdaptor
+        .getElementInfoByName(context, elementContext, null, ElementType.VspModel.name())
+        .orElseThrow(() -> new IllegalStateException(
+            String.format(VSP_MODEL_NOT_EXIST, vspId, version.getId())))
+        .getId().getValue();
+
+    updateVspModelId(vspId, version, vspModelId);
+  }
+
+  @Override
+  public void updateVspModelConflictResolution(String vspId, Version version,
+                                               Resolution resolution) {
+    accessor.updateModelResolution(resolution, getUser(), vspId, version.getId());
+  }
+
+  @Override
+  public void applyVspModelConflictResolution(String vspId, Version version) {
+    //called only when no conflicts
+
+    SessionContext context = createSessionContext();
+    ElementContext elementContext = new ElementContext(new Id(vspId), new Id(version.getId()));
+
+    List<ElementInfo> vspModels = listVspModels(context, elementContext);
+    if (vspModels.size() == 1) {
+      updateVspModelId(vspId, version, vspModels.iterator().next().getId().getValue());
+      return;
+    }
+
+    if (vspModels.size() != 2) {
+      return;
+    }
+
+    String user = getUser();
+    Row row = accessor.getModelMergeHint(user, vspId, version.getId()).one();
+    if (row == null) {
+      throw new IllegalStateException(
+          "Vsp model id must exists if its conflict is being resolved");
+    }
+    String resolutionValue = row.getString("model_resolution");
+    if (resolutionValue == null) {
+      return; // model conflict is not resolved yet
+    }
+
+    Resolution resolution = Resolution.valueOf(resolutionValue);
+    String localModelId = row.getString("model_id");
+
+    String chosenModelId =
+        keepOnlyChosenVspModel(context, elementContext, vspModels, resolution, localModelId);
+
+    accessor.updateModelMergeHint(chosenModelId, null, user, vspId, version.getId());
+  }
+
+  private String keepOnlyChosenVspModel(SessionContext context, ElementContext elementContext,
+                                        List<ElementInfo> vspModels, Resolution resolution,
+                                        String localModelId) {
+    String newLocalModelId = null;
+    for (ElementInfo vspModel : vspModels) {
+      if (isRedundantModel(vspModel.getId().getValue(), localModelId, resolution)) {
+        zusammenAdaptor
+            .saveElement(context, elementContext, buildElement(vspModel.getId(), Action.DELETE),
+                "Delete Redundant Vsp Model");
+      } else {
+        newLocalModelId = vspModel.getId().getValue();
+      }
+    }
+    return newLocalModelId;
+  }
+
+  private boolean isRedundantModel(String modelId, String localModelId, Resolution resolution) {
+    return resolution == Resolution.THEIRS && modelId.equals(localModelId) ||
+        resolution == Resolution.YOURS && !modelId.equals(localModelId);
+  }
+
+  private List<ElementInfo> listVspModels(SessionContext context, ElementContext elementContext) {
+    return zusammenAdaptor.listElements(context, elementContext, null).stream()
+        .filter(elementInfo -> ElementType.VspModel.name().equals(elementInfo.getInfo().getName()))
+        .collect(Collectors.toList());
+  }
+
+  private void updateVspModelId(String vspId, Version version, String vspModelId) {
+    accessor.updateModelMergeHint(vspModelId, null, getUser(), vspId, version.getId());
+  }
+
+  private String getUser() {
+    return SessionContextProviderFactory.getInstance().createInterface()
+        .get().getUser().getUserId();
+  }
+
+  @Accessor
+  interface VspMergeHintAccessor {
+
+    @Query("UPDATE vsp_merge_hint SET model_id=?, model_resolution=? " +
+        "WHERE space=? AND item_id=? AND version_id=?")
+    void updateModelMergeHint(String vspModelId, Resolution modelResolution, String space,
+                              String itemId, String versionId);
+
+    @Query(
+        "UPDATE vsp_merge_hint SET model_resolution=? WHERE space=? AND item_id=? AND version_id=?")
+    void updateModelResolution(Resolution modelResolution, String space, String itemId,
+                               String versionId);
+
+    @Query("SELECT model_id, model_resolution FROM vsp_merge_hint " +
+        "WHERE space=? AND item_id=? AND version_id=?")
+    ResultSet getModelMergeHint(String space, String itemId, String versionId);
+  }
+}