JdbcTemplate工具類

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;
    }
}
相關文章
相關標籤/搜索