递归生成唯一推荐码是否计算成本高昂？百万用户场景如何优化？

嘿，这个问题问得很到位——百万用户规模下，你的实现确实可能遇到性能瓶颈，咱们一步步拆解来看：

原实现的成本分析 #

首先得明确：你的方案的开销高低，核心取决于两个点：referral_code字段有没有加唯一索引，以及推荐码的碰撞概率。

• 如果没加唯一索引：每次exists()查询都是全表扫描，百万级数据下这绝对是性能灾难，单次查询就能拖慢注册流程；
• 如果加了唯一索引：单次查询是O(log n)的开销，在8位随机字符串（假设是大小写字母+数字，共62种字符，总组合数≈2.1e14）的情况下，百万用户的碰撞概率极低，大部分时候一次生成就成功，开销其实不大。

但原实现有个更致命的问题：递归调用的风险。如果某天碰撞概率突然上升（比如生成逻辑出问题，或者用户量逼近组合数阈值），递归深度过大可能直接导致栈溢出，这比查询开销更影响稳定性。

更高效的实现方案 #

1. 数据库唯一约束+循环重试（替代递归） #

把referral_code设为唯一索引，然后用循环替代递归，把唯一性校验交给数据库（它做这个最专业），捕获唯一约束冲突的异常后重试：

public function generateUniqueReferralCode() {
    do {
        $referral_code = str_random(8);
        try {
            // 假设是给当前用户赋值并保存
            $this->referral_code = $referral_code;
            $this->save();
            return $referral_code;
        } catch (\Illuminate\Database\QueryException $e) {
            // 只处理唯一约束冲突的异常（MySQL错误码是1062，其他数据库需对应调整）
            if ($e->getCode() !== '1062') {
                throw $e;
            }
            // 冲突则继续循环生成新码
        }
    } while (true);
}

这个方案既保留了随机码的无关联性，又避免了递归的栈溢出风险，性能在百万级规模下完全够用。

2. 基于用户ID的短码转换（零碰撞+零查询） #

这是性能拉满的方案——利用用户自增ID的唯一性，把它转换成短字符串作为推荐码，完全不需要数据库查询，也不存在碰撞：

public function generateUniqueReferralCode() {
    $id = $this->id; // 确保用户已经生成了自增ID
    $chars = '0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ';
    $code = '';
    
    do {
        $code .= $chars[$id % 62];
        $id = floor($id / 62);
    } while ($id > 0);
    
    // 可选：补前导字符固定8位长度，避免短码暴露用户ID顺序
    return str_pad(strrev($code), 8, 'X', STR_PAD_LEFT);
}

如果不想让别人通过推荐码反推用户ID，可以给ID加一个固定的混淆因子（比如$id += 123456789）再转码，既保证唯一性，又增加了随机性。

3. 预生成推荐码池（高并发场景） #

如果你的系统注册并发量极高，实时生成+重试的开销还是有点大，可以提前批量生成一批推荐码存在单独的表中，用户注册时直接取未使用的码：

public function generateUniqueReferralCode() {
    // 用行锁+跳过已锁行保证并发安全（MySQL 8.0+支持SKIP LOCKED）
    $referralCode = ReferralCode::where('used', false)
        ->lockForUpdate()
        ->skipLocked()
        ->first();
    
    if (!$referralCode) {
        // 码池耗尽时批量补充
        $this->generateBatchReferralCodes(1000);
        return $this->generateUniqueReferralCode();
    }
    
    $referralCode->used = true;
    $referralCode->save();
    return $referralCode->code;
}

private function generateBatchReferralCodes($count) {
    $codes = [];
    for ($i = 0; $i < $count; $i++) {
        $codes[] = ['code' => str_random(8), 'used' => false];
    }
    // 利用insertOrIgnore自动过滤生成时的重复码
    ReferralCode::insertOrIgnore($codes);
}

这个方案把生成开销提前到后台，注册时只需要简单的查询和更新，性能极其稳定。

内容的提问来源于stack exchange，提问作者user3574492