1 files changed, 68 insertions, 0 deletions
diff --git a/infra/test-utils/test-tools/src/main/java/io/fd/honeycomb/test/tools/YangDataProcessorRegistry.java b/infra/test-utils/test-tools/src/main/java/io/fd/honeycomb/test/tools/YangDataProcessorRegistry.java
new file mode 100644
index 000000000..d23cd59c1
--- /dev/null
+++ b/infra/test-utils/test-tools/src/main/java/io/fd/honeycomb/test/tools/YangDataProcessorRegistry.java
@@ -0,0 +1,68 @@
+/*
+ * Copyright (c) 2017 Cisco and/or its affiliates.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at:
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package io.fd.honeycomb.test.tools;
+
+import org.opendaylight.controller.md.sal.binding.impl.BindingToNormalizedNodeCodec;
+import org.opendaylight.yangtools.yang.binding.DataObject;
+import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
+import org.opendaylight.yangtools.yang.model.api.SchemaContext;
+
+import javax.annotation.Nonnull;
+import java.util.LinkedList;
+import java.util.List;
+import java.util.stream.Collectors;
+
+import static com.google.common.base.Preconditions.checkState;
+
+/**
+ * Process yang data from json to BA Objects
+ */
+final class YangDataProcessorRegistry {
+
+    private final List<YangDataProcessor> processors;
+
+    private YangDataProcessorRegistry(@Nonnull final SchemaContext context,
+                                      @Nonnull final BindingToNormalizedNodeCodec codec) {
+        // linked should be faster for iteration
+        processors = new LinkedList<>();
+        processors.add(new ListNodeDataProcessor(context, codec));
+        processors.add(new ContainerNodeDataProcessor(context, codec));
+    }
+
+    static YangDataProcessorRegistry create(@Nonnull final SchemaContext context,
+                                                   @Nonnull final BindingToNormalizedNodeCodec codec) {
+        return new YangDataProcessorRegistry(context, codec);
+    }
+
+    @Nonnull
+    DataObject getNodeData(@Nonnull final YangInstanceIdentifier yangInstanceIdentifier,
+                                  @Nonnull final String resourcePath) {
+        return pickProcessor(yangInstanceIdentifier).getNodeData(yangInstanceIdentifier, resourcePath);
+    }
+
+    private YangDataProcessor pickProcessor(final YangInstanceIdentifier yangInstanceIdentifier) {
+        final List<YangDataProcessor> eligibleProcessors = processors.stream()
+                .filter(processors -> processors.canProcess(yangInstanceIdentifier))
+                .collect(Collectors.toList());
+
+        // canProcess should be exclusive for node type, but just in case
+        checkState(eligibleProcessors.size() == 1,
+                "No single eligible processor found, matches=[%s]", eligibleProcessors);
+
+        return eligibleProcessors.get(0);
+    }
+}