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.
win.doyto.query.core.CrudBuilder Maven / Gradle / Ivy
package win.doyto.query.core;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.apache.commons.lang3.reflect.FieldUtils;
import win.doyto.query.entity.Persistable;
import java.lang.reflect.Field;
import java.util.*;
import java.util.stream.Collectors;
import java.util.stream.IntStream;
import javax.persistence.Column;
import javax.persistence.Id;
import javax.persistence.Table;
import static org.apache.commons.lang3.StringUtils.SPACE;
import static win.doyto.query.core.CommonUtil.*;
/**
* CrudBuilder
*
* @author f0rb
*/
@SuppressWarnings("squid:CommentedOutCodeLine")
@Slf4j
final class CrudBuilder extends QueryBuilder {
private static final String EQUALS_REPLACE_HOLDER = " = " + REPLACE_HOLDER;
private final Field idField;
private final String tableName;
private final List fields;
private final int fieldsSize;
private final boolean isDynamicTable;
private final String wildInsertValue; // ?, ?, ?
private final String insertColumns;
private final String wildSetClause; // column1 = ?, column2 = ?
private final String whereId;
public CrudBuilder(Class entityClass) {
tableName = entityClass.getAnnotation(Table.class).name();
idField = FieldUtils.getFieldsWithAnnotation(entityClass, Id.class)[0];
whereId = " WHERE " + resolveColumn(idField) + EQUALS_REPLACE_HOLDER;
isDynamicTable = isDynamicTable(tableName);
// init fields
Field[] allFields = FieldUtils.getAllFields(entityClass);
List tempFields = new ArrayList<>(allFields.length);
Arrays.stream(allFields).filter(field -> !ignoreField(field)).forEachOrdered(tempFields::add);
fields = Collections.unmodifiableList(tempFields);
fieldsSize = fields.size();
wildInsertValue = wrapWithParenthesis(StringUtils.join(IntStream.range(0, fieldsSize).mapToObj(i -> REPLACE_HOLDER).collect(Collectors.toList()), SEPARATOR));
List columnList = fields.stream().map(CrudBuilder::resolveColumn).collect(Collectors.toList());
insertColumns = wrapWithParenthesis(StringUtils.join(columnList, SEPARATOR));
wildSetClause = StringUtils.join(columnList.stream().map(c -> c + EQUALS_REPLACE_HOLDER).collect(Collectors.toList()), SEPARATOR);
}
static String resolveColumn(Field field) {
Column column = field.getAnnotation(Column.class);
if (column != null && !column.name().isEmpty()) {
return column.name();
} else {
return convertColumn(field.getName());
}
}
private static String buildInsertSql(String table, String columns, String fields) {
ArrayList insertList = new ArrayList<>();
insertList.add("INSERT INTO");
insertList.add(table);
insertList.add(columns);
insertList.add("VALUES");
insertList.add(fields);
return StringUtils.join(insertList, SPACE);
}
private static String buildUpdateSql(String tableName, String setClauses) {
ArrayList updateList = new ArrayList<>();
updateList.add("UPDATE");
updateList.add(tableName);
updateList.add("SET");
updateList.add(setClauses);
return StringUtils.join(updateList, SPACE);
}
private static void readValueToArgList(List fields, Object entity, List argList) {
fields.stream().map(field -> readFieldGetter(field, entity)).forEach(argList::add);
}
private static void readValueToArgList(List fields, Object entity, List argList, List setClauses) {
for (Field field : fields) {
Object o = readFieldGetter(field, entity);
if (o != null) {
setClauses.add(resolveColumn(field) + EQUALS_REPLACE_HOLDER);
argList.add(o);
}
}
}
private String resolveTableName(E entity) {
return isDynamicTable ? replaceTableName(entity, tableName) : tableName;
}
public String buildCreateAndArgs(E entity, List argList) {
String table = resolveTableName(entity);
readValueToArgList(fields, entity, argList);
return buildInsertSql(table, insertColumns, wildInsertValue);
}
public SqlAndArgs buildCreateAndArgs(Iterable entities) {
Iterator iterator = entities.iterator();
E next = iterator.next();
String table = resolveTableName(next);
StringBuilder insertSql = new StringBuilder(buildInsertSql(table, insertColumns, wildInsertValue));
ArrayList argList = new ArrayList<>();
readValueToArgList(fields, next, argList);
while (iterator.hasNext()) {
E entity = iterator.next();
readValueToArgList(fields, entity, argList);
insertSql.append(", ").append(wildInsertValue);
}
return new SqlAndArgs(insertSql.toString(), argList);
}
public String buildUpdateAndArgs(E entity, List argList) {
String table = resolveTableName(entity);
readValueToArgList(fields, entity, argList);
argList.add(readField(idField, entity));
return buildUpdateSql(table, wildSetClause) + whereId;
}
public SqlAndArgs buildUpdateAndArgs(E entity) {
ArrayList argList = new ArrayList<>();
return new SqlAndArgs(buildUpdateAndArgs(entity, argList), argList);
}
public String buildPatchAndArgs(E entity, List argList) {
String table = resolveTableName(entity);
List setClauses = new ArrayList<>(fieldsSize);
readValueToArgList(fields, entity, argList, setClauses);
return buildUpdateSql(table, StringUtils.join(setClauses, SEPARATOR));
}
public String buildPatchAndArgsWithId(E entity, List argList) {
String sql = buildPatchAndArgs(entity, argList) + whereId;
argList.add(readField(idField, entity));
return sql;
}
public SqlAndArgs buildPatchAndArgsWithId(E entity) {
ArrayList argList = new ArrayList<>();
return new SqlAndArgs(buildPatchAndArgsWithId(entity, argList), argList);
}
public String buildPatchAndArgsWithQuery(E entity, Object query, List argList) {
String sql = buildPatchAndArgs(entity, argList);
return buildWhere(sql, query, argList);
}
public SqlAndArgs buildPatchAndArgsWithQuery(E entity, Object query) {
ArrayList argList = new ArrayList<>();
String sql = buildPatchAndArgsWithQuery(entity, query, argList);
return new SqlAndArgs(sql, argList);
}
public String buildSelectById() {
return "SELECT * FROM " + tableName + whereId;
}
public String buildDeleteById() {
return "DELETE FROM " + tableName + whereId;
}
protected String buildSelectById(E e) {
return "SELECT * FROM " + replaceTableName(e, tableName) + whereId;
}
protected String buildDeleteById(E e) {
return "DELETE FROM " + replaceTableName(e, tableName) + whereId;
}
}
