ActiveRecord-Import 项目中复合主键导入问题的解决方案

问题背景

在 Rails 7.1 中使用 ActiveRecord-Import 这个 gem 进行批量数据导入时，当遇到带有复合主键(Composite Primary Key)的模型时，会出现无法正常导入的问题。这个问题表现为当尝试使用模型实例进行批量导入时，系统会抛出"can't quote Array"的错误。

技术细节分析

复合主键是指由多个列共同组成的主键，而不是传统的单一列主键。在 Rails 7.1 中，ActiveRecord 开始原生支持复合主键功能。然而，activerecord-import 这个 gem 在处理这种新型主键时出现了兼容性问题。

问题的核心在于：

1. 当模型定义了复合主键(如 self.primary_key = %i[id author_id])
2. 尝试使用模型实例进行批量导入(如 Book.import books)
3. gem 内部在生成 SQL 时会尝试引用主键序列(sequence)
4. 但复合主键情况下，主键是一个数组，无法被正确引用

解决方案比较

在实际应用中，开发者发现了几个可行的变通方案：

1. 使用哈希数组导入：将数据转换为哈希数组形式进行导入，这种方式可以绕过复合主键的问题

   books = []
   2.times do |i|
     books << {author_id: author.id, title: "book #{i}"}
   end
   Book.import books
   

2. 指定列名导入：明确指定要导入的列名，避免自动处理主键

   Book.import [:author_id, :title], [[author.id, "book 1"], [author.id, "book 2"]]
   

3. 等待官方修复：社区贡献者最终提交了一个修复补丁，解决了复合主键情况下的导入问题

最佳实践建议

对于使用复合主键的项目，建议：

1. 如果项目允许，优先考虑使用哈希数组方式进行批量导入，这是目前最稳定的方式
2. 如果需要使用模型实例导入(例如需要触发回调)，可以升级到包含修复补丁的版本
3. 对于关键业务系统，建议先在小规模测试环境中验证批量导入功能
4. 考虑使用 Rails 7.1 内置的 upsert 功能作为替代方案，特别是对于复合主键场景

技术实现原理

修复补丁主要解决了以下技术问题：

1. 正确处理复合主键情况下的序列名称查询
2. 避免将数组类型的主键直接传递给 SQL 引用函数
3. 在生成批量插入 SQL 时，正确识别和处理复合主键字段

这个问题的解决不仅修复了功能缺陷，也为 ActiveRecord-Import 项目更好地支持 Rails 7.1 的新特性铺平了道路。