opendaylight/netconf/netconf-cli/src/main/java/org/opendaylight/controller/netconf/cli/writer/impl/NodeCliSerializerDispatcher.java

   1 /*
   2  * Copyright (c) 2013 Cisco Systems, Inc. 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.controller.netconf.cli.writer.impl;
   9
  10 import com.google.common.base.Preconditions;
  11 import com.google.common.collect.Iterables;
  12 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
  13 import org.opendaylight.yangtools.yang.data.api.schema.AugmentationNode;
  14 import org.opendaylight.yangtools.yang.data.api.schema.ChoiceNode;
  15 import org.opendaylight.yangtools.yang.data.api.schema.ContainerNode;
  16 import org.opendaylight.yangtools.yang.data.api.schema.DataContainerChild;
  17 import org.opendaylight.yangtools.yang.data.api.schema.LeafNode;
  18 import org.opendaylight.yangtools.yang.data.api.schema.LeafSetNode;
  19 import org.opendaylight.yangtools.yang.data.api.schema.MapNode;
  20 import org.opendaylight.yangtools.yang.data.api.schema.MixinNode;
  21 import org.opendaylight.yangtools.yang.data.impl.schema.transform.FromNormalizedNodeSerializerFactory;
  22 import org.opendaylight.yangtools.yang.data.impl.schema.transform.base.serializer.NodeSerializerDispatcher;
  23 import org.opendaylight.yangtools.yang.model.api.AugmentationSchema;
  24 import org.opendaylight.yangtools.yang.model.api.ChoiceSchemaNode;
  25 import org.opendaylight.yangtools.yang.model.api.ContainerSchemaNode;
  26 import org.opendaylight.yangtools.yang.model.api.LeafListSchemaNode;
  27 import org.opendaylight.yangtools.yang.model.api.LeafSchemaNode;
  28 import org.opendaylight.yangtools.yang.model.api.ListSchemaNode;
  29
  30 public class NodeCliSerializerDispatcher implements NodeSerializerDispatcher<String> {
  31     private final FromNormalizedNodeSerializerFactory<String> factory;
  32
  33     public NodeCliSerializerDispatcher(final FromNormalizedNodeSerializerFactory<String> factory) {
  34         this.factory = Preconditions.checkNotNull(factory);
  35     }
  36
  37     @Override
  38     public final Iterable<String> dispatchChildElement(final Object childSchema,
  39             final DataContainerChild<? extends YangInstanceIdentifier.PathArgument, ?> dataContainerChild) {
  40         if (dataContainerChild instanceof ContainerNode) {
  41             return onContainerNode(childSchema, dataContainerChild);
  42         } else if (dataContainerChild instanceof LeafNode<?>) {
  43             return onLeafNode(childSchema, dataContainerChild);
  44         } else if (dataContainerChild instanceof MixinNode) {
  45             if (dataContainerChild instanceof LeafSetNode<?>) {
  46                 return onLeafListNode(childSchema, dataContainerChild);
  47             } else if (dataContainerChild instanceof MapNode) {
  48                 return onListNode(childSchema, dataContainerChild);
  49             } else if (dataContainerChild instanceof ChoiceNode) {
  50                 return onChoiceNode(childSchema, dataContainerChild);
  51             } else if (dataContainerChild instanceof AugmentationNode) {
  52                 return onAugmentationSchema(childSchema, dataContainerChild);
  53             }
  54         }
  55         throw new IllegalArgumentException("Unable to serialize " + childSchema);
  56     }
  57
  58     private Iterable<String> onAugmentationSchema(final Object childSchema,
  59             final DataContainerChild<? extends YangInstanceIdentifier.PathArgument, ?> dataContainerChild) {
  60         checkSchemaCompatibility(childSchema, AugmentationSchema.class, dataContainerChild);
  61         return factory.getAugmentationNodeSerializer().serialize((AugmentationSchema) childSchema,
  62                 (AugmentationNode) dataContainerChild);
  63     }
  64
  65     private Iterable<String> onChoiceNode(final Object childSchema,
  66             final DataContainerChild<? extends YangInstanceIdentifier.PathArgument, ?> dataContainerChild) {
  67         checkSchemaCompatibility(childSchema, ChoiceSchemaNode.class, dataContainerChild);
  68         return factory.getChoiceNodeSerializer().serialize(
  69                 (ChoiceSchemaNode) childSchema, (ChoiceNode) dataContainerChild);
  70     }
  71
  72     private Iterable<String> onListNode(final Object childSchema,
  73             final DataContainerChild<? extends YangInstanceIdentifier.PathArgument, ?> dataContainerChild) {
  74         checkSchemaCompatibility(childSchema, ListSchemaNode.class, dataContainerChild);
  75         return factory.getMapNodeSerializer().serialize((ListSchemaNode) childSchema, (MapNode) dataContainerChild);
  76     }
  77
  78     private Iterable<String> onLeafListNode(final Object childSchema,
  79             final DataContainerChild<? extends YangInstanceIdentifier.PathArgument, ?> dataContainerChild) {
  80         checkSchemaCompatibility(childSchema, LeafListSchemaNode.class, dataContainerChild);
  81         return factory.getLeafSetNodeSerializer().serialize((LeafListSchemaNode) childSchema,
  82                 (LeafSetNode<?>) dataContainerChild);
  83     }
  84
  85     private Iterable<String> onLeafNode(final Object childSchema,
  86             final DataContainerChild<? extends YangInstanceIdentifier.PathArgument, ?> dataContainerChild) {
  87         checkSchemaCompatibility(childSchema, LeafSchemaNode.class, dataContainerChild);
  88         final Iterable<String> elements = factory.getLeafNodeSerializer().serialize((LeafSchemaNode) childSchema,
  89                 (LeafNode<?>) dataContainerChild);
  90         checkOnlyOneSerializedElement(elements, dataContainerChild);
  91         return elements;
  92     }
  93
  94     private static void checkOnlyOneSerializedElement(final Iterable<?> elements,
  95             final DataContainerChild<? extends YangInstanceIdentifier.PathArgument, ?> dataContainerChild) {
  96         final int size = Iterables.size(elements);
  97         Preconditions.checkArgument(size == 1,
  98                 "Unexpected count of elements for entry serialized from: %s, should be 1, was: %s", dataContainerChild,
  99                 size);
 100     }
 101
 102     private Iterable<String> onContainerNode(final Object childSchema,
 103             final DataContainerChild<? extends YangInstanceIdentifier.PathArgument, ?> dataContainerChild) {
 104         checkSchemaCompatibility(childSchema, ContainerSchemaNode.class, dataContainerChild);
 105
 106         final Iterable<String> elements = factory.getContainerNodeSerializer().serialize(
 107                 (ContainerSchemaNode) childSchema, (ContainerNode) dataContainerChild);
 108         checkOnlyOneSerializedElement(elements, dataContainerChild);
 109         return elements;
 110     }
 111
 112     private static void checkSchemaCompatibility(final Object childSchema, final Class<?> containerSchemaNodeClass,
 113             final DataContainerChild<? extends YangInstanceIdentifier.PathArgument, ?> dataContainerChild) {
 114         Preconditions.checkArgument(containerSchemaNodeClass.isAssignableFrom(childSchema.getClass()),
 115                 "Incompatible schema: %s with node: %s, expected: %s", childSchema, dataContainerChild,
 116                 containerSchemaNodeClass);
 117     }
 118 }