diff --git a/contrib/pig/src/java/org/apache/cassandra/hadoop/pig/CassandraStorage.java b/contrib/pig/src/java/org/apache/cassandra/hadoop/pig/CassandraStorage.java index b1af1b5e83..63758abbe7 100644 --- a/contrib/pig/src/java/org/apache/cassandra/hadoop/pig/CassandraStorage.java +++ b/contrib/pig/src/java/org/apache/cassandra/hadoop/pig/CassandraStorage.java @@ -491,8 +491,18 @@ public class CassandraStorage extends LoadFunc implements StoreFuncInterface, Lo if (o == null) return (ByteBuffer)o; if (o instanceof java.lang.String) - o = new DataByteArray((String)o); - return ByteBuffer.wrap(((DataByteArray) o).get()); + return new ByteBuffer.wrap(DataByteArray((String)o).get()); + if (o instanceof Integer) + return IntegerType.instance.decompose((BigInteger)o); + if (o instanceof Long) + return LongType.instance.decompose((Long)o); + if (o instanceof Float) + return FloatType.instance.decompose((Float)o); + if (o instanceof Double) + return DoubleType.instance.decompose((Double)o); + if (o instanceof UUID) + return ByteBuffer.wrap(UUIDGen.decompose((UUID) o)); + return null; } public void putNext(Tuple t) throws ExecException, IOException