dev.miku.r2dbc.mysql.TextQuery Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2018-2020 the original author or authors.
 *
 * 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.
 */

package dev.miku.r2dbc.mysql;

import java.util.List;
import java.util.Map;
import java.util.Set;

import static dev.miku.r2dbc.mysql.util.AssertUtils.require;
import static dev.miku.r2dbc.mysql.util.AssertUtils.requireNonNull;

/**
 * An implementation of {@link Query} considers client-preparing statements.
 */
final class TextQuery extends Query {

    private final Map nameKeyedIndex;

    private final List sqlParts;

    TextQuery(Map nameKeyedIndex, List sqlParts) {
        requireNonNull(nameKeyedIndex, "named parameter map must not be null");
        requireNonNull(sqlParts, "sql parts must not be null");
        require(sqlParts.size() > 1, "sql parts need least 2 parts");

        this.nameKeyedIndex = nameKeyedIndex;
        this.sqlParts = sqlParts;
    }

    @Override
    int getParameters() {
        return sqlParts.size() - 1;
    }

    @Override
    ParameterIndex getIndexes(String identifier) {
        ParameterIndex index = nameKeyedIndex.get(identifier);

        if (index == null) {
            throw new IllegalArgumentException(String.format("No such parameter with identifier '%s'", identifier));
        }

        return index;
    }

    /**
     * Visible for unit tests.
     *
     * @return parameter name set
     */
    Set getParameterNames() {
        return nameKeyedIndex.keySet();
    }

    /**
     * Visible for unit tests.
     *
     * @return split SQL parts
     */
    List getSqlParts() {
        return sqlParts;
    }

    @Override
    public boolean equals(Object o) {
        if (this == o) {
            return true;
        }
        if (!(o instanceof TextQuery)) {
            return false;
        }

        TextQuery textQuery = (TextQuery) o;

        if (!nameKeyedIndex.equals(textQuery.nameKeyedIndex)) {
            return false;
        }
        return sqlParts.equals(textQuery.sqlParts);
    }

    @Override
    public int hashCode() {
        return 31 * nameKeyedIndex.hashCode() + sqlParts.hashCode();
    }

    @Override
    public String toString() {
        return String.format("TextQuery{nameKeyedIndex=%s, sqlParts=%s}", nameKeyedIndex, sqlParts);
    }
}