Rename CaseNodeCodecContext
[mdsal.git] / binding / mdsal-binding-dom-codec / src / main / java / org / opendaylight / mdsal / binding / dom / codec / impl / LeafNodeCodecContext.java
index 0baae812bf8edce53c3e9c316ebc8faa19fe6be0..bb3250c87e31755f475521d9356a0d13c3ff861e 100644 (file)
@@ -1,5 +1,5 @@
 /*
- * Copyright (c) 2014 Cisco Systems, Inc. and others.  All rights reserved.
+ * Copyright (c) 2019 PANTHEON.tech, s.r.o. and others.  All rights reserved.
  *
  * This program and the accompanying materials are made available under the
  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
@@ -9,92 +9,109 @@ package org.opendaylight.mdsal.binding.dom.codec.impl;
 
 import static java.util.Objects.requireNonNull;
 
-import java.lang.reflect.Method;
-import java.util.ArrayList;
-import java.util.Collection;
-import java.util.List;
-import java.util.Optional;
-import java.util.Set;
-import org.opendaylight.yangtools.concepts.Codec;
+import org.eclipse.jdt.annotation.NonNull;
+import org.opendaylight.mdsal.binding.dom.codec.api.BindingTypeObjectCodecTreeNode;
+import org.opendaylight.yangtools.yang.binding.TypeObject;
 import org.opendaylight.yangtools.yang.common.QName;
-import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifier;
-import org.opendaylight.yangtools.yang.data.api.schema.LeafNode;
-import org.opendaylight.yangtools.yang.data.api.schema.LeafSetEntryNode;
-import org.opendaylight.yangtools.yang.data.api.schema.LeafSetNode;
 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
 import org.opendaylight.yangtools.yang.data.impl.codec.TypeDefinitionAwareCodec;
+import org.opendaylight.yangtools.yang.data.impl.schema.ImmutableNodes;
 import org.opendaylight.yangtools.yang.model.api.DataSchemaNode;
+import org.opendaylight.yangtools.yang.model.api.EffectiveModelContext;
 import org.opendaylight.yangtools.yang.model.api.LeafSchemaNode;
 import org.opendaylight.yangtools.yang.model.api.Module;
 import org.opendaylight.yangtools.yang.model.api.ModuleImport;
-import org.opendaylight.yangtools.yang.model.api.SchemaContext;
 import org.opendaylight.yangtools.yang.model.api.TypeDefinition;
-import org.opendaylight.yangtools.yang.model.api.TypedDataSchemaNode;
 import org.opendaylight.yangtools.yang.model.api.type.IdentityrefTypeDefinition;
 
-// FIXME: MDSAL-436: this class should be specialized for Leaf and LeafSet
-final class LeafNodeCodecContext extends NodeCodecContext implements NodeContextSupplier {
-    private final NodeIdentifier yangIdentifier;
-    private final Codec<Object, Object> valueCodec;
-    private final Method getter;
-    private final TypedDataSchemaNode schema;
-    private final Object defaultObject;
-
-    LeafNodeCodecContext(final TypedDataSchemaNode schema, final Codec<Object, Object> codec, final Method getter,
-                final SchemaContext schemaContext) {
-        this.yangIdentifier = NodeIdentifier.create(schema.getQName());
-        this.valueCodec = requireNonNull(codec);
-        this.getter = getter;
-        this.schema = requireNonNull(schema);
-
-        this.defaultObject = createDefaultObject(schema, valueCodec, schemaContext);
+sealed class LeafNodeCodecContext extends ValueNodeCodecContext.WithCodec {
+    static final class OfTypeObject<T extends TypeObject> extends LeafNodeCodecContext
+            implements BindingTypeObjectCodecTreeNode<T> {
+        private final @NonNull Class<T> bindingClass;
+
+        OfTypeObject(final LeafSchemaNode schema, final ValueCodec<Object, Object> codec, final String getterName,
+                final EffectiveModelContext schemaContext, final Class<T> bindingClass) {
+            super(schema, codec, getterName, schemaContext);
+            this.bindingClass = requireNonNull(bindingClass);
+        }
+
+        @Override
+        public Class<T> getBindingClass() {
+            return bindingClass;
+        }
+
+        @Override
+        public T deserialize(final NormalizedNode data) {
+            return bindingClass.cast(deserializeObject(data));
+        }
+
+        @Override
+        public NormalizedNode serialize(final T data) {
+            return ImmutableNodes.leafNode(getDomPathArgument(), getValueCodec().serialize(data));
+        }
     }
 
-    private static Object createDefaultObject(final DataSchemaNode schema, final Codec<Object, Object> codec,
-                                              final SchemaContext schemaContext) {
-        if (schema instanceof LeafSchemaNode) {
-            Optional<? extends Object> defaultValue = ((LeafSchemaNode) schema).getType().getDefaultValue();
-            TypeDefinition<?> type = ((LeafSchemaNode) schema).getType();
-            if (defaultValue.isPresent()) {
-                if (type instanceof IdentityrefTypeDefinition) {
-                    return qnameDomValueFromString(codec, schema, (String) defaultValue.get(), schemaContext);
-                }
-                return domValueFromString(codec, type, defaultValue.get());
-            }
+    LeafNodeCodecContext(final LeafSchemaNode schema, final ValueCodec<Object, Object> codec, final String getterName,
+            final EffectiveModelContext schemaContext) {
+        super(schema, codec, getterName, createDefaultObject(schema, codec, schemaContext));
+    }
 
-            while (type.getBaseType() != null && !type.getDefaultValue().isPresent()) {
-                type = type.getBaseType();
+    static LeafNodeCodecContext of(final LeafSchemaNode schema, final ValueCodec<Object, Object> codec,
+            final String getterName, final Class<?> valueType, final EffectiveModelContext schemaContext) {
+        return TypeObject.class.isAssignableFrom(valueType)
+                ? new OfTypeObject<>(schema, codec, getterName, schemaContext, valueType.asSubclass(TypeObject.class))
+                        : new LeafNodeCodecContext(schema, codec, getterName, schemaContext);
+    }
+
+    @Override
+    protected Object deserializeObject(final NormalizedNode normalizedNode) {
+        return normalizedNode != null ? getValueCodec().deserialize(normalizedNode.body()) : null;
+    }
+
+    private static Object createDefaultObject(final LeafSchemaNode schema, final ValueCodec<Object, Object> codec,
+            final EffectiveModelContext schemaContext) {
+        var optDefaultValue = schema.getType().getDefaultValue();
+        TypeDefinition<?> type = schema.getType();
+        if (optDefaultValue.isPresent()) {
+            final var defaultValue = optDefaultValue.orElseThrow();
+            if (type instanceof IdentityrefTypeDefinition) {
+                return qnameDomValueFromString(codec, schema, (String) defaultValue, schemaContext);
             }
+            return domValueFromString(codec, type, defaultValue);
+        }
 
-            defaultValue = type.getDefaultValue();
-            if (defaultValue.isPresent()) {
-                if (type instanceof IdentityrefTypeDefinition) {
-                    return qnameDomValueFromString(codec, schema, (String) defaultValue.get(), schemaContext);
-                }
-                return domValueFromString(codec, type, defaultValue);
+        while (type.getBaseType() != null && type.getDefaultValue().isEmpty()) {
+            type = type.getBaseType();
+        }
+
+        optDefaultValue = type.getDefaultValue();
+        if (optDefaultValue.isPresent()) {
+            final var defaultValue = optDefaultValue.orElseThrow();
+            if (type instanceof IdentityrefTypeDefinition) {
+                return qnameDomValueFromString(codec, schema, (String) defaultValue, schemaContext);
             }
+            return domValueFromString(codec, type, defaultValue);
         }
         return null;
     }
 
-    private static Object qnameDomValueFromString(final Codec<Object, Object> codec, final DataSchemaNode schema,
-                                                  final String defaultValue, final SchemaContext schemaContext) {
+    private static Object qnameDomValueFromString(final ValueCodec<Object, Object> codec, final DataSchemaNode schema,
+            final String defaultValue, final EffectiveModelContext schemaContext) {
         int prefixEndIndex = defaultValue.indexOf(':');
         QName qname;
         if (prefixEndIndex != -1) {
             String defaultValuePrefix = defaultValue.substring(0, prefixEndIndex);
 
-            Module module = schemaContext.findModule(schema.getQName().getModule()).get();
+            Module module = schemaContext.findModule(schema.getQName().getModule()).orElseThrow();
             if (module.getPrefix().equals(defaultValuePrefix)) {
                 qname = QName.create(module.getQNameModule(), defaultValue.substring(prefixEndIndex + 1));
                 return codec.deserialize(qname);
             }
 
-            Set<ModuleImport> imports = module.getImports();
-            for (ModuleImport moduleImport : imports) {
+            for (ModuleImport moduleImport : module.getImports()) {
                 if (moduleImport.getPrefix().equals(defaultValuePrefix)) {
-                    Module importedModule = schemaContext.findModule(moduleImport.getModuleName(),
-                        moduleImport.getRevision()).get();
+                    Module importedModule = schemaContext.findModule(moduleImport.getModuleName().getLocalName(),
+                        moduleImport.getRevision()).orElseThrow();
                     qname = QName.create(importedModule.getQNameModule(), defaultValue.substring(prefixEndIndex + 1));
                     return codec.deserialize(qname);
                 }
@@ -106,7 +123,7 @@ final class LeafNodeCodecContext extends NodeCodecContext implements NodeContext
         return codec.deserialize(qname);
     }
 
-    private static Object domValueFromString(final Codec<Object, Object> codec, final TypeDefinition<?> type,
+    private static Object domValueFromString(final ValueCodec<Object, Object> codec, final TypeDefinition<?> type,
             final Object defaultValue) {
         TypeDefinitionAwareCodec<?, ?> typeDefAwareCodec = TypeDefinitionAwareCodec.from(type);
         if (typeDefAwareCodec != null) {
@@ -117,49 +134,4 @@ final class LeafNodeCodecContext extends NodeCodecContext implements NodeContext
         //                 getDefaultValue() implementation.
         return null;
     }
-
-    @Override
-    protected NodeIdentifier getDomPathArgument() {
-        return yangIdentifier;
-    }
-
-    protected Codec<Object, Object> getValueCodec() {
-        return valueCodec;
-    }
-
-    @Override
-    public NodeCodecContext get() {
-        return this;
-    }
-
-    Method getGetter() {
-        return getter;
-    }
-
-    @Override
-    protected Object deserializeObject(final NormalizedNode<?, ?> normalizedNode) {
-        if (normalizedNode instanceof LeafNode<?>) {
-            return valueCodec.deserialize(normalizedNode.getValue());
-        }
-        if (normalizedNode instanceof LeafSetNode<?>) {
-            @SuppressWarnings("unchecked")
-            final Collection<LeafSetEntryNode<Object>> domValues = ((LeafSetNode<Object>) normalizedNode).getValue();
-            final List<Object> result = new ArrayList<>(domValues.size());
-            for (final LeafSetEntryNode<Object> valueNode : domValues) {
-                result.add(valueCodec.deserialize(valueNode.getValue()));
-            }
-            return result;
-        }
-        return null;
-    }
-
-    @Override
-    public TypedDataSchemaNode getSchema() {
-        return schema;
-    }
-
-    @Override
-    Object defaultObject() {
-        return defaultObject;
-    }
 }