Rails中如何实现Excel的COUNTA功能统计指定列非空单元格？

嘿，这个需求我太熟了！在Rails里确实有灵活又简便的方式来实现类似Excel COUNTA的统计，不管涉及多少列都能轻松搞定。下面分两种常用场景给你详细说：

一、数据库层面统计（推荐，效率优先） #

如果数据量较大，直接在数据库层面完成统计是最优选择，不用把所有数据加载到内存里处理。

1. 基础写法（针对少量列） #

直接通过select配合SQL的CASE WHEN语句，给每个用户计算出对应的统计值：

# 查询所有用户，并附带socio_count和personality_count
users = User.select(
  'users.*',
  # 统计job和address的非空数量
  'CASE WHEN job IS NOT NULL AND job != "" THEN 1 ELSE 0 END + CASE WHEN address IS NOT NULL AND address != "" THEN 1 ELSE 0 END AS socio_count',
  # 统计looks和wealth的非空数量
  'CASE WHEN looks IS NOT NULL AND looks != "" THEN 1 ELSE 0 END + CASE WHEN wealth IS NOT NULL AND wealth != "" THEN 1 ELSE 0 END AS personality_count'
)

# 使用时直接调用属性
users.each do |user|
  puts "用户#{user.id}的socio统计值：#{user.socio_count}"
  puts "用户#{user.id}的personality统计值：#{user.personality_count}"
end

2. 动态生成SQL（适配多列场景） #

如果涉及的列很多，手动写每个CASE WHEN太麻烦，可以把列名放到数组里，动态拼接SQL：

# 定义需要统计的列组
socio_columns = %w[job address education phone] # 可按需添加更多列
personality_columns = %w[looks wealth intelligence charm]

# 动态生成socio_count的SQL片段
socio_count_sql = socio_columns.map do |col|
  "CASE WHEN #{col} IS NOT NULL AND #{col} != '' THEN 1 ELSE 0 END"
end.join(' + ')

# 动态生成personality_count的SQL片段
personality_count_sql = personality_columns.map do |col|
  "CASE WHEN #{col} IS NOT NULL AND #{col} != '' THEN 1 ELSE 0 END"
end.join(' + ')

# 执行查询
users = User.select('users.*', "#{socio_count_sql} AS socio_count", "#{personality_count_sql} AS personality_count")

这样不管列数增加多少，只要把列名加到对应数组里就行，非常灵活。

二、模型实例方法（适合内存数据） #

如果已经把用户数据加载到内存中（比如做批量处理或小数据量场景），可以给User模型添加实例方法，直接在Ruby层面计算：

class User < ApplicationRecord
  # 统计socio类列的非空数量
  def socio_count
    # 使用present?判断：自动排除nil、空字符串，若要排除空格可改成send(col).to_s.strip.present?
    %w[job address].count { |col| send(col).present? }
  end

  # 统计personality类列的非空数量
  def personality_count
    %w[looks wealth].count { |col| send(col).present? }
  end
end

# 使用时直接调用方法
user = User.first
puts user.socio_count # 输出该用户job+address的非空数量
puts user.personality_count # 输出该用户looks+wealth的非空数量

这种写法更直观，代码可读性强，适合小数据量场景。

额外：统计全表的总非空数量 #

如果需要统计整个表中指定列的总非空单元格数，也可以用类似思路：

socio_columns = %w[job address]
# 数据库层面一次查询得到总和，效率最高
total_socio_sql = socio_columns.map { |col| "COUNT(CASE WHEN #{col} IS NOT NULL AND #{col} != '' THEN 1 END)" }.join(' + ')
total_socio = User.select("#{total_socio_sql} AS total").first.total

puts "全表socio列总非空数量：#{total_socio}"

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