/**
* 根据权重抽奖通用方法
* @param list 参与抽奖的奖品集合
* @param weightFieldName 奖品对象中的权重字段
* @return 抽取到的奖品对象
*/
public static <E> E lottery(List<E> list,String weightFieldName) {
if (list == null || list.isEmpty()) {
return null;
}
//先移除掉权重为null的项,不参与抽奖
try {
for (int i = 0; i < list.size(); i++) {
Field f = list.get(i).getClass().getDeclaredField(weightFieldName);
f.setAccessible(true);
String winRate = String.valueOf(f.get(list.get(i)));
if(winRate==null||"null".equals(winRate)){
list.remove(i);
}
}
} catch (Exception e) {
e.printStackTrace();
}
List<BigDecimal> Rates = new ArrayList<BigDecimal>(
list.size());
try {
for (int i = 0; i < list.size(); i++) {
Field f = list.get(i).getClass().getDeclaredField(weightFieldName);
f.setAccessible(true);
String winRate = String.valueOf(f.get(list.get(i)));
Rates.add(new BigDecimal(winRate));
}
} catch (Exception e) {
e.printStackTrace();
}
if (Rates == null || Rates.isEmpty()) {
return null;
}
int size = Rates.size();
// 计算总概率,这样可以保证不一定总概率是1
BigDecimal sumRate = new BigDecimal("0.0");
for (BigDecimal rate : Rates) {
sumRate = sumRate.add(rate);
}
// 计算每个物品在总概率的基础下的概率情况
List<BigDecimal> sortOrignalRates = new ArrayList<BigDecimal>(size);
BigDecimal tempSumRate = new BigDecimal("0.0");
for (BigDecimal rate : Rates) {
tempSumRate = tempSumRate.add(rate);
sortOrignalRates.add(tempSumRate.divide(sumRate,2,BigDecimal.ROUND_FLOOR));
}
// 根据区块值来获取抽取到的物品索引
BigDecimal nextDouble = new BigDecimal(String.valueOf(Math.random()));
sortOrignalRates.add(nextDouble);
Collections.sort(sortOrignalRates);
return list.get(sortOrignalRates.indexOf(nextDouble));
}
Java通用抽奖方法
最新推荐文章于 2024-08-24 17:30:53 发布