io.druid.query.extraction.UpperExtractionFn.java Source code

Introduction

Here is the source code for io.druid.query.extraction.UpperExtractionFn.java

Source

/*
 *
 *  Licensed to Metamarkets Group Inc. (Metamarkets) under one
 *  or more contributor license agreements. See the NOTICE file
 *  distributed with this work for additional information
 *  regarding copyright ownership. Metamarkets 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 io.druid.query.extraction;

import com.fasterxml.jackson.annotation.JsonProperty;
import com.fasterxml.jackson.annotation.JsonTypeName;
import com.google.common.base.Strings;
import io.druid.java.util.common.StringUtils;

import javax.annotation.Nullable;
import java.nio.ByteBuffer;
import java.util.Locale;

@JsonTypeName("upper")
public class UpperExtractionFn extends DimExtractionFn {
    private final Locale locale;

    @JsonProperty
    private final String localeString;

    public UpperExtractionFn(@JsonProperty("locale") String localeString) {
        this.localeString = localeString;
        this.locale = localeString == null ? Locale.getDefault() : Locale.forLanguageTag(localeString);
    }

    /**
     * @param key string input of extraction function
     *
     * @return new string with all of the characters in {@code key} as an upper case  or <tt>null</tt> if {@code key} is empty or null
     */
    @Nullable
    @Override
    public String apply(String key) {
        if (Strings.isNullOrEmpty(key)) {
            return null;
        }
        return key.toUpperCase(locale);
    }

    @Override
    public boolean preservesOrdering() {
        return false;
    }

    @Override
    public ExtractionType getExtractionType() {
        return ExtractionType.MANY_TO_ONE;
    }

    @Override
    public byte[] getCacheKey() {
        byte[] localeBytes = StringUtils.toUtf8(Strings.nullToEmpty(localeString));
        return ByteBuffer.allocate(2 + localeBytes.length).put(ExtractionCacheHelper.CACHE_TYPE_ID_UPPER)
                .put((byte) 0XFF).put(localeBytes).array();
    }
}