Examples with SQLNotNullConstraint com.alibaba.druid.sql.ast.statement.SQLNotNullConstraint used on opensource projects

Example 1 with SQLNotNullConstraint

use of com.alibaba.druid.sql.ast.statement.SQLNotNullConstraint in project druid by alibaba.

the class SQLExprParser method parseColumnRest.

public SQLColumnDefinition parseColumnRest(SQLColumnDefinition column) {
    if (lexer.token() == Token.DEFAULT) {
        lexer.nextToken();
        column.setDefaultExpr(bitOr());
        return parseColumnRest(column);
    }
    if (lexer.token() == Token.NOT) {
        lexer.nextToken();
        accept(Token.NULL);
        column.addConstraint(new SQLNotNullConstraint());
        return parseColumnRest(column);
    }
    if (lexer.token() == Token.NULL) {
        lexer.nextToken();
        column.getConstraints().add(new SQLNullConstraint());
        return parseColumnRest(column);
    }
    if (lexer.token == Token.PRIMARY) {
        lexer.nextToken();
        accept(Token.KEY);
        column.addConstraint(new SQLColumnPrimaryKey());
        return parseColumnRest(column);
    }
    if (lexer.token == Token.UNIQUE) {
        lexer.nextToken();
        if (lexer.token() == Token.KEY) {
            lexer.nextToken();
        }
        column.addConstraint(new SQLColumnUniqueKey());
        return parseColumnRest(column);
    }
    if (lexer.token == Token.CONSTRAINT) {
        lexer.nextToken();
        SQLName name = this.name();
        if (lexer.token() == Token.PRIMARY) {
            lexer.nextToken();
            accept(Token.KEY);
            SQLColumnPrimaryKey pk = new SQLColumnPrimaryKey();
            pk.setName(name);
            column.addConstraint(pk);
            return parseColumnRest(column);
        }
        if (lexer.token() == Token.UNIQUE) {
            lexer.nextToken();
            SQLColumnUniqueKey uk = new SQLColumnUniqueKey();
            uk.setName(name);
            column.addConstraint(uk);
            return parseColumnRest(column);
        }
        if (lexer.token() == Token.REFERENCES) {
            lexer.nextToken();
            SQLColumnReference ref = new SQLColumnReference();
            ref.setName(name);
            ref.setTable(this.name());
            accept(Token.LPAREN);
            this.names(ref.getColumns(), ref);
            accept(Token.RPAREN);
            column.addConstraint(ref);
            return parseColumnRest(column);
        }
        if (lexer.token() == Token.NOT) {
            lexer.nextToken();
            accept(Token.NULL);
            SQLNotNullConstraint notNull = new SQLNotNullConstraint();
            notNull.setName(name);
            column.addConstraint(notNull);
            return parseColumnRest(column);
        }
        if (lexer.token == Token.CHECK) {
            SQLColumnCheck check = parseColumnCheck();
            check.setName(name);
            check.setParent(column);
            column.addConstraint(check);
            return parseColumnRest(column);
        }
        if (lexer.token == Token.DEFAULT) {
            lexer.nextToken();
            SQLExpr expr = this.expr();
            column.setDefaultExpr(expr);
            return parseColumnRest(column);
        }
        throw new ParserException("TODO : " + lexer.token() + " " + lexer.stringVal());
    }
    if (lexer.token == Token.CHECK) {
        SQLColumnCheck check = parseColumnCheck();
        column.addConstraint(check);
        return parseColumnRest(column);
    }
    if (lexer.token() == Token.COMMENT) {
        lexer.nextToken();
        column.setComment(primary());
        return parseColumnRest(column);
    }
    return column;
}

Example 2 with SQLNotNullConstraint

use of com.alibaba.druid.sql.ast.statement.SQLNotNullConstraint in project canal by alibaba.

the class MemoryTableMeta method processTableElement.

