Merge "Fix for Bug 3"
[controller.git] / opendaylight / md-sal / sal-rest-connector / src / main / java / org / opendaylight / controller / sal / rest / impl / XmlMapper.java
1 package org.opendaylight.controller.sal.rest.impl;
2
3 import javax.activation.UnsupportedDataTypeException;
4
5 import org.opendaylight.yangtools.yang.data.api.CompositeNode;
6 import org.opendaylight.yangtools.yang.data.api.codec.LeafrefCodec;
7 import org.opendaylight.yangtools.yang.data.impl.codec.TypeDefinitionAwareCodec;
8 import org.opendaylight.yangtools.yang.data.impl.codec.xml.XmlCodecProvider;
9 import org.opendaylight.yangtools.yang.data.impl.codec.xml.XmlDocumentUtils;
10 import org.opendaylight.yangtools.yang.model.api.DataNodeContainer;
11 import org.opendaylight.yangtools.yang.model.api.TypeDefinition;
12 import org.opendaylight.yangtools.yang.model.api.type.LeafrefTypeDefinition;
13 import org.opendaylight.yangtools.yang.model.util.Leafref;
14 import org.w3c.dom.Document;
15
16 import com.google.common.base.Optional;
17
18 public class XmlMapper {
19     private static final LeafrefCodecImpl LEAFREF_DEFAULT_CODEC = new LeafrefCodecImpl(
20             Optional.<LeafrefTypeDefinition> absent());
21
22     private static class LeafrefCodecImpl extends TypeDefinitionAwareCodec<Object, LeafrefTypeDefinition> implements
23             LeafrefCodec<String> {
24
25         protected LeafrefCodecImpl(Optional<LeafrefTypeDefinition> typeDef) {
26             super(typeDef, Object.class);
27         }
28
29         @Override
30         public String serialize(Object data) {
31             return String.valueOf(data);
32         }
33
34         @Override
35         public Object deserialize(String data) {
36             return data;
37         }
38     }
39
40     private static class XmlCodecProviderImpl implements XmlCodecProvider {
41         @Override
42         public TypeDefinitionAwareCodec<Object, ? extends TypeDefinition<?>> codecFor(TypeDefinition<?> baseType) {
43             TypeDefinitionAwareCodec<Object, ? extends TypeDefinition<?>> codec = TypeDefinitionAwareCodec
44                     .from(baseType);
45
46             if (codec == null) {
47                 if (baseType instanceof Leafref) {
48                     return LEAFREF_DEFAULT_CODEC;
49                 }
50             }
51             return codec;
52         }
53     }
54
55     private static final XmlCodecProvider XML_CODEC_PROVIDER_IMPL = new XmlCodecProviderImpl();
56
57     public Document write(CompositeNode data, DataNodeContainer schema) throws UnsupportedDataTypeException {
58         return XmlDocumentUtils.toDocument(data, schema, XML_CODEC_PROVIDER_IMPL);
59     }
60 }