Bug 3151: Fix Not a Proxy Instance Exception
[yangtools.git] / code-generator / binding-data-codec / src / main / java / org / opendaylight / yangtools / binding / data / codec / impl / LazyDataObject.java
index 6a22641e1353c8e6c5b19dc4020d64731aa4956d..38884816a13c01d049f2edb40e6234eecb083e53 100644 (file)
@@ -7,7 +7,8 @@
  */
 package org.opendaylight.yangtools.binding.data.codec.impl;
 
-import com.google.common.base.Objects.ToStringHelper;
+import com.google.common.base.MoreObjects;
+import com.google.common.base.MoreObjects.ToStringHelper;
 import com.google.common.base.Optional;
 import com.google.common.base.Preconditions;
 import com.google.common.collect.ImmutableMap;
@@ -28,7 +29,7 @@ import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNodeContainer;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
-class LazyDataObject implements InvocationHandler, AugmentationReader {
+class LazyDataObject<D extends DataObject> implements InvocationHandler, AugmentationReader {
 
     private static final Logger LOG = LoggerFactory.getLogger(LazyDataObject.class);
     private static final String GET_IMPLEMENTED_INTERFACE = "getImplementedInterface";
@@ -36,38 +37,34 @@ class LazyDataObject implements InvocationHandler, AugmentationReader {
     private static final String EQUALS = "equals";
     private static final String GET_AUGMENTATION = "getAugmentation";
     private static final String HASHCODE = "hashCode";
+    private static final String AUGMENTATIONS = "augmentations";
     private static final Object NULL_VALUE = new Object();
 
     private final ConcurrentHashMap<Method, Object> cachedData = new ConcurrentHashMap<>();
     private final NormalizedNodeContainer<?, PathArgument, NormalizedNode<?, ?>> data;
-    private final DataObjectCodecContext<?> context;
+    private final DataObjectCodecContext<D,?> context;
 
     private volatile ImmutableMap<Class<? extends Augmentation<?>>, Augmentation<?>> cachedAugmentations = null;
     private volatile Integer cachedHashcode = null;
 
     @SuppressWarnings({ "rawtypes", "unchecked" })
-    private LazyDataObject(final DataObjectCodecContext<?> ctx, final NormalizedNodeContainer data) {
+    LazyDataObject(final DataObjectCodecContext<D,?> ctx, final NormalizedNodeContainer data) {
         this.context = Preconditions.checkNotNull(ctx, "Context must not be null");
         this.data = Preconditions.checkNotNull(data, "Data must not be null");
     }
 
-    @SuppressWarnings("rawtypes")
-    static DataObject create(final DataObjectCodecContext ctx, final NormalizedNodeContainer<?, ?, ?> data) {
-        Class<?> bindingClass = ctx.bindingClass();
-        return (DataObject) Proxy.newProxyInstance(bindingClass.getClassLoader(), new Class<?>[] { bindingClass },
-                new LazyDataObject(ctx, data));
-    }
-
     @Override
     public Object invoke(final Object proxy, final Method method, final Object[] args) throws Throwable {
         if (method.getParameterTypes().length == 0) {
-            String name = method.getName();
+            final String name = method.getName();
             if (GET_IMPLEMENTED_INTERFACE.equals(name)) {
-                return context.bindingClass();
+                return context.getBindingClass();
             } else if (TO_STRING.equals(name)) {
                 return bindingToString();
             } else if (HASHCODE.equals(name)) {
                 return bindingHashCode();
+            } else if (AUGMENTATIONS.equals(name)) {
+                return getAugmentationsImpl();
             }
             return getBindingData(method);
         } else if (GET_AUGMENTATION.equals(method.getName())) {
@@ -82,13 +79,13 @@ class LazyDataObject implements InvocationHandler, AugmentationReader {
         if (other == null) {
             return false;
         }
-        if (!context.bindingClass().isAssignableFrom(other.getClass())) {
+        if (!context.getBindingClass().isAssignableFrom(other.getClass())) {
             return false;
         }
         try {
-            for (Method m : context.getHashCodeAndEqualsMethods()) {
-                Object thisValue = getBindingData(m);
-                Object otherValue = m.invoke(other);
+            for (final Method m : context.getHashCodeAndEqualsMethods()) {
+                final Object thisValue = getBindingData(m);
+                final Object otherValue = m.invoke(other);
                 if(!Objects.equals(thisValue, otherValue)) {
                     return false;
                 }
@@ -101,19 +98,19 @@ class LazyDataObject implements InvocationHandler, AugmentationReader {
     }
 
     private Integer bindingHashCode() {
-        Integer ret = cachedHashcode;
+        final Integer ret = cachedHashcode;
         if (ret != null) {
             return ret;
         }
 
         final int prime = 31;
         int result = 1;
-        for (Method m : context.getHashCodeAndEqualsMethods()) {
-            Object value = getBindingData(m);
-            result += prime * result + ((value == null) ? 0 : value.hashCode());
+        for (final Method m : context.getHashCodeAndEqualsMethods()) {
+            final Object value = getBindingData(m);
+            result = prime * result + ((value == null) ? 0 : value.hashCode());
         }
-        if (Augmentation.class.isAssignableFrom(context.bindingClass())) {
-            result += prime * result + (getAugmentations(this).hashCode());
+        if (Augmentation.class.isAssignableFrom(context.getBindingClass())) {
+            result = prime * result + (getAugmentations(this).hashCode());
         }
         cachedHashcode = result;
         return result;
@@ -122,7 +119,7 @@ class LazyDataObject implements InvocationHandler, AugmentationReader {
     private Object getBindingData(final Method method) {
         Object cached = cachedData.get(method);
         if (cached == null) {
-            Object readedValue = context.getBindingChildValue(method, data);
+            final Object readedValue = context.getBindingChildValue(method, data);
             if (readedValue == null) {
                 cached = NULL_VALUE;
             } else {
@@ -163,23 +160,25 @@ class LazyDataObject implements InvocationHandler, AugmentationReader {
             return aug.get(cls);
         }
         Preconditions.checkNotNull(cls,"Supplied augmentation must not be null.");
-        final Optional<DataContainerCodecContext<?>> augCtx= context.getPossibleStreamChild(cls);
+
+        @SuppressWarnings({"unchecked","rawtypes"})
+        final Optional<DataContainerCodecContext<?,?>> augCtx= context.possibleStreamChild((Class) cls);
         if(augCtx.isPresent()) {
             final Optional<NormalizedNode<?, ?>> augData = data.getChild(augCtx.get().getDomPathArgument());
             if (augData.isPresent()) {
-                return augCtx.get().dataFromNormalizedNode(augData.get());
+                return augCtx.get().deserialize(augData.get());
             }
         }
         return null;
     }
 
     public String bindingToString() {
-        ToStringHelper helper = com.google.common.base.Objects.toStringHelper(context.bindingClass()).omitNullValues();
+        final ToStringHelper helper = MoreObjects.toStringHelper(context.getBindingClass()).omitNullValues();
 
-        for (Method m :context.getHashCodeAndEqualsMethods()) {
+        for (final Method m :context.getHashCodeAndEqualsMethods()) {
             helper.add(m.getName(), getBindingData(m));
         }
-        if (Augmentable.class.isAssignableFrom(context.bindingClass())) {
+        if (Augmentable.class.isAssignableFrom(context.getBindingClass())) {
             helper.add("augmentations", getAugmentationsImpl());
         }
         return helper.toString();
@@ -205,7 +204,7 @@ class LazyDataObject implements InvocationHandler, AugmentationReader {
         if (getClass() != obj.getClass()) {
             return false;
         }
-        LazyDataObject other = (LazyDataObject) obj;
+        final LazyDataObject<?> other = (LazyDataObject<?>) obj;
         if (context == null) {
             if (other.context != null) {
                 return false;
@@ -222,5 +221,4 @@ class LazyDataObject implements InvocationHandler, AugmentationReader {
         }
         return true;
     }
-
 }