Java SQLUpdateStatement类代码示例

import com.alibaba.druid.sql.ast.statement.SQLUpdateStatement; //导入依赖的package包/类
public SQLUpdateStatement createSQLUpdateStatement() {
    if (JdbcConstants.MYSQL.equals(dbType)) {
        return new MySqlUpdateStatement();    
    }
    
    if (JdbcConstants.ORACLE.equals(dbType)) {
        return new OracleUpdateStatement();    
    }
    
    if (JdbcConstants.POSTGRESQL.equals(dbType)) {
        return new PGUpdateStatement();    
    }
    
    if (JdbcConstants.SQL_SERVER.equals(dbType)) {
        return new SQLServerUpdateStatement();    
    }
    
    return new SQLUpdateStatement();
}
 

import com.alibaba.druid.sql.ast.statement.SQLUpdateStatement; //导入依赖的package包/类
public boolean visit(SQLUpdateStatement x) {
    setAliasMap();

    setMode(x, Mode.Update);

    SQLName identName = x.getTableName();
    if (identName != null) {
        String ident = identName.toString();
        String alias = x.getTableSource().getAlias();
        setCurrentTable(ident);

        TableStat stat = getTableStat(ident);
        stat.incrementUpdateCount();

        Map<String, String> aliasMap = getAliasMap();
        
        aliasMap.put(ident, ident);
        if(alias != null) {
        	aliasMap.put(alias, ident);
        }
    } else {
        x.getTableSource().accept(this);
    }

    accept(x.getItems());
    accept(x.getWhere());

    return false;
}
 

import com.alibaba.druid.sql.ast.statement.SQLUpdateStatement; //导入依赖的package包/类
public void throwExceptionParse(String sql, boolean throwException) throws NoSuchMethodException {
    MySqlStatementParser parser = new MySqlStatementParser(sql);
    List<SQLStatement> statementList = parser.parseStatementList();
    SQLStatement sqlStatement = statementList.get(0);
    MySqlUpdateStatement update = (MySqlUpdateStatement) sqlStatement;
    SchemaConfig schemaConfig = mock(SchemaConfig.class);
    Map<String, TableConfig> tables = mock(Map.class);
    TableConfig tableConfig = mock(TableConfig.class);
    String tableName = "hotnews";
    when((schemaConfig).getTables()).thenReturn(tables);
    when(tables.get(tableName)).thenReturn(tableConfig);
    when(tableConfig.getParentTC()).thenReturn(null);
    RouteResultset routeResultset = new RouteResultset(sql, 11);
    Class c = DruidUpdateParser.class;
    Method method = c.getDeclaredMethod("confirmShardColumnNotUpdated", new Class[]{SQLUpdateStatement.class, SchemaConfig.class, String.class, String.class, String.class, RouteResultset.class});
    method.setAccessible(true);
    try {
        method.invoke(c.newInstance(), update, schemaConfig, tableName, "ID", "", routeResultset);
        if (throwException) {
            System.out.println("未抛异常，解析通过则不对！");
            Assert.assertTrue(false);
        } else {
            System.out.println("未抛异常，解析通过，此情况分片字段可能在update语句中但是实际不会被更新");
            Assert.assertTrue(true);
        }
    } catch (Exception e) {
        if (throwException) {
            System.out.println(e.getCause().getClass());
            Assert.assertTrue(e.getCause() instanceof SQLNonTransientException);
            System.out.println("抛异常原因为SQLNonTransientException则正确");
        } else {
            System.out.println("抛异常，需要检查");
            Assert.assertTrue(false);
        }
    }
}
 

import com.alibaba.druid.sql.ast.statement.SQLUpdateStatement; //导入依赖的package包/类
public SQLUpdateBuilderImpl(String sql, String dbType){
    List<SQLStatement> stmtList = SQLUtils.parseStatements(sql, dbType);

    if (stmtList.size() == 0) {
        throw new IllegalArgumentException("not support empty-statement :" + sql);
    }

    if (stmtList.size() > 1) {
        throw new IllegalArgumentException("not support multi-statement :" + sql);
    }

    SQLUpdateStatement stmt = (SQLUpdateStatement) stmtList.get(0);
    this.stmt = stmt;
    this.dbType = dbType;
}
 

