Cassandra Custom Collection Mapping - collections

Question

I am trying to create a custom codec which maps a Java Map to Cassandra Set<UDTShift>. UDTShift is a user defined type. Below is my test custom codec class according to the instruction: http://docs.datastax.com/en/developer/java-driver/3.1/manual/custom_codecs/
I have UDTShift which is called "udt+shift" in the cassandra database and have the ORM pojo class UDTShift class in Java as well.
I am having trouble register this codec in the codecRegistry and I am not sure if this is the right way to create codec.
package com.sick.il.analytics.system.configuration.cassandracodec;
import java.nio.ByteBuffer;
import java.util.Date;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;
import com.datastax.driver.core.ProtocolVersion;
import com.datastax.driver.core.TypeCodec;
import com.datastax.driver.core.UserType;
import com.datastax.driver.core.exceptions.InvalidTypeException;
import com.sick.il.analytics.system.configuration.model.UDTAssignedShift;
public class ShiftCodec extends TypeCodec < Map < String, Date[] >> {
private final TypeCodec < Set < UDTAssignedShift >> innerCodec;
private final UserType userType;
public ShiftCodec(TypeCodec < Set < UDTAssignedShift >> innerCodec, Class < Map < String, Date[] >> javaClass) {
super(innerCodec.getCqlType(), javaClass);
this.innerCodec = innerCodec;
this.userType = (UserType) innerCodec.getCqlType();
}
#Override
public ByteBuffer serialize(Map < String, Date[] > value, ProtocolVersion protocolVersion)
throws InvalidTypeException {
return innerCodec.serialize(toCassandraSet(value), protocolVersion);
}
#Override
public Map < String,
Date[] > deserialize(ByteBuffer bytes, ProtocolVersion protocolVersion)
throws InvalidTypeException {
// TODO Auto-generated method stub
return toJavaMap(innerCodec.deserialize(bytes, protocolVersion));
}
#Override
public Map < String,
Date[] > parse(String value) throws InvalidTypeException {
return value == null || value.isEmpty() || value.equals(null) ? null : toJavaMap(innerCodec.parse(value));
}
#Override
public String format(Map < String, Date[] > value) throws InvalidTypeException {
return value == null ? null : innerCodec.format(toCassandraSet(value));
}
protected Map < String,
Date[] > toJavaMap(Set < UDTAssignedShift > shiftSets) {
Date[] dates = new Date[6];
Map < String, Date[] > map = new HashMap < > ();
map.put("Monday", dates);
return map;
}
protected Set < UDTAssignedShift > toCassandraSet(Map < String, Date[] > shiftMap) {
Set < UDTAssignedShift > sets = new HashSet < > ();
UDTAssignedShift shift = new UDTAssignedShift();
shift.setShiftName("Monday_01");
shift.setStartDate("15:30:00:000");
shift.setEndDate("19:30:00:000");
sets.add(shift);
return sets;
}
}