Class Update

src/main/java/com/sql/parse/statement/impl/druid/Update.java:26–281

@date 2019年4月30日上午09:58:02 @desc Update语法操作类

 * @desc Update语法操作类
 */
public class Update extends com.sql.parse.statement.Update {
    private SQLUpdateStatement update;

    public Update(SQLUpdateStatement update) {
        this.update = update;
    }

    public Update(String tableName) {
        SQLUpdateStatement update = new SQLUpdateStatement();
        SQLExprTableSource table = new SQLExprTableSource(tableName);
        update.setTableSource(table);
        this.update = update;
    }

    @Override
    public String getSql(boolean safe) {
        Update myUpdate = this.clone();
        /*where条件为空时，添加1=2的条件表达式*/
        if (safe && myUpdate.getUpdate().getWhere() == null) {
            myUpdate.where("1 = 2");
        }
        return SQLUtils.toSQLString(myUpdate.getUpdate());
    }

    public SQLUpdateStatement getUpdate() {
        return this.update;
    }

    /**
     * @title: addColumn
     * @desc 新增update数据
     */
    public Update addColumn(UpdateItem updateItem) {
        SQLUpdateSetItem setItem = new SQLUpdateSetItem();
        setItem.setColumn(ExpressionBuilder.parse(updateItem.getColumnName()));
        setItem.setValue(ExpressionBuilder.parse(updateItem.getExpression()));
        update.addItem(setItem);
        return this;
    }

    public Update addColumn(String column, String value) {
        return addColumn(new UpdateItem(column, value));
    }

    /**
     * @title: where
     * @desc "WHERE"符号
     */
    public Update where(String expr) {
        SQLExpr expression = ExpressionBuilder.parse(expr);
        where(expression);
        return this;
    }

    public Update where(SQLExpr where) {
        SQLExpr oldWhere = update.getWhere();
        if (oldWhere == null) {
            update.setWhere(where);