1 files changed, 110 insertions, 0 deletions
diff --git a/aai-core/src/main/java/org/onap/aai/introspection/sideeffect/DataLinkWriter.java b/aai-core/src/main/java/org/onap/aai/introspection/sideeffect/DataLinkWriter.java
new file mode 100644
index 00000000..f98e07c4
--- /dev/null
+++ b/aai-core/src/main/java/org/onap/aai/introspection/sideeffect/DataLinkWriter.java
@@ -0,0 +1,110 @@
+/**
+ * ============LICENSE_START=======================================================
+ * org.onap.aai
+ * ================================================================================
+ * Copyright © 2017 AT&T Intellectual Property. All rights reserved.
+ * ================================================================================
+ * 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=========================================================
+ *
+ * ECOMP is a trademark and service mark of AT&T Intellectual Property.
+ */
+package org.onap.aai.introspection.sideeffect;
+
+import org.apache.tinkerpop.gremlin.structure.Vertex;
+import org.onap.aai.db.props.AAIProperties;
+import org.onap.aai.exceptions.AAIException;
+import org.onap.aai.introspection.Introspector;
+import org.onap.aai.introspection.sideeffect.exceptions.AAIMultiplePropertiesException;
+import org.onap.aai.parsers.query.QueryParser;
+import org.onap.aai.parsers.uri.URIToObject;
+import org.onap.aai.restcore.util.URITools;
+import org.onap.aai.schema.enums.PropertyMetadata;
+import org.onap.aai.serialization.db.DBSerializer;
+import org.onap.aai.serialization.engines.TransactionalGraphEngine;
+
+import javax.ws.rs.core.MultivaluedMap;
+import java.io.UnsupportedEncodingException;
+import java.net.URI;
+import java.net.URISyntaxException;
+import java.util.List;
+import java.util.Map.Entry;
+import java.util.Optional;
+
+public class DataLinkWriter extends SideEffect {
+
+	public DataLinkWriter(Introspector obj, Vertex self, TransactionalGraphEngine dbEngine, DBSerializer serializer) {
+		super(obj, self, dbEngine, serializer);
+	}
+
+	@Override
+	protected PropertyMetadata getPropertyMetadata() {
+		return PropertyMetadata.DATA_LINK;
+	}
+
+	@Override
+	protected void processURI(Optional<String> completeUri, Entry<String, String> entry)
+			throws URISyntaxException, UnsupportedEncodingException, AAIException {
+		if (completeUri.isPresent()) {
+			URI uri = new URI(completeUri.get());
+			MultivaluedMap<String, String> map = URITools.getQueryMap(uri);
+			QueryParser uriQuery = dbEngine.getQueryBuilder(this.latestLoader).createQueryFromURI(uri, map);
+			List<Vertex> results = uriQuery.getQueryBuilder().toList();
+			if (results.size() == 1) {
+				if (results.get(0).<Boolean>property(AAIProperties.LINKED).orElse(false) && obj.getValue(entry.getKey()) == null) {
+					//delete vertex because property was removed
+					serializer.delete(results.get(0), "", false);
+				} else {
+					//link vertex that already exists
+					this.addLinkedProperty(results.get(0));
+				}
+ 			} else {
+				if (results.isEmpty()) {
+					//locate previously linked vertex
+					List<Vertex> linkedVertices = uriQuery.getQueryBuilder().getContainerQuery().getVerticesByProperty(AAIProperties.LINKED, true).toList();
+					if (!linkedVertices.isEmpty()) {
+						if (linkedVertices.size() > 1) {
+							throw new AAIMultiplePropertiesException("multiple vertices found for single cardinality propery found when searching " + uri);
+						} else {
+							//found one, remove the linked property because it didn't match the uri
+							linkedVertices.get(0).property(AAIProperties.LINKED).remove();
+						}
+					}
+					if (obj.getValue(entry.getKey()) != null) {
+						//add new vertex to database if we have values
+						URIToObject parser = new URIToObject(this.latestLoader, uri);
+						Introspector resultObj = parser.getEntity();
+						Vertex newV = serializer.createNewVertex(resultObj);
+						serializer.serializeToDb(resultObj, newV, uriQuery, completeUri.get(), this.latestLoader.getVersion().toString());
+						this.addLinkedProperty(newV);
+					}
+				} else if (results.size() > 1) {
+					throw new AAIMultiplePropertiesException("multiple values of " + entry.getKey() + " found when searching " + uri);
+				}
+			}
+		} else {
+			//skip processing because no required properties were specified
+		}
+	}
+	
+	@Override
+	protected boolean replaceWithWildcard() {
+		return true;
+	}
+	
+	private void addLinkedProperty(Vertex v) {
+		v.property(AAIProperties.LINKED, true);
+	}
+
+
+}