8cd6bd1acd908f24c622ae33c505b2aeeb599809
[sdc.git] /
1 package org.openecomp.sdc.vendorsoftwareproduct.dao.impl.zusammen;
2
3 import com.amdocs.zusammen.adaptor.inbound.api.types.item.Element;
4 import com.amdocs.zusammen.adaptor.inbound.api.types.item.ElementInfo;
5 import com.amdocs.zusammen.adaptor.inbound.api.types.item.ZusammenElement;
6 import com.amdocs.zusammen.datatypes.Id;
7 import com.amdocs.zusammen.datatypes.SessionContext;
8 import com.amdocs.zusammen.datatypes.item.Action;
9 import com.amdocs.zusammen.datatypes.item.ElementContext;
10 import com.amdocs.zusammen.datatypes.item.Info;
11 import org.openecomp.core.zusammen.api.ZusammenAdaptor;
12 import org.openecomp.sdc.datatypes.model.ElementType;
13 import org.openecomp.sdc.vendorsoftwareproduct.dao.NetworkDao;
14 import org.openecomp.sdc.vendorsoftwareproduct.dao.impl.zusammen.convertor.ElementToNetworkConvertor;
15 import org.openecomp.sdc.vendorsoftwareproduct.dao.type.NetworkEntity;
16 import org.openecomp.sdc.versioning.dao.types.Version;
17 import org.openecomp.types.ElementPropertyName;
18
19 import java.io.ByteArrayInputStream;
20 import java.util.ArrayList;
21 import java.util.Collection;
22 import java.util.Collections;
23 import java.util.Optional;
24 import java.util.stream.Collectors;
25
26 import static org.openecomp.core.zusammen.api.ZusammenUtil.*;
27
28 public class NetworkDaoZusammenImpl implements NetworkDao {
29
30   private ZusammenAdaptor zusammenAdaptor;
31
32   public NetworkDaoZusammenImpl(ZusammenAdaptor zusammenAdaptor) {
33     this.zusammenAdaptor = zusammenAdaptor;
34   }
35
36   @Override
37   public void registerVersioning(String versionableEntityType) {
38     // registerVersioning not implemented for NetworkDaoZusammenImpl
39   }
40
41   @Override
42   public Collection<NetworkEntity> list(NetworkEntity network) {
43     SessionContext context = createSessionContext();
44     ElementContext elementContext =
45         new ElementContext(network.getVspId(), network.getVersion().getId());
46
47     Optional<ElementInfo> vspModel = zusammenAdaptor
48         .getElementInfoByName(context, elementContext, null, ElementType.VspModel.name());
49     if (!vspModel.isPresent()) {
50       return new ArrayList<>();
51     }
52
53     return zusammenAdaptor.listElementsByName(context, elementContext, vspModel.get().getId(),
54         ElementType.Networks.name()).stream()
55         .map(new ElementToNetworkConvertor()::convert)
56         .map(entity -> {
57           entity.setVspId(network.getVspId());
58           entity.setVersion(network.getVersion());
59           return entity;
60         })
61         .collect(Collectors.toList());
62   }
63
64   @Override
65   public void create(NetworkEntity network) {
66     ZusammenElement networkElement = buildNetworkElement(network, Action.CREATE);
67
68     ZusammenElement networksElement = buildStructuralElement(ElementType.Networks, Action.IGNORE);
69     networksElement.setSubElements(Collections.singletonList(networkElement));
70
71     ZusammenElement vspModel = buildStructuralElement(ElementType.VspModel, Action.IGNORE);
72     vspModel.addSubElement(networksElement);
73
74     SessionContext context = createSessionContext();
75     Element savedElement = zusammenAdaptor
76         .saveElement(context, new ElementContext(network.getVspId(), network.getVersion().getId()),
77             vspModel, "Create network");
78     network.setId(savedElement.getSubElements().iterator().next()
79         .getSubElements().iterator().next().getElementId().getValue());
80   }
81
82   @Override
83   public void update(NetworkEntity network) {
84     ZusammenElement networkElement = buildNetworkElement(network, Action.UPDATE);
85
86     SessionContext context = createSessionContext();
87     zusammenAdaptor
88         .saveElement(context, new ElementContext(network.getVspId(), network.getVersion().getId()),
89             networkElement, String.format("Update network with id %s", network.getId()));
90   }
91
92   @Override
93   public NetworkEntity get(NetworkEntity network) {
94     SessionContext context = createSessionContext();
95
96     Optional<Element> element =
97         zusammenAdaptor.getElement(context,
98             new ElementContext(network.getVspId(), network.getVersion().getId()), network.getId());
99
100     if (element.isPresent()) {
101       ElementToNetworkConvertor convertor = new ElementToNetworkConvertor();
102       NetworkEntity entity = convertor.convert(element.get());
103       entity.setVspId(network.getVspId());
104       entity.setVersion(network.getVersion());
105       return entity;
106     } else {
107       return null;
108     }
109   }
110
111   @Override
112   public void delete(NetworkEntity network) {
113     ZusammenElement networkElement = buildElement(new Id(network.getId()), Action.DELETE);
114
115     SessionContext context = createSessionContext();
116     zusammenAdaptor
117         .saveElement(context, new ElementContext(network.getVspId(), network.getVersion().getId()),
118             networkElement, String.format("Delete network with id %s", network.getId()));
119   }
120
121
122   @Override
123   public void deleteAll(String vspId, Version version) {
124     SessionContext context = createSessionContext();
125     ElementContext elementContext = new ElementContext(vspId, version.getId());
126
127     Collection<ElementInfo> vspModelSubs = zusammenAdaptor
128         .listElementsByName(context, elementContext, null, ElementType.VspModel.name());
129
130     Optional<ElementInfo> networksElement = vspModelSubs.stream()
131         .filter(elementInfo -> elementInfo.getInfo() != null
132             && ElementType.Networks.name().equals(elementInfo.getInfo().getName()))
133         .findFirst();
134     if (!networksElement.isPresent()) {
135       return;
136     }
137
138     ZusammenElement networks = buildElement(networksElement.get().getId(), Action.IGNORE);
139     networks.setSubElements(networksElement.get().getSubElements().stream()
140         .map(network -> buildElement(network.getId(), Action.DELETE))
141         .collect(Collectors.toList()));
142
143     zusammenAdaptor.saveElement(context, elementContext, networks, "Delete all networks");
144   }
145
146   private ZusammenElement buildNetworkElement(NetworkEntity network, Action action) {
147     ZusammenElement networkElement =
148         buildElement(network.getId() == null ? null : new Id(network.getId()), action);
149     Info info = new Info();
150     info.addProperty(ElementPropertyName.elementType.name(), ElementType.Network);
151     info.addProperty(ElementPropertyName.compositionData.name(), network.getCompositionData());
152     networkElement.setInfo(info);
153     networkElement.setData(new ByteArrayInputStream(network.getCompositionData().getBytes()));
154     return networkElement;
155   }
156 }