Java ValuesClause类代码示例

import com.alibaba.druid.sql.ast.statement.SQLInsertStatement.ValuesClause; //导入依赖的package包/类
public void cloneTo(SQLInsertInto x) {
    if (tableSource != null) {
        x.setTableSource(tableSource.clone());
    }
    for (SQLExpr column : columns) {
        SQLExpr column2 = column.clone();
        column2.setParent(x);
        x.columns.add(column2);
    }
    if (query != null) {
        x.setQuery(query.clone());
    }
    for (ValuesClause v : valuesList) {
        ValuesClause v2 = v.clone();
        v2.setParent(x);
        x.valuesList.add(v2);
    }
}
 

import com.alibaba.druid.sql.ast.statement.SQLInsertStatement.ValuesClause; //导入依赖的package包/类
public void setValues(ValuesClause values) {
    if (valuesList.size() == 0) {
        valuesList.add(values);
    } else {
        valuesList.set(0, values);
    }
}
 

import com.alibaba.druid.sql.ast.statement.SQLInsertStatement.ValuesClause; //导入依赖的package包/类
public boolean visit(SQLNullExpr x) {
    if (this.parameterized
            && x.getParent() instanceof ValuesClause) {
        print('?');
        incrementReplaceCunt();

        if(this.parameters != null){
            this.getParameters().add(null);
        }
        return false;
    }

    print0(ucase ? "NULL" : "null");
    return false;
}
 

import com.alibaba.druid.sql.ast.statement.SQLInsertStatement.ValuesClause; //导入依赖的package包/类
private static Set<Object> evalInsert(SQLParsedResult parseResult, String column, List<Object> params) {
		Set<Object> evalSet = new LinkedHashSet<Object>();
		MySqlInsertStatement stmt = (MySqlInsertStatement) parseResult.getStmt();

		List<SQLExpr> columns = stmt.getColumns();
		List<ValuesClause> valuesList = stmt.getValuesList();

		if (valuesList.size() > 1) {
			throw new ShardRouterException("Multipal rows insertion is currently unsupported!");
		}

		ValuesClause values = valuesList.get(0);
		for (int i = 0; i < columns.size(); i++) {
			SQLName columnObj = (SQLName) columns.get(i);
			if (evalColumn(columnObj.getSimpleName(), column)) {
				SQLExpr sqlExpr = values.getValues().get(i);
				if (sqlExpr instanceof SQLVariantRefExpr) {
					SQLVariantRefExpr ref = (SQLVariantRefExpr) sqlExpr;
					evalSet.add(params.get(ref.getIndex()));
//					evalSet.add(params.get(i));
				} else if (sqlExpr instanceof SQLValuableExpr) {
					evalSet.add(((SQLValuableExpr) sqlExpr).getValue());
				}
				break;
			}
		}

		return evalSet;
	}
 

import com.alibaba.druid.sql.ast.statement.SQLInsertStatement.ValuesClause; //导入依赖的package包/类
@Override
public void route(SystemConfig sysConfig, SchemaConfig schema, int sqlType,
		String realSQL, String charset, ServerConnection sc,
		LayerCachePool cachePool) {
	int rs = ServerParse.parse(realSQL);
	this.sqltype = rs & 0xff;
	this.sysConfig=sysConfig; 
	this.schema=schema;
	this.charset=charset; 
	this.sc=sc;	
	this.cachePool=cachePool;	
	
	try {
		MySqlStatementParser parser = new MySqlStatementParser(realSQL);	 
		SQLStatement statement = parser.parseStatement();
		MySqlInsertStatement insert = (MySqlInsertStatement)statement;
		if(insert.getValuesList()!=null){
			String tableName = StringUtil.getTableName(realSQL).toUpperCase();
			TableConfig tableConfig = schema.getTables().get(tableName);
			String primaryKey = tableConfig.getPrimaryKey();//获得表的主键字段
			
			SQLIdentifierExpr sqlIdentifierExpr = new SQLIdentifierExpr();
			sqlIdentifierExpr.setName(primaryKey);
			insert.getColumns().add(sqlIdentifierExpr);
			
			if(sequenceHandler == null){
				int seqHandlerType = MycatServer.getInstance().getConfig().getSystem().getSequnceHandlerType();
				switch(seqHandlerType){
					case SystemConfig.SEQUENCEHANDLER_MYSQLDB:
						sequenceHandler = IncrSequenceMySQLHandler.getInstance();
						break;
					case SystemConfig.SEQUENCEHANDLER_LOCALFILE:
						sequenceHandler = IncrSequencePropHandler.getInstance();
						break;
					case SystemConfig.SEQUENCEHANDLER_LOCAL_TIME:
						sequenceHandler = IncrSequenceTimeHandler.getInstance();
						break;
					case SystemConfig.SEQUENCEHANDLER_ZK_DISTRIBUTED:
						sequenceHandler = DistributedSequenceHandler.getInstance(MycatServer.getInstance().getConfig().getSystem());
						break;
					case SystemConfig.SEQUENCEHANDLER_ZK_GLOBAL_INCREMENT:
						sequenceHandler = IncrSequenceZKHandler.getInstance();
						break;
					default:
						throw new java.lang.IllegalArgumentException("Invalid sequnce handler type "+seqHandlerType);
				}
			}
			
			for(ValuesClause vc : insert.getValuesList()){
				SQLIntegerExpr sqlIntegerExpr = new SQLIntegerExpr();
				long value = sequenceHandler.nextId(tableName.toUpperCase());
				sqlIntegerExpr.setNumber(value);//插入生成的sequence值
				vc.addValue(sqlIntegerExpr);
			}
			
			String insertSql = insert.toString();
			this.executeSql = insertSql;
		}
		
	} catch (Exception e) {
		LOGGER.error("BatchInsertSequence.route(......)",e);
	}
}
 

