我想出了一种使用幻象映射 UDT 的方法。但不确定这是正确的方法。
我有一个专栏
batsmen_data map<text, frozen<bat_card>>
为了映射它,我们编写下面的代码
object batsmenData extends MapColumn[ScoreCardData, ScoreCard, String, Batting](this) with CustomPrimitives {
override lazy val name="batsmen_data"
}
编译时会显示错误No Primitive found for class Batting
这是因为 Phantom 已经为 String、Int 等原生类型定义了 Primitives。为避免您必须为 Batting 类定义原语,如下所示(您必须在类中扩展 CustomPrimitives 特征,否则您将得到相同的错误)。
trait CustomPrimitives extends Config {
implicit object BattingPrimitive extends Primitive[Batting]{
override type PrimitiveType = Batting
override def clz: Class[Batting] = classOf[Batting]
override def cassandraType: String = {
Connector.session.getCluster.getMetadata.getKeyspace(cassandraKeyspace).getUserType("bat_card").toString()
}
override def fromRow(column: String, row: dsl.Row): Try[Batting] = ???
override def fromString(value: String): Batting = ???
override def asCql(value: Batting): String = ???
}
在此之后,将显示另一个错误,提示未找到请求操作的 Codec: [frozen <-> Batting]。这是因为 cassandra 需要自定义 UDT 类型的编解码器来序列化和反序列化数据。为了避免这种情况,您必须编写一个 CodecClass 来帮助将 UDT 值反序列化(因为我只需要反序列化)到自定义对象中。
public class BattingCodec extends TypeCodec<Batting>{
protected BattingCodec(TypeCodec<UDTValue> innerCodec, Class<Batting> javaType) {
super(innerCodec.getCqlType(), javaType);
}
@Override
public ByteBuffer serialize(Batting value, ProtocolVersion protocolVersion) throws InvalidTypeException {
return null;
}
@Override
public Batting deserialize(ByteBuffer bytes, ProtocolVersion protocolVersion) throws InvalidTypeException {
return null;
}
@Override
public Batting parse(String value) throws InvalidTypeException {
return null;
}
@Override
public String format(Batting value) throws InvalidTypeException {
return null;
}
}
一旦定义了编解码器,最后一步就是将此编解码器注册到编解码器注册表中。
val codecRegistry = CodecRegistry.DEFAULT_INSTANCE
val bat_card = Connector.session.getCluster.getMetadata.getKeyspace(cassandraKeyspace).getUserType("bat_card")
val batCodec = new BattingCodec(TypeCodec.userType(bat_card), classOf[Batting])
codecRegistry.register(batCodec)
现在使用BattingCodec 中的反序列化函数,我们可以将字节映射到所需的对象。
这种方法工作正常。但我不确定这是使用 Phantom 实现 UDT 功能的理想方式