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