Commit a6c0793f by Richard Ding Committed by Madhan Neethiraj

ATLAS-2076: update type-search to support filter by RELATIONSHIP type

parent b445a1d2
......@@ -668,6 +668,9 @@ public abstract class AtlasTypeDefGraphStore implements AtlasTypeDefStore {
case ENTITY:
ret = ((AtlasEntityType) type).getEntityDef();
break;
case RELATIONSHIP:
ret = ((AtlasRelationshipType) type).getRelationshipDef();
break;
case PRIMITIVE:
case OBJECT_ID_TYPE:
case ARRAY:
......
......@@ -115,6 +115,8 @@ public class FilterUtil {
return atlasType.getTypeCategory() == TypeCategory.STRUCT;
case "ENUM":
return atlasType.getTypeCategory() == TypeCategory.ENUM;
case "RELATIONSHIP":
return atlasType.getTypeCategory() == TypeCategory.RELATIONSHIP;
default:
// This shouldn't have happened
return false;
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment