Java SQLObject类代码示例

import com.alibaba.druid.sql.ast.SQLObject; //导入依赖的package包/类
public static Object eval(String dbType, SQLObject sqlObject, List<Object> parameters, boolean throwError) {
    SQLEvalVisitor visitor = createEvalVisitor(dbType);
    visitor.setParameters(parameters);

    Object value;
    if (sqlObject instanceof SQLValuableExpr) {
        value = ((SQLValuableExpr) sqlObject).getValue();
    } else {
        sqlObject.accept(visitor);

        value = getValue(sqlObject);

        if (value == null) {
            if (throwError && !sqlObject.getAttributes().containsKey(EVAL_VALUE)) {
                throw new DruidRuntimeException("eval error : " + SQLUtils.toSQLString(sqlObject, dbType));
            }
        }
    }

    return value;
}
 

import com.alibaba.druid.sql.ast.SQLObject; //导入依赖的package包/类
private Comparable<?> evalExpression(final DatabaseType databaseType, final SQLObject sqlObject,
		final List<Object> parameters) {
	if (sqlObject instanceof SQLMethodInvokeExpr) {
		// TODO 解析函数中的sharingValue不支持
		return null;
	}
	Object result = SQLEvalVisitorUtils.eval(databaseType.name().toLowerCase(), sqlObject, parameters, false);
	if (null == result) {
		return null;
	}
	if (result instanceof Comparable<?>) {
		return (Comparable<?>) result;
	}
	// TODO 对于NULL目前解析为空字符串,此处待考虑解决方法
	return "";
}
 

import com.alibaba.druid.sql.ast.SQLObject; //导入依赖的package包/类
public boolean visit(SQLPropertyExpr x) {
    String ownerName = x.getOwnernName();
    if (ownerName == null) {
        return super.visit(x);
    }

    for (SQLObject parent = x.getParent(); parent != null; parent = parent.getParent()) {
        if (parent instanceof SQLSelectQueryBlock) {
            SQLSelectQueryBlock queryBlock = (SQLSelectQueryBlock) parent;
            SQLTableSource tableSource = queryBlock.findTableSource(ownerName);
            if (tableSource == null) {
                continue;
            }

            String alias = tableSource.computeAlias();
            if (tableSource != null
                    && ownerName.equalsIgnoreCase(alias)
                    && !ownerName.equals(alias)) {
                x.setOwner(alias);
            }

            break;
        }
    }

    return super.visit(x);
}
 

import com.alibaba.druid.sql.ast.SQLObject; //导入依赖的package包/类
public boolean visit(SQLSubqueryTableSource x) {
    String subViewName = generateSubViewName();

    SQLObject parent = x.getParent();
    if(parent instanceof SQLSelectQueryBlock) {
        SQLSelectQueryBlock queryBlock = (SQLSelectQueryBlock) parent;
        queryBlock.setFrom(subViewName, x.getAlias());
    } else if(parent instanceof SQLJoinTableSource) {
        SQLJoinTableSource join = (SQLJoinTableSource) parent;
        if (join.getLeft() == x) {
            join.setLeft(subViewName, x.getAlias());
        } else if (join.getRight() == x) {
            join.setRight(subViewName, x.getAlias());
        }
    }

    SQLCreateViewStatement stmt = new SQLCreateViewStatement();

    stmt.setName(generateSubViewName());

    SQLSelect select = x.getSelect();
    stmt.setSubQuery(select);

    targetList.add(0, stmt);

    stmt.accept(new FromSubqueryResolver(targetList, viewName));

    return false;
}
 

import com.alibaba.druid.sql.ast.SQLObject; //导入依赖的package包/类
public static Object getValue(SQLObject sqlObject) {
    if (sqlObject instanceof SQLNumericLiteralExpr) {
        return ((SQLNumericLiteralExpr) sqlObject).getNumber();
    }

    return sqlObject.getAttributes().get(EVAL_VALUE);
}
 

import com.alibaba.druid.sql.ast.SQLObject; //导入依赖的package包/类
private Column getColumnByExpr(SQLBetweenExpr betweenExpr, Map<String, String> aliasMap) {
    if (betweenExpr.getTestExpr() != null) {
        String tableName = null;
        String column = null;
        if (betweenExpr.getTestExpr() instanceof SQLPropertyExpr) { //field has alias
            tableName = ((SQLIdentifierExpr) ((SQLPropertyExpr) betweenExpr.getTestExpr()).getOwner()).getName();
            column = ((SQLPropertyExpr) betweenExpr.getTestExpr()).getName();
            SQLObject query = this.subQueryMap.get(tableName);
            if (query == null) {
                if (aliasMap.containsKey(tableName)) {
                    tableName = aliasMap.get(tableName);
                }
                return new Column(tableName, column);
            }
            return handleSubQueryColumn(tableName, column);
        } else if (betweenExpr.getTestExpr() instanceof SQLIdentifierExpr) {
            column = ((SQLIdentifierExpr) betweenExpr.getTestExpr()).getName();
            tableName = getOwnerTableName(betweenExpr, column);
        }
        String table = tableName;
        if (aliasMap.containsKey(table)) {
            table = aliasMap.get(table);
        }

        if (variants.containsKey(table)) {
            return null;
        }

        if (table != null && !"".equals(table)) {
            return new Column(table, column);
        }
    }
    return null;
}
 

import com.alibaba.druid.sql.ast.SQLObject; //导入依赖的package包/类
/**
 * get table name of field in between expr
 *
 * @param betweenExpr
 * @param column
 * @return
 */
