io.micronaut.http.netty.NettyHttpParameters Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of micronaut-http-netty Show documentation
Natively Cloud Native
There is a newer version: 4.7.5
/*
 * Copyright 2017-2020 original 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
 *
 * https://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.micronaut.http.netty;

import io.micronaut.core.annotation.Internal;
import io.micronaut.core.annotation.Nullable;
import io.micronaut.core.convert.ArgumentConversionContext;
import io.micronaut.core.convert.ConversionService;
import io.micronaut.core.convert.value.ConvertibleMultiValuesMap;
import io.micronaut.core.util.CollectionUtils;
import io.micronaut.http.MutableHttpParameters;

import java.util.ArrayList;
import java.util.Collection;
import java.util.Collections;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import java.util.Optional;
import java.util.Set;
import java.util.function.BiConsumer;
import java.util.stream.Collectors;

/**
 * Implementation of {@link MutableHttpParameters} for Netty.
 *
 * @author Graeme Rocher
 * @since 1.0
 */
@Internal
public class NettyHttpParameters implements MutableHttpParameters {

    private final LinkedHashMap> valuesMap;
    private final ConvertibleMultiValuesMap values;
    private final BiConsumer> onChange;

    /**
     * @param parameters        The parameters
     * @param conversionService The conversion service
     * @param onChange A callback for changes
     */
    public NettyHttpParameters(Map> parameters,
                               ConversionService conversionService,
                               @Nullable BiConsumer> onChange) {
        this.valuesMap = CollectionUtils.newLinkedHashMap(parameters.size());
        this.values = new ConvertibleMultiValuesMap<>(valuesMap, conversionService);
        this.onChange = onChange;
        for (Map.Entry> entry : parameters.entrySet()) {
            valuesMap.put(entry.getKey(), Collections.unmodifiableList(entry.getValue()));
        }
    }

    @Override
    public Set names() {
        return values.names();
    }

    @Override
    public Collection> values() {
        return values.values();
    }

    @Override
    public List getAll(CharSequence name) {
        return values.getAll(name);
    }

    @Override
    public String get(CharSequence name) {
        return values.get(name);
    }

    @Override
    public  Optional get(CharSequence name, ArgumentConversionContext conversionContext) {
        return values.get(name, conversionContext);
    }

    @Override
    public MutableHttpParameters add(CharSequence name, List values) {
        List valueList = valuesMap.compute(name, (key, val) -> {
            List newValues = values.stream().map(v -> v == null ? null : v.toString()).collect(Collectors.toList());
            if (val == null) {
                val = new ArrayList<>(newValues.size());
            } else {
                //val is unmodifiable
                val = new ArrayList<>(val);
            }
            val.addAll(newValues);

            return Collections.unmodifiableList(val);
        });

        if (onChange != null) {
            onChange.accept(name, valueList);
        }
        return this;
    }

    @Override
    public ConversionService getConversionService() {
        return values.getConversionService();
    }

    @Override
    public void setConversionService(ConversionService conversionService) {
        values.setConversionService(conversionService);
    }
}