/**
* ============LICENSE_START=======================================================
* org.onap.aai
* ================================================================================
* Copyright © 2017-2018 AT&T Intellectual Property. All rights reserved.
* Copyright © 2017-2018 Amdocs
* ================================================================================
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
* ============LICENSE_END=========================================================
*/
package org.onap.aai.sparky.crossentityreference.sync;
import static java.util.concurrent.CompletableFuture.supplyAsync;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Deque;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentLinkedDeque;
import java.util.concurrent.ExecutorService;
import java.util.function.Supplier;
import org.onap.aai.cl.api.Logger;
import org.onap.aai.cl.eelf.LoggerFactory;
import org.onap.aai.cl.mdc.MdcContext;
import org.onap.aai.restclient.client.OperationResult;
import org.onap.aai.sparky.config.oxm.CrossEntityReference;
import org.onap.aai.sparky.config.oxm.CrossEntityReferenceDescriptor;
import org.onap.aai.sparky.config.oxm.CrossEntityReferenceLookup;
import org.onap.aai.sparky.config.oxm.OxmEntityDescriptor;
import org.onap.aai.sparky.config.oxm.OxmEntityLookup;
import org.onap.aai.sparky.config.oxm.SearchableEntityLookup;
import org.onap.aai.sparky.config.oxm.SearchableOxmEntityDescriptor;
import org.onap.aai.sparky.dal.ActiveInventoryAdapter;
import org.onap.aai.sparky.dal.NetworkTransaction;
import org.onap.aai.sparky.dal.rest.HttpMethod;
import org.onap.aai.sparky.logging.AaiUiMsgs;
import org.onap.aai.sparky.sync.AbstractEntitySynchronizer;
import org.onap.aai.sparky.sync.IndexSynchronizer;
import org.onap.aai.sparky.sync.SynchronizerConstants;
import org.onap.aai.sparky.sync.config.ElasticSearchSchemaConfig;
import org.onap.aai.sparky.sync.config.NetworkStatisticsConfig;
import org.onap.aai.sparky.sync.entity.IndexableCrossEntityReference;
import org.onap.aai.sparky.sync.entity.MergableEntity;
import org.onap.aai.sparky.sync.entity.SelfLinkDescriptor;
import org.onap.aai.sparky.sync.enumeration.OperationState;
import org.onap.aai.sparky.sync.enumeration.SynchronizerState;
import org.onap.aai.sparky.sync.task.PerformActiveInventoryRetrieval;
import org.onap.aai.sparky.sync.task.PerformElasticSearchPut;
import org.onap.aai.sparky.sync.task.PerformElasticSearchRetrieval;
import org.onap.aai.sparky.sync.task.PerformElasticSearchUpdate;
import org.onap.aai.sparky.util.NodeUtils;
import org.slf4j.MDC;
import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.ObjectReader;
import com.fasterxml.jackson.databind.node.ArrayNode;
/**
* The Class CrossEntityReferenceSynchronizer.
*/
public class CrossEntityReferenceSynchronizer extends AbstractEntitySynchronizer
implements IndexSynchronizer {
/**
* The Class RetryCrossEntitySyncContainer.
*/
private class RetryCrossEntitySyncContainer {
NetworkTransaction txn;
IndexableCrossEntityReference icer;
/**
* Instantiates a new retry cross entity sync container.
*
* @param txn the txn
* @param icer the icer
*/
public RetryCrossEntitySyncContainer(NetworkTransaction txn,
IndexableCrossEntityReference icer) {
this.txn = txn;
this.icer = icer;
}
public NetworkTransaction getNetworkTransaction() {
return txn;
}
public IndexableCrossEntityReference getIndexableCrossEntityReference() {
return icer;
}
}
private static final Logger LOG =
LoggerFactory.getInstance().getLogger(CrossEntityReferenceSynchronizer.class);
private static final String SERVICE_INSTANCE = "service-instance";
private Deque selflinks;
private Deque retryQueue;
private Map retryLimitTracker;
private boolean isAllWorkEnumerated;
protected ExecutorService esPutExecutor;
private CrossEntityReferenceLookup crossEntityReferenceLookup;
private OxmEntityLookup oxmEntityLookup;
private SearchableEntityLookup searchableEntityLookup;
/**
* Instantiates a new cross entity reference synchronizer.
*
* @throws Exception the exception
*/
public CrossEntityReferenceSynchronizer(ElasticSearchSchemaConfig schemaConfig,
int internalSyncWorkers, int aaiWorkers, int esWorkers, NetworkStatisticsConfig aaiStatConfig,
NetworkStatisticsConfig esStatConfig, CrossEntityReferenceLookup crossEntityReferenceLookup,
OxmEntityLookup oxmEntityLookup, SearchableEntityLookup searchableEntityLookup) throws Exception {
super(LOG, "CERS", internalSyncWorkers, aaiWorkers, esWorkers, schemaConfig.getIndexName(),
aaiStatConfig, esStatConfig);
this.crossEntityReferenceLookup = crossEntityReferenceLookup;
this.oxmEntityLookup = oxmEntityLookup;
this.searchableEntityLookup = searchableEntityLookup;
this.selflinks = new ConcurrentLinkedDeque();
this.retryQueue = new ConcurrentLinkedDeque();
this.retryLimitTracker = new ConcurrentHashMap();
this.synchronizerName = "Cross Reference Entity Synchronizer";
this.isAllWorkEnumerated = false;
this.esPutExecutor = NodeUtils.createNamedExecutor("CERS-ES-PUT", 5, LOG);
this.aaiEntityStats.intializeEntityCounters(
crossEntityReferenceLookup.getCrossReferenceEntityDescriptors().keySet());
this.esEntityStats.intializeEntityCounters(
crossEntityReferenceLookup.getCrossReferenceEntityDescriptors().keySet());
this.syncDurationInMs = -1;
}
/* (non-Javadoc)
* @see org.openecomp.sparky.synchronizer.IndexSynchronizer#doSync()
*/
@Override
public OperationState doSync() {
this.syncDurationInMs = -1;
String txnID = NodeUtils.getRandomTxnId();
MdcContext.initialize(txnID, "CrossEntitySynchronizer", "", "Sync", "");
resetCounters();
syncStartedTimeStampInMs = System.currentTimeMillis();
launchSyncFlow();
return OperationState.OK;
}
@Override
public SynchronizerState getState() {
if (!isSyncDone()) {
return SynchronizerState.PERFORMING_SYNCHRONIZATION;
}
return SynchronizerState.IDLE;
}
/* (non-Javadoc)
* @see org.openecomp.sparky.synchronizer.IndexSynchronizer#getStatReport(boolean)
*/
@Override
public String getStatReport(boolean showFinalReport) {
syncDurationInMs = System.currentTimeMillis() - syncStartedTimeStampInMs;
return getStatReport(syncDurationInMs, showFinalReport);
}
/* (non-Javadoc)
* @see org.openecomp.sparky.synchronizer.IndexSynchronizer#shutdown()
*/
@Override
public void shutdown() {
this.shutdownExecutors();
}
@Override
protected boolean isSyncDone() {
int totalWorkOnHand = aaiWorkOnHand.get() + esWorkOnHand.get();
if (totalWorkOnHand > 0 || !isAllWorkEnumerated) {
return false;
}
return true;
}
/**
* Launch sync flow.
*
* @return the operation state
*/
private OperationState launchSyncFlow() {
final Map contextMap = MDC.getCopyOfContextMap();
Map descriptorMap =
crossEntityReferenceLookup.getCrossReferenceEntityDescriptors();
if (descriptorMap.isEmpty()) {
LOG.error(AaiUiMsgs.ERROR_LOADING_OXM);
return OperationState.ERROR;
}
Collection syncTypes = descriptorMap.keySet();
try {
/*
* launch a parallel async thread to process the documents for each entity-type (to max the of
* the configured executor anyway)
*/
aaiWorkOnHand.set(syncTypes.size());
for (String key : syncTypes) {
supplyAsync(new Supplier() {
@Override
public Void get() {
MDC.setContextMap(contextMap);
OperationResult typeLinksResult = null;
try {
typeLinksResult = aaiAdapter.getSelfLinksByEntityType(key);
aaiWorkOnHand.decrementAndGet();
processEntityTypeSelfLinks(typeLinksResult);
} catch (Exception exc) {
LOG.error(AaiUiMsgs.ERROR_GENERIC,
"An error occurred processing entity selflinks. Error: " + exc.getMessage());
}
return null;
}
}, aaiExecutor).whenComplete((result, error) -> {
if (error != null) {
LOG.error(AaiUiMsgs.ERROR_GETTING_DATA_FROM_AAI, error.getMessage());
}
});
}
while (aaiWorkOnHand.get() != 0) {
if (LOG.isDebugEnabled()) {
LOG.debug(AaiUiMsgs.WAIT_FOR_ALL_SELFLINKS_TO_BE_COLLECTED);
}
Thread.sleep(1000);
}
aaiWorkOnHand.set(selflinks.size());
isAllWorkEnumerated = true;
performSync();
while (!isSyncDone()) {
performRetrySync();
Thread.sleep(1000);
}
/*
* Make sure we don't hang on to retries that failed which could cause issues during future
* syncs
*/
retryLimitTracker.clear();
} catch (Exception exc) {
LOG.error(AaiUiMsgs.ERROR_GENERIC,
"An error occurred during entity synchronization. Error: " + exc.getMessage());
}
return OperationState.OK;
}
/**
* Perform sync.
*/
private void performSync() {
while (selflinks.peek() != null) {
SelfLinkDescriptor linkDescriptor = selflinks.poll();
aaiWorkOnHand.decrementAndGet();
CrossEntityReferenceDescriptor descriptor = null;
if (linkDescriptor.getSelfLink() != null && linkDescriptor.getEntityType() != null) {
descriptor = crossEntityReferenceLookup.getCrossReferenceEntityDescriptors()
.get(linkDescriptor.getEntityType());
if (descriptor == null) {
LOG.error(AaiUiMsgs.MISSING_ENTITY_DESCRIPTOR, linkDescriptor.getEntityType());
// go to next element in iterator
continue;
}
if (descriptor.hasCrossEntityReferences()) {
NetworkTransaction txn = new NetworkTransaction();
txn.setDescriptor(descriptor);
txn.setLink(linkDescriptor.getSelfLink());
txn.setQueryParameters(linkDescriptor.getDepthModifier());
txn.setOperationType(HttpMethod.GET);
txn.setEntityType(linkDescriptor.getEntityType());
aaiWorkOnHand.incrementAndGet();
supplyAsync(new PerformActiveInventoryRetrieval(txn, aaiAdapter), aaiExecutor)
.whenComplete((result, error) -> {
aaiWorkOnHand.decrementAndGet();
if (error != null) {
LOG.error(AaiUiMsgs.SELF_LINK_GET, error.getLocalizedMessage());
} else {
if (result == null) {
LOG.error(AaiUiMsgs.SELF_LINK_CROSS_REF_SYNC);
} else {
updateActiveInventoryCounters(result);
fetchDocumentForUpsert(result);
}
}
});
}
}
}
}
/**
* Process entity type self links.
*
* @param operationResult the operation result
*/
private void processEntityTypeSelfLinks(OperationResult operationResult) {
final String jsonResult = operationResult.getResult();
if (jsonResult != null && jsonResult.length() > 0) {
try {
JsonNode rootNode = mapper.readTree(jsonResult);
JsonNode resultData = rootNode.get("result-data");
if (resultData.isArray()) {
ArrayNode resultDataArrayNode = (ArrayNode) resultData;
Iterator elementIterator = resultDataArrayNode.elements();
while (elementIterator.hasNext()) {
JsonNode element = elementIterator.next();
final String resourceType = NodeUtils.getNodeFieldAsText(element, "resource-type");
final String resourceLink = NodeUtils.getNodeFieldAsText(element, "resource-link");
if (resourceType != null && resourceLink != null) {
CrossEntityReferenceDescriptor descriptor = crossEntityReferenceLookup.getCrossReferenceEntityDescriptors().get(resourceType);
if (descriptor == null) {
LOG.error(AaiUiMsgs.MISSING_ENTITY_DESCRIPTOR, resourceType);
// go to next element in iterator
continue;
}
if (descriptor.hasCrossEntityReferences()) {
selflinks.add(new SelfLinkDescriptor(
resourceLink, SynchronizerConstants.DEPTH_ALL_MODIFIER, resourceType));
}
}
}
}
} catch (IOException exc) {
// TODO // TODO -> LOG, waht should be logged here?
}
}
}
/**
* By providing the entity type and a json node for the entity, determine the
* primary key name(s) + primary key value(s) sufficient to build an entity query string
* of the following format:
*
* .:
*
* @return - a composite string in the above format or null
*/
private String determineEntityQueryString(String entityType, JsonNode entityJsonNode) {
OxmEntityDescriptor entityDescriptor =
oxmEntityLookup.getEntityDescriptors().get(entityType);
String queryString = null;
if ( entityDescriptor != null ) {
final List primaryKeyNames = entityDescriptor.getPrimaryKeyAttributeNames();
final List keyValues = new ArrayList();
NodeUtils.extractFieldValuesFromObject(entityJsonNode, primaryKeyNames, keyValues);
queryString = entityType + "." + NodeUtils.concatArray(primaryKeyNames,"/") + ":" + NodeUtils.concatArray(keyValues);
}
return queryString;
}
/**
* Fetch document for upsert.
*
* @param txn the txn
*/
private void fetchDocumentForUpsert(NetworkTransaction txn) {
if (!txn.getOperationResult().wasSuccessful()) {
LOG.error(AaiUiMsgs.SELF_LINK_GET, txn.getOperationResult().getResult());
return;
}
CrossEntityReferenceDescriptor cerDescriptor = crossEntityReferenceLookup
.getCrossReferenceEntityDescriptors().get(txn.getDescriptor().getEntityName());
if (cerDescriptor != null && cerDescriptor.hasCrossEntityReferences()) {
final String jsonResult = txn.getOperationResult().getResult();
if (jsonResult != null && jsonResult.length() > 0) {
/**
* Here's what we are going to do:
*
* Extract primary key name and value from the parent type.
* Extract the primary key and value from the nested child instance.
* Build a generic query to discover the self-link for the nested-child-instance using
* parent and child.
* Set the self-link on the child.
* Generate the id that will allow the elastic-search upsert to work.
* Rinse and repeat.
*/
CrossEntityReference cerDefinition = cerDescriptor.getCrossEntityReference();
if (cerDefinition != null) {
JsonNode convertedNode = null;
try {
convertedNode = NodeUtils.convertJsonStrToJsonNode(txn.getOperationResult().getResult());
final String parentEntityQueryString = determineEntityQueryString(txn.getEntityType(), convertedNode);
List extractedParentEntityAttributeValues = new ArrayList();
NodeUtils.extractFieldValuesFromObject(convertedNode,
cerDefinition.getReferenceAttributes(),
extractedParentEntityAttributeValues);
List nestedTargetEntityInstances = new ArrayList();
NodeUtils.extractObjectsByKey(convertedNode, cerDefinition.getTargetEntityType(),
nestedTargetEntityInstances);
for (JsonNode targetEntityInstance : nestedTargetEntityInstances) {
if (cerDescriptor != null) {
String childEntityType = cerDefinition.getTargetEntityType();
OxmEntityDescriptor childDesciptor = oxmEntityLookup.getEntityDescriptors().get(childEntityType);
List childPrimaryKeyNames = null;
if (childDesciptor != null) {
childPrimaryKeyNames = childDesciptor.getPrimaryKeyAttributeNames();
} else {
childPrimaryKeyNames = new ArrayList();
}
List childKeyValues = new ArrayList();
NodeUtils.extractFieldValuesFromObject(targetEntityInstance, childPrimaryKeyNames, childKeyValues);
String childEntityQueryKeyString = childEntityType + "." + NodeUtils.concatArray(childPrimaryKeyNames,"/") + ":" + NodeUtils.concatArray(childKeyValues);
/**
* Build generic-query to query child instance self-link from AAI
*/
List orderedQueryKeyParams = new ArrayList();
/**
* At present, there is an issue with resolving the self-link using the
* generic-query with nothing more than the service-instance identifier and the
* service-subscription. There is another level of detail we don't have access to
* unless we parse it out of the service-subscription self-link, which is a
* coupling I would like to avoid. Fortunately, there is a workaround, but only
* for service-instances, which is presently our only use-case for the
* cross-entity-reference in R1707. Going forwards hopefully there will be other
* ways to resolve a child self-link using parental embedded meta data that we
* don't currently have.
*
* The work-around with the service-instance entity-type is that it's possible to
* request the self-link using only the service-instance-id because of a
* historical AAI functional query requirement that it be possible to query a
* service-instance only by it's service-instance-id. This entity type is the only
* one in the system that can be queried this way which makes it a very limited
* workaround, but good enough for the current release.
*/
if (SERVICE_INSTANCE.equals(childEntityType)) {
orderedQueryKeyParams.clear();
orderedQueryKeyParams.add(childEntityQueryKeyString);
} else {
orderedQueryKeyParams.add(parentEntityQueryString);
orderedQueryKeyParams.add(childEntityQueryKeyString);
}
String genericQueryStr = null;
try {
genericQueryStr = aaiAdapter.getGenericQueryForSelfLink(childEntityType, orderedQueryKeyParams);
if (genericQueryStr != null) {
aaiWorkOnHand.incrementAndGet();
OperationResult aaiQueryResult = aaiAdapter.queryActiveInventoryWithRetries(
genericQueryStr, "application/json",
aaiAdapter.getEndpointConfig().getNumRequestRetries());
aaiWorkOnHand.decrementAndGet();
if (aaiQueryResult!= null && aaiQueryResult.wasSuccessful()) {
Collection entityLinks = new ArrayList();
JsonNode genericQueryResult = null;
try {
genericQueryResult = NodeUtils.convertJsonStrToJsonNode(aaiQueryResult.getResult());
if ( genericQueryResult != null ) {
NodeUtils.extractObjectsByKey(genericQueryResult, "resource-link", entityLinks);
String selfLink = null;
if (entityLinks.size() != 1) {
/**
* an ambiguity exists where we can't reliably determine the self
* link, this should be a permanent error
*/
LOG.error(AaiUiMsgs.ENTITY_SYNC_FAILED_SELFLINK_AMBIGUITY, String.valueOf(entityLinks.size()));
} else {
selfLink = ((JsonNode) entityLinks.toArray()[0]).asText();
IndexableCrossEntityReference icer =
getPopulatedDocument(targetEntityInstance, cerDescriptor);
for (String parentCrossEntityReferenceAttributeValue : extractedParentEntityAttributeValues) {
icer.addCrossEntityReferenceValue(
parentCrossEntityReferenceAttributeValue);
}
icer.setLink(ActiveInventoryAdapter.extractResourcePath(selfLink));
icer.deriveFields();
String link = null;
try {
link = elasticSearchAdapter
.buildElasticSearchGetDocUrl(getIndexName(), icer.getId());
} catch (Exception exc) {
LOG.error(AaiUiMsgs.ES_FAILED_TO_CONSTRUCT_QUERY,
exc.getLocalizedMessage());
}
if (link != null) {
NetworkTransaction n2 = new NetworkTransaction();
n2.setLink(link);
n2.setEntityType(txn.getEntityType());
n2.setDescriptor(txn.getDescriptor());
n2.setOperationType(HttpMethod.GET);
esWorkOnHand.incrementAndGet();
supplyAsync(
new PerformElasticSearchRetrieval(n2, elasticSearchAdapter),
esExecutor).whenComplete((result, error) -> {
esWorkOnHand.decrementAndGet();
if (error != null) {
LOG.error(AaiUiMsgs.ES_RETRIEVAL_FAILED,
error.getLocalizedMessage());
} else {
updateElasticSearchCounters(result);
performDocumentUpsert(result, icer);
}
});
}
}
} else {
LOG.error(AaiUiMsgs.ENTITY_SYNC_FAILED_DURING_AAI_RESPONSE_CONVERSION);
}
} catch (Exception exc) {
LOG.error(AaiUiMsgs.JSON_CONVERSION_ERROR, JsonNode.class.toString(), exc.getLocalizedMessage());
}
} else {
String result = aaiQueryResult != null ? aaiQueryResult.getResult() : "unknown";
String message = "Entity sync failed because AAI query failed with error " + result;
LOG.error(AaiUiMsgs.ENTITY_SYNC_FAILED_QUERY_ERROR, message);
}
} else {
String message = "Entity Sync failed because generic query str could not be determined.";
LOG.error(AaiUiMsgs.ENTITY_SYNC_FAILED_QUERY_ERROR, message);
}
} catch (Exception exc) {
String message = "Failed to sync entity because generation of generic query failed with error = " + exc.getMessage();
LOG.error(AaiUiMsgs.ENTITY_SYNC_FAILED_QUERY_ERROR, message);
}
}
}
} catch (IOException ioe) {
LOG.error(AaiUiMsgs.JSON_PROCESSING_ERROR, ioe.getMessage());
}
}
}
} else {
LOG.error(AaiUiMsgs.ENTITY_SYNC_FAILED_DESCRIPTOR_NOT_FOUND, txn.getEntityType());
}
}
/**
* Perform document upsert.
*
* @param esGetResult the es get result
* @param icer the icer
*/
protected void performDocumentUpsert(NetworkTransaction esGetResult,
IndexableCrossEntityReference icer) {
/**
*
*
* As part of the response processing we need to do the following:
* - 1. Extract the version (if present), it will be the ETAG when we use the
* Search-Abstraction-Service
*
- 2. Spawn next task which is to do the PUT operation into elastic with or with the version
* tag
*
- a) if version is null or RC=404, then standard put, no _update with version tag
*
- b) if version != null, do PUT with _update?version= (versionNumber) in the URI to elastic
*
*
*/
String link = null;
try {
link = elasticSearchAdapter.buildElasticSearchGetDocUrl(getIndexName(), icer.getId());
} catch (Exception exc) {
LOG.error(AaiUiMsgs.ES_LINK_UPSERT, exc.getLocalizedMessage());
return;
}
boolean wasEntryDiscovered = false;
String versionNumber = null;
if (esGetResult.getOperationResult().getResultCode() == 404) {
LOG.info(AaiUiMsgs.ES_SIMPLE_PUT, icer.getEntityPrimaryKeyValue());
} else if (esGetResult.getOperationResult().getResultCode() == 200) {
wasEntryDiscovered = true;
try {
versionNumber = NodeUtils.extractFieldValueFromObject(
NodeUtils.convertJsonStrToJsonNode(esGetResult.getOperationResult().getResult()),
"_version");
} catch (IOException exc) {
LOG.error(AaiUiMsgs.ES_ABORT_CROSS_ENTITY_REF_SYNC, "version Number",
icer.getEntityPrimaryKeyValue(), exc.getLocalizedMessage());
return;
}
} else {
/*
* Not being a 200 does not mean a failure. eg 201 is returned for created. TODO -> Should we
* return.
*/
LOG.info(AaiUiMsgs.ES_OPERATION_RETURN_CODE,
String.valueOf(esGetResult.getOperationResult().getResultCode()));
return;
}
try {
String jsonPayload = null;
if (wasEntryDiscovered) {
try {
ArrayList sourceObject = new ArrayList();
NodeUtils.extractObjectsByKey(
NodeUtils.convertJsonStrToJsonNode(esGetResult.getOperationResult().getResult()),
"_source", sourceObject);
if (!sourceObject.isEmpty()) {
String responseSource = NodeUtils.convertObjectToJson(sourceObject.get(0), false);
MergableEntity me = mapper.readValue(responseSource, MergableEntity.class);
ObjectReader updater = mapper.readerForUpdating(me);
MergableEntity merged = updater.readValue(icer.getAsJson());
jsonPayload = mapper.writeValueAsString(merged);
}
} catch (IOException exc) {
LOG.error(AaiUiMsgs.ES_ABORT_CROSS_ENTITY_REF_SYNC, "source value",
icer.getEntityPrimaryKeyValue(), exc.getLocalizedMessage());
return;
}
} else {
jsonPayload = icer.getAsJson();
}
if (wasEntryDiscovered) {
if (versionNumber != null && jsonPayload != null) {
String requestPayload = elasticSearchAdapter.buildBulkImportOperationRequest(getIndexName(),
"default", icer.getId(), versionNumber, jsonPayload);
NetworkTransaction transactionTracker = new NetworkTransaction();
transactionTracker.setEntityType(esGetResult.getEntityType());
transactionTracker.setDescriptor(esGetResult.getDescriptor());
transactionTracker.setOperationType(HttpMethod.PUT);
esWorkOnHand.incrementAndGet();
supplyAsync(new PerformElasticSearchUpdate(elasticSearchAdapter.getBulkUrl(),
requestPayload, elasticSearchAdapter, transactionTracker), esPutExecutor)
.whenComplete((result, error) -> {
esWorkOnHand.decrementAndGet();
if (error != null) {
LOG.error(AaiUiMsgs.ES_CROSS_ENTITY_REF_PUT, error.getLocalizedMessage());
} else {
updateElasticSearchCounters(result);
processStoreDocumentResult(result, esGetResult, icer);
}
});
}
} else {
if (link != null && jsonPayload != null) {
NetworkTransaction updateElasticTxn = new NetworkTransaction();
updateElasticTxn.setLink(link);
updateElasticTxn.setEntityType(esGetResult.getEntityType());
updateElasticTxn.setDescriptor(esGetResult.getDescriptor());
updateElasticTxn.setOperationType(HttpMethod.PUT);
esWorkOnHand.incrementAndGet();
supplyAsync(new PerformElasticSearchPut(jsonPayload, updateElasticTxn, elasticSearchAdapter),
esPutExecutor).whenComplete((result, error) -> {
esWorkOnHand.decrementAndGet();
if (error != null) {
LOG.error(AaiUiMsgs.ES_CROSS_ENTITY_REF_PUT, error.getLocalizedMessage());
} else {
updateElasticSearchCounters(result);
processStoreDocumentResult(result, esGetResult, icer);
}
});
}
}
} catch (Exception exc) {
LOG.error(AaiUiMsgs.ES_CROSS_ENTITY_REF_PUT, exc.getLocalizedMessage());
}
}
/**
* Process store document result.
*
* @param esPutResult the es put result
* @param esGetResult the es get result
* @param icer the icer
*/
private void processStoreDocumentResult(NetworkTransaction esPutResult,
NetworkTransaction esGetResult, IndexableCrossEntityReference icer) {
OperationResult or = esPutResult.getOperationResult();
if (!or.wasSuccessful()) {
if (or.getResultCode() == VERSION_CONFLICT_EXCEPTION_CODE) {
if (shouldAllowRetry(icer.getId())) {
esWorkOnHand.incrementAndGet();
RetryCrossEntitySyncContainer rsc = new RetryCrossEntitySyncContainer(esGetResult, icer);
retryQueue.push(rsc);
LOG.warn(AaiUiMsgs.ES_CROSS_REF_SYNC_VERSION_CONFLICT);
}
} else {
LOG.error(AaiUiMsgs.ES_CROSS_REF_SYNC_FAILURE, String.valueOf(or.getResultCode()),
or.getResult());
}
}
}
/**
* Perform retry sync.
*/
private void performRetrySync() {
while (retryQueue.peek() != null) {
RetryCrossEntitySyncContainer rsc = retryQueue.poll();
if (rsc != null) {
IndexableCrossEntityReference icer = rsc.getIndexableCrossEntityReference();
NetworkTransaction txn = rsc.getNetworkTransaction();
String link = null;
try {
// In this retry flow the icer object has already
// derived its fields
link = elasticSearchAdapter.buildElasticSearchGetDocUrl(getIndexName(), icer.getId());
} catch (Exception exc) {
LOG.error(AaiUiMsgs.ES_FAILED_TO_CONSTRUCT_URI, exc.getLocalizedMessage());
}
if (link != null) {
NetworkTransaction retryTransaction = new NetworkTransaction();
retryTransaction.setLink(link);
retryTransaction.setEntityType(txn.getEntityType());
retryTransaction.setDescriptor(txn.getDescriptor());
retryTransaction.setOperationType(HttpMethod.GET);
/*
* IMPORTANT - DO NOT incrementAndGet the esWorkOnHand as this is a retry flow and we did
* that for this request already when queuing the failed PUT!
*/
supplyAsync(new PerformElasticSearchRetrieval(retryTransaction, elasticSearchAdapter),
esExecutor).whenComplete((result, error) -> {
esWorkOnHand.decrementAndGet();
if (error != null) {
LOG.error(AaiUiMsgs.ES_RETRIEVAL_FAILED_RESYNC, error.getLocalizedMessage());
} else {
updateElasticSearchCounters(result);
performDocumentUpsert(result, icer);
}
});
}
}
}
}
/**
* Should allow retry.
*
* @param id the id
* @return true, if successful
*/
private boolean shouldAllowRetry(String id) {
boolean isRetryAllowed = true;
if (retryLimitTracker.get(id) != null) {
Integer currentCount = retryLimitTracker.get(id);
if (currentCount.intValue() >= RETRY_COUNT_PER_ENTITY_LIMIT.intValue()) {
isRetryAllowed = false;
LOG.error(AaiUiMsgs.ES_CROSS_ENTITY_RESYNC_LIMIT, id);
} else {
Integer newCount = new Integer(currentCount.intValue() + 1);
retryLimitTracker.put(id, newCount);
}
} else {
Integer firstRetryCount = new Integer(1);
retryLimitTracker.put(id, firstRetryCount);
}
return isRetryAllowed;
}
/**
* Gets the populated document.
*
* @param entityNode the entity node
* @param resultDescriptor the result descriptor
* @return the populated document
* @throws JsonProcessingException the json processing exception
* @throws IOException Signals that an I/O exception has occurred.
*/
protected IndexableCrossEntityReference getPopulatedDocument(JsonNode entityNode,
OxmEntityDescriptor resultDescriptor) throws JsonProcessingException, IOException {
IndexableCrossEntityReference icer = new IndexableCrossEntityReference();
icer.setEntityType(resultDescriptor.getEntityName());
List primaryKeyValues = new ArrayList();
String pkeyValue = null;
for (String keyName : resultDescriptor.getPrimaryKeyAttributeNames()) {
pkeyValue = NodeUtils.getNodeFieldAsText(entityNode, keyName);
if (pkeyValue != null) {
primaryKeyValues.add(pkeyValue);
} else {
LOG.warn(AaiUiMsgs.ES_PKEYVALUE_NULL, resultDescriptor.getEntityName());
}
}
final String primaryCompositeKeyValue = NodeUtils.concatArray(primaryKeyValues, "/");
icer.setEntityPrimaryKeyValue(primaryCompositeKeyValue);
return icer;
}
}