data/yang-data-util/src/main/java/org/opendaylight/yangtools/yang/data/util/AbstractMountPointDataWithSchema.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 static com.google.common.base.Preconditions.checkState;
  11
  12 import com.google.common.annotations.Beta;
  13 import java.io.IOException;
  14 import org.opendaylight.yangtools.yang.common.MountPointLabel;
  15 import org.opendaylight.yangtools.yang.data.api.schema.MountPointContextFactory;
  16 import org.opendaylight.yangtools.yang.data.api.schema.stream.NormalizedNodeStreamWriter;
  17 import org.opendaylight.yangtools.yang.data.api.schema.stream.NormalizedNodeStreamWriter.MetadataExtension;
  18 import org.opendaylight.yangtools.yang.model.api.DataSchemaNode;
  19
  20 /**
  21  * A {@link CompositeNodeDataWithSchema} which can hold mount-point data. This data is manipulated through
  22  * {@link #getMountPointData(MountPointLabel, MountPointContextFactory)}.
  23  */
  24 @Beta
  25 public abstract sealed class AbstractMountPointDataWithSchema<T extends DataSchemaNode>
  26         extends CompositeNodeDataWithSchema<T> permits ContainerNodeDataWithSchema, ListEntryNodeDataWithSchema {
  27     private MountPointData mountedData;
  28
  29     AbstractMountPointDataWithSchema(final T schema) {
  30         super(schema);
  31     }
  32
  33     @Override
  34     public void write(final NormalizedNodeStreamWriter writer, final MetadataExtension metaWriter) throws IOException {
  35         super.write(writer, metaWriter);
  36         if (mountedData != null) {
  37             mountedData.write(writer);
  38         }
  39     }
  40
  41     public final MountPointData getMountPointData(final MountPointLabel label, final MountPointContextFactory factory) {
  42         if (mountedData != null) {
  43             final var existing = mountedData.label();
  44             checkState(label.equals(existing), "Mismatched mount label %s, already have %s", label, existing);
  45         } else {
  46             mountedData = new MountPointData(label, factory);
  47         }
  48         return mountedData;
  49     }
  50 }