UpdateSQLAnalyser.java

package org.opencloudb.mpp;

import java.sql.SQLSyntaxErrorException;
import java.util.HashMap;
import java.util.LinkedHashMap;
import java.util.Map;
import java.util.Set;

import com.akiban.sql.StandardException;
import com.akiban.sql.parser.ColumnReference;
import com.akiban.sql.parser.QueryTreeNode;
import com.akiban.sql.parser.ResultColumn;
import com.akiban.sql.parser.ResultColumnList;
import com.akiban.sql.parser.SelectNode;
import com.akiban.sql.parser.UpdateNode;
import com.akiban.sql.parser.ValueNode;

/**
 * update sql analyser
 * 
 * @author wuzhih
 * 
 */
// update A set A.qcye=B.qcye from B where A.kmdm=B.kmdm and A.fmonth=B.fmonth
// and A.fmonth=0
// UPDATE A SET HIGH=B.NEW FROM A LEFT JOIN B ON (A.HIGH=B.OLD)
// update a set HIGH=b.NEW from @SPEC1 a join @tmpDOT b on a.HIGH=b.OLD
// Update HouseInfo Set UpdateTime = '"&Now()&"',I_Valid='"&I_Valid&"' Where
// I_ID In ("&I_ID&")"
// update a set HIGH=b.NEW from SPEC1 a,tmpDOT b where a.high=b.old
// update a set high = (select new from tmpdot where old=a.high ) from spec1 a

public class UpdateSQLAnalyser {

	public static UpdateParsInf analyse(QueryTreeNode ast)
			throws SQLSyntaxErrorException {
		UpdateNode updateNode = (UpdateNode) ast;
		String targetTable = updateNode.getTargetTableName().getTableName()
				.toUpperCase();
		UpdateParsInf parsInf = new UpdateParsInf();
		ShardingParseInfo ctx = null;
		parsInf.tableName = targetTable;
		SelectNode selNode = (SelectNode) updateNode.getResultSetNode();
		ResultColumnList updateColumsLst = selNode.getResultColumns();
		int updateColumnSize = updateColumsLst.size();
		Map<String, String> colMap = new HashMap<String, String>(
				updateColumnSize);
		for (int i = 0; i < updateColumnSize; i++) {
			ResultColumn column = updateColumsLst.get(i);
			ColumnReference colRef = column.getReference();
			String colTableName = colRef.getTableName();
			if (colTableName != null
					&& !colTableName.toUpperCase().equals(targetTable)) {
				throw new SQLSyntaxErrorException(
						"update multi table not supported");
			}

			ValueNode valNode = column.getExpression();
			if (valNode instanceof ColumnReference) {
				String bColTableName = ((ColumnReference) valNode)
						.getTableName();
				if (bColTableName != null
						&& !bColTableName.equalsIgnoreCase(colTableName)) {
					// A.col=B.col
					if (ctx == null) {
						ctx = new ShardingParseInfo();
						parsInf.ctx = ctx;
					}
					// and B table info
					Map<String, Set<ColumnRoutePair>> tableCondMap = new LinkedHashMap<String, Set<ColumnRoutePair>>();
					ctx.tablesAndCondtions.put(bColTableName.toUpperCase(),
							tableCondMap);
				}
			}
			String columName = colRef.getColumnName().toUpperCase();
			colMap.put(columName, "?");
		}
		parsInf.columnPairMap = colMap;

		if (selNode.getWhereClause() != null) {
			// anlayse where condition
			if (ctx == null) {
				ctx = new ShardingParseInfo();
				parsInf.ctx = ctx;
			}
			Map<String, Set<ColumnRoutePair>> tableCondMap = new LinkedHashMap<String, Set<ColumnRoutePair>>();
			ctx.tablesAndCondtions.put(targetTable, tableCondMap);
			try {
				SelectSQLAnalyser.analyseWhereCondition(parsInf, false,
						targetTable, selNode.getWhereClause());
			} catch (StandardException e) {
				throw new SQLSyntaxErrorException(e);
			}
			parsInf.ctx = ctx;
		}
		return parsInf;
	}

}