Search in sources :

Example 6 with ConversionContext

use of com.datastax.oss.dsbulk.codecs.api.ConversionContext in project dsbulk by datastax.

the class JsonNodeToTupleCodecTest method setUp.

@BeforeEach
void setUp() {
    tupleType = DriverUtils.mockTupleType(DefaultProtocolVersion.V4, CodecRegistry.DEFAULT, DataTypes.TIMESTAMP, DataTypes.TEXT);
    ConversionContext context1 = new TextConversionContext().setNullStrings("NULL", "");
    ConversionContext context2 = new TextConversionContext().setAllowExtraFields(true);
    ConversionContext context3 = new TextConversionContext().setAllowMissingFields(true);
    codec1 = (JsonNodeToTupleCodec) new ConvertingCodecFactory(context1).<JsonNode, TupleValue>createConvertingCodec(tupleType, JSON_NODE_TYPE, true);
    codec2 = (JsonNodeToTupleCodec) new ConvertingCodecFactory(context2).<JsonNode, TupleValue>createConvertingCodec(tupleType, JSON_NODE_TYPE, true);
    codec3 = (JsonNodeToTupleCodec) new ConvertingCodecFactory(context3).<JsonNode, TupleValue>createConvertingCodec(tupleType, JSON_NODE_TYPE, true);
}
Also used : ConvertingCodecFactory(com.datastax.oss.dsbulk.codecs.api.ConvertingCodecFactory) TextConversionContext(com.datastax.oss.dsbulk.codecs.text.TextConversionContext) TextConversionContext(com.datastax.oss.dsbulk.codecs.text.TextConversionContext) ConversionContext(com.datastax.oss.dsbulk.codecs.api.ConversionContext) JsonNode(com.fasterxml.jackson.databind.JsonNode) TupleValue(com.datastax.oss.driver.api.core.data.TupleValue) BeforeEach(org.junit.jupiter.api.BeforeEach)

Example 7 with ConversionContext

use of com.datastax.oss.dsbulk.codecs.api.ConversionContext in project dsbulk by datastax.

the class JsonNodeToUDTCodecTest method setUp.

@BeforeEach
void setUp() {
    ConversionContext context1 = new TextConversionContext().setNullStrings("NULL", "");
    ConversionContext context2 = new TextConversionContext().setAllowExtraFields(true);
    ConversionContext context3 = new TextConversionContext().setAllowMissingFields(true);
    ConvertingCodecFactory codecFactory1 = new ConvertingCodecFactory(context1);
    ConvertingCodecFactory codecFactory2 = new ConvertingCodecFactory(context2);
    ConvertingCodecFactory codecFactory3 = new ConvertingCodecFactory(context3);
    udtCodec1 = (JsonNodeToUDTCodec) codecFactory1.<JsonNode, UdtValue>createConvertingCodec(udt1, JSON_NODE_TYPE, true);
    udtCodec2 = (JsonNodeToUDTCodec) codecFactory1.<JsonNode, UdtValue>createConvertingCodec(udt2, JSON_NODE_TYPE, true);
    udtCodec3 = (JsonNodeToUDTCodec) codecFactory2.<JsonNode, UdtValue>createConvertingCodec(udt3, JSON_NODE_TYPE, true);
    udtCodec4 = (JsonNodeToUDTCodec) codecFactory3.<JsonNode, UdtValue>createConvertingCodec(udt4, JSON_NODE_TYPE, true);
}
Also used : ConvertingCodecFactory(com.datastax.oss.dsbulk.codecs.api.ConvertingCodecFactory) UdtValue(com.datastax.oss.driver.api.core.data.UdtValue) TextConversionContext(com.datastax.oss.dsbulk.codecs.text.TextConversionContext) TextConversionContext(com.datastax.oss.dsbulk.codecs.text.TextConversionContext) ConversionContext(com.datastax.oss.dsbulk.codecs.api.ConversionContext) JsonNode(com.fasterxml.jackson.databind.JsonNode) BeforeEach(org.junit.jupiter.api.BeforeEach)

Example 8 with ConversionContext

use of com.datastax.oss.dsbulk.codecs.api.ConversionContext in project dsbulk by datastax.

the class StringToFloatCodecTest method setUp.

@BeforeEach
void setUp() {
    ConversionContext context = new TextConversionContext().setNullStrings("NULL").setFormatNumbers(true).setRoundingMode(HALF_EVEN);
    ConvertingCodecFactory codecFactory = new ConvertingCodecFactory(context);
    codec = (StringToFloatCodec) codecFactory.<String, Float>createConvertingCodec(DataTypes.FLOAT, GenericType.STRING, true);
}
Also used : ConvertingCodecFactory(com.datastax.oss.dsbulk.codecs.api.ConvertingCodecFactory) TextConversionContext(com.datastax.oss.dsbulk.codecs.text.TextConversionContext) TextConversionContext(com.datastax.oss.dsbulk.codecs.text.TextConversionContext) ConversionContext(com.datastax.oss.dsbulk.codecs.api.ConversionContext) BeforeEach(org.junit.jupiter.api.BeforeEach)

Example 9 with ConversionContext

