Rails多租户开发：使用Apartment gem跨租户复制关联数据的高效实现方法

这确实是Apartment多租户场景下常见的痛点——多层关联数据复制频繁切换租户会带来明显的性能损耗，尤其是数据量较大的时候。我之前处理过类似的需求，分享几个亲测有效的优化方案：

方案一：直接跨数据库连接操作（无需切换租户） #

Apartment本质是为每个租户维护独立的数据库连接，我们可以直接获取源租户和目标租户的连接，在不切换租户上下文的前提下完成数据读取和写入，完全避免租户切换的开销。

实现示例 #

# 定义源和目标租户名称
source_tenant = "tenant_a"
target_tenant = "tenant_b"

# 获取对应租户的数据库连接
source_conn = Apartment::Tenant.connection(source_tenant)
target_conn = Apartment::Tenant.connection(target_tenant)

# 在源连接下预加载所有关联数据，减少N+1查询
courses = Course.using(source_conn).includes(lessons: :attachments).where(id: [1, 2, 3]) # 替换为你的筛选条件

# 开启事务，确保数据一致性
Course.using(target_conn).transaction do
  courses.each do |course|
    # 过滤掉自动生成的字段（id、时间戳、租户标识等）
    course_attrs = course.attributes.except("id", "created_at", "updated_at", "tenant_id")
    # 在目标连接下创建课程
    new_course = Course.using(target_conn).create!(course_attrs)

    # 处理课程关联的课时
    course.lessons.each do |lesson|
      lesson_attrs = lesson.attributes.except("id", "created_at", "updated_at", "course_id")
      new_lesson = new_course.lessons.create!(lesson_attrs)

      # 处理课时关联的附件
      lesson.attachments.each do |attachment|
        attachment_attrs = attachment.attributes.except("id", "created_at", "updated_at", "lesson_id")
        new_lesson.attachments.create!(attachment_attrs)
      end
    end
  end
end

优点 #

• 完全避免租户切换带来的连接切换开销；
• 利用ORM的预加载机制减少查询次数；
• 代码逻辑清晰，容易维护和扩展。

适用场景 #

• 中等数据量的复制需求；
• 需要保留ORM的关联校验和回调逻辑。

方案二：数据库级批量导入导出（性能最优） #

如果你的数据量较大（比如上千条课程+关联数据），ORM层面的操作会有性能瓶颈，这时可以直接用数据库原生的导出/导入命令（比如PostgreSQL的COPY、MySQL的LOAD DATA INFILE），这是性能最高的方案。

实现思路（PostgreSQL为例） #

1. 从源租户导出需要复制的主表数据（如courses），并保留源记录的id用于后续关联映射；
2. 将主表数据导入目标租户，同时记录源id到目标id的映射关系；
3. 导出关联表数据（如lessons、attachments），替换外键为目标租户的新id；
4. 导入关联表数据到目标租户。

简化代码示例 #

source_tenant = "tenant_a"
target_tenant = "tenant_b"
course_ids = [1, 2, 3] # 要复制的课程ID

# 1. 导出源租户课程数据
Apartment::Tenant.switch(source_tenant) do
  ActiveRecord::Base.connection.execute(
    "COPY (SELECT id, name, description FROM courses WHERE id IN (#{course_ids.join(',')})) TO '/tmp/courses_export.csv' WITH CSV HEADER;"
  )
end

# 2. 导入课程到目标租户并建立ID映射
Apartment::Tenant.switch(target_tenant) do
  # 临时添加original_id字段存储源ID（后续可以删除）
  ActiveRecord::Base.connection.execute("ALTER TABLE courses ADD COLUMN original_id INTEGER;")
  # 导入数据
  ActiveRecord::Base.connection.execute(
    "COPY courses (original_id, name, description) FROM '/tmp/courses_export.csv' WITH CSV HEADER;"
  )
  # 获取源ID到目标ID的映射
  course_mapping = Course.pluck(:original_id, :id).to_h
  # 移除临时字段
  ActiveRecord::Base.connection.execute("ALTER TABLE courses DROP COLUMN original_id;")
end

# 3. 处理课时数据（类似步骤，替换course_id为映射后的ID）
# ... 同理处理attachments数据

优点 #

• 性能远超ORM操作，适合大数据量复制；
• 减少应用层的内存占用。

缺点 #

• 依赖数据库特定语法，跨数据库兼容性差；
• 需要手动处理关联外键的映射，逻辑相对复杂；
• 无法触发ORM的回调和校验逻辑（如果需要的话，要额外处理）。

适用场景 #

• 大数据量的跨租户复制需求；
• 对性能要求极高的场景。

方案三：序列化+批量创建（代码简洁） #

如果数据量不大，可以先将源数据序列化为包含关联的JSON结构，再在目标租户下批量创建，这种方式代码更简洁，只需要切换两次租户（一次读，一次写）。

实现示例 #

source_tenant = "tenant_a"
target_tenant = "tenant_b"

# 1. 在源租户下序列化数据（包含所有关联）
source_data = nil
Apartment::Tenant.switch(source_tenant) do
  source_data = Course.includes(lessons: :attachments).where(id: [1,2,3]).as_json(
    include: { lessons: { include: :attachments } }
  )
end

# 2. 在目标租户下批量创建
Apartment::Tenant.switch(target_tenant) do
  Course.transaction do
    source_data.each do |course_json|
      # 过滤自动字段
      course_attrs = course_json.except("id", "created_at", "updated_at")
      lessons_json = course_json.delete("lessons")

      new_course = Course.create!(course_attrs)

      lessons_json.each do |lesson_json|
        lesson_attrs = lesson_json.except("id", "created_at", "updated_at", "course_id")
        attachments_json = lesson_json.delete("attachments")

        new_lesson = new_course.lessons.create!(lesson_attrs)

        attachments_json.each do |attachment_json|
          attachment_attrs = attachment_json.except("id", "created_at", "updated_at", "lesson_id")
          new_lesson.attachments.create!(attachment_attrs)
        end
      end
    end
  end
end

优点 #

• 代码简洁，利用ActiveRecord的序列化功能快速处理关联；
• 只需要切换两次租户，开销远小于频繁切换。

适用场景 #

• 小数据量的复制需求；
• 追求代码简洁性的场景。

额外注意事项 #

• 事务处理：所有复制操作都要包裹在事务中，避免部分复制导致的数据不一致；
• 唯一性约束：确保复制后的对象不会违反目标租户的唯一性约束（比如课程名称唯一），必要时添加前缀或修改字段值；
• 索引优化：如果用数据库级导入，导入后建议重建目标表的索引，提升后续查询性能；
• 回调和校验：如果依赖ORM的回调（如after_create）或校验逻辑，优先选择方案一或方案三，数据库级导入不会触发这些逻辑。

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