org.simpleflatmapper.jdbc.impl.getter.ArrayResultSetGetter Maven / Gradle / Ivy

Go to download

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

Java library to map flat record - ResultSet, csv - to java object with minimum configuration and low footprint.

There is a newer version: 9.0.2

package org.simpleflatmapper.jdbc.impl.getter;

import org.simpleflatmapper.converter.Context;
import org.simpleflatmapper.map.getter.ContextualGetter;
import org.simpleflatmapper.reflect.Getter;

import java.sql.Array;
import java.sql.ResultSet;
import java.util.ArrayList;
import java.util.List;

public class ArrayResultSetGetter implements Getter, ContextualGetter {

    public static final int VALUE_INDEX = 2;

    private final int index;
    private final Getter getter;
    private final T[] emptyArray;

    public ArrayResultSetGetter(Class componentType, Getter getter, int index) {
        this.getter = getter;
        this.index = index;
        this.emptyArray = (T[]) java.lang.reflect.Array.newInstance(componentType, 0);
    }

    @Override
    public T[] get(ResultSet resultSet, Context context) throws Exception {
        return get(resultSet);
    }

    @Override
    public T[] get(ResultSet target) throws Exception {
        Array sqlArray = target.getArray(index);

        if (sqlArray != null) {
            List list = new ArrayList();
            ResultSet rs = sqlArray.getResultSet();
            try {
                while (rs.next()) {
                    list.add(getter.get(rs));
                }
            } finally {
                rs.close();
            }
            return list.toArray(emptyArray);
        }
        return null;
    }
}