Fixed dependent classes related to retrieving types by Category

repository/src/main/java/org/apache/hadoop/metadata/services/DefaultMetadataService.java

@@ -34,11 +34,7 @@ import org.apache.hadoop.metadata.typesystem.TypesDef;
 import org.apache.hadoop.metadata.typesystem.json.InstanceSerialization;
 import org.apache.hadoop.metadata.typesystem.json.Serialization$;
 import org.apache.hadoop.metadata.typesystem.json.TypesSerialization;
-import org.apache.hadoop.metadata.typesystem.types.ClassType;
-import org.apache.hadoop.metadata.typesystem.types.IDataType;
-import org.apache.hadoop.metadata.typesystem.types.Multiplicity;
-import org.apache.hadoop.metadata.typesystem.types.TraitType;
-import org.apache.hadoop.metadata.typesystem.types.TypeSystem;
+import org.apache.hadoop.metadata.typesystem.types.*;
 import org.codehaus.jettison.json.JSONException;
 import org.codehaus.jettison.json.JSONObject;
 import org.slf4j.Logger;
@@ -144,8 +140,8 @@ public class DefaultMetadataService implements MetadataService {
      * @return list of trait type names in the type system
      */
     @Override
-    public List<String> getTraitNamesList() throws MetadataException {
-        return typeSystem.getTraitsNames();
+    public List<String> getTypeNamesByCategory(DataTypes.TypeCategory typeCategory) throws MetadataException {
+        return typeSystem.getTypeNamesByCategory(typeCategory);
     }
 
     /**

repository/src/main/java/org/apache/hadoop/metadata/services/MetadataService.java

@@ -19,6 +19,7 @@
 package org.apache.hadoop.metadata.services;
 
 import org.apache.hadoop.metadata.MetadataException;
+import org.apache.hadoop.metadata.typesystem.types.DataTypes;
 import org.codehaus.jettison.json.JSONObject;
 
 import java.util.List;
@@ -57,7 +58,7 @@ public interface MetadataService {
      *
      * @return list of trait type names in the type system
      */
-    List<String> getTraitNamesList() throws MetadataException;
+    List<String> getTypeNamesByCategory(DataTypes.TypeCategory typeCategory) throws MetadataException;
 
     /**
      * Creates an entity, instance of the type.

typesystem/src/test/java/org/apache/hadoop/metadata/typesystem/types/TypeSystemTest.java

@@ -86,7 +86,7 @@ public class TypeSystemTest extends BaseTest {
                         soxTrait, secTrait, financeTrait),
                 ImmutableList.<HierarchicalTypeDefinition<ClassType>>of());
 
-        final ImmutableList<String> traitsNames = getTypeSystem().getTraitsNames();
+        final ImmutableList<String> traitsNames = getTypeSystem().getTypeNamesByCategory(DataTypes.TypeCategory.TRAIT);
         Assert.assertEquals(traitsNames.size(), 7);
         List traits = Arrays.asList(new String[]{
                 "Classification",