use of com.amazonaws.athena.connector.lambda.data.writers.extractors.DecimalExtractor in project aws-athena-query-federation by awslabs.
the class GeneratedRowWriter method makeFieldWriter.
private FieldWriter makeFieldWriter(FieldVector vector) {
Field field = vector.getField();
String fieldName = field.getName();
Types.MinorType fieldType = Types.getMinorTypeForArrowType(field.getType());
Extractor extractor = extractors.get(fieldName);
ConstraintProjector constraint = constraints.get(fieldName);
FieldWriterFactory factory = fieldWriterFactories.get(fieldName);
if (factory != null) {
return factory.create(vector, extractor, constraint);
}
if (extractor == null) {
throw new IllegalStateException("Missing extractor for field[" + fieldName + "]");
}
switch(fieldType) {
case INT:
return new IntFieldWriter((IntExtractor) extractor, (IntVector) vector, constraint);
case BIGINT:
return new BigIntFieldWriter((BigIntExtractor) extractor, (BigIntVector) vector, constraint);
case DATEMILLI:
return new DateMilliFieldWriter((DateMilliExtractor) extractor, (DateMilliVector) vector, constraint);
case DATEDAY:
return new DateDayFieldWriter((DateDayExtractor) extractor, (DateDayVector) vector, constraint);
case TINYINT:
return new TinyIntFieldWriter((TinyIntExtractor) extractor, (TinyIntVector) vector, constraint);
case SMALLINT:
return new SmallIntFieldWriter((SmallIntExtractor) extractor, (SmallIntVector) vector, constraint);
case FLOAT4:
return new Float4FieldWriter((Float4Extractor) extractor, (Float4Vector) vector, constraint);
case FLOAT8:
return new Float8FieldWriter((Float8Extractor) extractor, (Float8Vector) vector, constraint);
case DECIMAL:
return new DecimalFieldWriter((DecimalExtractor) extractor, (DecimalVector) vector, constraint);
case BIT:
return new BitFieldWriter((BitExtractor) extractor, (BitVector) vector, constraint);
case VARCHAR:
return new VarCharFieldWriter((VarCharExtractor) extractor, (VarCharVector) vector, constraint);
case VARBINARY:
return new VarBinaryFieldWriter((VarBinaryExtractor) extractor, (VarBinaryVector) vector, constraint);
default:
throw new RuntimeException(fieldType + " is not supported");
}
}
use of com.amazonaws.athena.connector.lambda.data.writers.extractors.DecimalExtractor in project aws-athena-query-federation by awslabs.
the class ExampleRecordHandler method makeExtractor.
/**
* Creates an Extractor for the given field. In this example the extractor just creates some random data.
*/
private Extractor makeExtractor(Field field, RowContext rowContext) {
Types.MinorType fieldType = Types.getMinorTypeForArrowType(field.getType());
// they need to match the split otherwise filtering will brake in unexpected ways.
if (field.getName().equals("year")) {
return (IntExtractor) (Object context, NullableIntHolder dst) -> {
dst.isSet = 1;
dst.value = rowContext.getYear();
};
} else if (field.getName().equals("month")) {
return (IntExtractor) (Object context, NullableIntHolder dst) -> {
dst.isSet = 1;
dst.value = rowContext.getMonth();
};
} else if (field.getName().equals("day")) {
return (IntExtractor) (Object context, NullableIntHolder dst) -> {
dst.isSet = 1;
dst.value = rowContext.getDay();
};
}
switch(fieldType) {
case INT:
return (IntExtractor) (Object context, NullableIntHolder dst) -> {
dst.isSet = 1;
dst.value = ((RowContext) context).seed * (((RowContext) context).negative ? -1 : 1);
};
case DATEMILLI:
return (DateMilliExtractor) (Object context, NullableDateMilliHolder dst) -> {
dst.isSet = 1;
dst.value = ((RowContext) context).seed * (((RowContext) context).negative ? -1 : 1);
};
case DATEDAY:
return (DateDayExtractor) (Object context, NullableDateDayHolder dst) -> {
dst.isSet = 1;
dst.value = ((RowContext) context).seed * (((RowContext) context).negative ? -1 : 1);
};
case TINYINT:
return (TinyIntExtractor) (Object context, NullableTinyIntHolder dst) -> {
dst.isSet = 1;
dst.value = (byte) ((((RowContext) context).seed % 4) * (((RowContext) context).negative ? -1 : 1));
};
case SMALLINT:
return (SmallIntExtractor) (Object context, NullableSmallIntHolder dst) -> {
dst.isSet = 1;
dst.value = (short) ((((RowContext) context).seed % 4) * (((RowContext) context).negative ? -1 : 1));
};
case FLOAT4:
return (Float4Extractor) (Object context, NullableFloat4Holder dst) -> {
dst.isSet = 1;
dst.value = ((float) ((RowContext) context).seed) * 1.1f * (((RowContext) context).negative ? -1f : 1f);
};
case FLOAT8:
return (Float8Extractor) (Object context, NullableFloat8Holder dst) -> {
dst.isSet = 1;
dst.value = ((double) ((RowContext) context).seed) * 1.1D;
};
case DECIMAL:
return (DecimalExtractor) (Object context, NullableDecimalHolder dst) -> {
dst.isSet = 1;
double d8Val = ((RowContext) context).seed * 1.1D * (((RowContext) context).negative ? -1d : 1d);
BigDecimal bdVal = new BigDecimal(d8Val);
dst.value = bdVal.setScale(((ArrowType.Decimal) field.getType()).getScale(), RoundingMode.HALF_UP);
};
case BIT:
return (BitExtractor) (Object context, NullableBitHolder dst) -> {
dst.isSet = 1;
dst.value = ((RowContext) context).seed % 2;
};
case BIGINT:
return (BigIntExtractor) (Object context, NullableBigIntHolder dst) -> {
dst.isSet = 1;
dst.value = ((RowContext) context).seed * 1L * (((RowContext) context).negative ? -1 : 1);
};
case VARCHAR:
return (VarCharExtractor) (Object context, NullableVarCharHolder dst) -> {
dst.isSet = 1;
dst.value = "VarChar" + ((RowContext) context).seed;
};
case VARBINARY:
return (VarBinaryExtractor) (Object context, NullableVarBinaryHolder dst) -> {
dst.isSet = 1;
dst.value = ("VarChar" + ((RowContext) context).seed).getBytes(Charsets.UTF_8);
};
default:
return null;
}
}
Aggregations