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;
}
}