All Downloads are FREE. Search and download functionalities are using the official Maven repository.
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.
com.neko233.toolchain.sql.ConditionGenerator Maven / Gradle / Ivy
package com.neko233.toolchain.sql;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.collections4.CollectionUtils;
import java.util.*;
import java.util.stream.Collectors;
/**
* SQL 条件生成器
*
* @author SolarisNeko on 2022-01-01
**/
@Slf4j
public class ConditionGenerator {
public static String condition(String columnName, SqlOperation operation, Object value) {
return condition(columnName, operation, value, true);
}
@SuppressWarnings("uncheck")
public static String condition(String columnName, SqlOperation operation, Object value, boolean addAnd) {
String operateWithSpace = chooseSqlOperate(operation);
String tempValue;
if (value instanceof Collection) {
checkCollectionTypeByIn(operation, operateWithSpace);
Collection collection = (Collection) value;
if (CollectionUtils.isEmpty(collection)) {
return "";
}
List itemConvertList = (List) collection.stream()
.map(item -> {
if (item instanceof String) {
String afterItem = ((String) item).replaceAll("'", "''");
return "'" + afterItem + "'";
} else {
return String.valueOf(item);
}
})
.collect(Collectors.toList());
tempValue = String.join(",", itemConvertList);
} else {
tempValue = String.valueOf(value);
if (value instanceof String) {
tempValue = "'" + value + "'";
}
}
String sql;
if (Objects.equals(SqlOperation.IN, operation) || Objects.equals(SqlOperation.NOT_IN, operation)) {
sql = columnName + operateWithSpace + "(" + tempValue + ")";
} else {
sql = columnName + operateWithSpace + tempValue;
}
if (addAnd) {
return "and " + sql;
} else {
return sql;
}
}
private static void checkCollectionTypeByIn(SqlOperation operation, String operateWithSpace) {
if (Objects.equals(operation, SqlOperation.IN) || Objects.equals(operation, SqlOperation.NOT_IN)) {
// ok, write hard
} else {
throw new RuntimeException(String.format("When your value type is Collection, SqlOperation must is 'in', 'not in'. your operation = %s", operateWithSpace));
}
}
public static String condition(String columnName, SqlOperation operation, Object... valueList) {
return condition(columnName, operation, Arrays.asList(valueList), true);
}
public static String condition(String columnName, SqlOperation operation, Object[] valueList, boolean addAnd) {
return condition(columnName, operation, Arrays.asList(valueList), addAnd);
}
public static String condition(String columnName, SqlOperation operation, Collection valueList, boolean addAnd) {
if (CollectionUtils.isEmpty(valueList)) {
log.error(" you input empty value list for column name = {}", columnName);
return "";
}
String operateWithSpace = chooseSqlOperate(operation);
checkCollectionTypeByIn(operation, operateWithSpace);
Set dataSet = valueList.stream()
.map(value -> {
if (value == null) {
return "null";
}
if (value instanceof String) {
return "'" + value + "'";
} else {
return String.valueOf(value);
}
}).collect(Collectors.toSet());
String tempContent = columnName + operateWithSpace + "(" + String.join(",", dataSet) + ") ";
if (addAnd) {
return "and " + tempContent;
} else {
return tempContent;
}
}
/**
* 映射, 工厂方法.
*
* @param operation 操作符
* @return 条件符号
*/
private static String chooseSqlOperate(SqlOperation operation) {
switch (operation) {
case EQ:
return " = ";
case NOT_EQ:
return " != ";
case GT:
return " > ";
case GTE:
return " >= ";
case LT:
return " < ";
case LTE:
return " <= ";
case IN:
return " in ";
case NOT_IN:
return " not in ";
default:
throw new RuntimeException("not found this operate = " + operation);
}
}
}
