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

Example 96 with SQLName

use of com.alibaba.druid.sql.ast.SQLName in project druid by alibaba.

the class WallVisitorUtils method checkReadOnly.

public static void checkReadOnly(WallVisitor visitor, SQLTableSource tableSource) {
    if (tableSource instanceof SQLExprTableSource) {
        String tableName = null;
        SQLExpr tableNameExpr = ((SQLExprTableSource) tableSource).getExpr();
        if (tableNameExpr instanceof SQLName) {
            tableName = ((SQLName) tableNameExpr).getSimpleName();
        }
        boolean readOnlyValid = visitor.getProvider().checkReadOnlyTable(tableName);
        if (!readOnlyValid) {
            addViolation(visitor, ErrorCode.READ_ONLY, "table readonly : " + tableName, tableSource);
        }
    } else if (tableSource instanceof SQLJoinTableSource) {
        SQLJoinTableSource join = (SQLJoinTableSource) tableSource;
        checkReadOnly(visitor, join.getLeft());
        checkReadOnly(visitor, join.getRight());
    }
}

Example 97 with SQLName

use of com.alibaba.druid.sql.ast.SQLName in project druid by alibaba.

the class WallVisitorUtils method check.

public static void check(WallVisitor visitor, SQLCreateTableStatement x) {
    String tableName = ((SQLName) x.getName()).getSimpleName();
    WallContext context = WallContext.current();
    if (context != null) {
        WallSqlTableStat tableStat = context.getTableStat(tableName);
        if (tableStat != null) {
            tableStat.incrementCreateCount();
        }
    }
}

Example 98 with SQLName

use of com.alibaba.druid.sql.ast.SQLName in project druid by alibaba.

the class WallVisitorUtils method check.

public static void check(WallVisitor visitor, SQLDropTableStatement x) {
    for (SQLTableSource item : x.getTableSources()) {
        if (item instanceof SQLExprTableSource) {
            SQLExpr expr = ((SQLExprTableSource) item).getExpr();
            String tableName = ((SQLName) expr).getSimpleName();
            WallContext context = WallContext.current();
            if (context != null) {
                WallSqlTableStat tableStat = context.getTableStat(tableName);
                if (tableStat != null) {
                    tableStat.incrementDropCount();
                }
            }
        }
    }
}

Example 99 with SQLName

use of com.alibaba.druid.sql.ast.SQLName in project druid by alibaba.

the class WallVisitorUtils method hasTableSource.

private static boolean hasTableSource(SQLTableSource x) {
    if (x == null) {
        return false;
    }
    if (x instanceof SQLExprTableSource) {
        SQLExpr fromExpr = ((SQLExprTableSource) x).getExpr();
        if (fromExpr instanceof SQLName) {
            String name = fromExpr.toString();
            name = form(name);
            if (name.equalsIgnoreCase("DUAL")) {
                return false;
            }
        }
        return true;
    } else if (x instanceof SQLJoinTableSource) {
        SQLJoinTableSource join = (SQLJoinTableSource) x;
        return hasTableSource(join.getLeft()) || hasTableSource(join.getRight());
    } else if (x instanceof SQLSubqueryTableSource) {
        return hasTableSource(((SQLSubqueryTableSource) x).getSelect().getQuery());
    }
    return false;
}

Example 100 with SQLName

use of com.alibaba.druid.sql.ast.SQLName in project druid by alibaba.

the class WallVisitorUtils method isTopFromDenySchema.

private static boolean isTopFromDenySchema(WallVisitor visitor, SQLMethodInvokeExpr x) {
    SQLObject parent = x.getParent();
    for (; ; ) {
        if (parent instanceof SQLExpr || parent instanceof Item || parent instanceof SQLSelectItem) {
            parent = parent.getParent();
        } else {
            break;
        }
    }
    if (parent instanceof SQLSelectQueryBlock) {
        SQLSelectQueryBlock queryBlock = (SQLSelectQueryBlock) parent;
        if (!(queryBlock.getParent() instanceof SQLSelect)) {
            return false;
        }
        SQLSelect select = (SQLSelect) queryBlock.getParent();
        if (!(select.getParent() instanceof SQLSelectStatement)) {
            return false;
        }
        SQLSelectStatement stmt = (SQLSelectStatement) select.getParent();
        if (stmt.getParent() != null) {
            return false;
        }
        SQLTableSource from = queryBlock.getFrom();
        if (from instanceof SQLExprTableSource) {
            SQLExpr fromExpr = ((SQLExprTableSource) from).getExpr();
            if (fromExpr instanceof SQLName) {
                String fromTableName = fromExpr.toString();
                return visitor.isDenyTable(fromTableName);
            }
        }
        return false;
    }
    return false;
}