#932, fix for enum type

Impetus · Apr 7, 2017 · 1584ae5 · 1584ae5
1 parent 674c7db
commit 1584ae5
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/...core/src/main/java/com/impetus/client/cassandra/datahandler/CassandraDataHandlerBase.java b/...core/src/main/java/com/impetus/client/cassandra/datahandler/CassandraDataHandlerBase.java
@@ -1421,7 +1421,12 @@ else if (type.getClass().getSimpleName().equals("SetType"))
                     serializer.validate(field);
 
                     Object finalValue = serializer.deserialize(field);
-                    PropertyAccessorHelper.set(entity, fieldToSet, finalValue);
+                    if (type.getClass().getSimpleName().equals("UTF8Type")){
+                        PropertyAccessorHelper.set(entity, fieldToSet, ((String) finalValue).getBytes());
+                    }
+                    else{
+                        PropertyAccessorHelper.set(entity, fieldToSet, finalValue);
+                    }
                 }
 
             }

diff --git a/...ore/src/main/java/com/impetus/client/cassandra/schemamanager/CassandraDataTranslator.java b/...ore/src/main/java/com/impetus/client/cassandra/schemamanager/CassandraDataTranslator.java
@@ -250,7 +250,9 @@ public static CassandraType get(String clazz)
      */
     public static CassandraType getCassandraDataTypeClass(Class clazz)
     {
-
+        if(clazz.isEnum()){
+            return CassandraType.STRING;
+        }
         return typeToClazz.get(clazz);
     }