openecomp-be/lib/openecomp-sdc-model-lib/openecomp-sdc-model-impl/src/main/java/org/openecomp/sdc/model/impl/zusammen/EnrichedServiceModelDaoZusammenImpl.java


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51

package org.openecomp.sdc.model.impl.zusammen;

import com.amdocs.zusammen.adaptor.inbound.api.types.item.ZusammenElement;
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 org.openecomp.core.model.dao.EnrichedServiceModelDao;
import org.openecomp.core.model.types.ServiceArtifact;
import org.openecomp.core.model.types.ServiceElement;
import org.openecomp.core.utilities.file.FileUtils;
import org.openecomp.core.zusammen.api.ZusammenAdaptor;
import org.openecomp.core.zusammen.api.ZusammenUtil;
import org.openecomp.sdc.logging.api.Logger;
import org.openecomp.sdc.logging.api.LoggerFactory;
import org.openecomp.sdc.tosca.datatypes.ToscaServiceModel;
import org.openecomp.sdc.versioning.dao.types.Version;

public class EnrichedServiceModelDaoZusammenImpl extends ServiceModelDaoZusammenImpl implements
    EnrichedServiceModelDao<ToscaServiceModel, ServiceElement> {

  private static final Logger logger =
      LoggerFactory.getLogger(EnrichedServiceModelDaoZusammenImpl.class);

  public EnrichedServiceModelDaoZusammenImpl(ZusammenAdaptor zusammenAdaptor) {
    super(zusammenAdaptor);
    this.name = StructureElement.EnrichedServiceModel.name();
  }

  @Override
  public void storeExternalArtifact(ServiceArtifact serviceArtifact) {
    ZusammenElement artifactElement = buildArtifactElement(serviceArtifact.getName(),
        FileUtils.toByteArray(serviceArtifact.getContent()), Action.CREATE);

    ZusammenElement artifactsElement =
        buildStructuralElement(StructureElement.Artifacts.name(), null);
    artifactsElement.addSubElement(artifactElement);

    ZusammenElement enrichedServiceModelElement = buildStructuralElement(name, null);
    enrichedServiceModelElement.addSubElement(artifactsElement);

    SessionContext context = ZusammenUtil.createSessionContext();
    Id itemId = new Id(serviceArtifact.getVspId());
    ElementContext elementContext = new ElementContext(itemId, getFirstVersionId(context, itemId));
    zusammenAdaptor
        .saveElement(context, elementContext, enrichedServiceModelElement, "add service artifact.");

    logger.info("Finished adding artifact to service model for vsp id -> " +
        elementContext.getItemId().getValue());
  }
}