ZhouXY108/hutool
1
0
Fork 0
mirror of https://gitee.com/chinabugotech/hutool.git synced 2025-04-19 03:01:48 +08:00
Code Issues Packages Projects Releases Wiki Activity

!151 bug fix 随机权重并发问题

Browse Source 
Merge pull request !151 from chengxian-yi/hutool-ycx
This commit is contained in:
Looly 2020-08-07 16:04:38 +08:00 committed by Gitee
commit ec8926745d
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
hutool-core/src/main/java/cn/hutool/core/lang/WeightRandom.java
View File

@ -32,7 +32,7 @@ public class WeightRandom<T> implements Serializable {
private static final long serialVersionUID = -8244697995702786499L;
private final TreeMap<Double, T> weightMap;
private final Random random;
/**
* 创建权重随机获取器
@ -50,7 +50,7 @@ public class WeightRandom<T> implements Serializable {
*/
public WeightRandom() {
weightMap = new TreeMap<>();
random = RandomUtil.getRandom();
}
/**
@ -141,6 +141,7 @@ public class WeightRandom<T> implements Serializable {
if(MapUtil.isEmpty(this.weightMap)) {
return null;
}
final Random random = RandomUtil.getRandom();
final double randomWeight = this.weightMap.lastKey() * random.nextDouble();
final SortedMap<Double, T> tailMap = this.weightMap.tailMap(randomWeight, false);
return this.weightMap.get(tailMap.firstKey());