Class Insert

src/main/java/com/sql/parse/statement/impl/druid/Insert.java:26–221

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

 * @desc Insert语法操作类
 */
public class Insert extends com.sql.parse.statement.Insert {
    private SQLInsertStatement insert;

    public Insert(SQLInsertStatement insert) {
        this.insert = insert;
    }

    public Insert(String tableName) {
        SQLInsertStatement insert = new SQLInsertStatement();
        SQLExprTableSource table = new SQLExprTableSource(tableName);
        insert.setTableSource(table);
        this.insert = insert;
    }

    @Override
    public String getSql(boolean safe) {
        if (this.insert.getQuery() == null && (this.insert.getValuesList() == null || this.insert.getValuesList().isEmpty())) {
            throw BaseException.errorCode(SqlParseConstant.CODE_014);
        } else {
            return SQLUtils.toSQLString(this.insert);
        }
    }

    public SQLInsertStatement getInsert() {
        return this.insert;
    }

    /**
     * @title: addColumn
     * @desc 新增insert数据列
     */
    public Insert addColumn(String column) {
        insert.addColumn(ExpressionBuilder.parse(column));
        return this;
    }

    /**
     * @title: addColumn
     * @desc 新增insert数据列
     */
    public Insert addColumn(String column, String value) {
        return addColumn(new InsertItem(column, value));
    }

    /**
     * @title: addColumn
     * @desc 新增insert数据列
     */
    public Insert addColumn(InsertItem insertItem) {
        if (insert.getQuery() == null) {
            insert.addColumn(ExpressionBuilder.parse(insertItem.getColumnName()));
            List<SQLInsertStatement.ValuesClause> values = insert.getValuesList();
            if (values.isEmpty()) {
                SQLInsertStatement.ValuesClause valuesClause = new SQLInsertStatement.ValuesClause();
                valuesClause.addValue(ExpressionBuilder.parse(insertItem.getExpression()));
                values.add(valuesClause);
            } else {
                for (SQLInsertStatement.ValuesClause valuesClause : values) {