Fetch CM handles by collection of xpaths
[cps.git] / cps-ri / src / main / java / org / onap / cps / spi / impl / CpsDataPersistenceServiceImpl.java
1 /*
2  *  ============LICENSE_START=======================================================
3  *  Copyright (C) 2021-2022 Nordix Foundation
4  *  Modifications Copyright (C) 2021 Pantheon.tech
5  *  Modifications Copyright (C) 2020-2022 Bell Canada.
6  *  Modifications Copyright (C) 2022 TechMahindra Ltd.
7  *  ================================================================================
8  *  Licensed under the Apache License, Version 2.0 (the "License");
9  *  you may not use this file except in compliance with the License.
10  *  You may obtain a copy of the License at
11  *
12  *        http://www.apache.org/licenses/LICENSE-2.0
13  *
14  *  Unless required by applicable law or agreed to in writing, software
15  *  distributed under the License is distributed on an "AS IS" BASIS,
16  *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
17  *  See the License for the specific language governing permissions and
18  *  limitations under the License.
19  *
20  *  SPDX-License-Identifier: Apache-2.0
21  *  ============LICENSE_END=========================================================
22  */
23
24 package org.onap.cps.spi.impl;
25
26 import com.google.common.collect.ImmutableSet;
27 import com.google.common.collect.ImmutableSet.Builder;
28 import java.io.Serializable;
29 import java.util.ArrayList;
30 import java.util.Collection;
31 import java.util.Collections;
32 import java.util.HashMap;
33 import java.util.HashSet;
34 import java.util.List;
35 import java.util.Map;
36 import java.util.Set;
37 import java.util.regex.Matcher;
38 import java.util.regex.Pattern;
39 import java.util.stream.Collectors;
40 import javax.transaction.Transactional;
41 import lombok.RequiredArgsConstructor;
42 import lombok.extern.slf4j.Slf4j;
43 import org.hibernate.StaleStateException;
44 import org.onap.cps.cpspath.parser.CpsPathQuery;
45 import org.onap.cps.cpspath.parser.CpsPathUtil;
46 import org.onap.cps.cpspath.parser.PathParsingException;
47 import org.onap.cps.spi.CpsDataPersistenceService;
48 import org.onap.cps.spi.FetchDescendantsOption;
49 import org.onap.cps.spi.entities.AnchorEntity;
50 import org.onap.cps.spi.entities.DataspaceEntity;
51 import org.onap.cps.spi.entities.FragmentEntity;
52 import org.onap.cps.spi.entities.FragmentEntityArranger;
53 import org.onap.cps.spi.entities.FragmentExtract;
54 import org.onap.cps.spi.exceptions.AlreadyDefinedException;
55 import org.onap.cps.spi.exceptions.AlreadyDefinedExceptionBatch;
56 import org.onap.cps.spi.exceptions.ConcurrencyException;
57 import org.onap.cps.spi.exceptions.CpsAdminException;
58 import org.onap.cps.spi.exceptions.CpsPathException;
59 import org.onap.cps.spi.exceptions.DataNodeNotFoundException;
60 import org.onap.cps.spi.model.DataNode;
61 import org.onap.cps.spi.model.DataNodeBuilder;
62 import org.onap.cps.spi.repository.AnchorRepository;
63 import org.onap.cps.spi.repository.DataspaceRepository;
64 import org.onap.cps.spi.repository.FragmentQueryBuilder;
65 import org.onap.cps.spi.repository.FragmentRepository;
66 import org.onap.cps.spi.utils.SessionManager;
67 import org.onap.cps.utils.JsonObjectMapper;
68 import org.springframework.dao.DataIntegrityViolationException;
69 import org.springframework.stereotype.Service;
70
71 @Service
72 @Slf4j
73 @RequiredArgsConstructor
74 public class CpsDataPersistenceServiceImpl implements CpsDataPersistenceService {
75
76     private final DataspaceRepository dataspaceRepository;
77     private final AnchorRepository anchorRepository;
78     private final FragmentRepository fragmentRepository;
79     private final JsonObjectMapper jsonObjectMapper;
80     private final SessionManager sessionManager;
81
82     private static final String REG_EX_FOR_OPTIONAL_LIST_INDEX = "(\\[@[\\s\\S]+?]){0,1})";
83
84     @Override
85     public void addChildDataNode(final String dataspaceName, final String anchorName, final String parentNodeXpath,
86                                  final DataNode newChildDataNode) {
87         addNewChildDataNode(dataspaceName, anchorName, parentNodeXpath, newChildDataNode);
88     }
89
90     @Override
91     public void addChildDataNodes(final String dataspaceName, final String anchorName,
92                                   final String parentNodeXpath, final Collection<DataNode> dataNodes) {
93         addChildrenDataNodes(dataspaceName, anchorName, parentNodeXpath, dataNodes);
94     }
95
96     @Override
97     public void addListElements(final String dataspaceName, final String anchorName, final String parentNodeXpath,
98                                 final Collection<DataNode> newListElements) {
99         addChildrenDataNodes(dataspaceName, anchorName, parentNodeXpath, newListElements);
100     }
101
102     @Override
103     public void addMultipleLists(final String dataspaceName, final String anchorName, final String parentNodeXpath,
104                                  final Collection<Collection<DataNode>> newLists) {
105         final Collection<String> failedXpaths = new HashSet<>();
106         newLists.forEach(newList -> {
107             try {
108                 addChildrenDataNodes(dataspaceName, anchorName, parentNodeXpath, newList);
109             } catch (final AlreadyDefinedExceptionBatch e) {
110                 failedXpaths.addAll(e.getAlreadyDefinedXpaths());
111             }
112         });
113
114         if (!failedXpaths.isEmpty()) {
115             throw new AlreadyDefinedExceptionBatch(failedXpaths);
116         }
117
118     }
119
120     private void addNewChildDataNode(final String dataspaceName, final String anchorName,
121                                      final String parentNodeXpath, final DataNode newChild) {
122         final FragmentEntity parentFragmentEntity =
123             getFragmentWithoutDescendantsByXpath(dataspaceName, anchorName, parentNodeXpath);
124         final FragmentEntity newChildAsFragmentEntity =
125                 convertToFragmentWithAllDescendants(parentFragmentEntity.getDataspace(),
126                         parentFragmentEntity.getAnchor(), newChild);
127         newChildAsFragmentEntity.setParentId(parentFragmentEntity.getId());
128         try {
129             fragmentRepository.save(newChildAsFragmentEntity);
130         } catch (final DataIntegrityViolationException e) {
131             throw AlreadyDefinedException.forDataNode(newChild.getXpath(), anchorName, e);
132         }
133
134     }
135
136     private void addChildrenDataNodes(final String dataspaceName, final String anchorName, final String parentNodeXpath,
137                                       final Collection<DataNode> newChildren) {
138         final FragmentEntity parentFragmentEntity =
139             getFragmentWithoutDescendantsByXpath(dataspaceName, anchorName, parentNodeXpath);
140         final List<FragmentEntity> fragmentEntities = new ArrayList<>(newChildren.size());
141         try {
142             newChildren.forEach(newChildAsDataNode -> {
143                 final FragmentEntity newChildAsFragmentEntity =
144                         convertToFragmentWithAllDescendants(parentFragmentEntity.getDataspace(),
145                                 parentFragmentEntity.getAnchor(), newChildAsDataNode);
146                 newChildAsFragmentEntity.setParentId(parentFragmentEntity.getId());
147                 fragmentEntities.add(newChildAsFragmentEntity);
148             });
149             fragmentRepository.saveAll(fragmentEntities);
150         } catch (final DataIntegrityViolationException e) {
151             log.warn("Exception occurred : {} , While saving : {} children, retrying using individual save operations",
152                     e, fragmentEntities.size());
153             retrySavingEachChildIndividually(dataspaceName, anchorName, parentNodeXpath, newChildren);
154         }
155     }
156
157     private void retrySavingEachChildIndividually(final String dataspaceName, final String anchorName,
158                                                   final String parentNodeXpath,
159                                                   final Collection<DataNode> newChildren) {
160         final Collection<String> failedXpaths = new HashSet<>();
161         for (final DataNode newChild : newChildren) {
162             try {
163                 addNewChildDataNode(dataspaceName, anchorName, parentNodeXpath, newChild);
164             } catch (final AlreadyDefinedException e) {
165                 failedXpaths.add(newChild.getXpath());
166             }
167         }
168         if (!failedXpaths.isEmpty()) {
169             throw new AlreadyDefinedExceptionBatch(failedXpaths);
170         }
171     }
172
173     @Override
174     public void storeDataNode(final String dataspaceName, final String anchorName, final DataNode dataNode) {
175         storeDataNodes(dataspaceName, anchorName, Collections.singletonList(dataNode));
176     }
177
178     @Override
179     public void storeDataNodes(final String dataspaceName, final String anchorName,
180                                final Collection<DataNode> dataNodes) {
181         final DataspaceEntity dataspaceEntity = dataspaceRepository.getByName(dataspaceName);
182         final AnchorEntity anchorEntity = anchorRepository.getByDataspaceAndName(dataspaceEntity, anchorName);
183         final List<FragmentEntity> fragmentEntities = new ArrayList<>(dataNodes.size());
184         try {
185             for (final DataNode dataNode: dataNodes) {
186                 final FragmentEntity fragmentEntity = convertToFragmentWithAllDescendants(dataspaceEntity, anchorEntity,
187                         dataNode);
188                 fragmentEntities.add(fragmentEntity);
189             }
190             fragmentRepository.saveAll(fragmentEntities);
191         } catch (final DataIntegrityViolationException exception) {
192             log.warn("Exception occurred : {} , While saving : {} data nodes, Retrying saving data nodes individually",
193                     exception, dataNodes.size());
194             storeDataNodesIndividually(dataspaceName, anchorName, dataNodes);
195         }
196     }
197
198     private void storeDataNodesIndividually(final String dataspaceName, final String anchorName,
199                                            final Collection<DataNode> dataNodes) {
200         final DataspaceEntity dataspaceEntity = dataspaceRepository.getByName(dataspaceName);
201         final AnchorEntity anchorEntity = anchorRepository.getByDataspaceAndName(dataspaceEntity, anchorName);
202         final Collection<String> failedXpaths = new HashSet<>();
203         for (final DataNode dataNode: dataNodes) {
204             try {
205                 final FragmentEntity fragmentEntity = convertToFragmentWithAllDescendants(dataspaceEntity, anchorEntity,
206                         dataNode);
207                 fragmentRepository.save(fragmentEntity);
208             } catch (final DataIntegrityViolationException e) {
209                 failedXpaths.add(dataNode.getXpath());
210             }
211         }
212         if (!failedXpaths.isEmpty()) {
213             throw new AlreadyDefinedExceptionBatch(failedXpaths);
214         }
215     }
216
217     /**
218      * Convert DataNode object into Fragment and places the result in the fragments placeholder. Performs same action
219      * for all DataNode children recursively.
220      *
221      * @param dataspaceEntity       dataspace
222      * @param anchorEntity          anchorEntity
223      * @param dataNodeToBeConverted dataNode
224      * @return a Fragment built from current DataNode
225      */
226     private FragmentEntity convertToFragmentWithAllDescendants(final DataspaceEntity dataspaceEntity,
227                                                                final AnchorEntity anchorEntity,
228                                                                final DataNode dataNodeToBeConverted) {
229         final FragmentEntity parentFragment = toFragmentEntity(dataspaceEntity, anchorEntity, dataNodeToBeConverted);
230         final Builder<FragmentEntity> childFragmentsImmutableSetBuilder = ImmutableSet.builder();
231         for (final DataNode childDataNode : dataNodeToBeConverted.getChildDataNodes()) {
232             final FragmentEntity childFragment =
233                     convertToFragmentWithAllDescendants(parentFragment.getDataspace(), parentFragment.getAnchor(),
234                             childDataNode);
235             childFragmentsImmutableSetBuilder.add(childFragment);
236         }
237         parentFragment.setChildFragments(childFragmentsImmutableSetBuilder.build());
238         return parentFragment;
239     }
240
241     private FragmentEntity toFragmentEntity(final DataspaceEntity dataspaceEntity,
242                                             final AnchorEntity anchorEntity, final DataNode dataNode) {
243         return FragmentEntity.builder()
244                 .dataspace(dataspaceEntity)
245                 .anchor(anchorEntity)
246                 .xpath(dataNode.getXpath())
247                 .attributes(jsonObjectMapper.asJsonString(dataNode.getLeaves()))
248                 .build();
249     }
250
251     @Override
252     public DataNode getDataNode(final String dataspaceName, final String anchorName, final String xpath,
253                                 final FetchDescendantsOption fetchDescendantsOption) {
254         final FragmentEntity fragmentEntity = getFragmentByXpath(dataspaceName, anchorName, xpath,
255             fetchDescendantsOption);
256         return toDataNode(fragmentEntity, fetchDescendantsOption);
257     }
258
259     @Override
260     public Collection<DataNode> getDataNodes(final String dataspaceName, final String anchorName,
261                                              final Collection<String> xpaths,
262                                              final FetchDescendantsOption fetchDescendantsOption) {
263         final DataspaceEntity dataspaceEntity = dataspaceRepository.getByName(dataspaceName);
264         final AnchorEntity anchorEntity = anchorRepository.getByDataspaceAndName(dataspaceEntity, anchorName);
265         final List<FragmentEntity> fragmentEntities =
266                 fragmentRepository.findByAnchorAndMultipleCpsPaths(anchorEntity.getId(), xpaths);
267         final Collection<DataNode> dataNodesCollection = new ArrayList<>(fragmentEntities.size());
268         for (final FragmentEntity fragmentEntity : fragmentEntities) {
269             dataNodesCollection.add(toDataNode(fragmentEntity, fetchDescendantsOption));
270         }
271         return dataNodesCollection;
272     }
273
274     private FragmentEntity getFragmentWithoutDescendantsByXpath(final String dataspaceName,
275                                                                 final String anchorName,
276                                                                 final String xpath) {
277         return getFragmentByXpath(dataspaceName, anchorName, xpath, FetchDescendantsOption.OMIT_DESCENDANTS);
278     }
279
280     private FragmentEntity getFragmentByXpath(final String dataspaceName, final String anchorName,
281                                               final String xpath, final FetchDescendantsOption fetchDescendantsOption) {
282         final DataspaceEntity dataspaceEntity = dataspaceRepository.getByName(dataspaceName);
283         final AnchorEntity anchorEntity = anchorRepository.getByDataspaceAndName(dataspaceEntity, anchorName);
284         final FragmentEntity fragmentEntity;
285         if (isRootXpath(xpath)) {
286             final List<FragmentExtract> fragmentExtracts = fragmentRepository.getTopLevelFragments(dataspaceEntity,
287                     anchorEntity);
288             fragmentEntity = FragmentEntityArranger.toFragmentEntityTrees(anchorEntity, fragmentExtracts)
289                 .stream().findFirst().orElse(null);
290         } else {
291             final String normalizedXpath = getNormalizedXpath(xpath);
292             if (FetchDescendantsOption.OMIT_DESCENDANTS.equals(fetchDescendantsOption)) {
293                 fragmentEntity =
294                     fragmentRepository.getByDataspaceAndAnchorAndXpath(dataspaceEntity, anchorEntity, normalizedXpath);
295             } else {
296                 fragmentEntity = buildFragmentEntitiesFromFragmentExtracts(anchorEntity, normalizedXpath)
297                     .stream().findFirst().orElse(null);
298             }
299         }
300         if (fragmentEntity == null) {
301             throw new DataNodeNotFoundException(dataspaceEntity.getName(), anchorEntity.getName(), xpath);
302         }
303         return fragmentEntity;
304
305     }
306
307     private Collection<FragmentEntity> buildFragmentEntitiesFromFragmentExtracts(final AnchorEntity anchorEntity,
308                                                                                  final String normalizedXpath) {
309         final List<FragmentExtract> fragmentExtracts =
310                 fragmentRepository.findByAnchorIdAndParentXpath(anchorEntity.getId(), normalizedXpath);
311         log.debug("Fetched {} fragment entities by anchor {} and cps path {}.",
312                 fragmentExtracts.size(), anchorEntity.getName(), normalizedXpath);
313         return FragmentEntityArranger.toFragmentEntityTrees(anchorEntity, fragmentExtracts);
314
315     }
316
317     @Override
318     public List<DataNode> queryDataNodes(final String dataspaceName, final String anchorName, final String cpsPath,
319                                          final FetchDescendantsOption fetchDescendantsOption) {
320         final DataspaceEntity dataspaceEntity = dataspaceRepository.getByName(dataspaceName);
321         final AnchorEntity anchorEntity = anchorRepository.getByDataspaceAndName(dataspaceEntity, anchorName);
322         final CpsPathQuery cpsPathQuery;
323         try {
324             cpsPathQuery = CpsPathUtil.getCpsPathQuery(cpsPath);
325         } catch (final PathParsingException e) {
326             throw new CpsPathException(e.getMessage());
327         }
328
329         Collection<FragmentEntity> fragmentEntities;
330         if (canUseRegexQuickFind(fetchDescendantsOption, cpsPathQuery)) {
331             return getDataNodesUsingRegexQuickFind(fetchDescendantsOption, anchorEntity, cpsPathQuery);
332         }
333         fragmentEntities = fragmentRepository.findByAnchorAndCpsPath(anchorEntity.getId(), cpsPathQuery);
334         if (cpsPathQuery.hasAncestorAxis()) {
335             fragmentEntities = getAncestorFragmentEntities(anchorEntity.getId(), cpsPathQuery, fragmentEntities);
336         }
337         return createDataNodesFromProxiedFragmentEntities(fetchDescendantsOption, anchorEntity, fragmentEntities);
338     }
339
340     private static boolean canUseRegexQuickFind(final FetchDescendantsOption fetchDescendantsOption,
341                                                 final CpsPathQuery cpsPathQuery) {
342         return fetchDescendantsOption.equals(FetchDescendantsOption.INCLUDE_ALL_DESCENDANTS)
343             && !cpsPathQuery.hasLeafConditions()
344             && !cpsPathQuery.hasTextFunctionCondition();
345     }
346
347     private List<DataNode> getDataNodesUsingRegexQuickFind(final FetchDescendantsOption fetchDescendantsOption,
348                                                            final AnchorEntity anchorEntity,
349                                                            final CpsPathQuery cpsPathQuery) {
350         Collection<FragmentEntity> fragmentEntities;
351         final String xpathRegex = FragmentQueryBuilder.getXpathSqlRegex(cpsPathQuery, true);
352         final List<FragmentExtract> fragmentExtracts =
353             fragmentRepository.quickFindWithDescendants(anchorEntity.getId(), xpathRegex);
354         fragmentEntities = FragmentEntityArranger.toFragmentEntityTrees(anchorEntity, fragmentExtracts);
355         if (cpsPathQuery.hasAncestorAxis()) {
356             fragmentEntities = getAncestorFragmentEntities(anchorEntity.getId(), cpsPathQuery, fragmentEntities);
357         }
358         return createDataNodesFromFragmentEntities(fetchDescendantsOption, fragmentEntities);
359     }
360
361     private Collection<FragmentEntity> getAncestorFragmentEntities(final int anchorId,
362                                                                    final CpsPathQuery cpsPathQuery,
363                                                                    final Collection<FragmentEntity> fragmentEntities) {
364         final Collection<String> ancestorXpaths = processAncestorXpath(fragmentEntities, cpsPathQuery);
365         return ancestorXpaths.isEmpty() ? Collections.emptyList()
366             : fragmentRepository.findByAnchorAndMultipleCpsPaths(anchorId, ancestorXpaths);
367     }
368
369     private List<DataNode> createDataNodesFromProxiedFragmentEntities(
370                                             final FetchDescendantsOption fetchDescendantsOption,
371                                             final AnchorEntity anchorEntity,
372                                             final Collection<FragmentEntity> proxiedFragmentEntities) {
373         final List<DataNode> dataNodes = new ArrayList<>(proxiedFragmentEntities.size());
374         for (final FragmentEntity proxiedFragmentEntity : proxiedFragmentEntities) {
375             if (FetchDescendantsOption.OMIT_DESCENDANTS.equals(fetchDescendantsOption)) {
376                 dataNodes.add(toDataNode(proxiedFragmentEntity, fetchDescendantsOption));
377             } else {
378                 final String normalizedXpath = getNormalizedXpath(proxiedFragmentEntity.getXpath());
379                 final Collection<FragmentEntity> unproxiedFragmentEntities =
380                     buildFragmentEntitiesFromFragmentExtracts(anchorEntity, normalizedXpath);
381                 for (final FragmentEntity unproxiedFragmentEntity : unproxiedFragmentEntities) {
382                     dataNodes.add(toDataNode(unproxiedFragmentEntity, fetchDescendantsOption));
383                 }
384             }
385         }
386         return Collections.unmodifiableList(dataNodes);
387     }
388
389     private List<DataNode> createDataNodesFromFragmentEntities(final FetchDescendantsOption fetchDescendantsOption,
390                                                                final Collection<FragmentEntity> fragmentEntities) {
391         final List<DataNode> dataNodes = new ArrayList<>(fragmentEntities.size());
392         for (final FragmentEntity fragmentEntity : fragmentEntities) {
393             dataNodes.add(toDataNode(fragmentEntity, fetchDescendantsOption));
394         }
395         return Collections.unmodifiableList(dataNodes);
396     }
397
398     private static String getNormalizedXpath(final String xpathSource) {
399         final String normalizedXpath;
400         try {
401             normalizedXpath = CpsPathUtil.getNormalizedXpath(xpathSource);
402         } catch (final PathParsingException e) {
403             throw new CpsPathException(e.getMessage());
404         }
405         return normalizedXpath;
406     }
407
408     @Override
409     public String startSession() {
410         return sessionManager.startSession();
411     }
412
413     @Override
414     public void closeSession(final String sessionId) {
415         sessionManager.closeSession(sessionId, SessionManager.WITH_COMMIT);
416     }
417
418     @Override
419     public void lockAnchor(final String sessionId, final String dataspaceName,
420                            final String anchorName, final Long timeoutInMilliseconds) {
421         sessionManager.lockAnchor(sessionId, dataspaceName, anchorName, timeoutInMilliseconds);
422     }
423
424     private static Set<String> processAncestorXpath(final Collection<FragmentEntity> fragmentEntities,
425                                                     final CpsPathQuery cpsPathQuery) {
426         final Set<String> ancestorXpath = new HashSet<>();
427         final Pattern pattern =
428                 Pattern.compile("([\\s\\S]*\\/" + Pattern.quote(cpsPathQuery.getAncestorSchemaNodeIdentifier())
429                         + REG_EX_FOR_OPTIONAL_LIST_INDEX + "\\/[\\s\\S]*");
430         for (final FragmentEntity fragmentEntity : fragmentEntities) {
431             final Matcher matcher = pattern.matcher(fragmentEntity.getXpath());
432             if (matcher.matches()) {
433                 ancestorXpath.add(matcher.group(1));
434             }
435         }
436         return ancestorXpath;
437     }
438
439     private DataNode toDataNode(final FragmentEntity fragmentEntity,
440                                 final FetchDescendantsOption fetchDescendantsOption) {
441         final List<DataNode> childDataNodes = getChildDataNodes(fragmentEntity, fetchDescendantsOption);
442         Map<String, Serializable> leaves = new HashMap<>();
443         if (fragmentEntity.getAttributes() != null) {
444             leaves = jsonObjectMapper.convertJsonString(fragmentEntity.getAttributes(), Map.class);
445         }
446         return new DataNodeBuilder()
447                 .withXpath(fragmentEntity.getXpath())
448                 .withLeaves(leaves)
449                 .withChildDataNodes(childDataNodes).build();
450     }
451
452     private List<DataNode> getChildDataNodes(final FragmentEntity fragmentEntity,
453                                              final FetchDescendantsOption fetchDescendantsOption) {
454         if (fetchDescendantsOption.hasNext()) {
455             return fragmentEntity.getChildFragments().stream()
456                     .map(childFragmentEntity -> toDataNode(childFragmentEntity, fetchDescendantsOption.next()))
457                     .collect(Collectors.toList());
458         }
459         return Collections.emptyList();
460     }
461
462     @Override
463     public void updateDataLeaves(final String dataspaceName, final String anchorName, final String xpath,
464                                  final Map<String, Serializable> leaves) {
465         final FragmentEntity fragmentEntity = getFragmentWithoutDescendantsByXpath(dataspaceName, anchorName, xpath);
466         fragmentEntity.setAttributes(jsonObjectMapper.asJsonString(leaves));
467         fragmentRepository.save(fragmentEntity);
468     }
469
470     @Override
471     public void updateDataNodeAndDescendants(final String dataspaceName, final String anchorName,
472                                              final DataNode dataNode) {
473         final FragmentEntity fragmentEntity =
474             getFragmentWithoutDescendantsByXpath(dataspaceName, anchorName, dataNode.getXpath());
475         updateFragmentEntityAndDescendantsWithDataNode(fragmentEntity, dataNode);
476         try {
477             fragmentRepository.save(fragmentEntity);
478         } catch (final StaleStateException staleStateException) {
479             throw new ConcurrencyException("Concurrent Transactions",
480                     String.format("dataspace :'%s', Anchor : '%s' and xpath: '%s' is updated by another transaction.",
481                             dataspaceName, anchorName, dataNode.getXpath()));
482         }
483     }
484
485     @Override
486     public void updateDataNodesAndDescendants(final String dataspaceName,
487                                               final String anchorName,
488                                               final List<DataNode> dataNodes) {
489
490         final Map<DataNode, FragmentEntity> dataNodeFragmentEntityMap = dataNodes.stream()
491                 .collect(Collectors.toMap(
492                         dataNode -> dataNode,
493                         dataNode ->
494                             getFragmentWithoutDescendantsByXpath(dataspaceName, anchorName, dataNode.getXpath())));
495         dataNodeFragmentEntityMap.forEach(
496                 (dataNode, fragmentEntity) -> updateFragmentEntityAndDescendantsWithDataNode(fragmentEntity, dataNode));
497         try {
498             fragmentRepository.saveAll(dataNodeFragmentEntityMap.values());
499         } catch (final StaleStateException staleStateException) {
500             retryUpdateDataNodesIndividually(dataspaceName, anchorName, dataNodeFragmentEntityMap.values());
501         }
502     }
503
504     private void retryUpdateDataNodesIndividually(final String dataspaceName, final String anchorName,
505                                                   final Collection<FragmentEntity> fragmentEntities) {
506         final Collection<String> failedXpaths = new HashSet<>();
507
508         fragmentEntities.forEach(dataNodeFragment -> {
509             try {
510                 fragmentRepository.save(dataNodeFragment);
511             } catch (final StaleStateException e) {
512                 failedXpaths.add(dataNodeFragment.getXpath());
513             }
514         });
515
516         if (!failedXpaths.isEmpty()) {
517             final String failedXpathsConcatenated = String.join(",", failedXpaths);
518             throw new ConcurrencyException("Concurrent Transactions", String.format(
519                     "DataNodes : %s in Dataspace :'%s' with Anchor : '%s'  are updated by another transaction.",
520                     failedXpathsConcatenated, dataspaceName, anchorName));
521         }
522     }
523
524     private void updateFragmentEntityAndDescendantsWithDataNode(final FragmentEntity existingFragmentEntity,
525                                                                 final DataNode newDataNode) {
526
527         existingFragmentEntity.setAttributes(jsonObjectMapper.asJsonString(newDataNode.getLeaves()));
528
529         final Map<String, FragmentEntity> existingChildrenByXpath = existingFragmentEntity.getChildFragments().stream()
530                 .collect(Collectors.toMap(FragmentEntity::getXpath, childFragmentEntity -> childFragmentEntity));
531
532         final Collection<FragmentEntity> updatedChildFragments = new HashSet<>();
533
534         for (final DataNode newDataNodeChild : newDataNode.getChildDataNodes()) {
535             final FragmentEntity childFragment;
536             if (isNewDataNode(newDataNodeChild, existingChildrenByXpath)) {
537                 childFragment = convertToFragmentWithAllDescendants(
538                         existingFragmentEntity.getDataspace(), existingFragmentEntity.getAnchor(), newDataNodeChild);
539             } else {
540                 childFragment = existingChildrenByXpath.get(newDataNodeChild.getXpath());
541                 updateFragmentEntityAndDescendantsWithDataNode(childFragment, newDataNodeChild);
542             }
543             updatedChildFragments.add(childFragment);
544         }
545
546         existingFragmentEntity.getChildFragments().clear();
547         existingFragmentEntity.getChildFragments().addAll(updatedChildFragments);
548     }
549
550     @Override
551     @Transactional
552     public void replaceListContent(final String dataspaceName, final String anchorName, final String parentNodeXpath,
553                                    final Collection<DataNode> newListElements) {
554         final FragmentEntity parentEntity =
555             getFragmentWithoutDescendantsByXpath(dataspaceName, anchorName, parentNodeXpath);
556         final String listElementXpathPrefix = getListElementXpathPrefix(newListElements);
557         final Map<String, FragmentEntity> existingListElementFragmentEntitiesByXPath =
558                 extractListElementFragmentEntitiesByXPath(parentEntity.getChildFragments(), listElementXpathPrefix);
559         deleteListElements(parentEntity.getChildFragments(), existingListElementFragmentEntitiesByXPath);
560         final Set<FragmentEntity> updatedChildFragmentEntities = new HashSet<>();
561         for (final DataNode newListElement : newListElements) {
562             final FragmentEntity existingListElementEntity =
563                     existingListElementFragmentEntitiesByXPath.get(newListElement.getXpath());
564             final FragmentEntity entityToBeAdded = getFragmentForReplacement(parentEntity, newListElement,
565                     existingListElementEntity);
566
567             updatedChildFragmentEntities.add(entityToBeAdded);
568         }
569         parentEntity.getChildFragments().addAll(updatedChildFragmentEntities);
570         fragmentRepository.save(parentEntity);
571     }
572
573     @Override
574     @Transactional
575     public void deleteDataNodes(final String dataspaceName, final String anchorName) {
576         final DataspaceEntity dataspaceEntity = dataspaceRepository.getByName(dataspaceName);
577         anchorRepository.findByDataspaceAndName(dataspaceEntity, anchorName)
578                 .ifPresent(
579                         anchorEntity -> fragmentRepository.deleteByAnchorIn(Set.of(anchorEntity)));
580     }
581
582     @Override
583     @Transactional
584     public void deleteListDataNode(final String dataspaceName, final String anchorName,
585                                    final String targetXpath) {
586         deleteDataNode(dataspaceName, anchorName, targetXpath, true);
587     }
588
589     @Override
590     @Transactional
591     public void deleteDataNode(final String dataspaceName, final String anchorName, final String targetXpath) {
592         deleteDataNode(dataspaceName, anchorName, targetXpath, false);
593     }
594
595     private void deleteDataNode(final String dataspaceName, final String anchorName, final String targetXpath,
596                                 final boolean onlySupportListNodeDeletion) {
597         final String parentNodeXpath;
598         FragmentEntity parentFragmentEntity = null;
599         boolean targetDeleted;
600         if (isRootXpath(targetXpath)) {
601             deleteDataNodes(dataspaceName, anchorName);
602             targetDeleted = true;
603         } else {
604             if (isRootContainerNodeXpath(targetXpath)) {
605                 parentNodeXpath = targetXpath;
606             } else {
607                 parentNodeXpath = CpsPathUtil.getNormalizedParentXpath(targetXpath);
608             }
609             parentFragmentEntity = getFragmentWithoutDescendantsByXpath(dataspaceName, anchorName, parentNodeXpath);
610             if (CpsPathUtil.isPathToListElement(targetXpath)) {
611                 targetDeleted = deleteDataNode(parentFragmentEntity, targetXpath);
612             } else {
613                 targetDeleted = deleteAllListElements(parentFragmentEntity, targetXpath);
614                 final boolean tryToDeleteDataNode = !targetDeleted && !onlySupportListNodeDeletion;
615                 if (tryToDeleteDataNode) {
616                     targetDeleted = deleteDataNode(parentFragmentEntity, targetXpath);
617                 }
618             }
619         }
620         if (!targetDeleted) {
621             final String additionalInformation = onlySupportListNodeDeletion
622                     ? "The target is probably not a List." : "";
623             throw new DataNodeNotFoundException(parentFragmentEntity.getDataspace().getName(),
624                     parentFragmentEntity.getAnchor().getName(), targetXpath, additionalInformation);
625         }
626     }
627
628     private boolean deleteDataNode(final FragmentEntity parentFragmentEntity, final String targetXpath) {
629         final String normalizedTargetXpath = CpsPathUtil.getNormalizedXpath(targetXpath);
630         if (parentFragmentEntity.getXpath().equals(normalizedTargetXpath)) {
631             fragmentRepository.delete(parentFragmentEntity);
632             return true;
633         }
634         if (parentFragmentEntity.getChildFragments()
635                 .removeIf(fragment -> fragment.getXpath().equals(normalizedTargetXpath))) {
636             fragmentRepository.save(parentFragmentEntity);
637             return true;
638         }
639         return false;
640     }
641
642     private boolean deleteAllListElements(final FragmentEntity parentFragmentEntity, final String listXpath) {
643         final String normalizedListXpath = CpsPathUtil.getNormalizedXpath(listXpath);
644         final String deleteTargetXpathPrefix = normalizedListXpath + "[";
645         if (parentFragmentEntity.getChildFragments()
646                 .removeIf(fragment -> fragment.getXpath().startsWith(deleteTargetXpathPrefix))) {
647             fragmentRepository.save(parentFragmentEntity);
648             return true;
649         }
650         return false;
651     }
652
653     private static void deleteListElements(
654             final Collection<FragmentEntity> fragmentEntities,
655             final Map<String, FragmentEntity> existingListElementFragmentEntitiesByXPath) {
656         fragmentEntities.removeAll(existingListElementFragmentEntitiesByXPath.values());
657     }
658
659     private static String getListElementXpathPrefix(final Collection<DataNode> newListElements) {
660         if (newListElements.isEmpty()) {
661             throw new CpsAdminException("Invalid list replacement",
662                     "Cannot replace list elements with empty collection");
663         }
664         final String firstChildNodeXpath = newListElements.iterator().next().getXpath();
665         return firstChildNodeXpath.substring(0, firstChildNodeXpath.lastIndexOf('[') + 1);
666     }
667
668     private FragmentEntity getFragmentForReplacement(final FragmentEntity parentEntity,
669                                                      final DataNode newListElement,
670                                                      final FragmentEntity existingListElementEntity) {
671         if (existingListElementEntity == null) {
672             return convertToFragmentWithAllDescendants(
673                     parentEntity.getDataspace(), parentEntity.getAnchor(), newListElement);
674         }
675         if (newListElement.getChildDataNodes().isEmpty()) {
676             copyAttributesFromNewListElement(existingListElementEntity, newListElement);
677             existingListElementEntity.getChildFragments().clear();
678         } else {
679             updateFragmentEntityAndDescendantsWithDataNode(existingListElementEntity, newListElement);
680         }
681         return existingListElementEntity;
682     }
683
684     private static boolean isNewDataNode(final DataNode replacementDataNode,
685                                          final Map<String, FragmentEntity> existingListElementsByXpath) {
686         return !existingListElementsByXpath.containsKey(replacementDataNode.getXpath());
687     }
688
689     private static boolean isRootContainerNodeXpath(final String xpath) {
690         return 0 == xpath.lastIndexOf('/');
691     }
692
693     private void copyAttributesFromNewListElement(final FragmentEntity existingListElementEntity,
694                                                   final DataNode newListElement) {
695         final FragmentEntity replacementFragmentEntity =
696                 FragmentEntity.builder().attributes(jsonObjectMapper.asJsonString(
697                         newListElement.getLeaves())).build();
698         existingListElementEntity.setAttributes(replacementFragmentEntity.getAttributes());
699     }
700
701     private static Map<String, FragmentEntity> extractListElementFragmentEntitiesByXPath(
702             final Set<FragmentEntity> childEntities, final String listElementXpathPrefix) {
703         return childEntities.stream()
704                 .filter(fragmentEntity -> fragmentEntity.getXpath().startsWith(listElementXpathPrefix))
705                 .collect(Collectors.toMap(FragmentEntity::getXpath, fragmentEntity -> fragmentEntity));
706     }
707
708     private static boolean isRootXpath(final String xpath) {
709         return "/".equals(xpath) || "".equals(xpath);
710     }
711 }