private void processTableElement(SQLTableElement element, TableMeta tableMeta) {
    if (element instanceof SQLColumnDefinition) {
        FieldMeta fieldMeta = new FieldMeta();
        SQLColumnDefinition column = (SQLColumnDefinition) element;
        String name = getSqlName(column.getName());
        // String charset = getSqlName(column.getCharsetExpr());
        SQLDataType dataType = column.getDataType();
        String dataTypStr = dataType.getName();
        if (StringUtils.equalsIgnoreCase(dataTypStr, "float")) {
            if (dataType.getArguments().size() == 1) {
                int num = Integer.valueOf(dataType.getArguments().get(0).toString());
                if (num > 24) {
                    dataTypStr = "double";
                }
            }
        }
        if (dataType.getArguments().size() > 0) {
            dataTypStr += "(";
            for (int i = 0; i < column.getDataType().getArguments().size(); i++) {
                if (i != 0) {
                    dataTypStr += ",";
                }
                SQLExpr arg = column.getDataType().getArguments().get(i);
                dataTypStr += arg.toString();
            }
            dataTypStr += ")";
        }
        if (dataType instanceof SQLDataTypeImpl) {
            SQLDataTypeImpl dataTypeImpl = (SQLDataTypeImpl) dataType;
            if (dataTypeImpl.isUnsigned()) {
                dataTypStr += " unsigned";
            }
            if (dataTypeImpl.isZerofill()) {
                // 如果设置了zerofill，自动给列添加unsigned属性
                if (!dataTypeImpl.isUnsigned()) {
                    dataTypStr += " unsigned";
                }
                dataTypStr += " zerofill";
            }
        }
        if (column.getDefaultExpr() == null || column.getDefaultExpr() instanceof SQLNullExpr) {
            fieldMeta.setDefaultValue(null);
        } else {
            fieldMeta.setDefaultValue(DruidDdlParser.unescapeQuotaName(getSqlName(column.getDefaultExpr())));
        }
        fieldMeta.setColumnName(name);
        fieldMeta.setColumnType(dataTypStr);
        fieldMeta.setNullable(true);
        List<SQLColumnConstraint> constraints = column.getConstraints();
        for (SQLColumnConstraint constraint : constraints) {
            if (constraint instanceof SQLNotNullConstraint) {
                fieldMeta.setNullable(false);
            } else if (constraint instanceof SQLNullConstraint) {
                fieldMeta.setNullable(true);
            } else if (constraint instanceof SQLColumnPrimaryKey) {
                fieldMeta.setKey(true);
                fieldMeta.setNullable(false);
            } else if (constraint instanceof SQLColumnUniqueKey) {
                fieldMeta.setUnique(true);
            }
        }
        tableMeta.addFieldMeta(fieldMeta);
    } else if (element instanceof MySqlPrimaryKey) {
        MySqlPrimaryKey column = (MySqlPrimaryKey) element;
        List<SQLSelectOrderByItem> pks = column.getColumns();
        for (SQLSelectOrderByItem pk : pks) {
            String name = getSqlName(pk.getExpr());
            FieldMeta field = tableMeta.getFieldMetaByName(name);
            field.setKey(true);
            field.setNullable(false);
        }
    } else if (element instanceof MySqlUnique) {
        MySqlUnique column = (MySqlUnique) element;
        List<SQLSelectOrderByItem> uks = column.getColumns();
        for (SQLSelectOrderByItem uk : uks) {
            String name = getSqlName(uk.getExpr());
            FieldMeta field = tableMeta.getFieldMetaByName(name);
            field.setUnique(true);
        }
    }
}

Example 3 with SQLNotNullConstraint

use of com.alibaba.druid.sql.ast.statement.SQLNotNullConstraint in project druid by alibaba.

the class SQLASTVisitorAdapterTest method test_adapter.

public void test_adapter() throws Exception {
    SQLASTVisitorAdapter adapter = new SQLASTVisitorAdapter();
    new SQLBinaryOpExpr().accept(adapter);
    new SQLInListExpr().accept(adapter);
    new SQLSelectQueryBlock().accept(adapter);
    new SQLDropTableStatement().accept(adapter);
    new SQLCreateTableStatement().accept(adapter);
    new SQLDeleteStatement().accept(adapter);
    new SQLCurrentOfCursorExpr().accept(adapter);
    new SQLInsertStatement().accept(adapter);
    new SQLUpdateStatement().accept(adapter);
    new SQLNotNullConstraint().accept(adapter);
    new SQLMethodInvokeExpr().accept(adapter);
    new SQLCallStatement().accept(adapter);
    new SQLSomeExpr().accept(adapter);
    new SQLAnyExpr().accept(adapter);
    new SQLAllExpr().accept(adapter);
    new SQLDefaultExpr().accept(adapter);
    new SQLCommentStatement().accept(adapter);
    new SQLDropViewStatement().accept(adapter);
    new SQLSavePointStatement().accept(adapter);
    new SQLReleaseSavePointStatement().accept(adapter);
    new SQLCreateDatabaseStatement().accept(adapter);
    new SQLAlterTableDropIndex().accept(adapter);
    new SQLOver().accept(adapter);
    new SQLWithSubqueryClause().accept(adapter);
    new SQLAlterTableAlterColumn().accept(adapter);
    new SQLAlterTableStatement().accept(adapter);
    new SQLAlterTableDisableConstraint().accept(adapter);
    new SQLAlterTableEnableConstraint().accept(adapter);
    new SQLColumnCheck().accept(adapter);
    new SQLExprHint().accept(adapter);
    new SQLAlterTableDropConstraint().accept(adapter);
}