com.codingapi.springboot.framework.dto.request.SearchRequest Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of springboot-starter Show documentation
Show all versions of springboot-starter Show documentation
springboot-starter project for Spring Boot
package com.codingapi.springboot.framework.dto.request;
import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import jakarta.servlet.http.HttpServletRequest;
import lombok.Getter;
import lombok.Setter;
import org.springframework.data.domain.Sort;
import org.springframework.util.StringUtils;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;
import java.lang.reflect.Field;
import java.util.ArrayList;
import java.util.Base64;
import java.util.Enumeration;
import java.util.List;
/**
* HttpServletRequest 请求参数解析成 PageRequest对象
*/
public class SearchRequest {
@Getter
private int current;
@Getter
private int pageSize;
private final HttpServletRequest request;
private final List removeKeys = new ArrayList<>();
private final PageRequest pageRequest;
public SearchRequest() {
ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.currentRequestAttributes();
this.request = attributes.getRequest();
this.pageRequest = new PageRequest();
}
public void setCurrent(int current) {
this.current = current - 1;
this.removeKeys.add("current");
}
public void setPageSize(int pageSize) {
this.pageSize = pageSize;
this.removeKeys.add("pageSize");
}
public void addSort(Sort sort) {
pageRequest.addSort(sort);
}
public void removeFilter(String key) {
pageRequest.removeFilter(key);
this.removeKeys.add(key);
}
public String getParameter(String key) {
return request.getParameter(key);
}
public String[] getParameterValues(String key) {
return request.getParameterValues(key);
}
public List getParameterNames() {
Enumeration enumeration = request.getParameterNames();
List list = new ArrayList<>();
while (enumeration.hasMoreElements()) {
list.add(enumeration.nextElement());
}
return list;
}
public PageRequest addFilter(String key, Relation relation, Object... value) {
return pageRequest.addFilter(key, relation, value);
}
public PageRequest addFilter(String key, Object... value) {
return pageRequest.addFilter(key, value);
}
public PageRequest andFilter(Filter... filters) {
return pageRequest.andFilter(filters);
}
public PageRequest orFilters(Filter... filters) {
return pageRequest.orFilters(filters);
}
private String decode(String value) {
return new String(Base64.getDecoder().decode(value));
}
static class ClassContent {
private final Class clazz;
private final PageRequest pageRequest;
public ClassContent(Class clazz, PageRequest pageRequest) {
this.clazz = clazz;
this.pageRequest = pageRequest;
}
public void addFilter(String key, Relation relation, String value) {
Class keyClass = getKeyType(key);
Object v = parseObject(value, keyClass);
pageRequest.addFilter(key, relation, v);
}
public void addFilter(String key, String value) {
Class keyClass = getKeyType(key);
Object v = parseObject(value, keyClass);
pageRequest.addFilter(key, Relation.EQUAL, v);
}
private Object parseObject(String value, Class keyClass) {
if (value.getClass().equals(keyClass)) {
return value;
}
return JSON.parseObject(value, keyClass);
}
public void addFilter(String key, List value) {
Class keyClass = getKeyType(key);
pageRequest.addFilter(key, Relation.IN, value.stream()
.map(v -> parseObject(v, keyClass))
.toArray()
);
}
private Class getKeyType(String key) {
String[] keys = key.split("\\.");
Class keyClass = clazz;
for (String k : keys) {
Field[] fields = keyClass.getDeclaredFields();
for (Field field : fields) {
if (field.getName().equals(k)) {
keyClass = field.getType();
break;
}
}
}
return keyClass;
}
}
@Setter
@Getter
static class ParamOperation {
private String key;
private String type;
public Relation getOperation() {
return Relation.valueOf(type);
}
}
private List loadParamOperations() {
String params = request.getParameter("params");
if (StringUtils.hasLength(params)) {
params = decode(params);
if (JSON.isValid(params)) {
removeKeys.add("params");
return JSON.parseArray(params, ParamOperation.class);
}
}
return null;
}
public PageRequest toPageRequest(Class clazz) {
pageRequest.setCurrent(current);
pageRequest.setPageSize(pageSize);
ClassContent content = new ClassContent(clazz, pageRequest);
List loadParams = loadParamOperations();
String sort = request.getParameter("sort");
if (StringUtils.hasLength(sort)) {
sort = decode(sort);
if (JSON.isValid(sort)) {
removeKeys.add("sort");
JSONObject jsonObject = JSON.parseObject(sort);
for (String key : jsonObject.keySet()) {
String value = jsonObject.getString(key);
if ("ascend".equals(value)) {
pageRequest.addSort(Sort.by(key).ascending());
} else {
pageRequest.addSort(Sort.by(key).descending());
}
}
}
}
String filter = request.getParameter("filter");
if (StringUtils.hasLength(filter)) {
filter = decode(filter);
if (JSON.isValid(filter)) {
removeKeys.add("filter");
JSONObject jsonObject = JSON.parseObject(filter);
for (String key : jsonObject.keySet()) {
JSONArray value = jsonObject.getJSONArray(key);
if (value != null && !value.isEmpty()) {
List values = value.stream().map(Object::toString).toList();
content.addFilter(key, values);
}
}
}
}
request.getParameterNames().asIterator().forEachRemaining(key -> {
if (!removeKeys.contains(key)) {
String value = request.getParameter(key);
if (StringUtils.hasLength(value)) {
if (loadParams != null) {
ParamOperation operation = loadParams.stream()
.filter(paramOperation -> paramOperation.getKey().equals(key))
.findFirst()
.orElse(null);
if (operation != null) {
content.addFilter(key, operation.getOperation(), value);
} else {
content.addFilter(key, value);
}
} else {
content.addFilter(key, value);
}
}
}
});
return pageRequest;
}
}