yang/yang-data-util/src/main/java/org/opendaylight/yangtools/yang/data/util/OpaqueAnydataNodeDataWithSchema.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.yangtools.yang.data.util;
   9
  10 import com.google.common.annotations.Beta;
  11 import com.google.common.collect.ImmutableMap;
  12 import java.io.IOException;
  13 import java.util.List;
  14 import org.opendaylight.yangtools.rfc7952.data.api.NormalizedMetadata;
  15 import org.opendaylight.yangtools.rfc7952.data.api.NormalizedMetadataStreamWriter;
  16 import org.opendaylight.yangtools.rfc7952.data.api.OpaqueAnydataStreamWriter;
  17 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifier;
  18 import org.opendaylight.yangtools.yang.data.api.schema.opaque.OpaqueData;
  19 import org.opendaylight.yangtools.yang.data.api.schema.opaque.OpaqueDataContainer;
  20 import org.opendaylight.yangtools.yang.data.api.schema.opaque.OpaqueDataList;
  21 import org.opendaylight.yangtools.yang.data.api.schema.opaque.OpaqueDataNode;
  22 import org.opendaylight.yangtools.yang.data.api.schema.opaque.OpaqueDataValue;
  23 import org.opendaylight.yangtools.yang.data.api.schema.stream.NormalizedNodeStreamWriter;
  24 import org.opendaylight.yangtools.yang.data.api.schema.stream.OpaqueAnydataExtension;
  25 import org.opendaylight.yangtools.yang.data.api.schema.stream.OpaqueAnydataExtension.StreamWriter;
  26 import org.opendaylight.yangtools.yang.model.api.AnyDataSchemaNode;
  27
  28 @Beta
  29 public final class OpaqueAnydataNodeDataWithSchema extends AbstractAnydataNodeDataWithSchema<OpaqueData> {
  30     public OpaqueAnydataNodeDataWithSchema(final AnyDataSchemaNode dataSchemaNode) {
  31         super(dataSchemaNode);
  32     }
  33
  34     @Override
  35     protected Class<OpaqueData> objectModelClass() {
  36         return OpaqueData.class;
  37     }
  38
  39     @Override
  40     protected void write(final NormalizedNodeStreamWriter writer, final NormalizedMetadataStreamWriter metaWriter)
  41             throws IOException {
  42         final OpaqueAnydataExtension ext = writer.getExtensions().getInstance(OpaqueAnydataExtension.class);
  43         if (ext != null) {
  44             writer.nextDataSchemaNode(getSchema());
  45             streamOpaqueAnydataNode(ext, provideNodeIdentifier(), getValue(), getMetadata());
  46             writer.endNode();
  47         }
  48     }
  49
  50     private static void streamOpaqueAnydataNode(final OpaqueAnydataExtension ext, final NodeIdentifier name,
  51             final OpaqueData data, final NormalizedMetadata metadata) throws IOException {
  52         if (metadata != null) {
  53             final StreamWriter dataWriter = ext.startOpaqueAnydataNode(name, data.hasAccurateLists());
  54             final OpaqueDataNode node = data.getRoot();
  55             if (dataWriter instanceof OpaqueAnydataStreamWriter) {
  56                 streamOpaqueDataNode((OpaqueAnydataStreamWriter) dataWriter, node, metadata);
  57             } else {
  58                 dataWriter.streamOpaqueDataNode(node);
  59             }
  60         } else {
  61             ext.streamOpaqueAnydataNode(name, data);
  62         }
  63     }
  64
  65     private static void streamOpaqueDataNode(final OpaqueAnydataStreamWriter writer, final OpaqueDataNode node,
  66             final NormalizedMetadata metadata) throws IOException {
  67         if (node instanceof OpaqueDataValue) {
  68             writer.startOpaqueContainer(node.getIdentifier(), 0);
  69             writeMetadata(writer, metadata);
  70             writer.opaqueValue(((OpaqueDataValue) node).getValue());
  71             writer.endOpaqueNode();
  72             return;
  73         }
  74         final List<? extends OpaqueDataNode> children;
  75         if (node instanceof OpaqueDataList) {
  76             children = ((OpaqueDataList) node).getChildren();
  77             writer.startOpaqueList(node.getIdentifier(), children.size());
  78         } else if (node instanceof OpaqueDataContainer) {
  79             children = ((OpaqueDataContainer) node).getChildren();
  80             writer.startOpaqueContainer(node.getIdentifier(), children.size());
  81         } else {
  82             throw new IllegalStateException("Unhandled node " + node);
  83         }
  84
  85         writeMetadata(writer, metadata);
  86         for (OpaqueDataNode child : children) {
  87             streamOpaqueDataNode(writer, child, metadata.getChildren().get(child.getIdentifier()));
  88         }
  89         writer.endOpaqueNode();
  90     }
  91
  92     private static void writeMetadata(final OpaqueAnydataStreamWriter writer, final NormalizedMetadata metadata)
  93             throws IOException {
  94         if (metadata != null) {
  95             writer.metadata(ImmutableMap.copyOf(metadata.getAnnotations()));
  96         }
  97     }
  98 }