All Downloads are FREE. Search and download functionalities are using the official Maven repository.
Please wait. This can take some minutes ...
Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
net.snowflake.client.core.arrow.DecimalToScaledFixedConverter Maven / Gradle / Ivy
/*
* Copyright (c) 2012-2019 Snowflake Computing Inc. All rights reserved.
*/
package net.snowflake.client.core.arrow;
import net.snowflake.client.core.DataConversionContext;
import net.snowflake.client.core.SFException;
import net.snowflake.client.jdbc.ErrorCode;
import net.snowflake.client.jdbc.SnowflakeType;
import org.apache.arrow.vector.DecimalVector;
import org.apache.arrow.vector.ValueVector;
import java.math.BigDecimal;
/**
* Data vector whose snowflake logical type is fixed while represented as a
* BigDecimal value vector
*/
public class DecimalToScaledFixedConverter extends AbstractArrowVectorConverter
{
protected DecimalVector decimalVector;
public DecimalToScaledFixedConverter(ValueVector fieldVector, int vectorIndex, DataConversionContext context)
{
super(String.format("%s(%s,%s)", SnowflakeType.FIXED,
fieldVector.getField().getMetadata().get("precision"),
fieldVector.getField().getMetadata().get("scale")),
fieldVector,
vectorIndex,
context);
decimalVector = (DecimalVector) fieldVector;
}
@Override
public byte[] toBytes(int index)
{
if (isNull(index))
{
return null;
}
else
{
return toBigDecimal(index).toBigInteger().toByteArray();
}
}
@Override
public byte toByte(int index) throws SFException
{
if (isNull(index))
{
return 0;
}
BigDecimal bigDecimal = toBigDecimal(index);
if (bigDecimal.scale() == 0)
{
byte byteVal = bigDecimal.byteValue();
if (byteVal == bigDecimal.longValue())
{
return byteVal;
}
else
{
throw new SFException(ErrorCode.INVALID_VALUE_CONVERT, logicalTypeStr,
"Byte", bigDecimal.toPlainString());
}
}
else
{
throw new SFException(ErrorCode.INVALID_VALUE_CONVERT, logicalTypeStr,
"Byte", bigDecimal.toPlainString());
}
}
@Override
public short toShort(int index) throws SFException
{
if (isNull(index))
{
return 0;
}
BigDecimal bigDecimal = toBigDecimal(index);
if (bigDecimal.scale() == 0)
{
short shortValue = bigDecimal.shortValue();
if (bigDecimal.compareTo(BigDecimal.valueOf((long) shortValue)) == 0)
{
return shortValue;
}
else
{
throw new SFException(ErrorCode.INVALID_VALUE_CONVERT, logicalTypeStr,
"Short", bigDecimal.toPlainString());
}
}
else
{
throw new SFException(ErrorCode.INVALID_VALUE_CONVERT, logicalTypeStr,
"Short", bigDecimal.toPlainString());
}
}
@Override
public int toInt(int index) throws SFException
{
if (isNull(index))
{
return 0;
}
BigDecimal bigDecimal = toBigDecimal(index);
if (bigDecimal.scale() == 0)
{
int intValue = bigDecimal.intValue();
if (bigDecimal.compareTo(BigDecimal.valueOf((long) intValue)) == 0)
{
return intValue;
}
else
{
throw new SFException(ErrorCode.INVALID_VALUE_CONVERT, logicalTypeStr,
"Int", bigDecimal.toPlainString());
}
}
else
{
throw new SFException(ErrorCode.INVALID_VALUE_CONVERT, logicalTypeStr,
"Int", bigDecimal.toPlainString());
}
}
@Override
public long toLong(int index) throws SFException
{
if (isNull(index))
{
return 0;
}
BigDecimal bigDecimal = toBigDecimal(index);
if (bigDecimal.scale() == 0)
{
long longValue = bigDecimal.longValue();
if (bigDecimal.compareTo(BigDecimal.valueOf(longValue)) == 0)
{
return longValue;
}
else
{
throw new SFException(ErrorCode.INVALID_VALUE_CONVERT, logicalTypeStr,
"Long", bigDecimal.toPlainString());
}
}
else
{
throw new SFException(ErrorCode.INVALID_VALUE_CONVERT, logicalTypeStr,
"Long", bigDecimal.toPlainString());
}
}
@Override
public float toFloat(int index) throws SFException
{
if (isNull(index))
{
return 0;
}
BigDecimal bigDecimal = toBigDecimal(index);
return bigDecimal.floatValue();
}
@Override
public double toDouble(int index) throws SFException
{
if (isNull(index))
{
return 0;
}
BigDecimal bigDecimal = toBigDecimal(index);
return bigDecimal.doubleValue();
}
@Override
public BigDecimal toBigDecimal(int index)
{
return decimalVector.getObject(index);
}
@Override
public Object toObject(int index) throws SFException
{
return toBigDecimal(index);
}
@Override
public String toString(int index)
{
BigDecimal bigDecimal = toBigDecimal(index);
return bigDecimal == null ? null : bigDecimal.toPlainString();
}
@Override
public boolean toBoolean(int index) throws SFException
{
if (isNull(index))
{
return false;
}
BigDecimal val = toBigDecimal(index);
if (val.compareTo(BigDecimal.ZERO) == 0)
{
return false;
}
else if (val.compareTo(BigDecimal.ONE) == 0)
{
return true;
}
else
{
throw new SFException(ErrorCode.INVALID_VALUE_CONVERT, logicalTypeStr,
"Boolean", val.toPlainString());
}
}
}