binding/mdsal-binding-dom-codec/src/main/java/org/opendaylight/mdsal/binding/dom/codec/impl/AugmentationCodecContext.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 com.google.common.base.Throwables;
  11 import com.google.common.collect.ImmutableSet;
  12 import java.lang.invoke.MethodHandle;
  13 import java.lang.invoke.MethodHandles;
  14 import java.lang.invoke.MethodType;
  15 import java.util.List;
  16 import java.util.Map;
  17 import org.opendaylight.mdsal.binding.dom.codec.api.BindingAugmentationCodecTreeNode;
  18 import org.opendaylight.mdsal.binding.runtime.api.AugmentRuntimeType;
  19 import org.opendaylight.yangtools.yang.binding.Augmentation;
  20 import org.opendaylight.yangtools.yang.binding.DataObject;
  21 import org.opendaylight.yangtools.yang.binding.DataObjectStep;
  22 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifier;
  23 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.PathArgument;
  24 import org.opendaylight.yangtools.yang.data.api.schema.DataContainerNode;
  25 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
  26
  27 final class AugmentationCodecContext<D extends DataObject & Augmentation<?>>
  28         extends AbstractDataObjectCodecContext<D, AugmentRuntimeType> implements BindingAugmentationCodecTreeNode<D> {
  29     private static final MethodType CONSTRUCTOR_TYPE = MethodType.methodType(void.class,
  30         AbstractDataObjectCodecContext.class, DataContainerNode.class);
  31     private static final MethodType DATAOBJECT_TYPE = MethodType.methodType(DataObject.class,
  32         AugmentationCodecContext.class, DataContainerNode.class);
  33
  34     private final MethodHandle proxyConstructor;
  35
  36     private AugmentationCodecContext(final AugmentationCodecPrototype prototype,
  37             final DataContainerAnalysis<AugmentRuntimeType> analysis) {
  38         super(prototype, analysis);
  39
  40         final var bindingClass = CodecDataObjectGenerator.generate(prototype.contextFactory().getLoader(),
  41             prototype.javaClass(), analysis.leafContexts, analysis.daoProperties, null);
  42
  43         final MethodHandle ctor;
  44         try {
  45             ctor = MethodHandles.publicLookup().findConstructor(bindingClass, CONSTRUCTOR_TYPE);
  46         } catch (NoSuchMethodException | IllegalAccessException e) {
  47             throw new LinkageError("Failed to find contructor for class " + bindingClass, e);
  48         }
  49
  50         proxyConstructor = ctor.asType(DATAOBJECT_TYPE);
  51     }
  52
  53     AugmentationCodecContext(final AugmentationCodecPrototype prototype) {
  54         this(prototype, new DataContainerAnalysis<>(prototype, CodecItemFactory.of()));
  55     }
  56
  57     @Override
  58     public PathArgument serializePathArgument(final DataObjectStep<?> step) {
  59         if (!bindingArg().equals(step)) {
  60             throw new IllegalArgumentException("Unexpected argument " + step);
  61         }
  62         return null;
  63     }
  64
  65     @Override
  66     public DataObjectStep<?> deserializePathArgument(final PathArgument arg) {
  67         if (arg != null) {
  68             throw new IllegalArgumentException("Unexpected argument " + arg);
  69         }
  70         return bindingArg();
  71     }
  72
  73     @SuppressWarnings("checkstyle:illegalCatch")
  74     @Override
  75     public D filterFrom(final DataContainerNode parentData) {
  76         for (var childArg : ((AugmentationCodecPrototype) prototype()).getChildArgs()) {
  77             if (parentData.childByArg(childArg) != null) {
  78                 try {
  79                     return (D) proxyConstructor.invokeExact(this, parentData);
  80                 } catch (final Throwable e) {
  81                     Throwables.throwIfUnchecked(e);
  82                     throw new IllegalStateException(e);
  83                 }
  84             }
  85         }
  86         return null;
  87     }
  88
  89     @Override
  90     public ImmutableSet<NodeIdentifier> childPathArguments() {
  91         return byYangKeySet();
  92     }
  93
  94     @Override
  95     protected Object deserializeObject(final NormalizedNode normalizedNode) {
  96         return filterFrom(checkDataArgument(DataContainerNode.class, normalizedNode));
  97     }
  98
  99     @Override
 100     void addYangPathArgument(final List<PathArgument> builder, final DataObjectStep<?> step) {
 101         // No-op
 102     }
 103
 104     @Override
 105     protected NodeIdentifier getDomPathArgument() {
 106         return null;
 107     }
 108
 109     @Override
 110     Map<Class<? extends Augmentation<?>>, Augmentation<?>> getAllAugmentationsFrom(final DataContainerNode data) {
 111         return Map.of();
 112     }
 113 }