如何在Rails Console中根据列格式过滤Candidate模型数据

直接在数据库层完成过滤，避免全量拉取数据到Ruby内存筛选，性能更高，以下是两种可落地的实现：

方案1：正则查询作用域（快速实现，适合低/中频查询场景） #

根据你使用的数据库，在Candidate模型中添加对应匹配范围：

MySQL 语法 #

# app/models/candidate.rb
class Candidate < ApplicationRecord
  # 筛选SSN格式（xxx-xx-xxxx）数据
  scope :ssn_format, -> { where("ssn_or_ein REGEXP ?", '^[0-9]{3}-[0-9]{2}-[0-9]{4}$') }
  # 筛选EIN格式（xx-xxxxxxx）数据
  scope :ein_format, -> { where("ssn_or_ein REGEXP ?", '^[0-9]{2}-[0-9]{7}$') }
end

PostgreSQL 语法 #

# app/models/candidate.rb
class Candidate < ApplicationRecord
  scope :ssn_format, -> { where("ssn_or_ein ~ ?", '^[0-9]{3}-[0-9]{2}-[0-9]{4}$') }
  scope :ein_format, -> { where("ssn_or_ein ~ ?", '^[0-9]{2}-[0-9]{7}$') }
end

使用方式：

• 查所有SSN类型数据：Candidate.ssn_format
• 查所有EIN类型数据：Candidate.ein_format
• 支持链式拼接其他查询条件，比如Candidate.ein_format.where(status: :active)

方案2：冗余类型字段（性能最优，适合高频查询场景） #

如果这个过滤逻辑是业务高频使用的功能，可以新增冗余字段避免每次查询都跑正则：

1. 生成迁移新增字段：

rails g migration AddSsnOrEinTypeToCandidates ssn_or_ein_type:string
rails db:migrate

2. 模型加自动赋值回调：

# app/models/candidate.rb
class Candidate < ApplicationRecord
  before_save :assign_ssn_or_ein_type, if: :ssn_or_ein_changed?

  private
  def assign_ssn_or_ein_type
    self.ssn_or_ein_type = if ssn_or_ein.match?(/^\d{3}-\d{2}-\d{4}$/)
      'ssn'
    elsif ssn_or_ein.match?(/^\d{2}-\d{7}$/)
      'ein'
    end
  end
end

3. 补全历史数据类型标记：

rails runner "Candidate.where(ssn_or_ein_type: nil).find_each(&:save)"

后续查询直接走字段匹配即可，还可以给ssn_or_ein_type加索引进一步优化查询速度：

Candidate.where(ssn_or_ein_type: 'ssn')

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