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 java.io.IOException;
13 import java.util.function.Function;
14 import org.eclipse.jdt.annotation.NonNull;
15 import org.opendaylight.yangtools.yang.common.QName;
16 import org.opendaylight.yangtools.yang.common.QNameModule;
17 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
18 import org.opendaylight.yangtools.yang.data.api.schema.stream.NormalizedNodeStreamWriter;
19 import org.opendaylight.yangtools.yang.data.api.schema.stream.NormalizedNodeWriter;
20 import org.opendaylight.yangtools.yang.data.impl.schema.ImmutableNormalizedNodeStreamWriter;
21 import org.opendaylight.yangtools.yang.data.impl.schema.NormalizedNodeResult;
24 public final class NormalizedNodeTransformations {
25 private NormalizedNodeTransformations() {
29 public static NormalizedNode<?, ?> transformQNames(final NormalizedNode<?, ?> original,
30 final Function<QName, QName> mapping) {
31 NormalizedNodeResult result = new NormalizedNodeResult();
32 NormalizedNodeStreamWriter nodeWriter = ImmutableNormalizedNodeStreamWriter.from(result);
33 NormalizedNodeStreamWriter transformWriter = QNameTransformingStreamWriter.fromFunction(nodeWriter, mapping);
35 NormalizedNodeWriter.forStreamWriter(transformWriter).write(original);
36 return result.getResult();
37 } catch (IOException e) {
38 throw new RuntimeException(e);
43 * Returns a {@link NormalizedNode} with QNames replaced by supplied mapping.
45 * @param original Original Normalized Node
46 * @param mapping Map of QNames to transform. Not listed QNames are preserved.
47 * @return Normalized Node with replaced QNames.
49 public static NormalizedNode<?, ?> replaceQNames(final @NonNull NormalizedNode<?, ?> original,
50 final @NonNull Map<QName, QName> mapping) {
51 return transformQNames(original, new QNameReplacementFunction(mapping));
55 * Returns a {@link NormalizedNode} with QNameModules replaced by supplied mapping.
57 * @param original Original Normalized Node
58 * @param mapping Map of QNameModules to transform. Not listed QNameModules are preserved.
59 * @return Normalized Node with replaced QNameModules.
61 public static NormalizedNode<?, ?> replaceQNameModules(final @NonNull NormalizedNode<?, ?> original,
62 final @NonNull Map<QNameModule, QNameModule> mapping) {
63 return transformQNames(original, new QNameModuleReplacementFunction(mapping));