io.github.jinghui70.rainbow.dbaccess.mapper.SingleColumnFieldRowMapper Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of rainbow-dbaccess Show documentation

Rainbow DBAccess

There is a newer version: 5.2.1

package io.github.jinghui70.rainbow.dbaccess.mapper;

import io.github.jinghui70.rainbow.dbaccess.fieldmapper.FieldMapper;
import org.springframework.jdbc.IncorrectResultSetColumnCountException;
import org.springframework.jdbc.core.RowMapper;

import java.sql.ResultSet;
import java.sql.ResultSetMetaData;
import java.sql.SQLException;

public class SingleColumnFieldRowMapper implements RowMapper {

    private final FieldMapper fieldMapper;

    public SingleColumnFieldRowMapper(FieldMapper fieldMapper) {
        this.fieldMapper = fieldMapper;
    }

    public T mapRow(ResultSet rs, int rowNum) throws SQLException {
        // Validate column count.
        ResultSetMetaData metaData = rs.getMetaData();
        int nrOfColumns = metaData.getColumnCount();
        if (nrOfColumns != 1) {
            throw new IncorrectResultSetColumnCountException(1, nrOfColumns);
        }
        // Extract column value from JDBC ResultSet.
        return fieldMapper.formDB(rs, 1);
    }
}