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