[SDC] Onboarding 1710 rebase.
[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
1 package org.openecomp.sdc.vendorlicense.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.core.zusammen.api.ZusammenUtil;
13 import org.openecomp.sdc.vendorlicense.dao.LimitDao;
14 import org.openecomp.sdc.vendorlicense.dao.types.AggregationFunction;
15 import org.openecomp.sdc.vendorlicense.dao.types.EntitlementMetric;
16 import org.openecomp.sdc.vendorlicense.dao.types.EntitlementTime;
17 import org.openecomp.sdc.vendorlicense.dao.types.LimitEntity;
18 import org.openecomp.sdc.vendorlicense.dao.types.LimitType;
19 import org.openecomp.sdc.versioning.dao.types.Version;
20
21 import java.util.Collection;
22 import java.util.Collections;
23 import java.util.Optional;
24 import java.util.stream.Collectors;
25
26 public class LimitZusammenDaoImpl implements LimitDao {
27
28   public static final String LIMT_TYPE = "type";
29   public static final String METRIC = "metric";
30   public static final String AGGREGATIONFUNCTION = "aggregationfunction";
31   public static final String TIME = "time";
32   public static final String UNIT = "unit";
33   public static final String VALUE = "value";
34   private ZusammenAdaptor zusammenAdaptor;
35
36   public LimitZusammenDaoImpl(ZusammenAdaptor zusammenAdaptor) {
37     this.zusammenAdaptor = zusammenAdaptor;
38   }
39
40   @Override
41   public void create(LimitEntity limitEntity) {
42     ZusammenElement limitElement = limitToZusammen(limitEntity, Action.CREATE);
43
44     ZusammenElement limitsElement =
45         VlmZusammenUtil.buildStructuralElement(StructureElement.Limits, null);
46     limitsElement.setSubElements(Collections.singletonList(limitElement));
47
48     ZusammenElement epLkgElement =
49         buildZusammenElement(new Id(limitEntity.getEpLkgId()), Action.IGNORE);
50     epLkgElement.setSubElements(Collections.singletonList(limitsElement));
51
52     SessionContext context = ZusammenUtil.createSessionContext();
53     Id itemId = new Id(limitEntity.getVendorLicenseModelId());
54     ElementContext elementContext = new ElementContext(itemId,
55         VlmZusammenUtil.getFirstVersionId(context, itemId, zusammenAdaptor));
56
57     Optional<Element> savedElement =
58         zusammenAdaptor.saveElement(context, elementContext, epLkgElement, "Create limit");
59     savedElement.ifPresent(element ->
60         limitEntity.setId(element.getSubElements().iterator().next()
61             .getSubElements().iterator().next().getElementId().getValue()));
62   }
63
64   @Override
65   public boolean isLimitPresent(LimitEntity limitEntity) {
66     SessionContext context = ZusammenUtil.createSessionContext();
67     Id itemId = new Id(limitEntity.getVendorLicenseModelId());
68     ElementContext elementContext = new ElementContext(itemId,
69         VlmZusammenUtil.getFirstVersionId(context, itemId, zusammenAdaptor),
70         VlmZusammenUtil.getVersionTag(limitEntity.getVersion()));
71
72     Collection<ElementInfo> elementInfos = zusammenAdaptor.listElementsByName(context,
73         elementContext, new Id(limitEntity.getEpLkgId()),StructureElement.Limits.name());
74
75     for (ElementInfo elementInfo : elementInfos) {
76       if (elementInfo.getId().getValue().equals(limitEntity.getId())) {
77         return true;
78       }
79     }
80
81     return false;
82   }
83
84   @Override
85   public Collection<LimitEntity> list(LimitEntity limitEntity) {
86     SessionContext context = ZusammenUtil.createSessionContext();
87     Id itemId = new Id(limitEntity.getVendorLicenseModelId());
88     ElementContext elementContext = new ElementContext(itemId,
89         VlmZusammenUtil.getFirstVersionId(context, itemId, zusammenAdaptor),
90         VlmZusammenUtil.getVersionTag(limitEntity.getVersion()));
91
92     return listLimits(context, elementContext, limitEntity);
93   }
94
95   private Collection<LimitEntity> listLimits(SessionContext context, ElementContext elementContext,
96                                            LimitEntity limitEntity) {
97     return zusammenAdaptor
98         .listElementsByName(context, elementContext, new Id(limitEntity.getEpLkgId()),
99             StructureElement.Limits.name())
100         .stream().map(elementInfo -> mapElementInfoToLimit(
101             limitEntity.getVendorLicenseModelId(), limitEntity.getVersion(),
102             limitEntity.getEpLkgId(), elementInfo))
103         .collect(Collectors.toList());
104   }
105
106   private LimitEntity mapElementInfoToLimit(String vlmId, Version version,
107                                         String epLkgId, ElementInfo elementInfo) {
108     LimitEntity limitEntity =
109         new LimitEntity(vlmId, version, epLkgId, elementInfo.getId().getValue());
110
111     limitEntity.setName(elementInfo.getInfo().getName());
112     limitEntity.setDescription(elementInfo.getInfo().getDescription());
113     limitEntity.setType( elementInfo.getInfo().getProperties().get(LIMT_TYPE) != null ?
114         LimitType.valueOf((String) elementInfo.getInfo().getProperties().get(LIMT_TYPE)) :
115         null);
116     limitEntity.setTime( elementInfo.getInfo().getProperties().get(TIME) != null ?
117         EntitlementTime.valueOf((String) elementInfo.getInfo().getProperties().get(TIME)) :
118         null);
119     limitEntity.setMetric( elementInfo.getInfo().getProperties().get(METRIC) != null ?
120         EntitlementMetric.valueOf((String) elementInfo.getInfo().getProperties().get(METRIC)) :
121         null);
122     limitEntity.setAggregationFunction( elementInfo.getInfo().getProperties().get
123         (AGGREGATIONFUNCTION) != null ?
124         AggregationFunction.valueOf((String) elementInfo.getInfo().getProperties()
125             .get(AGGREGATIONFUNCTION)) : null);
126     Object unit = elementInfo.getInfo().getProperties().get(UNIT);
127     limitEntity.setUnit(unit != null ? VlmZusammenUtil.toInteger(unit) : null);
128     Object value = elementInfo.getInfo().getProperties().get(VALUE);
129     limitEntity.setValue(value != null ? VlmZusammenUtil.toInteger(value) : null);
130
131     return limitEntity;
132   }
133
134   @Override
135   public void update(LimitEntity entity) {
136     ZusammenElement limitElement = limitToZusammen(entity, Action.UPDATE);
137
138     SessionContext context = ZusammenUtil.createSessionContext();
139     Id itemId = new Id(entity.getVendorLicenseModelId());
140     ElementContext elementContext =  new ElementContext(itemId,
141         VlmZusammenUtil.getFirstVersionId(context, itemId, zusammenAdaptor));
142
143     zusammenAdaptor.saveElement(context,elementContext, limitElement,
144         String.format("Update limit with id %s", entity.getId()));
145   }
146
147   @Override
148   public LimitEntity get(LimitEntity limitEntity) {
149     SessionContext context = ZusammenUtil.createSessionContext();
150     Id itemId = new Id(limitEntity.getVendorLicenseModelId());
151     ElementContext elementContext = new ElementContext(itemId,
152         VlmZusammenUtil.getFirstVersionId(context, itemId, zusammenAdaptor),
153         VlmZusammenUtil.getVersionTag(limitEntity.getVersion()));
154
155     return zusammenAdaptor.getElementInfo(context, elementContext, new Id(limitEntity.getId()))
156         .map(elementInfo -> mapElementInfoToLimit(
157             limitEntity.getVendorLicenseModelId(), limitEntity.getVersion(), limitEntity
158                 .getEpLkgId(), elementInfo))
159         .orElse(null);
160   }
161
162   @Override
163   public void delete(LimitEntity entity) {
164     SessionContext context = ZusammenUtil.createSessionContext();
165     ZusammenElement zusammenElement = new ZusammenElement();
166     zusammenElement.setAction(Action.DELETE);
167     zusammenElement.setElementId(new Id(entity.getId()));
168
169     Id itemId = new Id(entity.getVendorLicenseModelId());
170     ElementContext elementContext =
171             new ElementContext(itemId,
172                     VlmZusammenUtil.getFirstVersionId(context, itemId, zusammenAdaptor));
173     zusammenAdaptor.saveElement(context, elementContext, zusammenElement,
174             "delete limit Id:" + entity.getId() + ".");
175   }
176
177   @Override
178   public void registerVersioning(String versionableEntityType) {
179
180   }
181
182   private ZusammenElement limitToZusammen(LimitEntity limit,
183                                                      Action action) {
184     ZusammenElement limitElement = buildLimitElement(limit, action);
185     return limitElement;
186   }
187
188   private ZusammenElement buildLimitElement(LimitEntity limit,
189                                                        Action action) {
190     ZusammenElement limitElement = new ZusammenElement();
191     limitElement.setAction(action);
192     if (limit.getId() != null) {
193       limitElement.setElementId(new Id(limit.getId()));
194     }
195
196     Info info = new Info();
197     info.setName(limit.getName());
198     info.setDescription(limit.getDescription());
199     info.addProperty(LIMT_TYPE, limit.getType());
200     info.addProperty(METRIC, limit.getMetric());
201     info.addProperty(AGGREGATIONFUNCTION, limit.getAggregationFunction());
202     info.addProperty(TIME, limit.getTime());
203     info.addProperty(VALUE, limit.getValue());
204     info.addProperty(UNIT, limit.getUnit());
205     limitElement.setInfo(info);
206     return limitElement;
207   }
208
209   private ZusammenElement buildZusammenElement(Id elementId, Action action) {
210     ZusammenElement element = new ZusammenElement();
211     element.setElementId(elementId);
212     element.setAction(action);
213     return element;
214   }
215 }