com.github.bingoohuang.utils.lang.DefaultBeanMapper Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of java-utils Show documentation

java utilities

The newest version!

package com.github.bingoohuang.utils.lang;

import com.google.common.base.CaseFormat;
import com.google.common.collect.Maps;
import lombok.SneakyThrows;
import org.apache.commons.beanutils.BeanUtils;
import org.objenesis.ObjenesisStd;

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

public class DefaultBeanMapper implements Jdbc.BeanMapper {
    private final Class beanClass;

    public DefaultBeanMapper(Class beanClass) {
        this.beanClass = beanClass;
    }

    @Override @SneakyThrows
    public T map(ResultSet rs) throws SQLException {
        ResultSetMetaData metaData = rs.getMetaData();
        int columnCount = metaData.getColumnCount();
        Map map = Maps.newHashMap();

        for (int i = 1; i <= columnCount; ++i) {
            Object object = rs.getObject(i);

            String columnLabel = allUpperToLower(metaData.getColumnLabel(i));
            map.put(columnLabel, object);
            map.put(underscoreToCamel(columnLabel), object);


            String columnName = allUpperToLower(metaData.getColumnName(i));
            map.put(columnName, object);
            map.put(underscoreToCamel(columnName), object);
        }

        T instance = new ObjenesisStd().getInstantiatorOf(beanClass).newInstance();

        BeanUtils.populate(instance, map);

        return instance;
    }

    private String allUpperToLower(String str) {
        return (withoutLowercaseLetters(str)) ? str.toLowerCase() : str;
    }

    private boolean withoutLowercaseLetters(String str) {
        for (int i = 0, ii = str.length(); i < ii; ++i) {
            char ch = str.charAt(i);
            if (ch >= 'a' && ch <= 'z') return false;
        }
        return true;
    }

    private String underscoreToCamel(String str) {
        return CaseFormat.LOWER_UNDERSCORE.to(CaseFormat.LOWER_CAMEL, str);
    }
}