package com.walker.jdbc.sqlgen;
|
|
import com.walker.jdbc.SqlAndParameters;
|
import org.apache.commons.lang3.ArrayUtils;
|
|
import java.util.ArrayList;
|
import java.util.HashMap;
|
import java.util.List;
|
import java.util.Map;
|
|
/**
|
* update 语句生成工具
|
*
|
* @author 时克英
|
* @date 2017-3-32
|
*/
|
public class UpdateBuilder extends AbstractSqlBuilder {
|
private final String table;
|
private final List<String> setColumns = new ArrayList<>();
|
private final List<Object> setValues = new ArrayList<>();
|
|
private final List<String> whereColumns = new ArrayList<>();
|
private final List<Object> whereValues = new ArrayList<>();
|
|
/**
|
* 根据表名构造更新语句生成工具
|
*
|
* @param table 表名
|
*/
|
public UpdateBuilder(String table) {
|
this.table = table;
|
}
|
|
/**
|
* 设置update语句更新的字段值
|
*
|
* @param column 字段名
|
* @param value 字段值
|
* @return UpdateBuilder对象
|
*/
|
public UpdateBuilder set(String column, Object value) {
|
return this.set(column, value, true);
|
}
|
|
/**
|
* 设置update语句更新的字段值
|
*
|
* @param column 字段名
|
* @param value 字段值
|
* @param enable 是否更新
|
* @return UpdateBuilder对象
|
*/
|
public UpdateBuilder set(String column, Object value, boolean enable) {
|
if (enable) {
|
this.setColumns.add(column);
|
this.setValues.add(value);
|
}
|
return this;
|
}
|
|
/**
|
* 设置更新时的where条件字段
|
*
|
* @param column 条件字段名
|
* @param value 字段值
|
* @return UpdateBuilder对象
|
*/
|
public UpdateBuilder where(String column, Object value) {
|
this.whereColumns.add(column);
|
this.whereValues.add(value);
|
return this;
|
}
|
|
|
/**
|
* @see AbstractSqlBuilder#genMapSql()
|
*/
|
@Override
|
public SqlAndParameters<Map<String, Object>> genMapSql() {
|
Map<String, Object> parametersMap = new HashMap<>();
|
|
StringBuilder sb = new StringBuilder("update ").append(this.table);
|
sb.append(" set ");
|
for (int i = 0; i < this.setColumns.size(); i++) {
|
if (i > 0) {
|
sb.append(", ");
|
}
|
sb.append(this.setColumns.get(i)).append("=:").append(this.setColumns.get(i) + "_");
|
parametersMap.put(this.setColumns.get(i) + "_", getNotEmptyValue(this.setValues.get(i)));
|
}
|
sb.append(" where ");
|
for (int i = 0; i < this.whereColumns.size(); i++) {
|
if (i > 0) {
|
sb.append(" and ");
|
}
|
if (isSqlNull(whereValues.get(i))) {
|
sb.append(this.whereColumns.get(i)).append(" is null");
|
} else {
|
sb.append(this.whereColumns.get(i)).append("=:").append(this.whereColumns.get(i));
|
parametersMap.put(this.whereColumns.get(i), this.whereValues.get(i));
|
}
|
}
|
return new SqlAndParameters<>(sb.toString(), parametersMap);
|
}
|
|
/**
|
* @see AbstractSqlBuilder#genMapSql(String, Map< String, Object> )
|
*/
|
@Override
|
public SqlAndParameters<Map<String, Object>> genMapSql(String where, Map<String, Object> parametersMap) {
|
Map<String, Object> temp = new HashMap<>();
|
|
StringBuilder sb = new StringBuilder("update ").append(this.table);
|
sb.append(" set ");
|
for (int i = 0; i < this.setColumns.size(); i++) {
|
if (i > 0) {
|
sb.append(", ");
|
}
|
sb.append(this.setColumns.get(i)).append("=:").append(this.setColumns.get(i) + "_");
|
temp.put(this.setColumns.get(i) + "_", getNotEmptyValue(this.setValues.get(i)));
|
}
|
sb.append(" ").append(where);
|
|
temp.putAll(parametersMap);
|
|
return new SqlAndParameters<>(sb.toString(), temp);
|
}
|
|
/**
|
* @see AbstractSqlBuilder#genArraySql(String, Object[])
|
*/
|
@Override
|
public SqlAndParameters<Object[]> genArraySql(String where, Object[] parametersArray) {
|
Object[] temp = new Object[]{};
|
|
StringBuilder sb = new StringBuilder("update ").append(this.table);
|
sb.append(" set ");
|
for (int i = 0; i < this.setColumns.size(); i++) {
|
if (i > 0) {
|
sb.append(", ");
|
}
|
sb.append(this.setColumns.get(i)).append("=?");
|
temp = ArrayUtils.add(temp, getNotEmptyValue(this.setValues.get(i)));
|
}
|
sb.append(" ").append(where);
|
temp = ArrayUtils.addAll(temp, parametersArray);
|
|
return new SqlAndParameters<>(sb.toString(), temp);
|
}
|
|
/**
|
* @see AbstractSqlBuilder#genArraySql()
|
*/
|
@Override
|
public SqlAndParameters<Object[]> genArraySql() {
|
Object[] parametersArray = new Object[]{};
|
|
StringBuilder sb = new StringBuilder("update ").append(this.table);
|
sb.append(" set ");
|
for (int i = 0; i < this.setColumns.size(); i++) {
|
if (i > 0) {
|
sb.append(", ");
|
}
|
sb.append(this.setColumns.get(i)).append("=?");
|
parametersArray = ArrayUtils.add(parametersArray, getNotEmptyValue(this.setValues.get(i)));
|
}
|
sb.append(" where ");
|
for (int i = 0; i < this.whereColumns.size(); i++) {
|
if (i > 0) {
|
sb.append(" and ");
|
}
|
if (isSqlNull(whereValues.get(i))) {
|
sb.append(this.whereColumns.get(i)).append(" is null");
|
} else {
|
sb.append(this.whereColumns.get(i)).append("=?");
|
parametersArray = ArrayUtils.add(parametersArray, this.whereValues.get(i));
|
}
|
}
|
return new SqlAndParameters<>(sb.toString(), parametersArray);
|
}
|
}
|
