diff options
author | 2018-07-19 16:03:44 -0400 | |
---|---|---|
committer | 2018-07-19 16:08:20 -0400 | |
commit | 3bc6a702f2d3d8710c7aaa94cdc8c0ccf3deb759 (patch) | |
tree | 9703ebe15777acc73ad118520f6b1a219f60e107 /src/main/java/org/onap/schema/RelationshipSchema.java | |
parent | 194adee686ebb90488f739f2c637f6cb3def94d5 (diff) |
Auto-resolve edge type
Auto-resolve edge type based on EdgeRules file.
Change-Id: Ic6de47f5172bc410efcdd5f08c1ea5c4f822610e
Issue-ID: AAI-1396
Signed-off-by: sblimkie <steven.blimkie@amdocs.com>
Diffstat (limited to 'src/main/java/org/onap/schema/RelationshipSchema.java')
-rw-r--r-- | src/main/java/org/onap/schema/RelationshipSchema.java | 29 |
1 files changed, 29 insertions, 0 deletions
diff --git a/src/main/java/org/onap/schema/RelationshipSchema.java b/src/main/java/org/onap/schema/RelationshipSchema.java index 2aab7a9..0cf419e 100644 --- a/src/main/java/org/onap/schema/RelationshipSchema.java +++ b/src/main/java/org/onap/schema/RelationshipSchema.java @@ -23,7 +23,9 @@ package org.onap.schema; import java.io.IOException; import java.util.HashMap; +import java.util.HashSet; import java.util.Map; +import java.util.Set; import java.util.stream.Collectors; import javax.ws.rs.core.Response.Status; import org.codehaus.jackson.map.ObjectMapper; @@ -47,6 +49,10 @@ public class RelationshipSchema { private Map<String, Map<String, Class<?>>> relationTypes = new HashMap<>(); private Map<String, EdgeRule> relationshipRules = new HashMap<>(); + + // A map storing the list of valid edge types for a source/target pair + private Map<String, Set<String>> edgeTypesForNodePair = new HashMap<>(); + @SuppressWarnings("unchecked") public RelationshipSchema(Multimap<String, EdgeRule> rules, String props) throws CrudException, IOException { @@ -56,6 +62,16 @@ public class RelationshipSchema { for (EdgeRule rule : rules.values()) { String key = buildRelation(rule.getFrom(), rule.getTo(), rule.getLabel()); relationshipRules.put(key, rule); + + String nodePairKey = buildNodePairKey(rule.getFrom(), rule.getTo()); + if (edgeTypesForNodePair.get(nodePairKey) == null) { + Set<String> typeSet = new HashSet<String>(); + typeSet.add(rule.getLabel()); + edgeTypesForNodePair.put(nodePairKey, typeSet); + } + else { + edgeTypesForNodePair.get(nodePairKey).add(rule.getLabel()); + } } Map<String, Class<?>> edgeProps = properties.entrySet().stream().collect(Collectors.toMap(p -> p.getKey(), p -> { @@ -106,10 +122,23 @@ public class RelationshipSchema { return relationTypes.containsKey(type); } + public Set<String> getValidRelationTypes(String source, String target) { + Set<String> typeList = edgeTypesForNodePair.get(buildNodePairKey(source, target)); + if (typeList == null) { + return new HashSet<String>(); + } + + return typeList; + } + private String buildRelation(String source, String target, String relation) { return source + ":" + target + ":" + relation; } + + private String buildNodePairKey(String source, String target) { + return source + ":" + target; + } private Class<?> resolveClass(String type) throws CrudException, ClassNotFoundException { Class<?> clazz = Class.forName(type); |