컬렉션 자체인 항목의 속성으로 컬렉션을 인덱싱하는 가장 깔끔한 방법
-
20-12-2019 - |
문제
나는 List<Foo>
그리고 구아바 먹고 싶어 Multimap<String, Foo>
우리가 그룹화한 곳은 Foo
각 태그별로 Collection<String> getTags()
기능.
저는 Java 8을 사용하고 있으므로 람다 및 메소드 참조가 괜찮거나 권장됩니다.
예를 들어 다음과 같은 경우가 있습니다.
foo1, tags=a,b,c
foo2, tags=c,d
foo3, tags=a,c,e
나는 얻을 것이다 Multimap<String, Foo>
와 함께:
a -> foo1, foo3
b -> foo1
c -> foo1, foo2, foo3
d -> foo2
e -> foo3
해결책
이를 위해 사용자 정의 수집기를 사용할 수 있습니다.
Multimap<String, Foo> map = list.stream().collect(
ImmutableMultimap::builder,
(builder, value) -> value.getTags().forEach(tag -> builder.put(tag, value)),
(builder1, builder2) -> builder1.putAll(builder2.build())
).build();
이는 추가적인 부작용을 일으키지 않습니다(참조: 여기 이에 대해) 동시적이고 더 관용적입니다.
다음과 같이 이러한 임시 람다를 본격적인 수집기로 추출할 수도 있습니다.
public static <T, K> Collector<T, ?, Multimap<K, T>> toMultimapByKey(Function<? super T, ? extends Iterable<? extends K>> keysMapper) {
return new MultimapCollector<>(keysMapper);
}
private static class MultimapCollector<T, K> implements Collector<T, ImmutableMultimap.Builder<K, T>, Multimap<K, T>> {
private final Function<? super T, ? extends Iterable<? extends K>> keysMapper;
private MultimapCollector(Function<? super T, ? extends Iterable<? extends K>> keysMapper) {
this.keysMapper = keysMapper;
}
@Override
public Supplier<ImmutableMultimap.Builder<K, T>> supplier() {
return ImmutableMultimap::builder;
}
@Override
public BiConsumer<ImmutableMultimap.Builder<K, T>, T> accumulator() {
return (builder, value) -> keysMapper.apply(value).forEach(k -> builder.put(k, value));
}
@Override
public BinaryOperator<ImmutableMultimap.Builder<K, T>> combiner() {
return (b1, b2) -> b1.putAll(b2.build());
}
@Override
public Function<ImmutableMultimap.Builder<K, T>, Multimap<K, T>> finisher() {
return ImmutableMultimap.Builder<K, T>::build;
}
@Override
public Set<Characteristics> characteristics() {
return Collections.emptySet();
}
}
그러면 컬렉션은 다음과 같습니다.
Multimap<String, Foo> map = list.stream().collect(toMultimapByKey(Foo::getTags));
반품도 가능해요 EnumSet.of(Characteristics.UNORDERED)
~에서 characteristics()
순서가 중요하지 않은 경우 방법.이를 통해 특히 병렬 축소의 경우 내부 수집 기계가 보다 효율적으로 작동할 수 있습니다.
다른 팁
ImmutableMultimap.Builder<String, Foo> builder = ImmutableMultimap.builder();
list.forEach(foo -> foo.getTags().forEach(tag -> builder.put(tag, foo));
return builder.build();
이는 Java 8 스트림의 경우 약간 더 관용적입니다.
Multimap<String, Foo> map = list.stream()
//First build a stream of Pair<String, Foo>
.flatMap(f -> f.tags.stream().map(s -> new AbstractMap.SimpleImmutableEntry<>(s, f)))
//Then collect it up into a multimap.
.collect(
Multimaps.toMultimap(
x -> x.getKey(),
x -> x.getValue(),
MultimapBuilder.hashKeys().arrayListValues()::build
)
);
페어 클래스를 사용하는 것이 보기 흉하다는 건 알지만 제가 보여주고 싶은 핵심은 Multimaps.toMultmap
수집기.또한 Multmaps.flatteningToMultimap
다른 사용 사례의 경우.
제휴하지 않습니다 StackOverflow