package org.jeecg.modules.system.aspect;
|
|
import cn.hutool.core.util.ArrayUtil;
|
|
import java.util.*;
|
import java.util.function.BiConsumer;
|
|
/**
|
* 批量查询器
|
*/
|
public class FieldBatchQuery {
|
|
// key=查询要素=table+paramName， value=查询条件
|
private Map<String, QueryParam> queryBatch = new HashMap<>();
|
|
/**
|
* 便利查询要素
|
* @param action
|
*/
|
public void forEach(BiConsumer<? super String, ? super QueryParam> action) {
|
this.queryBatch.forEach(action);
|
}
|
|
/**
|
* 查询表数量
|
* @return
|
*/
|
public int size() {
|
return queryBatch.size();
|
}
|
|
/**
|
* 查找查询结果
|
* @param tableName
|
* @param paramName
|
* @param paramValue
|
* @param column
|
* @return
|
*/
|
public Object findResult(String tableName, String[] paramName, String[] paramValue, String column) {
|
String queryKey = tableName + "()" + ArrayUtil.join(paramName, ",");
|
if(!this.queryBatch.containsKey(queryKey)) return null;
|
|
return this.queryBatch.get(queryKey).findResult(paramValue, column);
|
}
|
|
/**
|
* 保存查询条件
|
* @param tableName 查询表名
|
* @param textName 查询字段名称
|
* @param paramName 查询参数名称
|
* @param paramValue 查询参数值
|
*/
|
public void put(String tableName, String textName, String[] paramName, String[] paramValue) {
|
String queryKey = tableName + "()" + ArrayUtil.join(paramName, ",");
|
|
QueryParam queryParams;
|
if(this.queryBatch.containsKey(queryKey)){
|
queryParams = this.queryBatch.get(queryKey);
|
}else{
|
queryParams = new QueryParam(tableName, paramName);
|
this.queryBatch.put(queryKey, queryParams);
|
}
|
|
// 添加查询结果column名称
|
queryParams.addTextName(textName);
|
// 添加查询参数
|
queryParams.addParamValue(paramValue);
|
}
|
|
public static class QueryParam {
|
private String queryTable;
|
private Set<String> columns = new HashSet<>();
|
private Set<String> paramValueSet = new HashSet<>();
|
private String[] paramName;
|
private List<Map> queryResult = new ArrayList<>();
|
private Map<String, Integer> resultIndex = new HashMap<>();
|
|
public QueryParam(String tableName, String[] column) {
|
this.queryTable = tableName;
|
this.paramName = column;
|
}
|
|
public void addTextName(String textName) {
|
this.columns.add(textName);
|
}
|
|
public void addParamValue(String[] paramValue) {
|
paramValueSet.add(ArrayUtil.join(paramValue, ","));
|
}
|
|
public String getQueryTable() {
|
return queryTable;
|
}
|
|
public String[] getColumns() {
|
String[] ret = new String[this.columns.size()];
|
this.columns.toArray(ret);
|
return ret;
|
}
|
|
public String getParamName() {
|
return "concat(" + ArrayUtil.join(this.paramName, ",',',") + ")";
|
}
|
|
public String[] getParamValues() {
|
String[] ret = new String[this.paramValueSet.size()];
|
this.paramValueSet.toArray(ret);
|
return ret;
|
}
|
|
public void setQueryResult(List<Map> queryResult) {
|
this.queryResult = queryResult;
|
}
|
|
public Object findResult(String[] paramValue, String column) {
|
if(paramValue == null) return null;
|
String value = ArrayUtil.join(paramValue, ",");
|
|
if(this.resultIndex.containsKey(value)){
|
return this.queryResult.get(this.resultIndex.get(value)).get(column);
|
}
|
|
for (int i = 0; i < this.queryResult.size(); i++) {
|
Map<String, Object> map = queryResult.get(i);
|
if(value.equals(map.get("paramValue"))){
|
this.resultIndex.put(value, i);
|
return map.get(column);
|
}
|
}
|
|
return null;
|
}
|
}
|
}
|
