org.jooq.impl.InCondition Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of payment-retries-plugin Show documentation
Kill Bill Payment Retries plugin
The newest version!
/*
 * 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;
        }
    }
}