Retrofit mount-point awareness into yang-data-util
[yangtools.git] / yang / 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.rfc7952.data.api.NormalizedMetadataStreamWriter;
15 import org.opendaylight.yangtools.yang.common.QName;
16 import org.opendaylight.yangtools.yang.data.api.schema.stream.NormalizedNodeStreamWriter;
17 import org.opendaylight.yangtools.yang.model.api.DataSchemaNode;
18
19 /**
20  * A {@link CompositeNodeDataWithSchema} which can hold mount-point data. This data is manipulated through
21  * {@link #getMountPointData(QName)}.
22  */
23 @Beta
24 public abstract class AbstractMountPointDataWithSchema<T extends DataSchemaNode>
25         extends CompositeNodeDataWithSchema<T> {
26     private MountPointData mountedData;
27
28     AbstractMountPointDataWithSchema(final T schema) {
29         super(schema);
30     }
31
32     @Override
33     public void write(final NormalizedNodeStreamWriter writer, final NormalizedMetadataStreamWriter metaWriter)
34             throws IOException {
35         super.write(writer, metaWriter);
36         if (mountedData != null) {
37             mountedData.write(writer);
38         }
39     }
40
41     public final MountPointData getMountPointData(final QName label) {
42         if (mountedData != null) {
43             final QName existing = mountedData.getIdentifier();
44             checkState(label.equals(existing), "Mismatched mount label {}, already have {}", label, existing);
45         } else {
46             mountedData = new MountPointData(label);
47         }
48         return mountedData;
49     }
50 }