org.jooq.impl.InCondition Maven / Gradle / Ivy
/*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*
* Other licenses:
* -----------------------------------------------------------------------------
* Commercial licenses for this work are available. These replace the above
* ASL 2.0 and offer limited warranties, support, maintenance, and commercial
* database integrations.
*
* For more information, please visit: http://www.jooq.org/licenses
*
*
*
*
*
*
*
*
*
*
*
*
*
*
*
*
*/
package org.jooq.impl;
import static java.lang.Boolean.TRUE;
import static java.util.Arrays.asList;
import static org.jooq.Clause.CONDITION;
import static org.jooq.Clause.CONDITION_IN;
import static org.jooq.Clause.CONDITION_NOT_IN;
import static org.jooq.Comparator.IN;
import static org.jooq.SQLDialect.FIREBIRD;
// ...
// ...
// ...
import static org.jooq.conf.ParamType.INDEXED;
import static org.jooq.impl.DSL.falseCondition;
import static org.jooq.impl.DSL.trueCondition;
import static org.jooq.impl.Keywords.K_AND;
import static org.jooq.impl.Keywords.K_OR;
import java.util.AbstractList;
import java.util.Arrays;
import java.util.List;
import org.jooq.Clause;
import org.jooq.Comparator;
import org.jooq.Context;
import org.jooq.Field;
/**
* @author Lukas Eder
*/
final class InCondition extends AbstractCondition {
private static final long serialVersionUID = -1653924248576930761L;
private static final int IN_LIMIT = 1000;
private static final Clause[] CLAUSES_IN = { CONDITION, CONDITION_IN };
private static final Clause[] CLAUSES_IN_NOT = { CONDITION, CONDITION_NOT_IN };
private final Field field;
private final Field[] values;
private final Comparator comparator;
InCondition(Field field, Field[] values, Comparator comparator) {
this.field = field;
this.values = values;
this.comparator = comparator;
}
@Override
public final Clause[] clauses(Context ctx) {
return comparator == IN ? CLAUSES_IN : CLAUSES_IN_NOT;
}
@Override
public final void accept(Context ctx) {
List> list = Arrays.asList(values);
if (list.size() == 0) {
if (comparator == IN)
ctx.visit(falseCondition());
else
ctx.visit(trueCondition());
}
else if (list.size() > IN_LIMIT) {
// [#798] Oracle and some other dialects can only hold 1000 values
// in an IN (...) clause
switch (ctx.family()) {
case FIREBIRD: {
ctx.sql('(')
.formatIndentStart()
.formatNewLine();
for (int i = 0; i < list.size(); i += IN_LIMIT) {
if (i > 0) {
// [#1515] The connector depends on the IN / NOT IN
// operator
if (comparator == Comparator.IN) {
ctx.formatSeparator()
.visit(K_OR)
.sql(' ');
}
else {
ctx.formatSeparator()
.visit(K_AND)
.sql(' ');
}
}
toSQLSubValues(ctx, padded(ctx, list.subList(i, Math.min(i + IN_LIMIT, list.size()))));
}
ctx.formatIndentEnd()
.formatNewLine()
.sql(')');
break;
}
// Most dialects can handle larger lists
default: {
toSQLSubValues(ctx, list);
break;
}
}
}
else {
toSQLSubValues(ctx, padded(ctx, list));
}
}
private static List> padded(Context ctx, List> list) {
return ctx.paramType() == INDEXED && TRUE.equals(ctx.settings().isInListPadding())
? new PaddedList>(list, asList(FIREBIRD).contains(ctx.family())
? IN_LIMIT
: Integer.MAX_VALUE)
: list;
}
/**
* Render the SQL for a sub-set of the IN
clause's values
*/
private void toSQLSubValues(Context ctx, List> subValues) {
ctx.visit(field)
.sql(' ')
.visit(comparator.toKeyword())
.sql(" (");
if (subValues.size() > 1) {
ctx.formatIndentStart()
.formatNewLine();
}
String separator = "";
for (Field value : subValues) {
ctx.sql(separator)
.formatNewLineAfterPrintMargin()
.visit(value);
separator = ", ";
}
if (subValues.size() > 1) {
ctx.formatIndentEnd()
.formatNewLine();
}
ctx.sql(')');
}
private static class PaddedList extends AbstractList {
private final List delegate;
private final int realSize;
private final int padSize;
PaddedList(List delegate, int maxPadding) {
this.delegate = delegate;
this.realSize = delegate.size();
int r = Integer.highestOneBit(realSize);
this.padSize = Math.min(maxPadding, r == realSize ? realSize : r << 1);
}
@Override
public T get(int index) {
return index < realSize ? delegate.get(index) : delegate.get(realSize - 1);
}
@Override
public int size() {
return padSize;
}
}
}