问题背景&思路
如果需要计算100的阶乘,那应该怎么做?
方法1:
for循环(默认,单线程)
方法2:
多线程,MapReduce思想
main线程开启多个子任务(个数=CPU核心数),放到线程池执行,每个子任务统计from ~ to的正整数,然后返回本次计算的结果;然后main线程拿到所有子任务的返回结果后再次统计
任务: SumTask implements Callable,返回统计结果Future
方法3:
ForkJoinPool,递归任务,MapReduce思想
任务:SumTask extends RecursiveTask,返回统计结果BigDecimal
任务中判断,当需要统计的数>某个阈值时,拆分成两个任务;否则直接执行并返回
这个阈值时多少合适?经过测试后,当计算10000的阶乘时,方法1略快于方法2,所以这个阈值就定位10000
上代码
为了方便测试,先定义了一个接口:
package com.wz.poc.forkjoin;
import java.math.BigDecimal;
/**
* @author liweizhi
* @date 2020/12/31
*/
public interface Calculator {
/**
* 求传进来数的阶乘
*
* @param number
* @return 总和
*/
BigDecimal factorial(long number);
}
单线程循环
package com.wz.poc.forkjoin;
import java.math.BigDecimal;
/**
* @author liweizhi
* @date 2020/12/31
*/
public class ForLoopCalculator implements Calculator {
@Override
public BigDecimal factorial(long number) {
BigDecimal ret = new BigDecimal(1);
for (long i = 1; i <= number; i++) {
ret = ret.multiply(BigDecimal.valueOf(i));
}
return ret;
}
}
普通线程池,多线程
package com.wz.poc.forkjoin;
import java.math.BigDecimal;
import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.Callable;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.Future;
/**
* @author liweizhi
* @date 2020/12/31
*/
public class ExecutorServiceCalculator implements Calculator {
@Override
public BigDecimal factorial(long number) {
List<Future<BigDecimal>> results = new ArrayList<>();