Rails 5中使用Devise实现注册前验证用户ID存在性的方法

这个需求我之前在项目里落地过，Devise的扩展性足够支撑这种特殊业务逻辑，咱们从用户体验和后端安全两方面一步步来实现：

1. 先做前置验证页面（两步式流程更贴合需求） #

既然要求「先验证ID才能继续注册」，最直观的方式是做一个前置页面让用户输入推荐ID，验证通过后再进入正式注册流程。

第一步：新增推荐验证控制器 #

创建一个专门处理推荐ID验证的控制器：

# app/controllers/referrals_controller.rb
class ReferralsController < ApplicationController
  def new
    # 渲染输入推荐ID的页面
  end

  def create
    @referral_user = User.find_by(id: params[:referral_id])
    if @referral_user
      # 验证通过，把有效ID存入session，跳转到注册页
      session[:valid_referral_id] = params[:referral_id]
      redirect_to new_user_registration_path, notice: "验证通过，请完成注册"
    else
      flash[:alert] = "提供的用户ID不存在，请检查后重试"
      render :new
    end
  end
end

第二步：配置路由 #

把前置验证路由放在Devise注册路由之前，确保用户必须先走验证流程：

# config/routes.rb
Rails.application.routes.draw do
  get "referral/verify", to: "referrals#new", as: :verify_referral
  post "referral/confirm", to: "referrals#create", as: :confirm_referral
  devise_for :users, controllers: { registrations: "users/registrations" }
  # 其他业务路由...
end

第三步：前置验证页面视图 #

创建输入推荐ID的页面：

# app/views/referrals/new.html.erb
<h2>请输入推荐用户ID</h2>
<%= form_tag confirm_referral_path do %>
  <div class="form-group">
    <%= label_tag :referral_id, "推荐用户ID" %>
    <%= text_field_tag :referral_id, params[:referral_id], required: true, class: "form-control" %>
  </div>
  <%= submit_tag "验证并继续", class: "btn btn-primary" %>
<% end %>
<% if flash[:alert] %>
  <div class="alert alert-danger mt-3"><%= flash[:alert] %></div>
<% end %>

2. 自定义Devise注册控制器，强制验证逻辑 #

接下来要确保只有通过推荐验证的用户才能访问注册页面，并且注册时自动关联推荐ID。

第一步：生成自定义注册控制器 #

先生成Devise的自定义注册控制器文件：

rails generate devise:controllers users -c registrations

第二步：修改注册控制器逻辑 #

添加前置验证，拦截未通过推荐验证的请求，同时自动填充推荐ID：

# app/controllers/users/registrations_controller.rb
class Users::RegistrationsController < Devise::RegistrationsController
  # 访问注册页和提交注册前，先验证是否有有效推荐ID
  before_action :check_valid_referral, only: [:new, :create]

  private

  def check_valid_referral
    unless session[:valid_referral_id].present? && User.exists?(id: session[:valid_referral_id])
      redirect_to verify_referral_path, alert: "请先验证推荐用户ID"
    end
  end

  # 允许注册参数包含推荐ID，同时从session中自动填充
  def sign_up_params
    params.require(:user).permit(:email, :password, :password_confirmation)
          .merge(referral_user_id: session[:valid_referral_id])
  end

  # 注册完成后清除session中的推荐ID，避免重复使用
  def after_sign_up_path_for(resource)
    session.delete(:valid_referral_id)
    root_path # 这里可以改成你想要的注册后跳转页
  end
end

3. （可选）在User模型中关联推荐关系 #

如果需要记录新用户的推荐来源，可以给User模型添加关联字段：

第一步：生成迁移文件 #

rails generate migration AddReferralUserIdToUsers referral_user_id:integer
rails db:migrate

第二步：模型中添加关联和双重验证 #

为了确保数据安全，即使前端和控制器做了验证，模型层也要再加一层校验：

# app/models/user.rb
class User < ApplicationRecord
  devise :database_authenticatable, :registerable,
         :recoverable, :rememberable, :validatable

  # 关联推荐用户
  belongs_to :referral_user, class_name: "User", optional: true

  # 注册时强制验证推荐ID存在且有效
  validates :referral_user_id, presence: true, on: :create
  validate :referral_user_is_valid, on: :create

  private

  def referral_user_is_valid
    return if User.exists?(id: referral_user_id)
    errors.add(:referral_user_id, "推荐用户ID无效，请重新验证")
  end
end

4. 调整注册视图（可选） #

因为推荐ID已经通过前置验证并存在session中，注册页面可以隐藏这个字段，不用让用户重复输入：

# app/views/users/registrations/new.html.erb
<!-- 保留原有的邮箱、密码输入框，添加隐藏字段自动填充推荐ID -->
<%= f.hidden_field :referral_user_id, value: session[:valid_referral_id] %>

额外小提示 #

• 如果不想用两步流程，也可以把推荐ID输入框放在注册表单里，提交时先验证ID有效性再创建用户，但两步流程的用户体验更贴合「先验证再继续」的需求。
• 生产环境可以给推荐ID验证接口加限流，防止恶意枚举用户ID。
• 可以在session过期时给出友好提示，引导用户重新验证推荐ID。

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