Examples with MutableCodecRegistry com.datastax.oss.driver.api.core.type.codec.registry.MutableCodecRegistry used on opensource projects

Example 1 with MutableCodecRegistry

use of com.datastax.oss.driver.api.core.type.codec.registry.MutableCodecRegistry in project zipkin by openzipkin.

the class DefaultSessionFactory method initializeUDTs.

static void initializeUDTs(CqlSession session, String keyspace) {
    KeyspaceMetadata ks = session.getMetadata().getKeyspace(keyspace).get();
    MutableCodecRegistry codecRegistry = (MutableCodecRegistry) session.getContext().getCodecRegistry();
    TypeCodec<UdtValue> annotationUDTCodec = codecRegistry.codecFor(ks.getUserDefinedType("annotation").get());
    codecRegistry.register(new AnnotationCodec(annotationUDTCodec));
    LOG.debug("Registering endpoint and annotation UDTs to keyspace {}", keyspace);
    TypeCodec<UdtValue> endpointUDTCodec = codecRegistry.codecFor(ks.getUserDefinedType("endpoint").get());
    codecRegistry.register(new EndpointCodec(endpointUDTCodec));
}