import com.szkunton.common.config.JdbcQuery; import java.util.ArrayList; import java.util.Collection; import java.util.HashMap; import java.util.List; import java.util.Map; import java.util.function.Function; import java.util.stream.Collectors; import org.springframework.util.CollectionUtils; public class QueryWhereBuilder { private Map<String, Object> argMap; private StringBuilder whereBuilder; public QueryWhereBuilder() { } public static QueryWhereBuilder where() { QueryWhereBuilder builder = new QueryWhereBuilder(); builder.setArgMap(new HashMap()); builder.setWhereBuilder(new StringBuilder()); return builder; } private void append(String w) { if (this.whereBuilder.length() == 0) { this.whereBuilder.append(" where ").append(w); } else { this.whereBuilder.append(" and ").append(w); } } public QueryWhereBuilder equal(boolean condition, String column, Object value) { if (condition) { this.append(column + " = :" + column); this.argMap.put(column, value); } return this; } public QueryWhereBuilder like(boolean condition, String column, Object value) { if (condition) { this.append(column + " like :" + column); this.argMap.put(column, value); } return this; } public QueryWhereBuilder or(boolean condition1, String value1, boolean condition2, String value2) { return this.or(condition1, value1, condition2, value2, false, (String)null); } public QueryWhereBuilder or(boolean condition1, String value1, boolean condition2, String value2, boolean condition3, String value3) { List<String> values = new ArrayList(); if (condition1) { values.add(value1); } if (condition2) { values.add(value2); } if (condition3) { values.add(value3); } if (values.size() > 0) { String v = (String)values.stream().collect(Collectors.joining(" or ")); this.append("(" + v + ")"); } return this; } public QueryWhereBuilder isNull(boolean condition, String column) { if (condition) { this.append(column + " is null "); } return this; } public QueryWhereBuilder isNotNull(boolean condition, String column) { if (condition) { this.append(column + " is not null "); } return this; } public <T> QueryWhereBuilder in(boolean condition, String column, Collection<T> value) { if (condition && !CollectionUtils.isEmpty(value)) { this.append(column + " in(':" + column + "')"); String in = (String)value.stream().map(String::valueOf).collect(Collectors.joining("','")); this.argMap.put(column, in); } return this; } public <T> QueryWhereBuilder greaterThan(boolean condition, String column, Object value) { if (condition && value != null) { this.append(column + " > :" + column); this.argMap.put(column, value); } return this; } public <T> QueryWhereBuilder greaterThanOrEqualTo(boolean condition, String column, Object value) { if (condition && value != null) { this.append(column + " >= :" + column); this.argMap.put(column, value); } return this; } public <T> QueryWhereBuilder lessThan(boolean condition, String column, Object value) { if (condition && value != null) { this.append(column + " < :" + column); this.argMap.put(column, value); } return this; } public <T> QueryWhereBuilder lessThanOrEqualTo(boolean condition, String column, Object value) { if (condition && value != null) { this.append(column + " <= :" + column); this.argMap.put(column, value); } return this; } public <T> QueryWhereBuilder orderBy(String column, String value) { this.whereBuilder.append(" order by " + column + " " + value); return this; } public <T> QueryWhereBuilder groupBy(String value) { this.whereBuilder.append(" group by " + value); return this; } public QueryWhereBuilder select(String value) { this.whereBuilder.insert(0, value); return this; } public JdbcQuery build() { return new JdbcQuery(this.whereBuilder.toString(), this.argMap); } public static void main(String[] args) { } public void setArgMap(final Map<String, Object> argMap) { this.argMap = argMap; } public void setWhereBuilder(final StringBuilder whereBuilder) { this.whereBuilder = whereBuilder; } }