use of io.shardingjdbc.core.parsing.parser.exception.SQLParsingException in project sharding-jdbc by shardingjdbc.
the class MySQLLimitClauseParser method getLimitWithOffset.
private Limit getLimitWithOffset(final int index, final int valueBeginPosition, final int value, final boolean isParameterForValue, final SelectStatement selectStatement) {
int offsetBeginPosition = lexerEngine.getCurrentToken().getEndPosition();
int offsetValue = -1;
int offsetIndex = -1;
boolean isParameterForOffset = false;
if (lexerEngine.equalAny(Literals.INT)) {
offsetValue = Integer.parseInt(lexerEngine.getCurrentToken().getLiterals());
offsetBeginPosition = offsetBeginPosition - (offsetValue + "").length();
} else if (lexerEngine.equalAny(Symbol.QUESTION)) {
offsetIndex = -1 == index ? selectStatement.getParametersIndex() : index + 1;
offsetBeginPosition--;
isParameterForOffset = true;
} else {
throw new SQLParsingException(lexerEngine);
}
lexerEngine.nextToken();
if (isParameterForOffset) {
selectStatement.increaseParametersIndex();
} else {
selectStatement.getSqlTokens().add(new OffsetToken(offsetBeginPosition, offsetValue));
}
if (isParameterForValue) {
selectStatement.increaseParametersIndex();
} else {
selectStatement.getSqlTokens().add(new RowCountToken(valueBeginPosition, value));
}
Limit result = new Limit(DatabaseType.MySQL);
result.setRowCount(new LimitValue(value, index, false));
result.setOffset(new LimitValue(offsetValue, offsetIndex, true));
return result;
}
use of io.shardingjdbc.core.parsing.parser.exception.SQLParsingException in project sharding-jdbc by shardingjdbc.
the class SQLServerTopClauseParser method parse.
/**
* Parse top.
*
* @param selectStatement select statement
*/
public void parse(final SelectStatement selectStatement) {
if (!lexerEngine.skipIfEqual(SQLServerKeyword.TOP)) {
return;
}
int beginPosition = lexerEngine.getCurrentToken().getEndPosition();
if (!lexerEngine.skipIfEqual(Symbol.LEFT_PAREN)) {
beginPosition = lexerEngine.getCurrentToken().getEndPosition() - lexerEngine.getCurrentToken().getLiterals().length();
}
SQLExpression sqlExpression = basicExpressionParser.parse(selectStatement);
lexerEngine.skipIfEqual(Symbol.RIGHT_PAREN);
LimitValue rowCountValue;
if (sqlExpression instanceof SQLNumberExpression) {
int rowCount = ((SQLNumberExpression) sqlExpression).getNumber().intValue();
rowCountValue = new LimitValue(rowCount, -1, false);
selectStatement.getSqlTokens().add(new RowCountToken(beginPosition, rowCount));
} else if (sqlExpression instanceof SQLPlaceholderExpression) {
rowCountValue = new LimitValue(-1, ((SQLPlaceholderExpression) sqlExpression).getIndex(), false);
} else {
throw new SQLParsingException(lexerEngine);
}
lexerEngine.unsupportedIfEqual(SQLServerKeyword.PERCENT);
lexerEngine.skipIfEqual(DefaultKeyword.WITH, SQLServerKeyword.TIES);
if (null == selectStatement.getLimit()) {
Limit limit = new Limit(DatabaseType.SQLServer);
limit.setRowCount(rowCountValue);
selectStatement.setLimit(limit);
} else {
selectStatement.getLimit().setRowCount(rowCountValue);
}
}
Aggregations