1 files changed, 128 insertions, 0 deletions
diff --git a/ajsc-aai/src/main/java/org/openecomp/aai/introspection/PojoLoader.java b/ajsc-aai/src/main/java/org/openecomp/aai/introspection/PojoLoader.java
new file mode 100644
index 0000000..91e88a1
--- /dev/null
+++ b/ajsc-aai/src/main/java/org/openecomp/aai/introspection/PojoLoader.java
@@ -0,0 +1,128 @@
+/*-
+ * ============LICENSE_START=======================================================
+ * org.openecomp.aai
+ * ================================================================================
+ * Copyright (C) 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=========================================================
+ */
+
+package org.openecomp.aai.introspection;
+
+import java.io.StringReader;
+
+import javax.xml.bind.JAXBContext;
+import javax.xml.bind.JAXBException;
+import javax.xml.bind.Unmarshaller;
+import javax.xml.transform.stream.StreamSource;
+
+import org.eclipse.persistence.dynamic.DynamicEntity;
+import org.eclipse.persistence.jaxb.UnmarshallerProperties;
+import org.openecomp.aai.db.AAIProperties;
+import org.openecomp.aai.exceptions.AAIException;
+import org.openecomp.aai.logging.AAILogger;
+import org.openecomp.aai.logging.LogLineBuilder;
+import org.openecomp.aai.rest.MediaType;
+import org.openecomp.aai.workarounds.NamingExceptions;
+
+import com.google.common.base.CaseFormat;
+
+public class PojoLoader extends Loader {
+
+	protected JAXBContext context;
+	private Unmarshaller unmarshaller = null;
+	private String POJO_CLASSPATH = "org.openecomp.aai.domain.yang";
+	private final String className = PojoLoader.class.getSimpleName();
+	private AAILogger aaiLogger = new AAILogger(PojoLoader.class.getName());
+	protected PojoLoader(Version version, LogLineBuilder llBuilder) {
+		super(version, ModelType.POJO, llBuilder);
+		try {
+			if (!version.equals(AAIProperties.LATEST)) {
+				POJO_CLASSPATH += "." + version; 
+			}
+			context = JAXBContext.newInstance(POJO_CLASSPATH);
+		} catch (JAXBException e) {
+			// TODO Auto-generated catch block
+			e.printStackTrace();
+		}
+		
+		
+		
+	}
+
+	@Override
+	public Introspector introspectorFromName(String name) {
+		Introspector result = null;
+		Object temp = this.objectFromName(name);
+		if (temp != null) {
+			result = IntrospectorFactory.newInstance(ModelType.POJO, temp, llBuilder);
+		}
+		return result;
+	}
+	
+	@Override
+	public Object objectFromName(String name) {
+		String upperCamel = "";
+
+		NamingExceptions exceptions = NamingExceptions.getInstance();
+		name = exceptions.getObjectName(name);
+		//Contains any uppercase, then assume it's upper camel
+		if (name.matches(".*[A-Z].*")) {
+			upperCamel = name;
+		} else {
+			upperCamel = CaseFormat.LOWER_HYPHEN.to(CaseFormat.UPPER_CAMEL, name);
+		}
+		
+		Object result = null;
+		try {
+			if (!upperCamel.contains(POJO_CLASSPATH)) {
+				upperCamel = POJO_CLASSPATH + "." + upperCamel;
+			}
+			result = Class.forName(upperCamel).newInstance();
+		} catch (InstantiationException | IllegalAccessException | ClassNotFoundException e) {
+			//entity does not exist
+		}
+		return result;
+	}
+
+	@Override
+	protected void process(Version version) {
+
+		
+	}
+
+	@Override
+	public Introspector unmarshal(String type, String json, MediaType mediaType) {
+		Introspector result = null;
+		Object obj = null;
+		Object clazz = this.objectFromName(type);
+		 try {
+			unmarshaller = context.createUnmarshaller();
+			if (mediaType.equals(MediaType.APPLICATION_JSON_TYPE)) {
+		        unmarshaller.setProperty(UnmarshallerProperties.MEDIA_TYPE, "application/json");
+		        unmarshaller.setProperty(UnmarshallerProperties.JSON_INCLUDE_ROOT, false);
+				unmarshaller.setProperty(UnmarshallerProperties.JSON_WRAPPER_AS_ARRAY_NAME, true);
+			}
+			
+			obj = unmarshaller.unmarshal(new StreamSource(new StringReader(json)), clazz.getClass()).getValue();
+			result = IntrospectorFactory.newInstance(ModelType.POJO, obj, llBuilder);
+		 } catch (JAXBException e) {
+			AAIException ex = new AAIException("AAI_4007", e);
+			aaiLogger.error(ex.getErrorObject(), llBuilder.build(className, "could not unmarshal"), e);
+		}
+		return result;
+	}
+	
+
+}