woojean的博客 模仿、熟练、超越

基于Redis实现简单的红包分配算法

2017-04-05

下载代码:https://github.com/woojean/demos/tree/master/redis-hongbao

<?php

/** 
* 生成红包函数
* $totalMoney 总金额
* $num 红包数量
* $min 红包最小值
*/
function gen($totalMoney, $num, $min='0.01'){
  // 分配结果
  $ret = [];
  
  // 剩余红包金额
  $remainMoney = $totalMoney;  
  
  for ( $i = 1; $i < $num; $i++) {
    // 剩余红包数量
    $remainNum = $num-$i;  
    
    // 当前可领取的红包的最大值
    $remainMax = $remainMoney-$remainNum*$min;  
    
    $allocateMoney = mt_rand($min*100, $remainMax*100)/100;
    $remainMoney = $remainMoney-$allocateMoney;
    $ret[$i] = array(
      'allocation' => $allocateMoney,
      'remainder' => $remainMoney
    );
  }

  // 处理最后一个
  $ret[$num] = [
    'allocation'=>$remainMoney,
    'remainder'=>0
  ];
  
  return $ret;
}


// ======================= demo ========================

// 红包总金额
$totalMoney = 10;  

// 红包总数
$num = 10;  

$redis = new Redis();
$redis->connect('127.0.0.1', 6379);

// 库存队列
$mapStock = 'queue_stock';  

// 已抢队列 uid => hongbao
$mapGrab = 'queue_grab';  

// 剩余红包索引 
$listIndexs = 'list_indexs';  


$allocated = $redis->hlen($mapGrab);
if($allocated == $num){
  echo '已抢光!';
  exit;
}

$inited = $redis->hlen($mapStock);

// 如果索引尚未生成,生成红包库存
if( 0 == $inited ){
  // 生成红包库存
  $stock = gen($totalMoney,$num);  

  // 存储红包索引
  foreach($stock as $index => $hongbao){
    $redis->hset($mapStock, $index, json_encode($hongbao));
    $redis->lpush($listIndexs,$index);
  }
}

// 获取用户ID
$uid = intval($_GET['uid']);
if( $uid < 1){
  echo '用户ID非法!';
  exit;
}

// 判断用户是否已经抢过红包
$participated = $redis->hexists($mapGrab,$uid);
if($participated){
  echo '不能重复参加!';
  exit;
}

// 分配红包给用户(原子操作,关键!)
$index = $redis->lpop($listIndexs);
if(intval($index) < 1){
  echo '已抢光!';
  exit;
}

// todo 操作失败后,将红包的索引push回去

$hongbao = $redis->hget($mapStock, $index);
$redis->hset($mapGrab, $uid, json_encode($hongbao));

echo $uid.' -> ' . json_encode($hongbao);

关键点:
* 正确的红包生成算法(给定总金额、红包数、每个红包的最小金额);
* 红包分配的的原子性、互斥性(红包领取失败的回滚);

上一篇 商品列表

公众号
文章目录