package com.zt.common.db.query;
|
|
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
import com.baomidou.mybatisplus.core.metadata.OrderItem;
|
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
import com.zt.common.constant.Constant;
|
import com.zt.common.db.constant.OP;
|
import com.zt.common.db.constant.OrderType;
|
import org.apache.commons.lang3.StringUtils;
|
|
import java.util.LinkedHashMap;
|
import java.util.List;
|
import java.util.Map;
|
|
public class QueryFilter {
|
/**
|
* 分页组件
|
*/
|
private Page page = new Page();
|
|
private QueryWrapper queryWrapper = new QueryWrapper();
|
/**
|
* 字段参数构建列表
|
*/
|
private Map<String, Object> params = new LinkedHashMap<>();
|
|
public QueryFilter() {
|
}
|
|
public QueryFilter(List<QueryParam> queryParams) {
|
try {
|
for (QueryParam param : queryParams) {
|
Object value = param.getValue();
|
if (value == null || (value instanceof String && StringUtils.isBlank((String) value))) {
|
continue;// k空值跳过
|
}
|
|
this.params.put(param.getName(), value);
|
|
/** 参数名称^条件 eg：a^EQ 则表示，a字段是varchar类型，条件是eq ^后第一个参数为数据类型 */
|
if (StringUtils.isNotBlank(param.getFormat())) {// 指定格式
|
String[] aryParamKey = param.getFormat().split("\\^");
|
if (aryParamKey.length != 2) {
|
return;
|
}
|
String name = aryParamKey[0], op = aryParamKey[1];
|
if (OP.EQ.equals(op)) { // 等于
|
queryWrapper.eq(name, value);
|
} else if (OP.LESS.equals(op)) { // 小于
|
queryWrapper.lt(name, value);
|
} else if (OP.GT.equals(op)) { // 大于
|
queryWrapper.gt(name, value);
|
} else if (OP.LE.equals(op)) { // 小于等于
|
queryWrapper.le(name, value);
|
} else if (OP.GE.equals(op)) { // 大于等于
|
queryWrapper.ge(name, value);
|
} else if (OP.LIKE.equals(op)) { // 相似
|
queryWrapper.like(name, value);
|
this.params.put(param.getName(), "%" + value.toString() + "%");
|
} else if (OP.LEFT_LIKE.equals(op)) { // 左相似
|
queryWrapper.likeLeft(name, value);
|
this.params.put(param.getName(), "%" + value.toString());
|
} else if (OP.RIGHT_LIKE.equals(op)) { // 右相似
|
queryWrapper.likeRight(name, value);
|
this.params.put(param.getName(), value.toString() + "%");
|
} else if (OP.IN.equals(op)) { // 在...中
|
queryWrapper.in(name, value.toString().split(","));
|
} else if (OP.NOT_IN.equals(op)) { // 不在...中
|
queryWrapper.notIn(name, value.toString().split(","));
|
}
|
}
|
}
|
|
String offset = String.valueOf(params.get(Constant.Q.PAGE));
|
String limit = String.valueOf(params.get(Constant.Q.LIMIT));
|
if (StringUtils.isNotBlank(offset) && StringUtils.isNotBlank(limit)) {
|
Page page = new Page(Long.parseLong(offset), Long.parseLong(limit));
|
// 分页参数
|
params.put(Constant.Q.PAGE, page);
|
// 分页对象
|
this.setPage(page);
|
}
|
|
// 排序字段
|
String orderField = (String) params.get(Constant.Q.ORDER_FIELD);
|
String order = (String) params.get(Constant.Q.ORDER);
|
|
// 前端字段排序
|
if (StringUtils.isNotBlank(orderField) && StringUtils.isNotBlank(order)) {
|
if (OrderType.fromString(order) == OrderType.ASC) {
|
queryWrapper.orderByAsc(orderField);
|
} else {
|
queryWrapper.orderByDesc(orderField);
|
}
|
}
|
} catch (Exception e) {
|
}
|
}
|
|
public Page getPage() {
|
return page;
|
}
|
|
public void setPage(Page page) {
|
this.page = page;
|
}
|
|
public void addParam(String key, Object obj) {
|
this.params.put(key, obj);
|
}
|
|
public Map<String, Object> getParams() {
|
return params;
|
}
|
|
/**
|
* 给工作流用的
|
*
|
* @return
|
*/
|
public Map<String, Object> getQueryParams() {
|
Map<String, Object> params = this.getParams();
|
|
// 构建动态条件SQL
|
String dynamicWhereSql = queryWrapper.getCustomSqlSegment();
|
if (StringUtils.isNotEmpty(dynamicWhereSql)) {
|
params.put("whereSql", dynamicWhereSql.replace("ew.paramNameValuePairs.", "").substring("where".length()));
|
}
|
params.putAll(queryWrapper.getParamNameValuePairs());
|
|
Page page = (Page) params.get(Constant.Q.PAGE);
|
if (page != null) {
|
List<OrderItem> orderItems = page.getOrders();
|
if (orderItems.size() > 0) {
|
StringBuilder sb = new StringBuilder();
|
for (OrderItem orderItem : orderItems) {
|
sb.append(orderItem.getColumn()).append(" ").append(orderItem.isAsc() ? "" : "desc").append(",");
|
}
|
sb.deleteCharAt(sb.length() - 1);
|
params.put("orderBySql", sb.toString());
|
}
|
}
|
return params;
|
}
|
|
public QueryWrapper getWrapper() {
|
return queryWrapper;
|
}
|
|
public <T> List<T> getPageList(List<T> list) {
|
return new PageList<>(list, page.getTotal());
|
}
|
|
public <T> List<T> getPageList(List<T> list, int total) {
|
return new PageList<>(list, total);
|
}
|
}
|
