com.jfirer.jsql.transfer.resultset.impl.EnumOrdinalTransfer Maven / Gradle / Ivy

Go to download
package com.jfirer.jsql.transfer.resultset.impl;

import com.jfirer.jsql.transfer.resultset.ResultSetTransfer;
import com.jfirer.baseutil.reflect.ReflectUtil;

import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.Map;

public class EnumOrdinalTransfer implements ResultSetTransfer
{
    private Enum[] instances;

    @Override
    public Object transfer(ResultSet resultSet) throws SQLException
    {
        int ordinal = resultSet.getInt(1);
        if ( resultSet.wasNull() )
        {
            return null;
        }
        return instances[ordinal];
    }

    @SuppressWarnings("unchecked")
    @Override
    public ResultSetTransfer initialize(Class type)
    {
        Map> allEnumInstances = ReflectUtil.getAllEnumInstances((Class>) type);
        Enum[] instances = new Enum[allEnumInstances.size()];
        for (Enum each : allEnumInstances.values())
        {
            instances[each.ordinal()] = each;
        }
        this.instances = instances;
        return this;
    }

}