可指定概率的随机对象生成工具-java

应用场景：最近做一个推广活动的小程序，用户通过链接分享点击数可以抽取奖励，点击数越高，抽取到大奖的概率越高，所以这里奖品的随机概率是确定并且动态变化的。

参考了HashMap的些许设计思路，代码如下：

package com.rc.components.common.utils;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.Map.Entry;
import java.util.Random;
import java.util.Set;

import lombok.Data;

/**
 * 概率随机工具
 * @author: rc
 * @date: 2018年2月5日 下午4:55:48
 * @version: V1.0
 * @review: rc/2018年2月5日 下午4:55:48
 */
public class ProbRandom {
    
    private final Integer randomBound; //随机边界范围
    
    private final Random random; //随机生成器
    
    private final HashMap originalMap; //随机元素的原始数据容器
    
    private ArrayList probDistributions; //元素分布概率的数组
    
    private boolean updateFlag; //原始数据容器是否有更新
    
    private Integer originalSum; //原始数据随机值求和
    
    private Integer zeroOriginalCount; //原始数据随机值为0(未设置随机值)的元素个数
    
    public ProbRandom(Integer randomBound) {
        this.originalMap = new HashMap<>();
        this.updateFlag = true;
        this.randomBound = randomBound;
        this.originalSum = 0;
        this.zeroOriginalCount = 0;
        this.random = new Random();
    }
    
    /**
     * 生成随机数
     * @return
     */
    public Object next(){
        distribute(); //计算元素分布的概率值
        if (null == probDistributions) {
            return null;
        }
        int nextInt = random.nextInt(randomBound);
        ProbDistribution prob = probDistributions.stream().filter(c->(c.floor<=nextInt&&c.ceil>nextInt)).findFirst().get();
        return prob.getKey();
    }
    
    /**
     * 添加随机元素和概率值，返回添加后的概率值
     * 如果prob=0 ，概率值计算为总体概率值确定的默认值
     * @param key
     * @param prob
     * @return
     */
    public void put(Object key, Integer prob){
        //校验
        if (null == key || prob == null ) {
            throw new NullPointerException("key或者prob不能为空");
        }
        if (prob < 0) {
            throw new RuntimeException("prob不能小于0");
        }
        //添加
        Integer previousProb = originalMap.put(key, prob);
        //更新
        if (prob == 0) {
            zeroOriginalCount ++ ;
        } else {
            originalSum += prob;
        }
        if (previousProb == null) {
            updateFlag = true;
        } else {
            if (previousProb == 0) {
                zeroOriginalCount -- ;
            } else {
                originalSum -= previousProb;
            }
            if (prob != previousProb) {
                updateFlag = true;
            }
        }
    }
    
    /**
     * 刪除已經添加的元素，返回元素設置的隨機值
     * 如果元素不存在，返回null
     * @param key
     * @return
     */
    public Integer remove(Object key){
        //校验
        if (null == key) {
            throw new NullPointerException("key或者prob不能为空");
        }
        //刪除
        Integer remove = originalMap.remove(key);
        if (null != remove) {
            updateFlag = true;
        }
        return remove;
    }
    
    /**
     * 等同于put(key,0)
     * @param key
     */
    public void put(Object key){
        put(key,0);
    }
    
    /**
     * 返回随机值边界randomBound
     * @return
     */
    public Integer getRandomBound(){
        return randomBound;
    }
    
    /**
     * 返回元素的随机值prob
     * @param key
     * @return
     */
    public Integer getProb(Object key){
        return originalMap.get(key);
    }
    
    /**
     * 返回元素的集合
     * @return
     */
    public Set