Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
org.springframework.data.r2dbc.support.DslUtils Maven / Gradle / Ivy
Spring Data DSL module for Web querying and Criteria building
package org.springframework.data.r2dbc.support;
import com.fasterxml.jackson.databind.JsonNode;
import kotlin.Pair;
import org.apache.commons.beanutils.ConvertUtils;
import org.apache.commons.codec.binary.Hex;
import org.apache.commons.lang3.ObjectUtils;
import org.apache.commons.lang3.StringUtils;
import org.springframework.data.domain.PageRequest;
import org.springframework.data.domain.Pageable;
import org.springframework.data.domain.Sort;
import org.springframework.data.r2dbc.query.Criteria;
import org.springframework.data.r2dbc.repository.query.Dsl;
import org.springframework.lang.Nullable;
import org.springframework.util.ReflectionUtils;
import java.lang.reflect.Field;
import java.math.BigInteger;
import java.nio.charset.StandardCharsets;
import java.time.*;
import java.util.*;
import java.util.stream.Collectors;
import java.util.stream.Stream;
import static org.apache.commons.lang3.StringUtils.SPACE;
import static org.sevenparadigms.kotlin.common.MethodExtensionsKt.has;
import static org.springframework.data.r2dbc.support.WordUtils.*;
/**
* Utilities for dsl interaction.
*
* @author Lao Tsing
*/
public abstract class DslUtils {
public static final String COMMANDS = "(\\^\\^|!\\^|==|!=|>>|>=|<<|<=|~~|@@)";
public static final String PREFIX = "(!|@|!@)";
public static final String CLEAN = "[^!=>\\^<~@]";
public static final String DOT = ".";
public static final String OR = "(";
public static final String GROUP_COMBINATOR = "\"";
public static final String COMBINATORS = "(\\(|\\)|\")";
public static final String DOT_REGEX = "\\.";
public static final String JSONB = "->>'";
public static final String HASH = "<->";
public enum Fields {createdAt, updatedAt, version, createdBy, updatedBy}
public static String toJsonbPath(final String path) {
if (path.contains(DOT)) {
final var paths = path.split(DOT_REGEX);
final var order = new StringBuilder(paths[0]);
for (int i = 1; i < paths.length; i++) {
if (i < paths.length - 1)
order.append("->'").append(paths[i]).append("'");
else
order.append(JSONB).append(paths[i]).append("'");
}
return order.toString();
} else return path;
}
public static String getJsonName(final String jsonPath) {
if (jsonPath.contains(JSONB)) {
return jsonPath.substring(jsonPath.indexOf(JSONB) + 4, jsonPath.length() - 1);
}
return jsonPath;
}
public static String toJsonbPath(final String dotter, final Class type) {
if (dotter.contains(DOT)) {
final var fieldName = dotter.split(DOT_REGEX)[0];
final var reflectionStorage = FastMethodInvoker.reflectionStorage(type);
for (var field : reflectionStorage) {
if (fieldName.equals(field.getName()) && field.getType() == JsonNode.class) {
return toJsonbPath(dotter);
}
if (fieldName.concat(StringUtils.capitalize(SqlField.id)).equals(field.getName())) {
var json = dotter.substring(dotter.indexOf(".") + 1);
return fieldName + "." + toJsonbPath(json);
}
}
}
return dotter;
}
public static List stringToObject(final String[] list, final String fieldName, final Class type) {
final var result = new ArrayList<>();
for (var it : list) {
result.add(stringToObject(it, fieldName, type));
}
return result;
}
public static Object stringToObject(final String it, final String fieldName, final Class type) {
final var reflectionStorage = FastMethodInvoker.reflectionStorage(type);
for (var field : reflectionStorage) {
if (sqlToCamel(lastOctet(fieldName)).equals(field.getName())) {
switch (field.getType().getSimpleName()) {
case "UUID":
return UUID.fromString(it);
case "OffsetDateTime":
var offset = it.replaceAll(" ", "+");
return OffsetDateTime.parse(offset.endsWith("Z") || offset.contains("+") ? offset : offset + "Z");
case "LocalDateTime":
return LocalDateTime.parse(it);
case "ZonedDateTime":
var zoned = it.replaceAll(" ", "+");
return ZonedDateTime.parse(zoned.endsWith("Z") || zoned.contains("+") ? zoned : zoned + "Z");
case "LocalDate":
return LocalDate.parse(it);
case "Short":
return Short.valueOf(it);
case "Integer":
case "int":
return Integer.valueOf(it);
case "Long":
return Long.parseLong(it);
case "Double":
return Double.valueOf(it);
case "BigInteger":
return BigInteger.valueOf(Long.parseLong(it));
case "Boolean":
return Boolean.valueOf(it);
case "byte[]":
return it.getBytes(StandardCharsets.UTF_8);
}
break;
}
}
return it;
}
public static String binding(String builder, Object target) {
var buildQuery = trimInline(builder);
for (var field : FastMethodInvoker.reflectionStorage(target.getClass())) {
if (buildQuery.contains(Dsl.COLON + field.getName() + Dsl.COLON) && !field.getName().equals(SqlField.id)) {
var value = FastMethodInvoker.getValue(target, field.getName());
String result = null;
if (value != null) {
result = objectToSql(value);
}
buildQuery = buildQuery.replaceAll(Dsl.COLON + field.getName() + Dsl.COLON, result == null ? "null" : result);
}
}
return buildQuery;
}
public static String objectToSql(@Nullable Object value) {
if (value == null) return "null";
if (value instanceof LocalDateTime) return "'" + value + "'::timestamp";
if (value instanceof ZonedDateTime || value instanceof OffsetDateTime) return "'" +
(value.toString().indexOf("[") > 0 ? value.toString().substring(0, value.toString().indexOf("[")) : value.toString()) + "'::timestamp with time zone";
if (value instanceof LocalDate) return "'" + value + "'::date";
if (value instanceof String) return "'" + value + "'";
if (value instanceof UUID) return "'" + value + "'::uuid";
if (value instanceof byte[]) return "decode('" + String.valueOf(Hex.encodeHex((byte[]) value)) + "', 'hex')";
if (value instanceof Enum) return "'" + ((Enum) value).name() + "'";
if (value instanceof JsonNode) return "'" + value.toString().replaceAll("'", "") + "'";
return ConvertUtils.convert(value, String.class).toString();
}
@Nullable
public static Criteria getCriteriaBy(Dsl dsl, Class type, @Nullable List jsonNodeFields) {
Criteria criteriaBy = null;
if (dsl.getQuery() != null && !dsl.getQuery().isEmpty()) {
String[] criterias = dsl.getQuery().split(Dsl.COMMA);
for (String criteria : criterias) {
String[] parts = criteria.split(COMMANDS);
String field = parts[0].replaceAll(PREFIX, "").replaceAll(COMBINATORS, "");
if (jsonNodeFields != null && jsonNodeFields.contains(field)) {
field = toJsonbPath(field, type);
}
Criteria.CriteriaStep step = getCriteriaByCombinators(criteriaBy, field, parts[0]);
String value = parts.length > 1 ? parts[1].replaceAll(COMBINATORS, "") : null;
switch (criteria.replaceAll(CLEAN, "")) {
case Dsl.in:
if (value != null) criteriaBy = step.in(stringToObject(value.split(SPACE), field, type));
break;
case Dsl.notIn:
if (value != null) criteriaBy = step.notIn(stringToObject(value.split(SPACE), field, type));
break;
case Dsl.equal:
if (value != null) criteriaBy = step.is(stringToObject(value, field, type));
break;
case Dsl.notEqual:
if (value != null) criteriaBy = step.not(stringToObject(value, field, type));
break;
case "":
criteriaBy = step.is(true);
break;
case Dsl.not:
criteriaBy = step.not(true);
break;
case Dsl.isNull:
criteriaBy = step.isNull();
break;
case Dsl.notNull:
criteriaBy = step.isNotNull();
break;
case Dsl.greater:
if (value != null) criteriaBy = step.greaterThan(stringToObject(value, field, type));
break;
case Dsl.greaterEqual:
if (value != null) criteriaBy = step.greaterThanOrEquals(stringToObject(value, field, type));
break;
case Dsl.less:
if (value != null) criteriaBy = step.lessThan(stringToObject(value, field, type));
break;
case Dsl.lessEqual:
if (value != null) criteriaBy = step.lessThanOrEquals(stringToObject(value, field, type));
break;
case Dsl.like:
if (value != null) criteriaBy = step.like("%" + value + "%");
break;
default:
criteriaBy = null;
}
}
}
return criteriaBy;
}
public static List getCriteriaFields(Dsl dsl) {
List list = new ArrayList<>();
if (dsl.getQuery() != null && !dsl.getQuery().isEmpty()) {
String[] criterias = dsl.getQuery().split(Dsl.COMMA);
for (String criteria : criterias) {
if (ObjectUtils.isEmpty(criteria) || criteria.contains(Dsl.fts)) continue;
String fieldName = criteria.split(COMMANDS)[0];
list.add(fieldName.replaceAll(PREFIX, "").replaceAll(COMBINATORS, ""));
}
}
return list;
}
public static Map getCriteriaMap(Dsl dsl) {
Map list = new HashMap<>();
if (dsl.getQuery() != null && !dsl.getQuery().isEmpty()) {
String[] criterias = dsl.getQuery().split(Dsl.COMMA);
for (String criteria : criterias) {
if (ObjectUtils.isEmpty(criteria) || criteria.contains(Dsl.fts)) continue;
String[] pair = criteria.split(COMMANDS);
list.put(pair[0].replaceAll(PREFIX, "").replaceAll(COMBINATORS, ""), pair.length > 1 ? pair[1] : null);
}
}
return list;
}
@Nullable
public static Pair getFtsPair(Dsl dsl, Class type) {
if (dsl.getQuery() != null && !dsl.getQuery().isEmpty()) {
String[] criterias = dsl.getQuery().split(Dsl.COMMA);
for (String criteria : criterias) {
if (criteria.contains(Dsl.fts)) {
String[] parts = criteria.split(COMMANDS);
if (parts[0].contains(DOT)) {
Field entityField = ReflectionUtils.findField(type, parts[0].split(DOT_REGEX)[0]);
if (entityField != null && entityField.getType() == JsonNode.class) {
parts[0] = toJsonbPath(parts[0], type);
}
}
return new Pair<>(parts[0], parts[1]);
}
}
}
return null;
}
public static Pageable getPageable(Dsl dsl) {
if (dsl.isPaged())
return PageRequest.of(dsl.getPage(), dsl.getSize(), getSorted(dsl));
else
return Pageable.unpaged();
}
public static Sort getSorted(Dsl dsl) {
if (dsl.isSorted()) {
return Sort.by(Stream.of(dsl.getSort().split(Dsl.COMMA)).map(it -> {
String[] parts = it.split(Dsl.COLON);
String name;
if (parts[0].contains(DOT)) {
name = toJsonbPath(parts[0]);
} else
name = "\"" + parts[0] + "\"";
return new Sort.Order(Sort.Direction.valueOf(parts[1].toUpperCase()), camelToSql(name));
}).collect(Collectors.toList()));
} else return Sort.unsorted();
}
public static void setVersion(Object objectToSave, Field version, Object versionValue) {
if (versionValue instanceof Number) {
var longValue = (Long) ConvertUtils.convert(versionValue, Long.class);
if (version.getType() == Long.class) {
FastMethodInvoker.setValue(objectToSave, version.getName(), longValue + 1);
}
if (version.getType() == Integer.class) {
var value = (Integer) ConvertUtils.convert(longValue + 1, Integer.class);
FastMethodInvoker.setValue(objectToSave, version.getName(), value);
}
}
setNowStamp(objectToSave, version);
}
public static void setNowStamp(Object objectToSave, Field field) {
if (field.getType() == LocalDateTime.class) {
FastMethodInvoker.setValue(objectToSave, field.getName(), LocalDateTime.now(ZoneOffset.UTC));
}
if (field.getType() == ZonedDateTime.class) {
FastMethodInvoker.setValue(objectToSave, field.getName(), ZonedDateTime.now(ZoneOffset.UTC));
}
if (field.getType() == OffsetDateTime.class) {
FastMethodInvoker.setValue(objectToSave, field.getName(), OffsetDateTime.now(ZoneOffset.UTC));
}
}
public static boolean compareDateTime(Object first, Object second) {
if (first instanceof LocalDateTime && second instanceof LocalDateTime) {
return ((LocalDateTime) first).isEqual(((LocalDateTime) second));
}
if (first instanceof ZonedDateTime && second instanceof ZonedDateTime) {
return ((ZonedDateTime) first).isEqual(((ZonedDateTime) second));
}
if (first instanceof OffsetDateTime && second instanceof OffsetDateTime) {
return ((OffsetDateTime) first).isEqual(((OffsetDateTime) second));
}
return dateTimeToString(first).equals(dateTimeToString(second));
}
public static boolean compareDateTime(@Nullable LocalDateTime first, @Nullable LocalDateTime second) {
return first != null && second != null && first.isEqual(second);
}
public static boolean compareDateTime(@Nullable ZonedDateTime first, @Nullable ZonedDateTime second) {
return first != null && second != null && first.isEqual(second);
}
public static boolean compareDateTime(@Nullable OffsetDateTime first, @Nullable OffsetDateTime second) {
return first != null && second != null && first.isEqual(second);
}
public static Set getFields(Object objectToSave, List> names, Class... cls) {
Set fields = new HashSet<>(FastMethodInvoker.getFields(objectToSave.getClass(), null, cls));
for (Enum enumName : names) {
if (has(objectToSave.getClass(), enumName.name())) {
fields.add(FastMethodInvoker.getField(objectToSave.getClass(), enumName.name()));
}
}
return fields;
}
public static Set getFields(Object objectToSave, Enum name, Class... cls) {
return FastMethodInvoker.getFields(objectToSave.getClass(), name.name(), cls);
}
public static Set getPropertyFields(Object objectToSave, String propertyString) {
var result = new HashSet();
if (!propertyString.isEmpty()) {
result.addAll(Arrays.stream(propertyString.split(Dsl.COMMA))
.map(name -> FastMethodInvoker.getField(objectToSave.getClass(), name)).filter(Objects::nonNull)
.collect(Collectors.toSet()));
}
return result;
}
public static String generateHash(Dsl dsl) {
return dsl.getQuery() + HASH + dsl.getPage() + HASH + dsl.getSize() + HASH + dsl.getSort();
}
public static Dsl generateDsl(String hash) {
var arr = hash.split(HASH);
return Dsl.create(arr[0]).pageable(Integer.parseInt(arr[1]), Integer.parseInt(arr[2])).sorting(arr[3]);
}
private static Criteria.CriteriaStep getCriteriaByCombinators(@Nullable Criteria criteriaBy,
String field,
String part) {
boolean hasOrCombinator = part.contains(OR);
if (criteriaBy != null) {
if (hasOrCombinator) {
return criteriaBy.or(camelToSql(field));
} else {
return criteriaBy.and(camelToSql(field));
}
}
return Criteria.where(camelToSql(field));
}
@Nullable
private static Criteria getCriteriaByGroupCombinators(@Nullable Criteria criteriaBy, String part) {
boolean hasOrCombinator = part.contains(OR);
boolean hasGroupCombinator = part.contains(GROUP_COMBINATOR);
List groupCriterias = new ArrayList<>();
if (criteriaBy != null && hasGroupCombinator) {
Criteria previousCriteria = criteriaBy.getPrevious();
while (previousCriteria != null) {
groupCriterias.add(previousCriteria);
previousCriteria = previousCriteria.getPrevious();
}
return hasOrCombinator ? criteriaBy.or(groupCriterias) : criteriaBy.and(groupCriterias);
}
return criteriaBy;
}
}
