如何在NodeJS中为注册用户创建唯一推荐链接？

嘿，不用找插件啦！用Mongoose自带的中间件就能轻松实现，而且完全可控。我给你一步步拆解：

核心思路 #

给每个用户生成唯一的推荐链接，关键是保证链接后缀的唯一性，最简单的方式就是利用Mongoose自动生成的_id（天生唯一），或者用轻量库生成短码。然后在用户注册时自动把链接存入数据库。

具体实现代码 #

1. 基础版：用用户_id生成链接 #

直接修改你的UserSchema，添加pre-save钩子（注册/保存前自动执行）：

const mongoose = require('mongoose');
const passportLocalMongoose = require('passport-local-mongoose');

var UserSchema = new mongoose.Schema({
  username: String,
  password: String,
  referralLink: String,
  // 可选：记录推荐成功的数量
  referralCount: { type: Number, default: 0 }
});

// 注册前自动生成推荐链接
UserSchema.pre('save', function(next) {
  // 只有当链接为空时才生成（避免更新用户时重复覆盖）
  if (!this.referralLink) {
    // 替换成你的网站域名
    const baseReferralUrl = 'https://your-site.com/ref/';
    // 用用户唯一_id拼接成链接
    this.referralLink = `${baseReferralUrl}${this._id}`;
  }
  next();
});

UserSchema.plugin(passportLocalMongoose);
module.exports = mongoose.model('User', UserSchema);

2. 进阶版：生成短码推荐链接 #

如果觉得_id太长不够友好，可以用nanoid生成短码（碰撞概率极低，几乎不用担心重复）：

首先安装依赖：

npm install nanoid

然后修改钩子：

const mongoose = require('mongoose');
const passportLocalMongoose = require('passport-local-mongoose');
const { nanoid } = require('nanoid');

var UserSchema = new mongoose.Schema({
  username: String,
  password: String,
  referralLink: String,
  referralCount: { type: Number, default: 0 }
});

UserSchema.pre('save', async function(next) {
  if (!this.referralLink) {
    const baseUrl = 'https://your-site.com/ref/';
    // 生成6位短码，长度可自定义（比如8位更安全）
    const shortCode = nanoid(6);
    
    // 保险起见：检查数据库是否已有相同短码（概率极低，但防万一）
    const existingUser = await this.constructor.findOne({ 
      referralLink: `${baseUrl}${shortCode}` 
    });
    
    if (existingUser) {
      // 重复就重新触发钩子生成新码
      return this.pre('save', next);
    }
    
    this.referralLink = `${baseUrl}${shortCode}`;
  }
  next();
});

UserSchema.plugin(passportLocalMongoose);
module.exports = mongoose.model('User', UserSchema);

后续：处理推荐链接的访问 #

当用户通过推荐链接访问时，你需要：

• 从URL中解析出_id或短码
• 找到对应的推荐用户
• 在新用户注册时，给新用户添加referrer字段（记录推荐人的_id）
• 可选：更新推荐人的referralCount计数

总结 #

完全不需要额外插件，用Mongoose的pre-save中间件就能搞定，既灵活又能自定义链接格式。如果后续需要更复杂的推荐逻辑（比如过期时间、多级推荐），再考虑扩展即可。

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