use of com.dangdang.ddframe.rdb.sharding.parser.result.merger.Limit in project sharding-jdbc by dangdangdotcom.
the class AbstractBaseParseTest method getDataParameter.
private static Object[] getDataParameter(final Assert assertObj) {
Object[] result = new Object[6];
result[0] = assertObj.getId();
result[1] = assertObj.getSql();
result[2] = assertObj.getExpectedSQL();
result[3] = Lists.transform(assertObj.getTables(), new Function<com.dangdang.ddframe.rdb.sharding.parser.jaxb.Table, Table>() {
@Override
public Table apply(final com.dangdang.ddframe.rdb.sharding.parser.jaxb.Table input) {
return new Table(input.getName(), input.getAlias());
}
});
if (null == assertObj.getConditionContexts()) {
result[4] = Collections.<ConditionContext>emptyList();
} else {
result[4] = Lists.transform(assertObj.getConditionContexts(), new Function<com.dangdang.ddframe.rdb.sharding.parser.jaxb.ConditionContext, ConditionContext>() {
@Override
public ConditionContext apply(final com.dangdang.ddframe.rdb.sharding.parser.jaxb.ConditionContext input) {
ConditionContext result = new ConditionContext();
if (null == input.getConditions()) {
return result;
}
for (com.dangdang.ddframe.rdb.sharding.parser.jaxb.Condition each : input.getConditions()) {
Condition condition = new Condition(new Column(each.getColumnName(), each.getTableName()), BinaryOperator.valueOf(each.getOperator().toUpperCase()));
condition.getValues().addAll(Lists.transform(each.getValues(), new Function<Value, Comparable<?>>() {
@Override
public Comparable<?> apply(final Value input) {
return input.getValueWithType();
}
}));
if (null != each.getValueIndices()) {
condition.getValueIndices().addAll(each.getValueIndices());
}
result.add(condition);
}
return result;
}
});
}
MergeContext mergeContext = new MergeContext();
if (null != assertObj.getOrderByColumns()) {
mergeContext.getOrderByColumns().addAll(Lists.transform(assertObj.getOrderByColumns(), new Function<com.dangdang.ddframe.rdb.sharding.parser.jaxb.OrderByColumn, OrderByColumn>() {
@Override
public OrderByColumn apply(final com.dangdang.ddframe.rdb.sharding.parser.jaxb.OrderByColumn input) {
return Strings.isNullOrEmpty(input.getName()) ? new OrderByColumn(input.getIndex(), OrderByType.valueOf(input.getOrderByType().toUpperCase())) : new OrderByColumn(Optional.fromNullable(input.getOwner()), input.getName(), Optional.fromNullable(input.getAlias()), OrderByType.valueOf(input.getOrderByType().toUpperCase()));
}
}));
}
if (null != assertObj.getGroupByColumns()) {
mergeContext.getGroupByColumns().addAll(Lists.transform(assertObj.getGroupByColumns(), new Function<com.dangdang.ddframe.rdb.sharding.parser.jaxb.GroupByColumn, GroupByColumn>() {
@Override
public GroupByColumn apply(final com.dangdang.ddframe.rdb.sharding.parser.jaxb.GroupByColumn input) {
return new GroupByColumn(Optional.fromNullable(input.getOwner()), input.getName(), Optional.fromNullable(input.getAlias()), OrderByType.valueOf(input.getOrderByType().toUpperCase()));
}
}));
}
if (null != assertObj.getAggregationColumns()) {
mergeContext.getAggregationColumns().addAll(Lists.transform(assertObj.getAggregationColumns(), new Function<com.dangdang.ddframe.rdb.sharding.parser.jaxb.AggregationColumn, AggregationColumn>() {
@Override
public AggregationColumn apply(final com.dangdang.ddframe.rdb.sharding.parser.jaxb.AggregationColumn input) {
AggregationColumn result = new AggregationColumn(input.getExpression(), AggregationType.valueOf(input.getAggregationType().toUpperCase()), Optional.fromNullable(input.getAlias()), Optional.fromNullable(input.getOption()));
if (null != input.getIndex()) {
result.setColumnIndex(input.getIndex());
}
for (com.dangdang.ddframe.rdb.sharding.parser.jaxb.AggregationColumn each : input.getDerivedColumns()) {
result.getDerivedColumns().add(new AggregationColumn(each.getExpression(), AggregationType.valueOf(each.getAggregationType().toUpperCase()), Optional.fromNullable(each.getAlias()), Optional.fromNullable(each.getOption())));
}
return result;
}
}));
}
if (null != assertObj.getLimit()) {
mergeContext.setLimit(new Limit(assertObj.getLimit().getOffset(), assertObj.getLimit().getRowCount(), assertObj.getLimit().getOffsetParameterIndex(), assertObj.getLimit().getRowCountParameterIndex()));
}
result[5] = mergeContext;
return result;
}
use of com.dangdang.ddframe.rdb.sharding.parser.result.merger.Limit in project sharding-jdbc by dangdangdotcom.
the class SQLParsedResultTest method generateMergeContext.
private void generateMergeContext(final MergeContext mergeContext) {
mergeContext.getAggregationColumns().add(new AggregationColumn("COUNT(id)", AggregationType.COUNT, Optional.of("c"), Optional.<String>absent()));
mergeContext.getOrderByColumns().add(new OrderByColumn(Optional.<String>absent(), "id", Optional.of("a"), OrderByType.DESC));
mergeContext.getGroupByColumns().add(new GroupByColumn(Optional.<String>absent(), "id", Optional.of("d"), OrderByType.ASC));
mergeContext.setLimit(new Limit(0, 10, -1, -1));
}
use of com.dangdang.ddframe.rdb.sharding.parser.result.merger.Limit in project sharding-jdbc by dangdangdotcom.
the class IteratorResultSetTest method assertNextWithLimitForPartData.
@Test
public void assertNextWithLimitForPartData() throws SQLException {
MergeContext mergeContext = new MergeContext();
mergeContext.setLimit(new Limit(1, 1, -1, -1));
ResultSet resultSet = ResultSetFactory.getResultSet(Arrays.<ResultSet>asList(new MockResultSet<>(1), new MockResultSet<>(2, 4), new MockResultSet<Integer>()), mergeContext);
int count = 0;
while (resultSet.next()) {
count++;
}
assertThat(count, is(1));
}
use of com.dangdang.ddframe.rdb.sharding.parser.result.merger.Limit in project sharding-jdbc by dangdangdotcom.
the class SQLRouteEngine method amendSQLAccordingToRouteResult.
private void amendSQLAccordingToRouteResult(final SQLParsedResult parsedResult, final List<Object> parameters, final SQLRouteResult result) {
boolean isVarious = result.getExecutionUnits().size() > 1;
Limit limit = result.getMergeContext().getLimit();
SQLBuilder sqlBuilder = parsedResult.getRouteContext().getSqlBuilder();
if (null != limit) {
limit.replaceSQL(sqlBuilder, isVarious);
limit.replaceParameters(parameters, isVarious);
}
if (!isVarious) {
sqlBuilder.removeDerivedSQL();
}
}
use of com.dangdang.ddframe.rdb.sharding.parser.result.merger.Limit in project sharding-jdbc by dangdangdotcom.
the class MySQLSelectVisitor method visit.
/**
* LIMIT 解析.
*
* @param x LIMIT表达式
* @return false 停止遍历AST
*/
@Override
public boolean visit(final MySqlSelectQueryBlock.Limit x) {
if (getParseContext().getParseContextIndex() > 0) {
return super.visit(x);
}
print("LIMIT ");
int offset = 0;
int offSetIndex = -1;
if (null != x.getOffset()) {
if (x.getOffset() instanceof SQLNumericLiteralExpr) {
offset = ((SQLNumericLiteralExpr) x.getOffset()).getNumber().intValue();
printToken(Limit.OFFSET_NAME, String.valueOf(offset));
print(", ");
} else {
offset = ((Number) getParameters().get(((SQLVariantRefExpr) x.getOffset()).getIndex())).intValue();
offSetIndex = ((SQLVariantRefExpr) x.getOffset()).getIndex();
print("?, ");
}
}
int rowCount;
int rowCountIndex = -1;
if (x.getRowCount() instanceof SQLNumericLiteralExpr) {
rowCount = ((SQLNumericLiteralExpr) x.getRowCount()).getNumber().intValue();
printToken(Limit.COUNT_NAME, String.valueOf(rowCount));
} else {
rowCount = ((Number) getParameters().get(((SQLVariantRefExpr) x.getRowCount()).getIndex())).intValue();
rowCountIndex = ((SQLVariantRefExpr) x.getRowCount()).getIndex();
print("?");
}
if (offset < 0 || rowCount < 0) {
throw new SQLParserException("LIMIT offset and row count can not be a negative value");
}
// "LIMIT {rowCount} OFFSET {offset}" will transform to "LIMIT {offset}, {rowCount}".So exchange parameter index
if (offSetIndex > -1 && rowCountIndex > -1 && offSetIndex > rowCountIndex) {
int tmp = rowCountIndex;
rowCountIndex = offSetIndex;
offSetIndex = tmp;
}
getParseContext().getParsedResult().getMergeContext().setLimit(new Limit(offset, rowCount, offSetIndex, rowCountIndex));
return false;
}
Aggregations