2 * Copyright (c) 2019 PANTHEON.tech, s.r.o. and others. All rights reserved.
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
8 package org.opendaylight.mdsal.binding.dom.codec.impl;
10 import static com.google.common.base.Preconditions.checkArgument;
12 import com.google.common.annotations.Beta;
13 import com.google.common.collect.ImmutableClassToInstanceMap;
14 import java.io.IOException;
15 import java.lang.reflect.InvocationHandler;
16 import java.lang.reflect.Proxy;
17 import java.util.Collection;
18 import java.util.List;
20 import java.util.Map.Entry;
21 import org.opendaylight.mdsal.binding.dom.codec.util.AugmentationReader;
22 import org.opendaylight.mdsal.binding.spec.reflect.BindingReflections;
23 import org.opendaylight.yangtools.yang.binding.Augmentable;
24 import org.opendaylight.yangtools.yang.binding.Augmentation;
25 import org.opendaylight.yangtools.yang.binding.BindingSerializer;
26 import org.opendaylight.yangtools.yang.binding.BindingStreamEventWriter;
27 import org.opendaylight.yangtools.yang.binding.DataContainer;
28 import org.opendaylight.yangtools.yang.binding.DataObject;
29 import org.opendaylight.yangtools.yang.binding.DataObjectSerializer;
30 import org.opendaylight.yangtools.yang.binding.DataObjectSerializerImplementation;
31 import org.opendaylight.yangtools.yang.binding.DataObjectSerializerRegistry;
32 import org.opendaylight.yangtools.yang.binding.Identifiable;
33 import org.slf4j.Logger;
34 import org.slf4j.LoggerFactory;
37 * Base superclass for all concrete streamers, that is objects which are able to turn a concrete DataObject into a
40 * @param <T> DataObject type
43 public abstract class DataObjectStreamer<T extends DataObject> implements DataObjectSerializerImplementation {
44 private static final Logger LOG = LoggerFactory.getLogger(DataObjectStreamer.class);
46 protected DataObjectStreamer() {
50 protected static final void streamAnyxml(final BindingStreamEventWriter writer, final String localName,
51 final Object value) throws IOException {
53 writer.anyxmlNode(localName, value);
57 protected static final void streamAugmentations(final DataObjectSerializerRegistry registry,
58 final BindingStreamEventWriter writer, final Augmentable<?> obj) throws IOException {
59 final Map<Class<? extends Augmentation<?>>, Augmentation<?>> augmentations;
60 if (registry instanceof AugmentationReader) {
61 augmentations = ((AugmentationReader) registry).getAugmentations(obj);
62 } else if (Proxy.isProxyClass(obj.getClass())) {
63 augmentations = getFromProxy(obj);
65 augmentations = BindingReflections.getAugmentations(obj);
67 for (final Entry<Class<? extends Augmentation<?>>, Augmentation<?>> aug : augmentations.entrySet()) {
68 emitAugmentation(aug.getKey(), aug.getValue(), writer, registry);
72 private static Map<Class<? extends Augmentation<?>>, Augmentation<?>> getFromProxy(final Augmentable<?> obj) {
73 final InvocationHandler proxy = Proxy.getInvocationHandler(obj);
74 if (proxy instanceof AugmentationReader) {
75 return ((AugmentationReader) proxy).getAugmentations(obj);
77 return ImmutableClassToInstanceMap.of();
80 protected static final <C extends DataContainer> void streamChoice(final Class<C> choiceClass,
81 final DataObjectSerializerRegistry registry, final BindingStreamEventWriter writer, final C value)
84 final Class<? extends DataContainer> caseClass = value.implementedInterface();
85 writer.startChoiceNode(choiceClass, BindingStreamEventWriter.UNKNOWN_SIZE);
86 final DataObjectSerializer caseStreamer = registry.getSerializer(caseClass.asSubclass(DataObject.class));
87 if (caseStreamer != null && tryCache(writer, (DataObject) value)) {
88 caseStreamer.serialize((DataObject) value, writer);
90 LOG.warn("No serializer for case {} is available in registry {}", caseClass, registry);
97 protected static final <C extends DataObject> void streamContainer(final DataObjectStreamer<C> childStreamer,
98 final DataObjectSerializerRegistry registry, final BindingStreamEventWriter writer, final C value)
100 if (value != null && tryCache(writer, value)) {
101 childStreamer.serialize(registry, value, writer);
105 protected static final void streamLeaf(final BindingStreamEventWriter writer, final String localName,
106 final Object value) throws IOException {
108 writer.leafNode(localName, value);
112 protected static final void streamLeafList(final BindingStreamEventWriter writer, final String localName,
113 final List<?> value) throws IOException {
115 writer.startLeafSet(localName, value.size());
116 commonStreamLeafset(writer, value);
120 protected static final void streamOrderedLeafList(final BindingStreamEventWriter writer,
121 final String localName, final List<?> value) throws IOException {
123 writer.startOrderedLeafSet(localName, value.size());
124 commonStreamLeafset(writer, value);
128 protected static final <E extends DataObject> void streamList(final Class<E> childClass,
129 final DataObjectStreamer<E> childStreamer, final DataObjectSerializerRegistry registry,
130 final BindingStreamEventWriter writer, final List<? extends E> value) throws IOException {
132 writer.startUnkeyedList(childClass, value.size());
133 commonStreamList(registry, writer, childStreamer, value);
137 protected static final <E extends DataObject & Identifiable<?>> void streamMap(final Class<E> childClass,
138 final DataObjectStreamer<E> childStreamer, final DataObjectSerializerRegistry registry,
139 final BindingStreamEventWriter writer, final List<? extends E> value) throws IOException {
141 writer.startMapNode(childClass, value.size());
142 commonStreamList(registry, writer, childStreamer, value);
146 protected static final <E extends DataObject & Identifiable<?>> void streamOrderedMap(final Class<E> childClass,
147 final DataObjectStreamer<E> childStreamer, final DataObjectSerializerRegistry registry,
148 final BindingStreamEventWriter writer, final List<? extends E> value) throws IOException {
150 writer.startOrderedMapNode(childClass, value.size());
151 commonStreamList(registry, writer, childStreamer, value);
155 private static <E extends DataObject> void commonStreamList(final DataObjectSerializerRegistry registry,
156 final BindingStreamEventWriter writer, final DataObjectStreamer<E> childStreamer,
157 final Collection<? extends E> value) throws IOException {
159 for (E entry : value) {
160 if (tryCache(writer, entry)) {
161 childStreamer.serialize(registry, entry, writer);
168 private static void commonStreamLeafset(final BindingStreamEventWriter writer, final List<?> value)
170 for (Object entry : value) {
171 writer.leafSetEntryNode(entry);
176 @SuppressWarnings("rawtypes")
177 private static void emitAugmentation(final Class type, final Augmentation<?> value,
178 final BindingStreamEventWriter writer, final DataObjectSerializerRegistry registry) throws IOException {
180 * Binding Specification allowed to insert augmentation with null for
181 * value, which effectively could be used to remove augmentation
182 * from builder / DTO.
185 checkArgument(value instanceof DataObject);
186 @SuppressWarnings("unchecked")
187 final DataObjectSerializer serializer = registry.getSerializer(type);
188 if (serializer != null) {
189 serializer.serialize((DataObject) value, writer);
191 LOG.warn("DataObjectSerializer is not present for {} in registry {}", type, registry);
196 private static boolean tryCache(final BindingStreamEventWriter writer, final DataObject value) {
197 return writer instanceof BindingSerializer ? ((BindingSerializer) writer).serialize(value) == null : true;