如何在ActiveRecord中检测MySQL表是否为分区表？

在MySQL环境下，我们可以通过查询系统元数据或解析建表语句两种方式，在运行时判断表是否为分区表，以下是具体实现方案：

方法一：查询information_schema系统表 #

MySQL的information_schema.PARTITIONS表存储了所有分区表的分区信息，非分区表不会在该表中生成带有有效分区名的记录。我们可以通过统计目标表的有效分区数量来判断：

class YourModel < ApplicationRecord
  def self.partitioned?
    query = <<~SQL
      SELECT COUNT(*)
      FROM information_schema.PARTITIONS
      WHERE TABLE_SCHEMA = DATABASE()
        AND TABLE_NAME = ?
        AND PARTITION_NAME IS NOT NULL
    SQL
    count = connection.execute(query, table_name).first[0]
    count > 0
  end
end

这个方法依赖数据库元数据，准确性更高，不会受建表语句格式、注释等因素干扰。

方法二：解析SHOW CREATE TABLE结果 #

通过获取表的建表语句，检查是否包含PARTITION BY关键字来判断：

class YourModel < ApplicationRecord
  def self.partitioned?
    create_table_result = connection.execute("SHOW CREATE TABLE #{table_name}").first
    create_table_sql = create_table_result[1]
    create_table_sql.downcase.include?('partition by')
  end
end

这种方法逻辑直观，但要注意如果建表语句的注释中包含PARTITION BY字符串，可能会导致误判，适合确认无特殊注释的场景。

使用示例 #

在业务逻辑中直接调用模型方法即可：

if YourModel.partitioned?
  # 执行分区表专属逻辑，比如分区查询优化
else
  # 执行非分区表逻辑
end

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