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 java.io.IOException;
  16 import java.lang.invoke.MethodHandles;
  17 import java.lang.invoke.VarHandle;
  18 import java.lang.reflect.Method;
  19 import java.lang.reflect.ParameterizedType;
  20 import java.util.List;
  21 import java.util.Map;
  22 import java.util.Optional;
  23 import java.util.Set;
  24 import org.eclipse.jdt.annotation.NonNull;
  25 import org.eclipse.jdt.annotation.Nullable;
  26 import org.opendaylight.mdsal.binding.dom.codec.api.BindingDataObjectCodecTreeNode;
  27 import org.opendaylight.mdsal.binding.dom.codec.api.BindingNormalizedNodeCachingCodec;
  28 import org.opendaylight.mdsal.binding.dom.codec.api.BindingStreamEventWriter;
  29 import org.opendaylight.mdsal.binding.dom.codec.api.IncorrectNestingException;
  30 import org.opendaylight.mdsal.binding.dom.codec.api.MissingClassInLoadingStrategyException;
  31 import org.opendaylight.mdsal.binding.dom.codec.api.MissingSchemaException;
  32 import org.opendaylight.mdsal.binding.dom.codec.api.MissingSchemaForClassException;
  33 import org.opendaylight.mdsal.binding.model.api.Type;
  34 import org.opendaylight.mdsal.binding.runtime.api.BindingRuntimeContext;
  35 import org.opendaylight.mdsal.binding.runtime.api.CompositeRuntimeType;
  36 import org.opendaylight.mdsal.binding.runtime.api.RuntimeTypeContainer;
  37 import org.opendaylight.yangtools.util.ClassLoaderUtils;
  38 import org.opendaylight.yangtools.yang.binding.Augmentation;
  39 import org.opendaylight.yangtools.yang.binding.BindingObject;
  40 import org.opendaylight.yangtools.yang.binding.DataContainer;
  41 import org.opendaylight.yangtools.yang.binding.DataObject;
  42 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier.PathArgument;
  43 import org.opendaylight.yangtools.yang.common.QName;
  44 import org.opendaylight.yangtools.yang.common.QNameModule;
  45 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
  46 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.AugmentationIdentifier;
  47 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
  48 import org.opendaylight.yangtools.yang.data.api.schema.stream.NormalizedNodeStreamWriter;
  49 import org.opendaylight.yangtools.yang.data.impl.schema.ImmutableNormalizedNodeStreamWriter;
  50 import org.opendaylight.yangtools.yang.data.impl.schema.NormalizedNodeResult;
  51 import org.slf4j.Logger;
  52 import org.slf4j.LoggerFactory;
  53
  54 abstract class DataContainerCodecContext<D extends DataObject, T extends RuntimeTypeContainer> extends NodeCodecContext
  55         implements BindingDataObjectCodecTreeNode<D>  {
  56     private static final Logger LOG = LoggerFactory.getLogger(DataContainerCodecContext.class);
  57     private static final VarHandle EVENT_STREAM_SERIALIZER;
  58
  59     static {
  60         try {
  61             EVENT_STREAM_SERIALIZER = MethodHandles.lookup().findVarHandle(DataContainerCodecContext.class,
  62                 "eventStreamSerializer", DataObjectSerializer.class);
  63         } catch (NoSuchFieldException | IllegalAccessException e) {
  64             throw new ExceptionInInitializerError(e);
  65         }
  66     }
  67
  68     private final @NonNull DataContainerCodecPrototype<T> prototype;
  69
  70     // Accessed via a VarHandle
  71     @SuppressWarnings("unused")
  72     private volatile DataObjectSerializer eventStreamSerializer;
  73
  74     DataContainerCodecContext(final DataContainerCodecPrototype<T> prototype) {
  75         this.prototype = requireNonNull(prototype);
  76     }
  77
  78     public final @NonNull T getType() {
  79         return prototype.getType();
  80     }
  81
  82     @Override
  83     public final ChildAddressabilitySummary getChildAddressabilitySummary() {
  84         return prototype.getChildAddressabilitySummary();
  85     }
  86
  87     protected final QNameModule namespace() {
  88         return prototype.getNamespace();
  89     }
  90
  91     protected final CodecContextFactory factory() {
  92         return prototype.getFactory();
  93     }
  94
  95     @Override
  96     protected YangInstanceIdentifier.PathArgument getDomPathArgument() {
  97         return prototype.getYangArg();
  98     }
  99
 100     /**
 101      * Returns nested node context using supplied YANG Instance Identifier.
 102      *
 103      * @param arg Yang Instance Identifier Argument
 104      * @return Context of child
 105      * @throws IllegalArgumentException If supplied argument does not represent valid child.
 106      */
 107     @Override
 108     public abstract NodeCodecContext yangPathArgumentChild(YangInstanceIdentifier.PathArgument arg);
 109
 110     /**
 111      * Returns nested node context using supplied Binding Instance Identifier
 112      * and adds YANG instance identifiers to supplied list.
 113      *
 114      * @param arg Binding Instance Identifier Argument
 115      * @return Context of child or null if supplied {@code arg} does not represent valid child.
 116      * @throws IllegalArgumentException If supplied argument does not represent valid child.
 117      */
 118     @Override
 119     public DataContainerCodecContext<?, ?> bindingPathArgumentChild(final PathArgument arg,
 120             final List<YangInstanceIdentifier.PathArgument> builder) {
 121         final DataContainerCodecContext<?,?> child = streamChild(arg.getType());
 122         if (builder != null) {
 123             child.addYangPathArgument(arg,builder);
 124         }
 125         return child;
 126     }
 127
 128     /**
 129      * Returns deserialized Binding Path Argument from YANG instance identifier.
 130      */
 131     protected PathArgument getBindingPathArgument(final YangInstanceIdentifier.PathArgument domArg) {
 132         return bindingArg();
 133     }
 134
 135     protected final PathArgument bindingArg() {
 136         return prototype.getBindingArg();
 137     }
 138
 139     @SuppressWarnings("unchecked")
 140     @Override
 141     public final Class<D> getBindingClass() {
 142         return Class.class.cast(prototype.getBindingClass());
 143     }
 144
 145     @Override
 146     public abstract <C extends DataObject> DataContainerCodecContext<C, ?> streamChild(Class<C> childClass);
 147
 148     /**
 149      * Returns child context as if it was walked by {@link BindingStreamEventWriter}. This means that to enter case, one
 150      * must issue getChild(ChoiceClass).getChild(CaseClass).
 151      *
 152      * @param childClass child class
 153      * @return Context of child or Optional.empty is supplied class is not applicable in context.
 154      */
 155     @Override
 156     public abstract <C extends DataObject> Optional<DataContainerCodecContext<C,?>> possibleStreamChild(
 157             Class<C> childClass);
 158
 159     @Override
 160     public String toString() {
 161         return getClass().getSimpleName() + " [" + prototype.getBindingClass() + "]";
 162     }
 163
 164     @Override
 165     public BindingNormalizedNodeCachingCodec<D> createCachingCodec(
 166             final ImmutableCollection<Class<? extends BindingObject>> cacheSpecifier) {
 167         if (cacheSpecifier.isEmpty()) {
 168             return new NonCachingCodec<>(this);
 169         }
 170         return new CachingNormalizedNodeCodec<>(this, ImmutableSet.copyOf(cacheSpecifier));
 171     }
 172
 173     @NonNull BindingStreamEventWriter createWriter(final NormalizedNodeStreamWriter domWriter) {
 174         return BindingToNormalizedStreamWriter.create(this, domWriter);
 175     }
 176
 177     protected final <V> @NonNull V childNonNull(final @Nullable V nullable,
 178             final YangInstanceIdentifier.PathArgument child, final String message, final Object... args) {
 179         if (nullable == null) {
 180             throw childNullException(extractName(child), message, args);
 181         }
 182         return nullable;
 183     }
 184
 185     protected final <V> @NonNull V childNonNull(final @Nullable V nullable, final QName child, final String message,
 186             final Object... args) {
 187         if (nullable == null) {
 188             throw childNullException(child, message, args);
 189         }
 190         return nullable;
 191     }
 192
 193     protected final <V> @NonNull V childNonNull(final @Nullable V nullable, final Class<?> childClass,
 194             final String message, final Object... args) {
 195         if (nullable == null) {
 196             throw childNullException(childClass, message, args);
 197         }
 198         return nullable;
 199     }
 200
 201     private IllegalArgumentException childNullException(final QName child, final String message, final Object... args) {
 202         final QNameModule module = child.getModule();
 203         if (!factory().getRuntimeContext().getEffectiveModelContext().findModule(module).isPresent()) {
 204             throw new MissingSchemaException("Module " + module + " is not present in current schema context.");
 205         }
 206         throw IncorrectNestingException.create(message, args);
 207     }
 208
 209     private IllegalArgumentException childNullException(final Class<?> childClass, final String message,
 210             final Object... args) {
 211         final BindingRuntimeContext runtimeContext = factory().getRuntimeContext();
 212         final CompositeRuntimeType schema;
 213         if (Augmentation.class.isAssignableFrom(childClass)) {
 214             schema = runtimeContext.getAugmentationDefinition(childClass.asSubclass(Augmentation.class));
 215         } else {
 216             schema = runtimeContext.getSchemaDefinition(childClass);
 217         }
 218         if (schema == null) {
 219             throw new MissingSchemaForClassException(childClass);
 220         }
 221
 222         try {
 223             runtimeContext.loadClass(Type.of(childClass));
 224         } catch (final ClassNotFoundException e) {
 225             throw new MissingClassInLoadingStrategyException(
 226                 "User supplied class " + childClass.getName() + " is not available in " + runtimeContext, e);
 227         }
 228
 229         throw IncorrectNestingException.create(message, args);
 230     }
 231
 232     private static QName extractName(final YangInstanceIdentifier.PathArgument child) {
 233         if (child instanceof AugmentationIdentifier) {
 234             final Set<QName> children = ((AugmentationIdentifier) child).getPossibleChildNames();
 235             checkArgument(!children.isEmpty(), "Augmentation without childs must not be used in data");
 236             return children.iterator().next();
 237         }
 238         return child.getNodeType();
 239     }
 240
 241     final DataObjectSerializer eventStreamSerializer() {
 242         final DataObjectSerializer existing = (DataObjectSerializer) EVENT_STREAM_SERIALIZER.getAcquire(this);
 243         return existing != null ? existing : loadEventStreamSerializer();
 244     }
 245
 246     // Split out to aid inlining
 247     private DataObjectSerializer loadEventStreamSerializer() {
 248         final DataObjectSerializer loaded = factory().getEventStreamSerializer(getBindingClass());
 249         final Object witness = EVENT_STREAM_SERIALIZER.compareAndExchangeRelease(this, null, loaded);
 250         return witness == null ? loaded : (DataObjectSerializer) witness;
 251     }
 252
 253     @Override
 254     public NormalizedNode serialize(final D data) {
 255         final NormalizedNodeResult result = new NormalizedNodeResult();
 256         // We create DOM stream writer which produces normalized nodes
 257         final NormalizedNodeStreamWriter domWriter = ImmutableNormalizedNodeStreamWriter.from(result);
 258         writeAsNormalizedNode(data, domWriter);
 259         return result.getResult();
 260     }
 261
 262     @Override
 263     public void writeAsNormalizedNode(final D data, final NormalizedNodeStreamWriter writer) {
 264         try {
 265             eventStreamSerializer().serialize(data, createWriter(writer));
 266         } catch (final IOException e) {
 267             throw new IllegalStateException("Failed to serialize Binding DTO",e);
 268         }
 269     }
 270
 271     static final <T extends NormalizedNode> @NonNull T checkDataArgument(final @NonNull Class<T> expectedType,
 272             final NormalizedNode data) {
 273         try {
 274             return expectedType.cast(requireNonNull(data));
 275         } catch (ClassCastException e) {
 276             throw new IllegalArgumentException("Expected " + expectedType.getSimpleName(), e);
 277         }
 278     }
 279
 280     // FIXME: MDSAL-780 replace this method with BindingRuntimeTypes-driven logic
 281     static final Optional<Class<? extends DataContainer>> getYangModeledReturnType(final Method method,
 282             final String prefix) {
 283         final String methodName = method.getName();
 284         if ("getClass".equals(methodName) || !methodName.startsWith(prefix) || method.getParameterCount() > 0) {
 285             return Optional.empty();
 286         }
 287
 288         final Class<?> returnType = method.getReturnType();
 289         if (DataContainer.class.isAssignableFrom(returnType)) {
 290             return optionalDataContainer(returnType);
 291         } else if (List.class.isAssignableFrom(returnType)) {
 292             return getYangModeledReturnType(method, 0);
 293         } else if (Map.class.isAssignableFrom(returnType)) {
 294             return getYangModeledReturnType(method, 1);
 295         }
 296         return Optional.empty();
 297     }
 298
 299     @SuppressWarnings("checkstyle:illegalCatch")
 300     private static Optional<Class<? extends DataContainer>> getYangModeledReturnType(final Method method,
 301             final int parameterOffset) {
 302         try {
 303             return ClassLoaderUtils.callWithClassLoader(method.getDeclaringClass().getClassLoader(),
 304                 () -> genericParameter(method.getGenericReturnType(), parameterOffset)
 305                     .flatMap(result -> result instanceof Class ? optionalCast((Class<?>) result) : Optional.empty()));
 306         } catch (Exception e) {
 307             /*
 308              * It is safe to log this this exception on debug, since this
 309              * method should not fail. Only failures are possible if the
 310              * runtime / backing.
 311              */
 312             LOG.debug("Unable to find YANG modeled return type for {}", method, e);
 313         }
 314         return Optional.empty();
 315     }
 316
 317     private static Optional<java.lang.reflect.Type> genericParameter(final java.lang.reflect.Type type,
 318             final int offset) {
 319         if (type instanceof ParameterizedType parameterized) {
 320             final var parameters = parameterized.getActualTypeArguments();
 321             if (parameters.length > offset) {
 322                 return Optional.of(parameters[offset]);
 323             }
 324         }
 325         return Optional.empty();
 326     }
 327
 328     private static Optional<Class<? extends DataContainer>> optionalCast(final Class<?> type) {
 329         return DataContainer.class.isAssignableFrom(type) ? optionalDataContainer(type) : Optional.empty();
 330     }
 331
 332
 333     // FIXME: MDSAL-780: remove this method
 334     static final Optional<Class<? extends DataContainer>> optionalDataContainer(final Class<?> type) {
 335         return Optional.of(type.asSubclass(DataContainer.class));
 336     }
 337 }