2 * Copyright (c) 2015 Cisco Systems, Inc. and others. All rights reserved.
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
8 package org.opendaylight.yangtools.transform;
10 import com.google.common.annotations.Beta;
11 import com.google.common.base.Throwables;
12 import java.io.IOException;
14 import java.util.function.Function;
15 import javax.annotation.Nonnull;
16 import org.opendaylight.yangtools.yang.common.QName;
17 import org.opendaylight.yangtools.yang.common.QNameModule;
18 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
19 import org.opendaylight.yangtools.yang.data.api.schema.stream.NormalizedNodeStreamWriter;
20 import org.opendaylight.yangtools.yang.data.api.schema.stream.NormalizedNodeWriter;
21 import org.opendaylight.yangtools.yang.data.impl.schema.ImmutableNormalizedNodeStreamWriter;
22 import org.opendaylight.yangtools.yang.data.impl.schema.NormalizedNodeResult;
25 public final class NormalizedNodeTransformations {
27 private NormalizedNodeTransformations() {
28 throw new UnsupportedOperationException("Utility class.");
31 public static NormalizedNode<?, ?> transformQNames(NormalizedNode<?, ?> original, Function<QName, QName> mapping) {
32 NormalizedNodeResult result = new NormalizedNodeResult();
33 NormalizedNodeStreamWriter nodeWriter = ImmutableNormalizedNodeStreamWriter.from(result);
34 NormalizedNodeStreamWriter transformWriter = QNameTransformingStreamWriter.fromFunction(nodeWriter, mapping);
36 NormalizedNodeWriter.forStreamWriter(transformWriter).write(original);
37 return result.getResult();
38 } catch (IOException e) {
39 throw Throwables.propagate(e);
45 * Returns a {@link NormalizedNode} with QNames replaced by supplied mapping.
47 * @param original Original Normalized Node
48 * @param mapping Map of QNames to transform. Not listed QNames are preserved.
49 * @return Normalized Node with replaced QNames.
51 public static NormalizedNode<?, ?> replaceQNames(@Nonnull NormalizedNode<?, ?> original,
52 @Nonnull Map<QName, QName> mapping) {
53 return transformQNames(original, new QNameReplacementFunction(mapping));
58 * Returns a {@link NormalizedNode} with QNameModules replaced by supplied mapping.
60 * @param original Original Normalized Node
61 * @param mapping Map of QNameModules to transform. Not listed QNameModules are preserved.
62 * @return Normalized Node with replaced QNameModules.
65 public static NormalizedNode<?, ?> replaceQNameModules(@Nonnull NormalizedNode<?, ?> original,
66 @Nonnull Map<QNameModule, QNameModule> mapping) {
67 return transformQNames(original, new QNameModuleReplacementFunction(mapping));