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