activerecord-multi-tenant 核心组件深度解析从 Model Extensions 到 Query Rewriter【免费下载链接】activerecord-multi-tenantRails/ActiveRecord support for distributed multi-tenant databases like PostgresCitus项目地址: https://gitcode.com/gh_mirrors/ac/activerecord-multi-tenantactiverecord-multi-tenant 是一个为 Rails/ActiveRecord 提供分布式多租户数据库支持的强大工具特别适用于 PostgresCitus 环境。本文将深入解析其核心组件帮助开发者理解其工作原理和实现方式。核心组件概览activerecord-multi-tenant 主要包含以下核心组件Model Extensions提供模型层面的多租户支持Query Rewriter负责查询语句的重写和租户过滤Arel Visitors处理 SQL AST (Abstract Syntax Tree) 的访问者Tenant Enforcement Clauses租户强制过滤条件这些组件协同工作确保应用程序中的所有数据库操作都正确地与当前租户上下文关联。Model Extensions模型层面的多租户支持Model Extensions 是 activerecord-multi-tenant 最基础也最重要的组件之一。它通过multi_tenant方法为 ActiveRecord 模型添加多租户功能。核心功能实现Model Extensions 的核心代码位于 lib/activerecord-multi-tenant/model_extensions.rb。它主要提供以下功能声明多租户模型通过multi_tenant方法声明模型为多租户模型def multi_tenant(tenant_name, options {}) # ...实现代码... end自动设置租户 ID在新建记录时自动设置当前租户 IDafter_initialize proc { |record| if MultiTenant.current_tenant_id (!record.attribute_present?(partition_key) || record.public_send(partition_key.to_sym).nil?) record.public_send(:#{partition_key}, MultiTenant.current_tenant_id) end }防止租户 ID 修改一旦记录创建禁止修改其租户 IDdefine_method #{partition_key} do |tenant_id| write_attribute(partition_key.to_s, tenant_id) was send(#{partition_key}_was) was_nil_or_skipped was.nil? || was.instance_of?(Object) if send(#{partition_key}_changed?) persisted? !was_nil_or_skipped raise MultiTenant::TenantIsImmutable end tenant_id end关联处理自动处理与租户模型的关联关系if MultiTenant.tenant_klass_defined?(tenant_name, options) belongs_to tenant_name, **options.slice(:class_name, :inverse_of, :optional) .merge(foreign_key: options[:partition_key]) end使用示例在 Rails 模型中使用 Model Extensions 非常简单class Product ApplicationRecord multi_tenant :organization end这行代码会自动为Product模型添加租户支持包括自动设置organization_id、防止修改租户 ID 等功能。Query Rewriter智能查询重写Query Rewriter 是 activerecord-multi-tenant 的另一个核心组件负责在查询执行前自动添加租户过滤条件。其核心实现位于 lib/activerecord-multi-tenant/query_rewriter.rb。工作原理Query Rewriter 通过以下步骤实现查询重写解析 SQL AST使用 Arel 访问者解析 SQL 的抽象语法树识别多租户表找出查询中涉及的所有多租户表添加过滤条件为每个多租户表添加租户过滤条件处理关联查询确保关联查询中也正确应用租户过滤核心实现Query Rewriter 通过重写build_arel方法来实现查询重写def build_arel(*) arel super unless MultiTenant.with_write_only_mode_enabled? visitor MultiTenant::ArelTenantVisitor.new(arel) visitor.contexts.each do |context| node context.arel_node context.unhandled_relations.each do |relation| model MultiTenant.multi_tenant_model_for_table(MultiTenant::TableNode.table_name(relation.arel_table)) if MultiTenant.current_tenant_id enforcement_clause MultiTenant::TenantEnforcementClause.new(relation.arel_table[model.partition_key]) # ...根据节点类型添加过滤条件... end # ...处理关联查询... end end end arel end租户过滤条件Query Rewriter 使用TenantEnforcementClause来创建租户过滤条件class TenantEnforcementClause BaseTenantEnforcementClause private def tenant_arel if defined?(Arel::Nodes::Quoted) tenant_attribute.eq(Arel::Nodes::Quoted.new(MultiTenant.current_tenant_id)) else tenant_attribute.eq(MultiTenant.current_tenant_id) end end end这个类会生成类似organization_id 123的 SQL 条件确保查询只返回当前租户的数据。Arel VisitorsSQL AST 处理Arel Visitors 是 activerecord-multi-tenant 处理 SQL AST 的关键组件。它基于 Arel 库的访问者模式遍历和修改 SQL 抽象语法树。核心实现Arel Visitors 的核心实现是ArelTenantVisitor类class ArelTenantVisitor if Arel::Visitors.const_defined?(:DepthFirst) Arel::Visitors::DepthFirst else ::MultiTenant::ArelVisitorsDepthFirst end def initialize(arel) super(proc {}) statement_node_id nil contexts [] current_context nil accept(arel.ast) end # ...访问者方法实现... end这个访问者会遍历 SQL AST识别其中的多租户表和已处理的关系为后续的查询重写做准备。实际应用与最佳实践基本使用流程安装 gem将 activerecord-multi-tenant 添加到 Gemfile配置租户设置默认租户和分区键模型声明在需要多租户支持的模型上使用multi_tenant方法设置当前租户在请求处理过程中设置当前租户注意事项迁移文件使用 lib/activerecord-multi-tenant/migrations.rb 提供的迁移扩展性能考虑确保租户列上有适当的索引测试策略使用 spec/ 目录中的测试用例作为参考总结activerecord-multi-tenant 通过 Model Extensions 和 Query Rewriter 等核心组件为 Rails 应用提供了强大而透明的多租户支持。它的实现巧妙地利用了 ActiveRecord 和 Arel 的扩展点实现了对多租户查询的自动处理大大简化了多租户应用的开发复杂度。无论是构建新的多租户应用还是将现有应用迁移到多租户架构activerecord-multi-tenant 都是一个值得考虑的优秀工具。通过深入理解其核心组件的工作原理开发者可以更好地利用这个工具构建高效、安全的多租户应用。官方文档docs/ 核心源码lib/activerecord-multi-tenant/【免费下载链接】activerecord-multi-tenantRails/ActiveRecord support for distributed multi-tenant databases like PostgresCitus项目地址: https://gitcode.com/gh_mirrors/ac/activerecord-multi-tenant创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考