import com.alibaba.druid.sql.ast.statement.SQLInsertStatement.ValuesClause; //导入依赖的package包/类
public ValuesClause getValues() {
    if (valuesList.size() == 0) {
        return null;
    }
    return valuesList.get(0);
}
 

import com.alibaba.druid.sql.ast.statement.SQLInsertStatement.ValuesClause; //导入依赖的package包/类
public List<ValuesClause> getValuesList() {
    return valuesList;
}
 

import com.alibaba.druid.sql.ast.statement.SQLInsertStatement.ValuesClause; //导入依赖的package包/类
public void addValueCause(ValuesClause valueClause) {
    if (valueClause != null) {
        valueClause.setParent(this);
    }
    valuesList.add(valueClause);
}
 

import com.alibaba.druid.sql.ast.statement.SQLInsertStatement.ValuesClause; //导入依赖的package包/类
@Override
public boolean visit(ValuesClause x) {
    return true;
}
 

import com.alibaba.druid.sql.ast.statement.SQLInsertStatement.ValuesClause; //导入依赖的package包/类
private String convertInsertSQL(SchemaInfo schemaInfo, MySqlInsertStatement insert, String originSql, TableConfig tc,
                                boolean isGlobalCheck) throws SQLNonTransientException {

    StructureMeta.TableMeta orgTbMeta = DbleServer.getInstance().getTmManager().getSyncTableMeta(schemaInfo.getSchema(), schemaInfo.getTable());
    if (orgTbMeta == null)
        return originSql;

    boolean isAutoIncrement = tc.isAutoIncrement();

    if (isGlobalCheck && !GlobalTableUtil.isInnerColExist(schemaInfo, orgTbMeta)) {
        if (!isAutoIncrement) {
            return originSql;
        } else {
            isGlobalCheck = false;
        }
    }

    StringBuilder sb = new StringBuilder(200);
    sb.append("insert ");
    if (insert.isIgnore()) {
        sb.append("ignore ");
    }
    sb.append("into ");
    sb.append(schemaInfo.getTable());

    List<SQLExpr> columns = insert.getColumns();

    int autoIncrement = -1;
    int idxGlobal = -1;
    int colSize;
    // insert without columns :insert into t values(xxx,xxx)
    if (columns == null || columns.size() <= 0) {
        if (isAutoIncrement) {
            autoIncrement = getPrimaryKeyIndex(schemaInfo, tc.getPrimaryKey());
        }
        colSize = orgTbMeta.getColumnsList().size();
        idxGlobal = getIdxGlobalByMeta(isGlobalCheck, orgTbMeta, sb, colSize);
    } else {
        genColumnNames(tc, isGlobalCheck, isAutoIncrement, sb, columns);
        colSize = columns.size();
        if (isAutoIncrement) {
            autoIncrement = columns.size();
            sb.append(",").append(tc.getPrimaryKey());
            colSize++;
        }
        if (isGlobalCheck) {
            idxGlobal = isAutoIncrement ? columns.size() + 1 : columns.size();
            sb.append(",").append(GlobalTableUtil.GLOBAL_TABLE_CHECK_COLUMN);
            colSize++;
        }
        sb.append(")");
    }

    sb.append(" values");
    String tableKey = StringUtil.getFullName(schemaInfo.getSchema(), schemaInfo.getTable());
    List<ValuesClause> vcl = insert.getValuesList();
    if (vcl != null && vcl.size() > 1) { // batch insert
        for (int j = 0; j < vcl.size(); j++) {
            if (j != vcl.size() - 1)
                appendValues(tableKey, vcl.get(j).getValues(), sb, autoIncrement, idxGlobal, colSize).append(",");
            else
                appendValues(tableKey, vcl.get(j).getValues(), sb, autoIncrement, idxGlobal, colSize);
        }
    } else {
        List<SQLExpr> values = insert.getValues().getValues();
        appendValues(tableKey, values, sb, autoIncrement, idxGlobal, colSize);
    }

    List<SQLExpr> dku = insert.getDuplicateKeyUpdate();
    if (dku != null && dku.size() > 0) {
        genDuplicate(isGlobalCheck, sb, dku);
    }
    return RouterUtil.removeSchema(sb.toString(), schemaInfo.getSchema());
}
 

import com.alibaba.druid.sql.ast.statement.SQLInsertStatement.ValuesClause; //导入依赖的package包/类
@Override
public void endVisit(ValuesClause x) {

}