import com.alibaba.druid.sql.ast.statement.SQLUpdateStatement; //导入依赖的package包/类
@Override
public SQLUpdateBuilderImpl where(String expr) {
    SQLUpdateStatement update = getSQLUpdateStatement();

    SQLExpr exprObj = SQLUtils.toSQLExpr(expr, dbType);
    update.setWhere(exprObj);

    return this;
}
 

import com.alibaba.druid.sql.ast.statement.SQLUpdateStatement; //导入依赖的package包/类
@Override
public SQLUpdateBuilderImpl whereAnd(String expr) {
    SQLUpdateStatement update = getSQLUpdateStatement();

    SQLExpr exprObj = SQLUtils.toSQLExpr(expr, dbType);
    SQLExpr newCondition = SQLUtils.buildCondition(SQLBinaryOperator.BooleanAnd, exprObj, false, update.getWhere());
    update.setWhere(newCondition);

    return this;
}
 

import com.alibaba.druid.sql.ast.statement.SQLUpdateStatement; //导入依赖的package包/类
@Override
public SQLUpdateBuilderImpl whereOr(String expr) {
    SQLUpdateStatement update = getSQLUpdateStatement();

    SQLExpr exprObj = SQLUtils.toSQLExpr(expr, dbType);
    SQLExpr newCondition = SQLUtils.buildCondition(SQLBinaryOperator.BooleanOr, exprObj, false, update.getWhere());
    update.setWhere(newCondition);

    return this;
}
 

import com.alibaba.druid.sql.ast.statement.SQLUpdateStatement; //导入依赖的package包/类
public SQLUpdateBuilderImpl set(String... items) {
    SQLUpdateStatement update = getSQLUpdateStatement();
    for (String item : items) {
        SQLUpdateSetItem updateSetItem = SQLUtils.toUpdateSetItem(item, dbType);
        update.addItem(updateSetItem);
    }
    
    return this;
}
 

import com.alibaba.druid.sql.ast.statement.SQLUpdateStatement; //导入依赖的package包/类
public SQLUpdateBuilderImpl setValue(String column, Object value) {
    SQLUpdateStatement update = getSQLUpdateStatement();
    
    SQLExpr columnExpr = SQLUtils.toSQLExpr(column, dbType);
    SQLExpr valueExpr = toSQLExpr(value, dbType);
    
    SQLUpdateSetItem item = new SQLUpdateSetItem();
    item.setColumn(columnExpr);
    item.setValue(valueExpr);
    update.addItem(item);
    
    return this;
}
 

import com.alibaba.druid.sql.ast.statement.SQLUpdateStatement; //导入依赖的package包/类
private void updateRecordLog(LogOperate operate,String sql,Object ...args){
    MySqlStatementParser mySqlStatementParser = new MySqlStatementParser(sql);
    SQLUpdateStatement sqlUpdateStatement = mySqlStatementParser.parseUpdateStatement();
    SQLTableSource tableSource = sqlUpdateStatement.getTableSource();
    Assert.notNull(tableSource,"不匹配的操作类型,日志记录失败!");

    List<SQLUpdateSetItem> items = sqlUpdateStatement.getItems();
    JSONObject log = new JSONObject();
    for (int i = 0; i < items.size(); i++) {
        SQLUpdateSetItem sqlUpdateSetItem = items.get(i);
        log.put(sqlUpdateSetItem.getColumn().toString(),args[i]);
    }

    logHandler.handle(operate,sqlUpdateStatement.getTableName().getSimpleName(),sqlUpdateStatement.getTableName().getSimpleName(),log);
}
 

