Switch NormalizedNode->Binding codegen to ByteBuddy
[mdsal.git] / binding / mdsal-binding-dom-codec / src / main / java / org / opendaylight / mdsal / binding / dom / codec / impl / OpaqueNodeCodecContext.java
1 /*
2  * Copyright (c) 2019 PANTHEON.tech, s.r.o. 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 com.google.common.base.Verify.verify;
12 import static java.util.Objects.requireNonNull;
13
14 import com.google.common.base.Throwables;
15 import java.lang.invoke.MethodHandle;
16 import java.lang.invoke.MethodHandles;
17 import java.lang.invoke.MethodType;
18 import java.lang.reflect.Constructor;
19 import java.lang.reflect.Method;
20 import javax.xml.transform.dom.DOMSource;
21 import net.bytebuddy.ByteBuddy;
22 import net.bytebuddy.dynamic.DynamicType.Builder;
23 import net.bytebuddy.jar.asm.Opcodes;
24 import org.eclipse.jdt.annotation.NonNull;
25 import org.opendaylight.mdsal.binding.dom.codec.api.BindingOpaqueObjectCodecTreeNode;
26 import org.opendaylight.mdsal.binding.dom.codec.loader.CodecClassLoader;
27 import org.opendaylight.mdsal.binding.dom.codec.loader.CodecClassLoader.GeneratorResult;
28 import org.opendaylight.yangtools.concepts.Codec;
29 import org.opendaylight.yangtools.yang.binding.OpaqueData;
30 import org.opendaylight.yangtools.yang.binding.OpaqueObject;
31 import org.opendaylight.yangtools.yang.data.api.schema.AnyXmlNode;
32 import org.opendaylight.yangtools.yang.data.api.schema.ForeignDataNode;
33 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
34 import org.opendaylight.yangtools.yang.data.impl.schema.Builders;
35 import org.opendaylight.yangtools.yang.model.api.AnyXmlSchemaNode;
36 import org.opendaylight.yangtools.yang.model.api.DataSchemaNode;
37
38 abstract class OpaqueNodeCodecContext<T extends OpaqueObject<T>> extends ValueNodeCodecContext
39         implements BindingOpaqueObjectCodecTreeNode<T> {
40     static final class AnyXml<T extends OpaqueObject<T>> extends OpaqueNodeCodecContext<T> {
41         AnyXml(final AnyXmlSchemaNode schema, final Method getter, final Class<T> bindingClass,
42                 final CodecClassLoader loader) {
43             super(schema, getter, bindingClass, loader);
44         }
45
46         @Override
47         ForeignDataNode<?, ?> serializedData(final OpaqueData<?> opaqueData) {
48             final Class<?> model = opaqueData.getObjectModel();
49             verify(DOMSource.class.isAssignableFrom(model), "Cannot just yet support object model %s", model);
50             return Builders.anyXmlBuilder().withNodeIdentifier(getDomPathArgument())
51                     .withValue((DOMSource) opaqueData.getData()).build();
52         }
53     }
54
55     private static final MethodType CONSTRUCTOR_TYPE = MethodType.methodType(OpaqueObject.class,
56         OpaqueData.class);
57     @SuppressWarnings("rawtypes")
58     private static final Builder<CodecOpaqueObject> TEMPLATE = new ByteBuddy().subclass(CodecOpaqueObject.class)
59             .modifiers(Opcodes.ACC_PUBLIC | Opcodes.ACC_FINAL | Opcodes.ACC_SYNTHETIC);
60
61     private final Codec<Object, Object> valueCodec = new Codec<Object, Object>() {
62         @Override
63         public Object serialize(final Object input) {
64             checkArgument(bindingClass.isInstance(input), "Unexpected input %s", input);
65             // FIXME: this works for DOMSource only, for generalization we need to pass down the object model, too
66             final OpaqueData<?> opaqueData = bindingClass.cast(input).getValue();
67             final Object data = opaqueData.getData();
68             checkArgument(data instanceof DOMSource, "Unexpected data %s", data);
69             return data;
70         }
71
72         @Override
73         public Object deserialize(final Object input) {
74             checkArgument(input instanceof NormalizedNode, "Unexpected input %s", input);
75             return OpaqueNodeCodecContext.this.deserializeObject((NormalizedNode<?, ?>) input);
76         }
77     };
78
79     private final MethodHandle proxyConstructor;
80     private final @NonNull Class<T> bindingClass;
81
82     OpaqueNodeCodecContext(final DataSchemaNode schema, final Method getter, final Class<T> bindingClass,
83             final CodecClassLoader loader) {
84         super(schema, getter, null);
85         this.bindingClass = requireNonNull(bindingClass);
86         proxyConstructor = createImpl(loader, bindingClass);
87     }
88
89     @Override
90     public final Class<T> getBindingClass() {
91         return bindingClass;
92     }
93
94     @Override
95     public final T deserialize(final NormalizedNode<?, ?> data) {
96         checkArgument(data instanceof ForeignDataNode, "Unexpected value %s", data);
97         final ForeignDataNode<?, ?> foreignData = (ForeignDataNode<?, ?>) data;
98         // Streaming cannot support anything but DOMSource-based AnyxmlNodes.
99         verify(foreignData instanceof AnyXmlNode, "Variable node %s not supported yet", foreignData);
100
101         return bindingClass.cast(createBindingProxy(new ForeignOpaqueData<>(foreignData)));
102     }
103
104     @Override
105     public final ForeignDataNode<?, ?> serialize(final T data) {
106         final OpaqueData<?> opaqueData = data.getValue();
107         return opaqueData instanceof ForeignOpaqueData ? ((ForeignOpaqueData<?>) opaqueData).domData()
108                 : serializedData(opaqueData);
109     }
110
111     @Override
112     protected final @NonNull Object deserializeObject(final NormalizedNode<?, ?> normalizedNode) {
113         return deserialize(normalizedNode);
114     }
115
116     @Override
117     Codec<Object, Object> getValueCodec() {
118         return valueCodec;
119     }
120
121     abstract @NonNull ForeignDataNode<?, ?> serializedData(OpaqueData<?> opaqueData);
122
123     @SuppressWarnings("checkstyle:illegalCatch")
124     private OpaqueObject<?> createBindingProxy(final OpaqueData<?> data) {
125         try {
126             return (OpaqueObject<?>) proxyConstructor.invokeExact(data);
127         } catch (final Throwable e) {
128             Throwables.throwIfUnchecked(e);
129             throw new IllegalStateException(e);
130         }
131     }
132
133     private static MethodHandle createImpl(final CodecClassLoader rootLoader, final Class<?> bindingClass) {
134         final Class<?> proxyClass = rootLoader.generateClass(bindingClass, "codecImpl",
135             (loader, fqcn, bindingInterface) -> GeneratorResult.of(TEMPLATE
136                 .name(fqcn)
137                 .implement(bindingInterface)
138                 .make()));
139
140         Constructor<?> ctor;
141         try {
142             ctor = proxyClass.getDeclaredConstructor(OpaqueData.class);
143         } catch (NoSuchMethodException e) {
144             throw new LinkageError("Failed to acquire constructor for prototype " + proxyClass, e);
145         }
146         try {
147             return MethodHandles.publicLookup().unreflectConstructor(ctor).asType(CONSTRUCTOR_TYPE);
148         } catch (IllegalAccessException e) {
149             throw new LinkageError("Failed to access constructor for prototype " + proxyClass, e);
150         }
151     }
152 }