/*
* 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<T> 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<T> field;
private final Field<?>[] values;
private final Comparator comparator;
InCondition(Field<T> 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<Field<?>> 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<Field<?>> padded(Context<?> ctx, List<Field<?>> list) {
return ctx.paramType() == INDEXED && TRUE.equals(ctx.settings().isInListPadding())
? new PaddedList<Field<?>>(list, asList(FIREBIRD).contains(ctx.family())
? IN_LIMIT
: Integer.MAX_VALUE)
: list;
}
/**
* Render the SQL for a sub-set of the <code>IN</code> clause's values
*/
private void toSQLSubValues(Context<?> ctx, List<Field<?>> 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<T> extends AbstractList<T> {
private final List<T> delegate;
private final int realSize;
private final int padSize;
PaddedList(List<T> delegate, int maxPadding) {
this.delegate = delegate;
this.realSize = delegate.size();
this.padSize = Math.min(maxPadding,
realSize <= 0x00000000 ? 0x00000000
: realSize <= 0x00000001 ? 0x00000001
: realSize <= 0x00000002 ? 0x00000002
: realSize <= 0x00000004 ? 0x00000004
: realSize <= 0x00000008 ? 0x00000008
: realSize <= 0x00000010 ? 0x00000010
: realSize <= 0x00000020 ? 0x00000020
: realSize <= 0x00000040 ? 0x00000040
: realSize <= 0x00000080 ? 0x00000080
: realSize <= 0x00000100 ? 0x00000100
: realSize <= 0x00000200 ? 0x00000200
: realSize <= 0x00000400 ? 0x00000400
: realSize <= 0x00000800 ? 0x00000800
: realSize <= 0x00001000 ? 0x00001000
: realSize <= 0x00002000 ? 0x00002000
: realSize <= 0x00004000 ? 0x00004000
: realSize <= 0x00008000 ? 0x00008000
: realSize <= 0x00010000 ? 0x00010000
: realSize <= 0x00020000 ? 0x00020000
: realSize <= 0x00040000 ? 0x00040000
: realSize <= 0x00080000 ? 0x00080000
: realSize <= 0x00100000 ? 0x00100000
: realSize <= 0x00200000 ? 0x00200000
: realSize <= 0x00400000 ? 0x00400000
: realSize <= 0x00800000 ? 0x00800000
: realSize <= 0x01000000 ? 0x01000000
: realSize <= 0x02000000 ? 0x02000000
: realSize <= 0x04000000 ? 0x04000000
: realSize <= 0x08000000 ? 0x08000000
: realSize <= 0x10000000 ? 0x10000000
: realSize <= 0x20000000 ? 0x20000000
: realSize <= 0x40000000 ? 0x40000000
: realSize <= 0x80000000 ? 0x80000000
: realSize);
}
@Override
public T get(int index) {
return index < realSize ? delegate.get(index) : delegate.get(realSize - 1);
}
@Override
public int size() {
return padSize;
}
}
}