在Java中,我們可以使用Collectors類來創建自定義的收集器實現方法。以下是一個示例,展示了如何創建一個自定義的收集器,用于計算一組數字的平均值:
import java.util.List;
import java.util.stream.Collector;
import java.util.stream.Collectors;
public class CustomCollector {
public static void main(String[] args) {
List<Integer> numbers = List.of(1, 2, 3, 4, 5);
double average = numbers.stream()
.collect(averageCollector());
System.out.println("Average: " + average);
}
public static Collector<Integer, AverageAccumulator, Double> averageCollector() {
return Collector.of(
AverageAccumulator::new,
AverageAccumulator::accumulate,
AverageAccumulator::combine,
AverageAccumulator::getAverage
);
}
static class AverageAccumulator {
private int sum;
private int count;
public AverageAccumulator() {
this.sum = 0;
this.count = 0;
}
public void accumulate(int number) {
sum += number;
count++;
}
public AverageAccumulator combine(AverageAccumulator other) {
this.sum += other.sum;
this.count += other.count;
return this;
}
public double getAverage() {
if (count == 0) {
return 0;
}
return (double) sum / count;
}
}
}
在這個示例中,我們創建了一個名為averageCollector
的自定義收集器方法,該方法返回一個Collector對象,該對象將使用AverageAccumulator
來計算一組數字的平均值。AverageAccumulator
類用于跟蹤累加的總和和數量,并提供計算平均值的方法。最后,我們將這個自定義的收集器應用到一個整數列表上,以計算其平均值并輸出結果。