com.speedment.runtime.core.util.CollectorUtil Maven / Gradle / Ivy
/**
*
* Copyright (c) 2006-2019, Speedment, Inc. All Rights Reserved.
*
* 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:
*
* http://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 com.speedment.runtime.core.util;
import com.speedment.common.mapstream.MapStream;
import java.util.*;
import java.util.function.*;
import java.util.stream.Collector;
import java.util.stream.Collectors;
import java.util.stream.Stream;
import static com.speedment.common.invariant.NullUtil.requireNonNullElements;
import static com.speedment.runtime.core.util.StaticClassUtil.instanceNotAllowed;
import static java.util.Objects.requireNonNull;
/**
* Utility methods for collecting Speedment streams in various ways.
*
* @author Per Minborg
* @author Emil Forslund
* @since 2.1.0
*/
public final class CollectorUtil {
private static final String NULL_TEXT = " must not be null";
@SafeVarargs
@SuppressWarnings({"unchecked", "varargs"})
public static T of(Supplier supplier, Consumer modifier, Consumer... additionalModifiers) {
requireNonNull(supplier, "supplier" + NULL_TEXT);
requireNonNull(modifier, "modifier" + NULL_TEXT);
requireNonNullElements(additionalModifiers, "additionalModifiers" + NULL_TEXT);
final T result = supplier.get();
modifier.accept(result);
Stream.of(additionalModifiers).forEach((Consumer c) -> {
c.accept(result);
});
return result;
}
public static T of(Supplier supplier, Consumer modifier, Function finisher) {
Objects.requireNonNull(supplier, "supplier" + NULL_TEXT);
Objects.requireNonNull(modifier, "modifier" + NULL_TEXT);
Objects.requireNonNull(finisher, "finisher" + NULL_TEXT);
final I intermediateResult = supplier.get();
modifier.accept(intermediateResult);
return finisher.apply(intermediateResult);
}
public static Collector, Set> toUnmodifiableSet() {
return Collector.of(HashSet::new, Set::add, (left, right) -> {
left.addAll(right);
return left;
}, Collections::unmodifiableSet, Collector.Characteristics.UNORDERED);
}
public static Collector> toReversedList() {
return Collectors.collectingAndThen(Collectors.toList(), l -> {
Collections.reverse(l);
return l;
});
}
@SafeVarargs
@SuppressWarnings({"unchecked", "varargs"})
public static Set unmodifiableSetOf(T... items) {
requireNonNullElements(items);
return Stream.of(items).collect(toUnmodifiableSet());
}
/**
* Similar to the
* {@link java.util.stream.Collectors#joining(java.lang.CharSequence, java.lang.CharSequence, java.lang.CharSequence) }
* method except that this method surrounds the result with the specified
* {@code prefix} and {@code suffix} even if the stream is empty.
*
* @param delimiter the delimiter to separate the strings
* @param prefix the prefix to put before the result
* @param suffix the suffix to put after the result
* @return a collector for joining string elements
*/
public static Collector joinIfNotEmpty(String delimiter, String prefix, String suffix) {
return new CollectorImpl<>(
() -> new StringJoiner(delimiter),
StringJoiner::add,
StringJoiner::merge,
s -> s.length() > 0
? prefix + s + suffix
: s.toString(),
Collections.emptySet()
);
}
/**
* Returns the specified string wrapped as an Optional. If the string was
* null or empty, the Optional will be empty.
*
* @param str the string to wrap
* @return the string wrapped as an optional
*/
public static Optional ifEmpty(String str) {
return Optional.ofNullable(str).filter(s -> !s.isEmpty());
}
/**
* Returns a new {@link MapStream} where the elements have been grouped
* together using the specified function.
*
* @param the stream element type
* @param the type of the key to group by
* @param grouper the function to use for grouping
* @return a {@link MapStream} grouped by key
*/
public static Collector>> groupBy(Function grouper) {
return new CollectorImpl<>(
() -> new GroupHolder<>(grouper),
GroupHolder::add,
GroupHolder::merge,
GroupHolder::finisher,
Collections.emptySet()
);
}
private static class GroupHolder {
private final Function grouper;
private final Map> elements;
private final Function> createList = c -> new ArrayList<>();
public GroupHolder(Function grouper) {
this.grouper = grouper;
this.elements = new HashMap<>();
}
public void add(T element) {
final C key = grouper.apply(element);
elements.computeIfAbsent(key, createList)
.add(element);
}
public GroupHolder merge(GroupHolder holder) {
holder.elements.entrySet().forEach(e
-> elements.computeIfAbsent(e.getKey(), createList)
.addAll(e.getValue())
);
return this;
}
public MapStream> finisher() {
return MapStream.of(elements);
}
}
/**
* Simple implementation class for {@code Collector}.
*
* @param the type of elements to be collected
* @param the type of the intermediate holder
* @param the type of the result
*/
static class CollectorImpl implements Collector {
private final Supplier supplier;
private final BiConsumer accumulator;
private final BinaryOperator combiner;
private final Function finisher;
private final Set characteristics;
CollectorImpl(Supplier supplier,
BiConsumer accumulator,
BinaryOperator combiner,
Function finisher,
Set characteristics) {
this.supplier = supplier;
this.accumulator = accumulator;
this.combiner = combiner;
this.finisher = finisher;
this.characteristics = characteristics;
}
@SuppressWarnings("unchecked")
CollectorImpl(Supplier supplier,
BiConsumer accumulator,
BinaryOperator combiner,
Set characteristics) {
this(supplier, accumulator, combiner, i -> (R) i, characteristics);
}
@Override
public BiConsumer accumulator() {
return accumulator;
}
@Override
public Supplier supplier() {
return supplier;
}
@Override
public BinaryOperator combiner() {
return combiner;
}
@Override
public Function finisher() {
return finisher;
}
@Override
public Set characteristics() {
return characteristics;
}
}
/**
* Utility classes should not be instantiated.
*/
private CollectorUtil() {
instanceNotAllowed(getClass());
}
}