唯一邀请码生成方案



按照随机策略生成一个不重复的邀请码(字母+数字),既要不重复又要保证性能。每个用户对应一个邀请码,必须做到唯一性。邀请码的需要手动输入所以长度不能太长,同时不能让用户猜到邀请码的生成逻辑,所以邀请的生成逻辑也必须要随机 。

一、使用 NanoID或自有规则算法生成(非并发场景)

根据生成库生成邀请码,在查询数据库,如果存在就继续生成直到成功。

//https://github.com/ai/nanoid
//composer require hidehalo/nanoid-php
use Hidehalo\Nanoid\Client;
use Hidehalo\Nanoid\GeneratorInterface;
/**
 * 生成邀请码
 * @return string
*/
function createInviteCode(){
        $client = new Client();
        do{
            $code = $client->formattedId('123456789ABCDEFGHJKLMNPQRSTUVWXYZ',5);
           //查询库里存在不。
            $existed_count = DbModel::where('invite_code',$code)->count();
        }while($existed_count > 0);
        return $code;
}

二、使用Redis生成唯一邀请码

方案来源:https://blog.csdn.net/JSPSEO/article/details/125359454

一个自增点会导致数量连续有规律,那么我们可以设置多个池子,每个池子维护一个自增点,随机去一个池子通过自增id生成,就能够做到随机了。因为邀请码的长度有限制,所以我们需要算出来邀请码最大的范围值有多少,然后将最大的范围值分成若干个池子。

6位(数字+字母)长度的邀请码,根据可取字符的数量,我们将邀请码的形式设置为36进制数,这样邀请码的范围值就是为000000~zzzzzz,也就是等于十进制的2176782335,相当于我们最多能够生成2176782335个邀请码。如果此数量不满足业务上的需求,可以根据业务要求适当增删。我们假设每个池子能够存储邀请码的数量为10000个,则会有217678个池子。我们每个自增点就等于:邀请码 = (池子number * 10000 + 自增点)转成32进制 的字符串。

要注意红色字体,每个池子有最大自增点限制,为该池子能够存储邀请的最大数量。通过上述过程就能做到生成一个唯一且无明显规律的邀请码。当然要想最大程度无规律的话,可以将前几个池子给剔除掉,这样子就不会出现000001这样子的邀请码。

<?php
class Code{
protected  Redis $redis;
private $codeIncrementSet = 'code_increment_set';
private $codeAvailableSet = 'code_available_set';
public function __construct(){
    $this->redis = new Redis();
    $this->init();
}
 
//初始化
private function init(){
   if (0 === $this->redis->exists($this->codeIncrementSet)){
    //添加1-217678序号的池子,并且自增点设置为0
    $this->redis->zAdd($this->codeIncrementSet,array_fill(1,217678),0);
    //保存1-217678序号的可用池子
    $this->redis->sAddArray($this->codeAvailableSet,range(1,217678))
   }
}
public function getCode(){
    //随机取一个可用的邀请码池子
    if (is_null($index = $this->redis->srandmember($codeAvailableSet))){
          throw new BusinessException('可用邀请码码数为空');
    }
    //获取对应需要池子的自增点
    $increment =  (int)$this->redis->zScore($this->codeIncrementSet,$index)
    
    //计算code值
    $number = (int)index * 10000 + $increment;
    
    //维护自增点
    $this->redis->zIncrBy($this->codeIncrementSet,1,$index);
    
    //该池子被用完了
    if (9999 === $increment){
         $this->redis->sRem($this->codeAvailableSet,$index);
    }
    //返回36进制的邀请码,用0部位
     return str_pad(base_convert($number, 10, 36), 6, '0', STR_PAD_LEFT);
}
}

如何做到高性能

生成用户邀请码是一个频繁的操作,如果每次实时生成的话性能会比较低。可用利用空间换时间的思维。预先生成一批邀请码然后在需要的时候直接拿出来用即可。这种思路在开发中很常见,异步处理,用空间换取时间。通过这种方式就能快速的生成邀请码。


我忘记了父亲所说的话,我忘记了母亲所说的话,整个城市充满了花,整个城市充满了雨。