Java模拟并发抽奖

import java.util.ArrayList;
import java.util.List;
import java.util.Random;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.LinkedBlockingQueue;
import java.util.concurrent.ThreadPoolExecutor;
import java.util.concurrent.TimeUnit;

/**
 * @author cc
 * @date 2024/7/5 9:58
 * @description 抽奖按照概率抽取奖品
 */
public class Lucky {
     public static void main(String[] args) {
         //所有奖品id
         List<Integer> prizeIdList = new ArrayList<Integer>();
         //添加奖品id
         prizeIdList.add(1);
         prizeIdList.add(2);
         //所有奖品数量 与奖品id对应
         List<Integer> prizeNumList = new ArrayList<>();
         prizeNumList.add(1);
         prizeNumList.add(99);
         Lottery lottery = new Lottery(prizeIdList, prizeNumList);

         //模拟并发抽奖
         ExecutorService executorService = new ThreadPoolExecutor(100, 200, 5, TimeUnit.MILLISECONDS, new LinkedBlockingQueue<Runnable>(50));
         for (int i = 0; i < 250; i++) {
             executorService.submit(new Human(lottery));
         }

         executorService.shutdown();
     }
}

/**
 * 抽奖类
 */
class Lottery{
    //所有奖品id
    private List<Integer> prizeIdList;
    //每种奖品的数量
    private List<Integer> prizeNumList;
    //剩余奖品总数 (避免每次抽奖遍历每种奖品的数量)
    private Integer prizeSum = 0;

    Lottery(List<Integer> prizeIdList, List<Integer> prizeNumList){
        this.prizeIdList = prizeIdList;
        this.prizeNumList = prizeNumList;
        //初始化所有奖品数量
        for (int i = 0; i < prizeNumList.size(); i++){
            prizeSum += prizeNumList.get(i);
        }
    }

    /**
     * 抽奖
     * @return 奖品id
     * @return null 奖池已空未抽到奖
     */
    public synchronized Integer getOne(){
        //过滤奖品已抽完
        if (prizeSum == 0){
            System.out.print("奖池已被抽完,");
            return null;
        }
        //随机到的奖品数字
        Integer num = randomInt(1, prizeSum);
        Integer temp = 0;
        //遍历到该奖品数字对应的prizeNumList下标
        for (int i = 0; i < prizeNumList.size(); i++){
            temp += prizeNumList.get(i);
            if (temp >= num){
                //总奖品剩余数-1
                prizeSum--;
                //该奖品剩余数-1
                prizeNumList.set(i, prizeNumList.get(i)-1);
                //返回奖品id
                return prizeIdList.get(i);
            }
        }
        return null;
    }
    /**
     * [min, max]
     * 生成max 到 min的随机数
     * @param min
     * @param max
     * @return 
     */
    public static int randomInt(int min, int max) {
        return new Random().nextInt(min, max+1);
    }

}
class Human implements Runnable{
    private final Lottery lottery;

    Human(Lottery lottery){
        this.lottery = lottery;
    }
    @Override
    public void run() {
        try {
            Thread.sleep(10);
        } catch (InterruptedException e) {
            throw new RuntimeException(e);
        }
        System.out.println(System.currentTimeMillis() + "抽到: " + lottery.getOne());
    }
}