cool.scx.jdbc.result_handler.MapHandler Maven / Gradle / Ivy

Go to download

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

SCX JDBC

There is a newer version: 2.7.4

package cool.scx.jdbc.result_handler;

import cool.scx.jdbc.dialect.Dialect;

import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.HashMap;
import java.util.Map;
import java.util.function.Supplier;

/**
 * BeanBuilder interface.
 *
 * @author scx567888
 * @version 0.2.1
 */
record MapHandler(Supplier> mapSupplier) implements ResultHandler> {

    static final MapHandler INSTANCE = new MapHandler();

    public MapHandler() {
        this(HashMap::new);
    }

    @Override
    public Map apply(ResultSet rs, Dialect dialect) throws SQLException {
        var rsm = rs.getMetaData();
        var count = rsm.getColumnCount();
        if (rs.next()) {
            var map = this.mapSupplier.get();
            for (int i = 1; i <= count; i = i + 1) {
                map.put(rsm.getColumnLabel(i), rs.getObject(i));
            }
            return map;
        } else {
            return null;
        }
    }

}