com.github.phantomthief.util.PairKeyRequestContextCache Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of retrieve-id-utils Show documentation
There is a newer version: 1.0.10
package com.github.phantomthief.util;

import static com.github.phantomthief.tuple.Tuple.tuple;
import static com.google.common.collect.Maps.newHashMapWithExpectedSize;

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

import com.github.phantomthief.tuple.TwoTuple;

/**
 * @author w.vela
 *
 * use https://github.com/PhantomThief/scope instead.
 */
@Deprecated
public class PairKeyRequestContextCache extends RequestContextCache, V> {

    public PairKeyRequestContextCache() {
        super();
    }

    public V getMultiKey(K1 key1, K2 key2) {
        return get(tuple(key1, key2));
    }

    public void setMultiKey(K1 key1, K2 key2, V value) {
        set(tuple(key1, key2), value);
    }

    public void removeMultiKey(K1 key1, K2 key2) {
        remove(tuple(key1, key2));
    }

    public IMultiDataAccess accessFixedSecondKey(K2 key2) {
        return new IMultiDataAccess() {

            @Override
            public Map get(Collection keys) {
                Map result = new HashMap<>();
                for (K1 k1 : keys) {
                    V v = PairKeyRequestContextCache.this.get(tuple(k1, key2));
                    if (v != null) {
                        result.put(k1, v);
                    }
                }
                return result;
            }

            @Override
            public void set(Map dataMap) {
                Map, V> mapToSet = newHashMapWithExpectedSize(dataMap.size());
                dataMap.entrySet().stream().filter(entry -> entry.getValue() != null).forEach(
                        entry -> mapToSet.put(tuple(entry.getKey(), key2), entry.getValue()));
                PairKeyRequestContextCache.this.set(mapToSet);
            }

        };
    }

    public IMultiDataAccess accessFixedFirstKey(K1 key1) {
        return new IMultiDataAccess() {

            @Override
            public Map get(Collection keys) {
                Map result = new HashMap<>();
                for (K2 k2 : keys) {
                    V v = PairKeyRequestContextCache.this.get(tuple(key1, k2));
                    if (v != null) {
                        result.put(k2, v);
                    }
                }
                return result;
            }

            @Override
            public void set(Map dataMap) {
                Map, V> mapToSet = newHashMapWithExpectedSize(dataMap.size());
                dataMap.entrySet().stream().filter(entry -> entry.getValue() != null).forEach(
                        entry -> mapToSet.put(tuple(key1, entry.getKey()), entry.getValue()));
                PairKeyRequestContextCache.this.set(mapToSet);
            }
        };
    }
}