group.rober.dataform.handler.impl.BeanDataHandler Maven / Gradle / Ivy

Go to download
package group.rober.dataform.handler.impl;

import group.rober.dataform.exception.DataFormException;
import group.rober.dataform.model.DataForm;
import group.rober.dataform.model.DataFormElement;
import group.rober.runtime.kit.*;
import group.rober.runtime.lang.PairBond;
import group.rober.sql.core.DataAccessor;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.jdbc.core.RowMapper;

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

public abstract class BeanDataHandler {
    protected Logger logger = LoggerFactory.getLogger(getClass());

    @Autowired
    protected DataAccessor dataAccessor;

    public DataAccessor getDataAccessor() {
        return dataAccessor;
    }

    public void setDataAccessor(DataAccessor dataAccessor) {
        this.dataAccessor = dataAccessor;
    }

    protected Class getFormClass(DataForm dataForm){
        ValidateKit.notNull(dataForm.getDataModel(),"显示模板{0}数据模型的数据类型为JavaBean，但没有配置数据实体类",dataForm.getId());
        Class clazz = null;
        try {
            clazz = (Class) Class.forName(dataForm.getDataModel());
        } catch (ClassNotFoundException e) {
            throw new DataFormException(e,"显示模板[{0}]的数据实体类[{1}]不存在",dataForm.getId(),dataForm.getDataModel());
        }
        return clazz;
    }

    private Map buildColumnIndexMap(Map columnIndexMap,ResultSetMetaData meta) throws SQLException {
        int count = meta.getColumnCount();
        for(int i=1;i<=count;i++){
            String column = StringKit.nvl(meta.getColumnName(i),meta.getColumnLabel(i));
            columnIndexMap.put(column,i);
        }
        return columnIndexMap;
    }

    protected RowMapper getRowMapper(DataForm dataForm,Class clazz){
        T rowTemplate = BeanUtils.instantiateClass(clazz);
        List  elements = dataForm.getElements();

        Map columnIndexMap = new HashMap();
        RowMapper rowMapper = new RowMapper() {
            public T mapRow(ResultSet rs, int rowNum) throws SQLException {
                if(columnIndexMap.isEmpty()){
                    buildColumnIndexMap(columnIndexMap,rs.getMetaData());
                }

                T row = BeanKit.deepClone(rowTemplate);

                elements.forEach(element->{
                    String code = StringKit.nvl(element.getCode(),StringKit.underlineToCamel(element.getColumn()));
//                    String column = StringKit.nvl(element.getColumn(),StringKit.camelToUnderline(code));
                    String column = element.getColumn();
                    if((SQLKit.isConstColumn(column) || StringKit.isBlank(column))
                            && StringKit.isNotBlank(code)){
                        column = StringKit.camelToUnderline(code);
                    }

                    if(SQLKit.isSelect(column)){    //使用子查询的情况
                        column = StringKit.camelToUnderline(code);
                    }

                    if(!columnIndexMap.containsKey(column))return;
                    Integer index = columnIndexMap.get(column);
                    try {
                        Object value = JdbcKit.getResultSetValue(rs,index);
                        BeanKit.setPropertyValue(row,code,value);
                    } catch (SQLException e) {
                        throw new DataFormException("",e);
                    }
                });
                //处理显示模板上没有的字段
//                fillBeanValueWithoutDataForm(dataForm,columnIndexMap,rs,row);

                return row;
            }
        };

        return rowMapper;
    }


    /**
     * 把显示模板上没有的字段，补到数值上来
     * @param dataForm
     */
    protected void fillBeanValueWithoutDataForm(DataForm dataForm,Map columnIndexMap,ResultSet rs,T row){
        List items = dataForm.getQuery().getSelectItems();
        items.forEach(item->{
            String alias = (String)item.getLeft();
            String column = (String)item.getRight();
            DataFormElement element = dataForm.getElement(alias);
            if(element == null)element = dataForm.getElement(column);
            //查出来，但是没有模板要素的处理
            if(!columnIndexMap.containsKey(column))return;
            Integer index = columnIndexMap.get(column);
            if(element == null){
                try {
                    Object value = JdbcKit.getResultSetValue(rs,index);
                    BeanKit.setPropertyValue(row,alias,value);
                } catch (SQLException e) {
                    throw new DataFormException("",e);
                }
            }

        });



    }

}