org.jooq.impl.InCondition Maven / Gradle / Ivy
/*
* Copyright (c) 2009-2016, Data Geekery GmbH (http://www.datageekery.com)
* All rights reserved.
*
* 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 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;
import org.jooq.tools.JooqLogger;
/**
* @author Lukas Eder
*/
final class InCondition extends AbstractCondition {
private static final JooqLogger log = JooqLogger.getLogger(InCondition.class);
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()
.keyword("or")
.sql(' ');
}
else {
ctx.formatSeparator()
.keyword("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(' ')
.keyword(comparator.toSQL())
.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();
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;
}
}
}