yang/yang-data-impl/src/main/java/org/opendaylight/yangtools/yang/data/impl/schema/ImmutableMountPointNormalizedNodeStreamWriter.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.impl.schema;
   9
  10 import com.google.common.annotations.Beta;
  11 import com.google.common.collect.ClassToInstanceMap;
  12 import com.google.common.collect.ImmutableClassToInstanceMap;
  13 import java.io.IOException;
  14 import org.opendaylight.yangtools.rcf8528.data.util.ImmutableMountPointNode;
  15 import org.opendaylight.yangtools.rfc8528.data.api.MountPointContext;
  16 import org.opendaylight.yangtools.rfc8528.data.api.MountPointIdentifier;
  17 import org.opendaylight.yangtools.rfc8528.data.api.StreamWriterMountPointExtension;
  18 import org.opendaylight.yangtools.yang.data.api.schema.ContainerNode;
  19 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
  20 import org.opendaylight.yangtools.yang.data.api.schema.stream.ForwardingNormalizedNodeStreamWriter;
  21 import org.opendaylight.yangtools.yang.data.api.schema.stream.NormalizedNodeStreamWriter;
  22 import org.opendaylight.yangtools.yang.data.api.schema.stream.NormalizedNodeStreamWriterExtension;
  23
  24 // FIXME: document usage of this
  25 @Beta
  26 public abstract class ImmutableMountPointNormalizedNodeStreamWriter extends ImmutableNormalizedNodeStreamWriter
  27         implements StreamWriterMountPointExtension {
  28     protected ImmutableMountPointNormalizedNodeStreamWriter(final NormalizedNodeResult result) {
  29         super(result);
  30     }
  31
  32     @Override
  33     public final ClassToInstanceMap<NormalizedNodeStreamWriterExtension> getExtensions() {
  34         return ImmutableClassToInstanceMap.of(StreamWriterMountPointExtension.class, this);
  35     }
  36
  37     @Override
  38     public final NormalizedNodeStreamWriter startMountPoint(final MountPointIdentifier mountId,
  39             final MountPointContext mountCtx) {
  40         final NormalizedNodeResult mountResult = new NormalizedNodeResult();
  41         final NormalizedNodeStreamWriter mountDelegate = ImmutableNormalizedNodeStreamWriter.from(mountResult);
  42
  43         return new ForwardingNormalizedNodeStreamWriter() {
  44             @Override
  45             protected NormalizedNodeStreamWriter delegate() {
  46                 return mountDelegate;
  47             }
  48
  49             @Override
  50             public void close() throws IOException {
  51                 super.close();
  52
  53                 final NormalizedNode<?, ?> data = mountResult.getResult();
  54                 if (!(data instanceof ContainerNode)) {
  55                     throw new IOException("Unhandled mount data " + data);
  56                 }
  57
  58                 writeChild(ImmutableMountPointNode.of(mountId, mountCtx, (ContainerNode) data));
  59             }
  60         };
  61     }
  62 }