import com.alibaba.druid.sql.ast.statement.SQLUpdateStatement; //导入依赖的package包/类
public void throwExceptionParse(String sql, boolean throwException) throws NoSuchMethodException {
    MySqlStatementParser parser = new MySqlStatementParser(sql);
    List<SQLStatement> statementList = parser.parseStatementList();
    SQLStatement sqlStatement = statementList.get(0);
    MySqlUpdateStatement update = (MySqlUpdateStatement) sqlStatement;
    SchemaConfig schemaConfig = mock(SchemaConfig.class);
    Map<String, TableConfig> tables = mock(Map.class);
    TableConfig tableConfig = mock(TableConfig.class);
    String tableName = "hotnews";
    when((schemaConfig).getTables()).thenReturn(tables);
    when(tables.get(tableName)).thenReturn(tableConfig);
    when(tableConfig.getParentTC()).thenReturn(null);
    RouteResultset routeResultset = new RouteResultset(sql, 11);
    Class c = DruidUpdateParser.class;
    Method method = c.getDeclaredMethod("confirmShardColumnNotUpdated", new Class[]{SQLUpdateStatement.class, SchemaConfig.class, String.class, String.class, String.class, RouteResultset.class});
    method.setAccessible(true);
    try {
        method.invoke(c.newInstance(), update, schemaConfig, tableName, "ID", "", routeResultset);
        if (throwException) {
            System.out.println("Not passed without exception is not correct");
            Assert.assertTrue(false);
        } else {
            System.out.println("Passed without exception. Maybe the partition key exists in update statement,but not update in fact");
            Assert.assertTrue(true);
        }
    } catch (Exception e) {
        if (throwException) {
            System.out.println(e.getCause().getClass());
            Assert.assertTrue(e.getCause() instanceof SQLNonTransientException);
            System.out.println("SQLNonTransientException is expected");
        } else {
            System.out.println("need checked");
            Assert.assertTrue(false);
        }
    }
}
 

import com.alibaba.druid.sql.ast.statement.SQLUpdateStatement; //导入依赖的package包/类
@Override
public boolean visit(SQLUpdateStatement x) {
    setAliasMap();

    setMode(x, Mode.Update);

    SQLName identName = x.getTableName();
    if (identName != null) {
        String ident = identName.toString();
        //
        String alias = x.getTableSource().getAlias();
        setCurrentTable(ident);

        TableStat stat = getTableStat(ident);
        stat.incrementUpdateCount();

        Map<String, String> aliasMap = getAliasMap();
        
        aliasMap.put(ident, ident);
        //
        if(alias != null) {
        	aliasMap.put(alias, ident);
        }
    } else {
        x.getTableSource().accept(this);
    }

    accept(x.getItems());
    accept(x.getWhere());

    return false;
}
 

import com.alibaba.druid.sql.ast.statement.SQLUpdateStatement; //导入依赖的package包/类
@Override
public SqlType getType() {
    if (sqlStatement instanceof SQLSelectStatement)
        return SqlType.SELECT;
    if (sqlStatement instanceof SQLUpdateStatement) return SqlType.UPDATE;

    if (sqlStatement instanceof SQLInsertStatement)

        return SqlType.INSERT;
    if (sqlStatement instanceof SQLDeleteStatement)
        return SqlType.DELETE;
    return null;
}
 

import com.alibaba.druid.sql.ast.statement.SQLUpdateStatement; //导入依赖的package包/类
public SQLStatementType getStatementType() {
    if (sqlStatement instanceof SQLSelectStatement) {
        return SQLStatementType.SELECT;
    }
    if (sqlStatement instanceof SQLInsertStatement) {
        return SQLStatementType.INSERT;
    }
    if (sqlStatement instanceof SQLUpdateStatement) {
        return SQLStatementType.UPDATE;
    }
    if (sqlStatement instanceof SQLDeleteStatement) {
        return SQLStatementType.DELETE;
    }
    throw new SQLParserException("Unsupported SQL statement: [%s]", sqlStatement);
}
 

import com.alibaba.druid.sql.ast.statement.SQLUpdateStatement; //导入依赖的package包/类
private static SQLASTOutputVisitor getSQLVisitor(DatabaseType databaseType,SQLStatement sqlStatement) {
    if (sqlStatement instanceof SQLSelectStatement) {
        return ClassUtil.newInstance(SQLVisitorRegistry.getSelectVistor(databaseType));
    }
    if (sqlStatement instanceof SQLInsertStatement) {
        return ClassUtil.newInstance(SQLVisitorRegistry.getInsertVistor(databaseType));
    }
    if (sqlStatement instanceof SQLUpdateStatement) {
        return ClassUtil.newInstance(SQLVisitorRegistry.getUpdateVistor(databaseType));
    }
    if (sqlStatement instanceof SQLDeleteStatement) {
        return ClassUtil.newInstance(SQLVisitorRegistry.getDeleteVistor(databaseType));
    }
    throw new SQLParserException("Unsupported SQL statement: [%s]", sqlStatement);
}