org.simpleflatmapper.datastax.impl.getter.DatastaxGenericBigIntegerGetter Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of sfm-datastax Show documentation

Cassandra Datastax SFM supports.

There is a newer version: 8.2.3

package org.simpleflatmapper.datastax.impl.getter;

import com.datastax.driver.core.DataType;
import com.datastax.driver.core.GettableByIndexData;
import org.simpleflatmapper.datastax.DataTypeHelper;
import org.simpleflatmapper.reflect.Getter;

import java.math.BigInteger;

public class DatastaxGenericBigIntegerGetter implements Getter {

    private final int index;
    private final DataType.Name dataTypeName;

    public DatastaxGenericBigIntegerGetter(int index, DataType dataType) {
        this.index = index;
        this.dataTypeName = validateName(dataType);
    }

    private DataType.Name validateName(DataType dataType) {
        if (DataTypeHelper.isNumber(dataType)) {
            return dataType.getName();
        }
        throw new IllegalArgumentException("DataType " + dataType + "is not a number");
    }

    @Override
    public BigInteger get(GettableByIndexData target) throws Exception {
        if (target.isNull(index)) {
            return null;
        }
        switch (dataTypeName) {
            case BIGINT:
            case COUNTER:
                return BigInteger.valueOf(target.getLong(index));
            case VARINT:
                return target.getVarint(index);
            case INT:
                return BigInteger.valueOf(target.getInt(index));
            case DECIMAL:
                return target.getDecimal(index).toBigInteger();
            case FLOAT:
                return BigInteger.valueOf((long) target.getFloat(index));
            case DOUBLE:
                return BigInteger.valueOf((long) target.getDouble(index));
            case SMALLINT:
                return BigInteger.valueOf((long) target.getShort(index));
            case TINYINT:
                return BigInteger.valueOf((long) target.getByte(index));
            case TIME:
                return BigInteger.valueOf((long) target.getTime(index));
        }

        return null;
    }
}