package rd.dao.type;

import lombok.extern.slf4j.Slf4j;
import org.apache.ibatis.type.BaseTypeHandler;
import org.apache.ibatis.type.JdbcType;
import org.apache.ibatis.type.MappedJdbcTypes;

import java.sql.*;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;

@Slf4j
//@MappedTypes(List<Short>.class)
@MappedJdbcTypes(JdbcType.ARRAY)
public class ShortListTypeHandler extends BaseTypeHandler<List<Short>> {

    @Override
    public void setNonNullParameter(PreparedStatement ps, int i, List<Short> parameter, JdbcType jdbcType) throws SQLException {
        String typeName = "integer";

        Connection conn = ps.getConnection();
        Array array = conn.createArrayOf(typeName, parameter.toArray());
        ps.setArray(i, array);
    }

    @Override
    public List<Short> getNullableResult(ResultSet rs, String columnName) throws SQLException {
        return getArray(rs.getArray(columnName));
    }

    @Override
    public List<Short> getNullableResult(ResultSet rs, int columnIndex) throws SQLException {
        return getArray(rs.getArray(columnIndex));
    }

    @Override
    public List<Short> getNullableResult(CallableStatement cs, int columnIndex) throws SQLException {
        return getArray(cs.getArray(columnIndex));
    }

    private List<Short> getArray(Array array) {
        if (array == null) {
            return null;
        }
        try {
            var arr = array.getArray();
            var result = new ArrayList<Short>();

            if (arr instanceof Short[] shortArr) {
                Collections.addAll(result, shortArr);
            } else if (arr instanceof Integer[] intArr) {
                for (Integer intVal : intArr) {
                    result.add(intVal.shortValue());
                }
            } else if (arr instanceof Long[] longArr) {
                for (Long longVal : longArr) {
                    result.add(longVal.shortValue());
                }
            }

            return result;
        } catch (Exception e) {
            log.error("类型转换出错：", e);
        }

        return null;
    }
}