use of com.datastax.oss.dsbulk.codecs.api.ConversionContext in project dsbulk by datastax.

the class StringToInstantCodecTest method setUpCodec1.

@BeforeEach
void setUpCodec1() {
    ConversionContext context1 = new TextConversionContext().setNullStrings("NULL");
    ConversionContext context2 = new TextConversionContext().setNullStrings("NULL").setTimestampFormat("yyyyMMddHHmmss");
    ConversionContext context3 = new TextConversionContext().setNullStrings("NULL").setTimeUnit(MINUTES).setEpoch(ZonedDateTime.parse("2000-01-01T00:00:00Z"));
    ConversionContext context4 = new TextConversionContext().setNullStrings("NULL").setTimeUnit(MINUTES).setEpoch(ZonedDateTime.parse("2000-01-01T00:00:00Z")).setTimestampFormat("UNITS_SINCE_EPOCH");
    codec1 = (StringToInstantCodec) new ConvertingCodecFactory(context1).<String, Instant>createConvertingCodec(DataTypes.TIMESTAMP, GenericType.STRING, true);
    codec2 = (StringToInstantCodec) new ConvertingCodecFactory(context2).<String, Instant>createConvertingCodec(DataTypes.TIMESTAMP, GenericType.STRING, true);
    codec3 = (StringToInstantCodec) new ConvertingCodecFactory(context3).<String, Instant>createConvertingCodec(DataTypes.TIMESTAMP, GenericType.STRING, true);
    codec4 = (StringToInstantCodec) new ConvertingCodecFactory(context4).<String, Instant>createConvertingCodec(DataTypes.TIMESTAMP, GenericType.STRING, true);
}
Also used : ConvertingCodecFactory(com.datastax.oss.dsbulk.codecs.api.ConvertingCodecFactory) TextConversionContext(com.datastax.oss.dsbulk.codecs.text.TextConversionContext) TextConversionContext(com.datastax.oss.dsbulk.codecs.text.TextConversionContext) ConversionContext(com.datastax.oss.dsbulk.codecs.api.ConversionContext) Instant(java.time.Instant) BeforeEach(org.junit.jupiter.api.BeforeEach)

Example 10 with ConversionContext

use of com.datastax.oss.dsbulk.codecs.api.ConversionContext in project dsbulk by datastax.

the class StringToMapCodecTest method setUp.

@BeforeEach
void setUp() {
    ConversionContext context = new TextConversionContext().setNullStrings("NULL").setFormatNumbers(true);
    ConvertingCodecFactory codecFactory = new ConvertingCodecFactory(context);
    codec = (StringToMapCodec<Double, List<String>>) codecFactory.<String, Map<Double, List<String>>>createConvertingCodec(DataTypes.mapOf(DataTypes.DOUBLE, DataTypes.listOf(DataTypes.TEXT)), GenericType.STRING, true);
}
Also used : ConvertingCodecFactory(com.datastax.oss.dsbulk.codecs.api.ConvertingCodecFactory) TextConversionContext(com.datastax.oss.dsbulk.codecs.text.TextConversionContext) TextConversionContext(com.datastax.oss.dsbulk.codecs.text.TextConversionContext) ConversionContext(com.datastax.oss.dsbulk.codecs.api.ConversionContext) List(java.util.List) ImmutableMap(com.datastax.oss.driver.shaded.guava.common.collect.ImmutableMap) LinkedHashMap(java.util.LinkedHashMap) Map(java.util.Map) BeforeEach(org.junit.jupiter.api.BeforeEach)

Aggregations

ConversionContext (com.datastax.oss.dsbulk.codecs.api.ConversionContext)32 ConvertingCodecFactory (com.datastax.oss.dsbulk.codecs.api.ConvertingCodecFactory)28 TextConversionContext (com.datastax.oss.dsbulk.codecs.text.TextConversionContext)28 BeforeEach (org.junit.jupiter.api.BeforeEach)28 JsonNode (com.fasterxml.jackson.databind.JsonNode)15 List (java.util.List)5 Instant (java.time.Instant)4 ImmutableMap (com.datastax.oss.driver.shaded.guava.common.collect.ImmutableMap)3 TupleValue (com.datastax.oss.driver.api.core.data.TupleValue)2 UdtValue (com.datastax.oss.driver.api.core.data.UdtValue)2 CustomType (com.datastax.oss.driver.api.core.type.CustomType)2 CodecNotFoundException (com.datastax.oss.driver.api.core.type.codec.CodecNotFoundException)2 CodecRegistry (com.datastax.oss.driver.api.core.type.codec.registry.CodecRegistry)2 ImmutableList (com.datastax.oss.driver.shaded.guava.common.collect.ImmutableList)2 ConvertingCodec (com.datastax.oss.dsbulk.codecs.api.ConvertingCodec)2 JdkCodecUtils.isUUID (com.datastax.oss.dsbulk.codecs.jdk.JdkCodecUtils.isUUID)2 NonNull (edu.umd.cs.findbugs.annotations.NonNull)2 Nullable (edu.umd.cs.findbugs.annotations.Nullable)2 BigDecimal (java.math.BigDecimal)2 BigInteger (java.math.BigInteger)2