org.apache.cassandra.index.sai.analyzer.NonTokenizingOptions Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of cassandra-all Show documentation
The Apache Cassandra Project develops a highly scalable second-generation distributed database, bringing together Dynamo's fully distributed design and Bigtable's ColumnFamily-based data model.
There is a newer version: 5.0.2
Show newest version
/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you 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 org.apache.cassandra.index.sai.analyzer;

import java.util.HashMap;
import java.util.Map;

import com.google.common.base.Strings;

import org.apache.cassandra.exceptions.InvalidRequestException;

public class NonTokenizingOptions
{
    public static final String NORMALIZE = "normalize";
    public static final String CASE_SENSITIVE = "case_sensitive";
    public static final String ASCII = "ascii";

    private boolean caseSensitive;
    private boolean normalized;
    private boolean ascii;

    boolean isCaseSensitive()
    {
        return caseSensitive;
    }

    void setCaseSensitive(boolean caseSensitive)
    {
        this.caseSensitive = caseSensitive;
    }
    
    boolean isNormalized()
    {
        return this.normalized;
    }

    void setAscii(boolean ascii)
    {
        this.ascii = ascii;
    }

    boolean isAscii()
    {
        return this.ascii;
    }
    
    void setNormalized(boolean normalized)
    {
        this.normalized = normalized;
    }

    static boolean hasOption(String option)
    {
        return option.equals(NORMALIZE) || option.equals(CASE_SENSITIVE) || option.equals(ASCII);
    }

    public static class OptionsBuilder
    {
        private boolean caseSensitive = true;
        private boolean normalized = false;
        private boolean ascii = false;

        OptionsBuilder() {}

        OptionsBuilder caseSensitive(boolean caseSensitive)
        {
            this.caseSensitive = caseSensitive;
            return this;
        }

        OptionsBuilder ascii(boolean ascii)
        {
            this.ascii = ascii;
            return this;
        }

        OptionsBuilder normalized(boolean normalized)
        {
            this.normalized = normalized;
            return this;
        }

        public NonTokenizingOptions build()
        {
            NonTokenizingOptions options = new NonTokenizingOptions();
            options.setCaseSensitive(caseSensitive);
            options.setNormalized(normalized);
            options.setAscii(ascii);
            return options;
        }
    }

    public static NonTokenizingOptions getDefaultOptions()
    {
        return fromMap(new HashMap<>(1));
    }

    public static NonTokenizingOptions fromMap(Map options)
    {
        OptionsBuilder builder = new OptionsBuilder();

        for (Map.Entry entry : options.entrySet())
        {
            switch (entry.getKey())
            {
                case CASE_SENSITIVE:
                {
                    boolean boolValue = validateBoolean(entry.getValue(), CASE_SENSITIVE);
                    builder = builder.caseSensitive(boolValue);
                    break;
                }
                
                case NORMALIZE:
                {
                    boolean boolValue = validateBoolean(entry.getValue(), NORMALIZE);
                    builder = builder.normalized(boolValue);
                    break;
                }

                case ASCII:
                {
                    boolean boolValue = validateBoolean(entry.getValue(), ASCII);
                    builder = builder.ascii(boolValue);
                    break;
                }
            }
        }
        return builder.build();
    }

    private static boolean validateBoolean(String value, String option)
    {
        if (Strings.isNullOrEmpty(value))
        {
            throw new InvalidRequestException("Empty value for boolean option '" + option + '\'');
        }

        if (!value.equalsIgnoreCase(Boolean.TRUE.toString()) && !value.equalsIgnoreCase(Boolean.FALSE.toString()))
        {
            throw new InvalidRequestException("Illegal value for boolean option '" + option + "': " + value);
        }

        return Boolean.parseBoolean(value);
    }
}