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