All Downloads are FREE. Search and download functionalities are using the official Maven repository.

titan.lightbatis.service.LightbatisQueryTemplate Maven / Gradle / Ivy

Go to download

通用数据访问构架

There is a newer version: 1.2.0
Show newest version
package titan.lightbatis.service;

import lombok.extern.slf4j.Slf4j;
import org.mybatis.spring.SqlSessionTemplate;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import titan.lightbatis.mapper.MapperManager;
import titan.lightbatis.mybatis.meta.StatementFragment;
import titan.lightbatis.result.Page;
import titan.lightbatis.result.PageList;

import java.util.*;

@Slf4j
@Service
public class LightbatisQueryTemplate {

    @Autowired
    private SqlSessionTemplate sessionTemplate = null;

    /**
     *
     * @param statementId Mybatis 的查询ID
     * @param params 查询的条件参数
     * @param page 分页
     * @param fetchMappingColumn 获取这个查询的映射信息。
     * @return
     */
    public List> list(String statementId, Map params, Page page, boolean fetchMappingColumn) {
        List> tmpList = new ArrayList<>();
        if (page == null) {
            tmpList = sessionTemplate.selectList(statementId, params);
        } else {
            tmpList = sessionTemplate.selectList(statementId, params, page);
        }
        final List> dataList = new ArrayList<>(tmpList);
        if (fetchMappingColumn) {
            //读取所有关联表
             List stmtList = MapperManager.getManager().getStatementMappingColumns(statementId);
            stmtList.forEach( stmt ->{
                String stmtId = stmt.getStatementId();
                Set values = new HashSet<>();

                dataList.forEach(item -> {
                    if (item.containsKey(stmt.getSourceColumn())) {
                        values.add(item.get(stmt.getSourceColumn()));
                    }
                });

                Map stmtParams = new HashMap<>();
                stmtParams.put("values", values);

                String fragmentStmtId = stmt.getStatementId();
                try{
                    //查询关联表的数据
                    List> stmtDataList = sessionTemplate.selectList(fragmentStmtId, stmtParams);
                    stmtDataList.forEach(fragment ->{
                        // 将数据关联到主表上去
                        dataList.stream().filter(data -> data.get(stmt.getSourceColumn()).toString().equals(fragment.get(stmt.getPkColumn()).toString())).forEach( item ->{
                            stmt.getMappingColumns().forEach(colMeta ->{
                               Object colValue = fragment.get(colMeta.getTargetColumn());
                               item.put(colMeta.getColumn(), colValue);
                            });
                        });
                    });
                }catch (Exception ex) {
                    ex.printStackTrace(System.err);
                }
            });
        }
        if (page == null) {
            return  dataList;
        } else {
            PageList plist = new PageList();
            plist.addAll(dataList);
            if (tmpList instanceof PageList) {
                PageList tmpPage = (PageList) tmpList;
                plist.setTotalSize(tmpPage.getTotalSize());
            }
            return plist;
        }

    }
}
    

    

    

            

    

            



    
    






    
© 2015 - 2024 Weber Informatics LLC | Privacy Policy