top.jfunc.http.holder.DefaultValuesMapHolder Maven / Gradle / Ivy

Go to download
package top.jfunc.http.holder;

import top.jfunc.common.utils.ArrayListMultiValueMap;
import top.jfunc.common.utils.MapUtil;
import top.jfunc.common.utils.MultiValueMap;

import java.util.Map;

/**
 * wrap of {@link MultiValueMap} and impl HeaderHolder
 * @see MultiValueMap
 * @author xiongshiyan
 */
public class DefaultValuesMapHolder implements ValuesMapHolder {
    /**
     * 懒加载
     */
    private MultiValueMap multiValueMap;

    @Override
    public MultiValueMap get() {
        return multiValueMap;
    }

    @Override
    public ValuesMapHolder set(MultiValueMap multiValueMap) {
        this.multiValueMap = multiValueMap;
        return this;
    }

    @Override
    public ValuesMapHolder set(Map map) {
        if(MapUtil.notEmpty(map)){
            this.multiValueMap = ArrayListMultiValueMap.fromMap(map);
        }
        return this;
    }

    @Override
    public ValuesMapHolder set(String key, String value) {
        initMultiValueMap();
        this.multiValueMap.set(key, value);
        return this;
    }

    @Override
    public ValuesMapHolder add(String key, String value){
        initMultiValueMap();
        this.multiValueMap.add(key, value);
        return this;
    }
    @Override
    public ValuesMapHolder add(String key, String value , String... values){
        initMultiValueMap();
        this.multiValueMap.add(key , value);
        for (String val : values) {
            this.multiValueMap.add(key , val);
        }
        return this;
    }
    @Override
    public ValuesMapHolder add(String key, Iterable values){
        initMultiValueMap();
        for (String value : values) {
            this.multiValueMap.add(key , value);
        }
        return this;
    }
    @Override
    public ValuesMapHolder add(Iterable>> multiValueEntries){
        for (Map.Entry> entry : multiValueEntries) {
            add(entry.getKey() , entry.getValue());
        }
        return this;
    }
    @SafeVarargs
    @Override
    public final ValuesMapHolder add(Map.Entry> entry, Map.Entry>... entries){
        add(entry.getKey() , entry.getValue());
        for (Map.Entry> h : entries) {
            add(h.getKey() , h.getValue());
        }
        return this;
    }
    private void initMultiValueMap(){
        if(null == this.multiValueMap){
            this.multiValueMap = new ArrayListMultiValueMap<>(2);
        }
    }
}