commit ce8f65c9c5f7f2cff3934829258ce3d2ebe84b60
author apoorvnaik <apoorvnaik@apache.org> Fri Aug 18 16:23:09 2017 -0700
committer Madhan Neethiraj <madhan@apache.org> Fri Aug 18 17:18:58 2017 -0700
tree 916eb27ff153d8bae0459137b46f30b0d4a8dccb
parent 2c84d7029808c090489b7a8cecdf691ffed71a93

ATLAS-2061: fixed basic-search handling of enum attributes in filter

Signed-off-by: Madhan Neethiraj <madhan@apache.org>

repository/src/main/java/org/apache/atlas/discovery/SearchProcessor.java

diff --git a/repository/src/main/java/org/apache/atlas/discovery/SearchProcessor.java b/repository/src/main/java/org/apache/atlas/discovery/SearchProcessor.java
index 15564fc..d5e3923 100644
--- a/repository/src/main/java/org/apache/atlas/discovery/SearchProcessor.java
+++ b/repository/src/main/java/org/apache/atlas/discovery/SearchProcessor.java
@@ -29,8 +29,10 @@
 import org.apache.atlas.repository.graphdb.AtlasVertex;
 import org.apache.atlas.repository.store.graph.v1.AtlasGraphUtilsV1;
 import org.apache.atlas.type.AtlasEntityType;
+import org.apache.atlas.type.AtlasEnumType;
 import org.apache.atlas.type.AtlasStructType;
 import org.apache.atlas.type.AtlasStructType.AtlasAttribute;
+import org.apache.atlas.type.AtlasType;
 import org.apache.atlas.util.AtlasGremlinQueryProvider;
 import org.apache.atlas.util.SearchPredicateUtil.VertexAttributePredicateGenerator;
 import org.apache.commons.collections.CollectionUtils;
@@ -409,9 +411,10 @@
         VertexAttributePredicateGenerator predicate = OPERATOR_PREDICATE_MAP.get(op);
 
         if (attribute != null && predicate != null) {
-            final String attributeType = attribute.getAttributeType().getTypeName().toLowerCase();
-            final Class  attrClass;
-            final Object attrValue;
+            final AtlasType attrType      = attribute.getAttributeType();
+            final String    attributeType = attrType.getTypeName().toLowerCase();
+            final Class     attrClass;
+            final Object    attrValue;
 
             switch (attributeType) {
                 case "string":
@@ -456,7 +459,11 @@
                     attrValue = new BigDecimal(attrVal);
                     break;
                 default:
-                    attrClass = Object.class;
+                    if (attrType instanceof AtlasEnumType) {
+                        attrClass = String.class;
+                    } else {
+                        attrClass = Object.class;
+                    }
                     attrValue = attrVal;
                     break;
             }