package cn.codeduck.guavatest.basic;

import com.google.common.collect.*;
import com.google.common.primitives.Ints;
import org.junit.Test;

import java.util.Map;
import java.util.Set;

// 集合工具类：提供作用于或生成 Multimap 的静态方法。
// 每一个有经验的Java程序员都曾实现过Map<K, List<V>>或Map<K, Set<V>>，并处理过这种结构的尴尬之处。
// 例如，Map<K, Set<V>>是表示无标记有向图的典型方法。
// Guava的Multimap框架可以很容易地处理从键到多个值的映射。Multimap是一种将键与任意多个值关联的通用方法
public class MultimapsTest {

    // 表兄弟关系: Maps.uniqueIndex, Multimaps.index(Iterable, Function)
    // index 分组，将iterable生成value,function生成key的map (例如 对集合分组)
    @Test
    public void index(){
        Set<String> set = ImmutableSet.of("zero", "one", "two", "three", "four", "five", "six", "seven", "eight", "nine");
        Multimap<Integer, String> map = Multimaps.index(set, s -> s != null ? s.length() : 0);
        System.out.println(map);    //{4=[zero, four, five, nine], 3=[one, two, six], 5=[three, seven, eight]}
    }

    // invertFrom 反转视图，由于Multimap可以将多个键映射到一个值，一个键可以映射到多个值，因此可以对Multimap进行反向操作。
    // 如果使用的是ImmutableMultimap，请考虑ImmutableMultimap.inverse()替代。
    @Test
    public void invertFrom(){
        ArrayListMultimap<String,Integer> listMultimap = ArrayListMultimap.create();
        listMultimap.putAll("b", Ints.asList(2,4,6));
        listMultimap.putAll("a", Ints.asList(4,2,1));
        listMultimap.putAll("c", Ints.asList(2,5,3));
        System.out.println(listMultimap);
        //{a=[4, 2, 1], b=[2, 4, 6], c=[2, 5, 3]}

        TreeMultimap<Integer, String> treeMultimap = Multimaps.invertFrom(listMultimap, TreeMultimap.create());
        System.out.println(treeMultimap);
        //{1=[a], 2=[a, b, c], 3=[c], 4=[a, b], 5=[c], 6=[b]}
    }

    // forMap 以SetMultimap的形式查看Map。这是特别有用的，例如，结合multiaps.invertfrom。
    @Test
    public void forMap(){
        Map<String, Integer> map = ImmutableMap.of("a", 1, "b", 1, "c", 2);
        SetMultimap<String, Integer> multimap = Multimaps.forMap(map);
//        multimap.put("d", 3);   //不可变map 不能放入元素
        System.out.println(multimap);
        //{a=[1], b=[1], c=[2]}

        HashMultimap<Integer, String> hashMultimap = Multimaps.invertFrom(multimap, HashMultimap.create());
        System.out.println(hashMultimap);
        //{1=[a, b], 2=[c]}
    }
}