commit 25bae4fe321194d27d205aa01ac49ab7a014013c
author Sarath Subramanian <ssubramanian@hortonworks.com> Sun Jul 01 17:07:00 2018 -0700
committer Madhan Neethiraj <madhan@apache.org> Mon Jul 02 00:54:30 2018 -0700
tree cb8fa9d81d73b51d014b940132814c1348b41044
parent eb22be8c36769d88c7ec342bf7f6195849dc13b2

ATLAS-2771: Add option to exclude internal types when retrieving typedefs via REST

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

webapp/src/main/java/org/apache/atlas/web/rest/TypesREST.java

diff --git a/webapp/src/main/java/org/apache/atlas/web/rest/TypesREST.java b/webapp/src/main/java/org/apache/atlas/web/rest/TypesREST.java
index 345250a..13e6512 100644
--- a/webapp/src/main/java/org/apache/atlas/web/rest/TypesREST.java
+++ b/webapp/src/main/java/org/apache/atlas/web/rest/TypesREST.java
@@ -414,13 +414,20 @@
      * @return
      */
     private SearchFilter getSearchFilter(HttpServletRequest httpServletRequest) {
-        SearchFilter ret = new SearchFilter();
-        Set<String> keySet = httpServletRequest.getParameterMap().keySet();
-        for (String key : keySet) {
-            ret.setParam(String.valueOf(key), String.valueOf(httpServletRequest.getParameter(key)));
+        SearchFilter ret    = new SearchFilter();
+        Set<String>  keySet = httpServletRequest.getParameterMap().keySet();
+
+        for (String k : keySet) {
+            String key   = String.valueOf(k);
+            String value = String.valueOf(httpServletRequest.getParameter(k));
+
+            if (key.equalsIgnoreCase("excludeInternalTypesAndReferences") && value.equalsIgnoreCase("true")) {
+                FilterUtil.addParamsToHideInternalType(ret);
+            } else {
+                ret.setParam(key, value);
+            }
         }
 
-        FilterUtil.addParamsToHideInternalType(ret);
         return ret;
     }
 }