binding/mdsal-binding-dom-codec/src/main/java/org/opendaylight/mdsal/binding/dom/codec/impl/DataContainerCodecContext.java

   1 /*
   2  * Copyright (c) 2014 Cisco Systems, Inc. and others.  All rights reserved.
   3  *
   4  * This program and the accompanying materials are made available under the
   5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
   6  * and is available at http://www.eclipse.org/legal/epl-v10.html
   7  */
   8 package org.opendaylight.mdsal.binding.dom.codec.impl;
   9
  10 import static com.google.common.base.Preconditions.checkArgument;
  11 import static java.util.Objects.requireNonNull;
  12
  13 import com.google.common.collect.ImmutableCollection;
  14 import com.google.common.collect.ImmutableSet;
  15 import edu.umd.cs.findbugs.annotations.CheckReturnValue;
  16 import java.io.IOException;
  17 import java.lang.invoke.MethodHandles;
  18 import java.lang.invoke.VarHandle;
  19 import java.lang.reflect.Method;
  20 import java.lang.reflect.Modifier;
  21 import java.lang.reflect.ParameterizedType;
  22 import java.util.Arrays;
  23 import java.util.HashSet;
  24 import java.util.List;
  25 import java.util.Map;
  26 import java.util.Optional;
  27 import java.util.Set;
  28 import org.eclipse.jdt.annotation.NonNull;
  29 import org.eclipse.jdt.annotation.Nullable;
  30 import org.opendaylight.mdsal.binding.dom.codec.api.BindingDataObjectCodecTreeNode;
  31 import org.opendaylight.mdsal.binding.dom.codec.api.BindingNormalizedNodeCachingCodec;
  32 import org.opendaylight.mdsal.binding.dom.codec.api.BindingStreamEventWriter;
  33 import org.opendaylight.mdsal.binding.dom.codec.api.IncorrectNestingException;
  34 import org.opendaylight.mdsal.binding.dom.codec.api.MissingClassInLoadingStrategyException;
  35 import org.opendaylight.mdsal.binding.dom.codec.api.MissingSchemaException;
  36 import org.opendaylight.mdsal.binding.dom.codec.api.MissingSchemaForClassException;
  37 import org.opendaylight.mdsal.binding.model.api.Type;
  38 import org.opendaylight.mdsal.binding.runtime.api.BindingRuntimeContext;
  39 import org.opendaylight.mdsal.binding.runtime.api.CompositeRuntimeType;
  40 import org.opendaylight.mdsal.binding.runtime.api.RuntimeTypeContainer;
  41 import org.opendaylight.mdsal.binding.spec.reflect.BindingReflections;
  42 import org.opendaylight.yangtools.util.ClassLoaderUtils;
  43 import org.opendaylight.yangtools.yang.binding.Augmentation;
  44 import org.opendaylight.yangtools.yang.binding.BindingObject;
  45 import org.opendaylight.yangtools.yang.binding.DataContainer;
  46 import org.opendaylight.yangtools.yang.binding.DataObject;
  47 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier.PathArgument;
  48 import org.opendaylight.yangtools.yang.common.QName;
  49 import org.opendaylight.yangtools.yang.common.QNameModule;
  50 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
  51 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.AugmentationIdentifier;
  52 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
  53 import org.opendaylight.yangtools.yang.data.api.schema.stream.NormalizedNodeStreamWriter;
  54 import org.opendaylight.yangtools.yang.data.impl.schema.ImmutableNormalizedNodeStreamWriter;
  55 import org.opendaylight.yangtools.yang.data.impl.schema.NormalizedNodeResult;
  56 import org.slf4j.Logger;
  57 import org.slf4j.LoggerFactory;
  58
  59 abstract class DataContainerCodecContext<D extends DataObject, T extends RuntimeTypeContainer> extends NodeCodecContext
  60         implements BindingDataObjectCodecTreeNode<D>  {
  61     private static final Logger LOG = LoggerFactory.getLogger(DataContainerCodecContext.class);
  62     private static final VarHandle EVENT_STREAM_SERIALIZER;
  63
  64     static {
  65         try {
  66             EVENT_STREAM_SERIALIZER = MethodHandles.lookup().findVarHandle(DataContainerCodecContext.class,
  67                 "eventStreamSerializer", DataObjectSerializer.class);
  68         } catch (NoSuchFieldException | IllegalAccessException e) {
  69             throw new ExceptionInInitializerError(e);
  70         }
  71     }
  72
  73     private final @NonNull DataContainerCodecPrototype<T> prototype;
  74
  75     // Accessed via a VarHandle
  76     @SuppressWarnings("unused")
  77     private volatile DataObjectSerializer eventStreamSerializer;
  78
  79     DataContainerCodecContext(final DataContainerCodecPrototype<T> prototype) {
  80         this.prototype = requireNonNull(prototype);
  81     }
  82
  83     public final @NonNull T getType() {
  84         return prototype.getType();
  85     }
  86
  87     @Override
  88     public final ChildAddressabilitySummary getChildAddressabilitySummary() {
  89         return prototype.getChildAddressabilitySummary();
  90     }
  91
  92     protected final QNameModule namespace() {
  93         return prototype.getNamespace();
  94     }
  95
  96     protected final CodecContextFactory factory() {
  97         return prototype.getFactory();
  98     }
  99
 100     @Override
 101     protected YangInstanceIdentifier.PathArgument getDomPathArgument() {
 102         return prototype.getYangArg();
 103     }
 104
 105     /**
 106      * Returns nested node context using supplied YANG Instance Identifier.
 107      *
 108      * @param arg Yang Instance Identifier Argument
 109      * @return Context of child
 110      * @throws IllegalArgumentException If supplied argument does not represent valid child.
 111      */
 112     @Override
 113     public abstract NodeCodecContext yangPathArgumentChild(YangInstanceIdentifier.PathArgument arg);
 114
 115     /**
 116      * Returns nested node context using supplied Binding Instance Identifier
 117      * and adds YANG instance identifiers to supplied list.
 118      *
 119      * @param arg Binding Instance Identifier Argument
 120      * @return Context of child or null if supplied {@code arg} does not represent valid child.
 121      * @throws IllegalArgumentException If supplied argument does not represent valid child.
 122      */
 123     @Override
 124     public DataContainerCodecContext<?, ?> bindingPathArgumentChild(final PathArgument arg,
 125             final List<YangInstanceIdentifier.PathArgument> builder) {
 126         final DataContainerCodecContext<?,?> child = streamChild(arg.getType());
 127         if (builder != null) {
 128             child.addYangPathArgument(arg,builder);
 129         }
 130         return child;
 131     }
 132
 133     /**
 134      * Returns deserialized Binding Path Argument from YANG instance identifier.
 135      */
 136     protected PathArgument getBindingPathArgument(final YangInstanceIdentifier.PathArgument domArg) {
 137         return bindingArg();
 138     }
 139
 140     protected final PathArgument bindingArg() {
 141         return prototype.getBindingArg();
 142     }
 143
 144     @SuppressWarnings("unchecked")
 145     @Override
 146     public final Class<D> getBindingClass() {
 147         return Class.class.cast(prototype.getBindingClass());
 148     }
 149
 150     @Override
 151     public abstract <C extends DataObject> DataContainerCodecContext<C, ?> streamChild(Class<C> childClass);
 152
 153     /**
 154      * Returns child context as if it was walked by {@link BindingStreamEventWriter}. This means that to enter case, one
 155      * must issue getChild(ChoiceClass).getChild(CaseClass).
 156      *
 157      * @param childClass child class
 158      * @return Context of child or Optional.empty is supplied class is not applicable in context.
 159      */
 160     @Override
 161     public abstract <C extends DataObject> Optional<DataContainerCodecContext<C,?>> possibleStreamChild(
 162             Class<C> childClass);
 163
 164     @Override
 165     public String toString() {
 166         return getClass().getSimpleName() + " [" + prototype.getBindingClass() + "]";
 167     }
 168
 169     @Override
 170     public BindingNormalizedNodeCachingCodec<D> createCachingCodec(
 171             final ImmutableCollection<Class<? extends BindingObject>> cacheSpecifier) {
 172         if (cacheSpecifier.isEmpty()) {
 173             return new NonCachingCodec<>(this);
 174         }
 175         return new CachingNormalizedNodeCodec<>(this, ImmutableSet.copyOf(cacheSpecifier));
 176     }
 177
 178     protected final <V> @NonNull V childNonNull(final @Nullable V nullable,
 179             final YangInstanceIdentifier.PathArgument child, final String message, final Object... args) {
 180         if (nullable == null) {
 181             throw childNullException(extractName(child), message, args);
 182         }
 183         return nullable;
 184     }
 185
 186     protected final <V> @NonNull V childNonNull(final @Nullable V nullable, final QName child, final String message,
 187             final Object... args) {
 188         if (nullable == null) {
 189             throw childNullException(child, message, args);
 190         }
 191         return nullable;
 192     }
 193
 194     protected final <V> @NonNull V childNonNull(final @Nullable V nullable, final Class<?> childClass,
 195             final String message, final Object... args) {
 196         if (nullable == null) {
 197             throw childNullException(childClass, message, args);
 198         }
 199         return nullable;
 200     }
 201
 202     @CheckReturnValue
 203     private IllegalArgumentException childNullException(final QName child, final String message, final Object... args) {
 204         final QNameModule module = child.getModule();
 205         if (!factory().getRuntimeContext().getEffectiveModelContext().findModule(module).isPresent()) {
 206             return new MissingSchemaException("Module " + module + " is not present in current schema context.");
 207         }
 208         return new IncorrectNestingException(message, args);
 209     }
 210
 211     @CheckReturnValue
 212     private @NonNull IllegalArgumentException childNullException(final Class<?> childClass, final String message,
 213             final Object... args) {
 214         return childNullException(factory().getRuntimeContext(), childClass, message, args);
 215     }
 216
 217     @CheckReturnValue
 218     private static @NonNull IllegalArgumentException childNullException(final BindingRuntimeContext runtimeContext,
 219             final Class<?> childClass, final String message, final Object... args) {
 220         final CompositeRuntimeType schema;
 221         if (Augmentation.class.isAssignableFrom(childClass)) {
 222             schema = runtimeContext.getAugmentationDefinition(childClass.asSubclass(Augmentation.class));
 223         } else {
 224             schema = runtimeContext.getSchemaDefinition(childClass);
 225         }
 226         if (schema == null) {
 227             return new MissingSchemaForClassException(childClass);
 228         }
 229
 230         try {
 231             runtimeContext.loadClass(Type.of(childClass));
 232         } catch (final ClassNotFoundException e) {
 233             return new MissingClassInLoadingStrategyException(
 234                 "User supplied class " + childClass.getName() + " is not available in " + runtimeContext, e);
 235         }
 236
 237         return new IncorrectNestingException(message, args);
 238     }
 239
 240     private static QName extractName(final YangInstanceIdentifier.PathArgument child) {
 241         if (child instanceof AugmentationIdentifier) {
 242             final Set<QName> children = ((AugmentationIdentifier) child).getPossibleChildNames();
 243             checkArgument(!children.isEmpty(), "Augmentation without childs must not be used in data");
 244             return children.iterator().next();
 245         }
 246         return child.getNodeType();
 247     }
 248
 249     final DataObjectSerializer eventStreamSerializer() {
 250         final DataObjectSerializer existing = (DataObjectSerializer) EVENT_STREAM_SERIALIZER.getAcquire(this);
 251         return existing != null ? existing : loadEventStreamSerializer();
 252     }
 253
 254     // Split out to aid inlining
 255     private DataObjectSerializer loadEventStreamSerializer() {
 256         final DataObjectSerializer loaded = factory().getEventStreamSerializer(getBindingClass());
 257         final Object witness = EVENT_STREAM_SERIALIZER.compareAndExchangeRelease(this, null, loaded);
 258         return witness == null ? loaded : (DataObjectSerializer) witness;
 259     }
 260
 261     @Override
 262     public NormalizedNode serialize(final D data) {
 263         final NormalizedNodeResult result = new NormalizedNodeResult();
 264         // We create DOM stream writer which produces normalized nodes
 265         final NormalizedNodeStreamWriter domWriter = ImmutableNormalizedNodeStreamWriter.from(result);
 266         writeAsNormalizedNode(data, domWriter);
 267         return result.getResult();
 268     }
 269
 270     @Override
 271     public void writeAsNormalizedNode(final D data, final NormalizedNodeStreamWriter writer) {
 272         try {
 273             eventStreamSerializer().serialize(data, new BindingToNormalizedStreamWriter(this, writer));
 274         } catch (final IOException e) {
 275             throw new IllegalStateException("Failed to serialize Binding DTO",e);
 276         }
 277     }
 278
 279     static final <T extends NormalizedNode> @NonNull T checkDataArgument(final @NonNull Class<T> expectedType,
 280             final NormalizedNode data) {
 281         try {
 282             return expectedType.cast(requireNonNull(data));
 283         } catch (ClassCastException e) {
 284             throw new IllegalArgumentException("Expected " + expectedType.getSimpleName(), e);
 285         }
 286     }
 287
 288     // FIXME: MDSAL-780 replace this method with BindingRuntimeTypes-driven logic
 289     static final Optional<Class<? extends DataContainer>> getYangModeledReturnType(final Method method,
 290             final String prefix) {
 291         final String methodName = method.getName();
 292         if ("getClass".equals(methodName) || !methodName.startsWith(prefix) || method.getParameterCount() > 0) {
 293             return Optional.empty();
 294         }
 295
 296         final Class<?> returnType = method.getReturnType();
 297         if (DataContainer.class.isAssignableFrom(returnType)) {
 298             return optionalDataContainer(returnType);
 299         } else if (List.class.isAssignableFrom(returnType)) {
 300             return getYangModeledReturnType(method, 0);
 301         } else if (Map.class.isAssignableFrom(returnType)) {
 302             return getYangModeledReturnType(method, 1);
 303         }
 304         return Optional.empty();
 305     }
 306
 307     @SuppressWarnings("checkstyle:illegalCatch")
 308     private static Optional<Class<? extends DataContainer>> getYangModeledReturnType(final Method method,
 309             final int parameterOffset) {
 310         try {
 311             return ClassLoaderUtils.callWithClassLoader(method.getDeclaringClass().getClassLoader(),
 312                 () -> genericParameter(method.getGenericReturnType(), parameterOffset)
 313                     .flatMap(result -> result instanceof Class ? optionalCast((Class<?>) result) : Optional.empty()));
 314         } catch (Exception e) {
 315             /*
 316              * It is safe to log this this exception on debug, since this
 317              * method should not fail. Only failures are possible if the
 318              * runtime / backing.
 319              */
 320             LOG.debug("Unable to find YANG modeled return type for {}", method, e);
 321         }
 322         return Optional.empty();
 323     }
 324
 325     private static Optional<java.lang.reflect.Type> genericParameter(final java.lang.reflect.Type type,
 326             final int offset) {
 327         if (type instanceof ParameterizedType parameterized) {
 328             final var parameters = parameterized.getActualTypeArguments();
 329             if (parameters.length > offset) {
 330                 return Optional.of(parameters[offset]);
 331             }
 332         }
 333         return Optional.empty();
 334     }
 335
 336     private static Optional<Class<? extends DataContainer>> optionalCast(final Class<?> type) {
 337         return DataContainer.class.isAssignableFrom(type) ? optionalDataContainer(type) : Optional.empty();
 338     }
 339
 340
 341     // FIXME: MDSAL-780: remove this method
 342     static final Optional<Class<? extends DataContainer>> optionalDataContainer(final Class<?> type) {
 343         return Optional.of(type.asSubclass(DataContainer.class));
 344     }
 345
 346
 347     /**
 348      * Determines if two augmentation classes or case classes represents same data.
 349      *
 350      * <p>
 351      * Two augmentations or cases could be substituted only if and if:
 352      * <ul>
 353      *   <li>Both implements same interfaces</li>
 354      *   <li>Both have same children</li>
 355      *   <li>If augmentations: Both have same augmentation target class. Target class was generated for data node in a
 356      *       grouping.</li>
 357      *   <li>If cases: Both are from same choice. Choice class was generated for data node in grouping.</li>
 358      * </ul>
 359      *
 360      * <p>
 361      * <b>Explanation:</b>
 362      * Binding Specification reuses classes generated for groupings as part of normal data tree, this classes from
 363      * grouping could be used at various locations and user may not be aware of it and may use incorrect case or
 364      * augmentation in particular subtree (via copy constructors, etc).
 365      *
 366      * @param potential Class which is potential substitution
 367      * @param target Class which should be used at particular subtree
 368      * @return true if and only if classes represents same data.
 369      * @throws NullPointerException if any argument is {@code null}
 370      */
 371     // FIXME: MDSAL-785: this really should live in BindingRuntimeTypes and should not be based on reflection. The only
 372     //                   user is binding-dom-codec and the logic could easily be performed on GeneratedType instead. For
 373     //                   a particular world this boils down to a matrix, which can be calculated either on-demand or
 374     //                   when we create BindingRuntimeTypes. Achieving that will bring us one step closer to being able
 375     //                   to have a pre-compiled Binding Runtime.
 376     @SuppressWarnings({ "rawtypes", "unchecked" })
 377     static boolean isSubstitutionFor(final Class potential, final Class target) {
 378         Set<Class> subImplemented = new HashSet<>(Arrays.asList(potential.getInterfaces()));
 379         Set<Class> targetImplemented = new HashSet<>(Arrays.asList(target.getInterfaces()));
 380         if (!subImplemented.equals(targetImplemented)) {
 381             return false;
 382         }
 383         if (Augmentation.class.isAssignableFrom(potential)
 384                 && !BindingReflections.findAugmentationTarget(potential).equals(
 385                         BindingReflections.findAugmentationTarget(target))) {
 386             return false;
 387         }
 388         for (Method potentialMethod : potential.getMethods()) {
 389             if (Modifier.isStatic(potentialMethod.getModifiers())) {
 390                 // Skip any static methods, as we are not interested in those
 391                 continue;
 392             }
 393
 394             try {
 395                 Method targetMethod = target.getMethod(potentialMethod.getName(), potentialMethod.getParameterTypes());
 396                 if (!potentialMethod.getReturnType().equals(targetMethod.getReturnType())) {
 397                     return false;
 398                 }
 399             } catch (NoSuchMethodException e) {
 400                 // Counterpart method is missing, so classes could not be substituted.
 401                 return false;
 402             } catch (SecurityException e) {
 403                 throw new IllegalStateException("Could not compare methods", e);
 404             }
 405         }
 406         return true;
 407     }
 408 }