使用java实现一个微信红包功能

/**
 * 红包类
 * @author mrchai
 *
 */
public class RedPacket {

 /**红包ID*/
 private int id;
 /**红包金额*/
 private BigDecimal money;
 
 public RedPacket() {
 }

 public RedPacket(int id, BigDecimal money) {
 super();
 this.id = id;
 this.money = money;
 }

 public int getId() {
 return id;
 }

 public void setId(int id) {
 this.id = id;
 }

 public BigDecimal getMoney() {
 return money;
 }

 public void setMoney(BigDecimal money) {
 this.money = money;
 }

 @Override
 public String toString() {
 return id+"号用户获得"+money+"元";
 }
 
}

/**
 * 红包异常
 * @author mrchai
 */
public class RedpacketException extends Exception{

 public RedpacketException() {
 // TODO Auto-generated constructor stub
 }
 
 public RedpacketException(String msg) {
 super(msg);
 }
}

package com.softeem.lesson18.RedPacket;

import java.math.BigDecimal;
import java.util.ArrayList;
import java.util.Random;

public class RedPacketManage {

 /** 设置每个红包最小金额 */
 static final BigDecimal MIN = new BigDecimal("0.01");

 /*
 * @double total 总金额
 * @int count 红包个数
 * @return 返回生成的所有红包金额集合
 */
 public static ArrayList<RedPacket> genRedPacket(double total, int count) throws RedPacketException {
 // 声明临时变量用于存储所有随机的红包对象
 ArrayList<RedPacket> packets = new ArrayList<RedPacket>();
 // 计算每个红包分配最低金额一共需要多少钱
 double min = MIN.multiply(new BigDecimal(count)).setScale(2, BigDecimal.ROUND_HALF_EVEN).doubleValue();
 if (min > total) {
  // 红包金额不够分配时，抛出异常
  throw new RedPacketException("每个红包金额不能少于0.01元");
 } else if (min == total) {
  // 红包金额恰好每人只够分配0.01元，则平均分配
  for (int i = 0; i < count; i++) {
  // 创建红包对象
  RedPacket item = new RedPacket(i + 1, new BigDecimal("0.01"));
  // 将红包加入集合
  packets.add(item);
  }
 } else {
  // 当总金额大于每人最少金额之和时，随机分配
  // 将总金额包装为BigDecimal
  BigDecimal totalMoney = new BigDecimal(total);
  
  //先为每人分配最低金额0.01元
  //避免因为总金额太少导致有红包金额为0
  for(int i=0;i<count;i++) {
  packets.add(new RedPacket(i+1, new BigDecimal("0.01")));
  }
  //将总金额设置为在原来基础上减去每人最低分配金额
  totalMoney = totalMoney.subtract(new BigDecimal(min));
  
  //声明临时变量统计当前分配的金额总数
  BigDecimal now = new BigDecimal(0);
  // 获取每个红包的比例
  double[] scale = randomScale(count);
  // 为前count-1个红包分配金额
  for (int i = 0; i < count - 1; i++) {
  // 获取当前比例红包需要分配的金额
  BigDecimal item = totalMoney.multiply(new BigDecimal(scale[i]))
      .setScale(2, BigDecimal.ROUND_HALF_EVEN);
  //为每人在最低金额基础上增加随机比例金额
  packets.get(i).setMoney(packets.get(i).getMoney().add(item));
  //累计已分配金额总数
  now = now.add(item);
  }
  // 剩余的金额给最后一个
  //获取剩余的金额
  BigDecimal last = totalMoney.subtract(now);
  //设置最后一个红包的金额为原来基础上增加剩余的总金额
  packets.get(count-1).setMoney(packets.get(count-1).getMoney().add(last).setScale(2, BigDecimal.ROUND_HALF_EVEN));
 }
 return packets;
 }

 /**
 * 随机红包金额比例
 * @param count 红包的份数
 * @return 每份红包的比例数组
 */
 private static double[] randomScale(int count) {
 // 临时数组存储所有红包的金额比例
 double[] scale = new double[count];
 Random r = new Random();
 double total = 0.0;
 for (int i = 0; i < count; i++) {
  // 为每一个元素设置一个1-100随机数
  scale[i] = r.nextInt(100) + 1;
  // 累计所有随机的数值
  total += scale[i];
 }
 // 循环计算每个红包的金额比例
 for (int i = 0; i < count; i++) {
  scale[i] = scale[i] / total;
 }
 return scale;
 }

 public static void main(String[] args) throws RedPacketException {
 ArrayList<RedPacket> list = genRedPacket(0.1, 5);
 BigDecimal t = new BigDecimal(0);
 for (RedPacket rp : list) {
  System.out.println(rp);
  t= t.add(rp.getMoney());
 }
 System.out.println(t);
 }
}

public static void main(String[] args) throws RedpacketException {
  ArrayList<RedPacket> list = genRedPacket(0.1, 5);
  BigDecimal t = new BigDecimal(0);
  for (RedPacket rp : list) {
    System.out.println(rp);
    t= t.add(rp.getMoney());
  }
  System.out.println(t);
}