private String getOwnerTableName(SQLBetweenExpr betweenExpr, String column) {
    if (tableStats.size() == 1) { //only has 1 table
        return tableStats.keySet().iterator().next().getName();
    } else if (tableStats.size() == 0) { //no table
        return "";
    } else { // multi tables
        for (Column col : columns.keySet()) {
            if (col.getName().equals(column)) {
                return col.getTable();
            }
        }

        //parser from parent
        SQLObject parent = betweenExpr.getParent();
        if (parent instanceof SQLBinaryOpExpr) {
            parent = parent.getParent();
        }

        if (parent instanceof MySqlSelectQueryBlock) {
            MySqlSelectQueryBlock select = (MySqlSelectQueryBlock) parent;
            if (select.getFrom() instanceof SQLJoinTableSource) {
                SQLJoinTableSource joinTableSource = (SQLJoinTableSource) select.getFrom();
                //FIXME :left as driven table
                return joinTableSource.getLeft().toString();
            } else if (select.getFrom() instanceof SQLExprTableSource) {
                return select.getFrom().toString();
            }
        } else if (parent instanceof SQLUpdateStatement) {
            SQLUpdateStatement update = (SQLUpdateStatement) parent;
            return update.getTableName().getSimpleName();
        } else if (parent instanceof SQLDeleteStatement) {
            SQLDeleteStatement delete = (SQLDeleteStatement) parent;
            return delete.getTableName().getSimpleName();
        }
    }
    return "";
}
 

import com.alibaba.druid.sql.ast.SQLObject; //导入依赖的package包/类
public static Object eval(String dbType, SQLObject sqlObject, List<Object> parameters, boolean throwError) {
    ActionMySqlEvalVisitorImpl visitor = new ActionMySqlEvalVisitorImpl();
    visitor.setParameters(parameters);
    sqlObject.accept(visitor);
    Object value = getValue(sqlObject);
    if (value == null && throwError && !sqlObject.getAttributes().containsKey("eval.value")) {
        throw new DruidRuntimeException("eval error : " + SQLUtils.toSQLString(sqlObject, dbType));
    } else {
        return value;
    }
}
 

import com.alibaba.druid.sql.ast.SQLObject; //导入依赖的package包/类
/**
 * 进行OR表达式的访问.
 *
 * @param sqlObject SQL对象
 * @return OR访问节点
 */
public Optional<AbstractOrASTNode> visitHandle(final SQLObject sqlObject) {
    reset();
    sqlObject.accept(this);
    postVisitHandle();
    return Optional.fromNullable(orASTNode);
}
 

import com.alibaba.druid.sql.ast.SQLObject; //导入依赖的package包/类
/**
     * 从between语句中获取字段所属的表名。
     * 对于容易出现ambiguous的（字段不知道到底属于哪个表），实际应用中必须使用别名来避免歧义
     * @param betweenExpr
     * @param column
     * @return
     */
    private String getOwnerTableName(SQLBetweenExpr betweenExpr,String column) {
        if(tableStats.size() == 1) {//只有一个表，直接返回这一个表名
            return tableStats.keySet().iterator().next().getName();
        } else if(tableStats.size() == 0) {//一个表都没有，返回空串
            return "";
        } else {//多个表名
            for (Column col : columns.keySet())
            {
                if(col.getName().equals(column)) {
                    return col.getTable();
                }
            }
//            for(Column col : columns) {//从columns中找表名
//                if(col.getName().equals(column)) {
//                    return col.getTable();
//                }
//            }

            //前面没找到表名的，自己从parent中解析

            SQLObject parent = betweenExpr.getParent();
            if(parent instanceof SQLBinaryOpExpr)
            {
                parent=parent.getParent();
            }

            if(parent instanceof MySqlSelectQueryBlock) {
                MySqlSelectQueryBlock select = (MySqlSelectQueryBlock) parent;
                if(select.getFrom() instanceof SQLJoinTableSource) {//多表连接
                    SQLJoinTableSource joinTableSource = (SQLJoinTableSource)select.getFrom();
                    return joinTableSource.getLeft().toString();//将left作为主表，此处有不严谨处，但也是实在没有办法，如果要准确，字段前带表名或者表的别名即可
                } else if(select.getFrom() instanceof SQLExprTableSource) {//单表
                    return select.getFrom().toString();
                }
            }
            else if(parent instanceof SQLUpdateStatement) {
                SQLUpdateStatement update = (SQLUpdateStatement) parent;
                return update.getTableName().getSimpleName();
            } else if(parent instanceof SQLDeleteStatement) {
                SQLDeleteStatement delete = (SQLDeleteStatement) parent;
                return delete.getTableName().getSimpleName();
            } else {
                
            }
        }
        return "";
    }
 

import com.alibaba.druid.sql.ast.SQLObject; //导入依赖的package包/类
public SQLObject getOn() {
    return on;
}
 

import com.alibaba.druid.sql.ast.SQLObject; //导入依赖的package包/类
public void setOn(SQLObject on) {
    this.on = on;
    on.setParent(this);
}
 

import com.alibaba.druid.sql.ast.SQLObject; //导入依赖的package包/类
public SQLObject getStorage() {
    return storage;
}
 

import com.alibaba.druid.sql.ast.SQLObject; //导入依赖的package包/类
public void setStorage(SQLObject storage) {
    if (storage != null) {
        storage.setParent(this);
    }
    this.storage = storage;
}
 

import com.alibaba.druid.sql.ast.SQLObject; //导入依赖的package包/类
public Map<String, SQLObject> getTableOptions() {
    return tableOptions;
}