activerecord-multi-tenant-next 2.5.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: 9c71faa57caf1f2b1168a7fb90edd30b1394f6e24812de44cf3df4ea9cf48b48
+  data.tar.gz: 878717495ee2392449f346ace840518fec7d963d914cc61f2663c704d6fbf64b
+SHA512:
+  metadata.gz: bc08983b8b2645b256241d5b6f43e60756338cf78c53ec79268644d2828a9a3fc3e0977a7e65589e14ce14c965293cc48efcc961fb569919696043af20f78b59
+  data.tar.gz: d017e6f20df585cefea35ffe8ea2b10e9a2db47634bd56f6c259e5f7b69586b5bedf1fb9d9ae98c2200287c0a6cb860ba658ca4a2a3074c32f917a7a82fdc37a

data/CHANGELOG.md

@@ -0,0 +1,264 @@
+# Changelog
+
+## 2.5.0      2026-10-02
+First release of the maintained fork, published as `activerecord-multi-tenant-next`.
+
+* Adds support for Rails 7.2 (#239), 8.0 (#260) and 8.1
+* Adds Ruby 3.4 to the test matrix, and makes CI actually install each appraisal gemfile (#264)
+* Fixes `update_all` writing `Arel.sql(...)` values as literal strings inside `MultiTenant.with` (#278)
+* Fixes `increment_counter`, `update_counters`, `increment!` and `decrement!` setting the column to `NULL` inside `MultiTenant.with` (#279, #282)
+* Bumps the optimistic locking column in tenant-scoped `update_all`, as Active Record does (#282)
+* Fixes `update_all` / `delete_all` inside `MultiTenant.with` raising on Active Record 6.0 and 8.1
+* Fixes joined `update_all` on Rails 8.1 failing on Citus with "complex joins are only supported when all distributed tables are co-located"
+* Fixes `NoMethodError: undefined method 'alias'` for Arel functions in conditions on Rails 8.1 (#284)
+* Fixes `TypeError: Cannot visit Arel::SelectManager` for Arel subqueries in conditions (#236)
+* Fixes `SystemStackError` in the schema dumper when another gem (e.g. strong_migrations) prepends to `ActiveRecord::SchemaDumper` (#271, also proposed in #245)
+* Fixes `Model.limit(n).delete_all` and `Model.limit(n).update_all` generating an incorrect query (#200)
+* Fixes tenant scoping for `update_all` (#223)
+* Fixes a `SyntaxError` in `undistribute_table` (#226)
+* Fixes errors when `table_name` is nil (#218)
+* Uses `prepend` instead of `alias` to patch Active Record (#216)
+
+## 2.4.0      2023-09-22
+* Adds citus 12 to test matrix (#210)
+* Adds Support for rails 7.1 (#208)
+* Fix missing scope in habtm.rb (#207)
+* Update logic inside the tenant_klass_defined? method (#202)
+
+## 2.3.0      2023-06-05
+* Adds has_and_belongs_to_many feature with tenant (#193)
+* Removes eol ruby versions 
+* Adds documentation in ReadTheDocs platform (#196)
+* Organizes badges in documentation and README.md (#197) 
+* Wrap ActiveRecord::Base with ActiveSupport.on_load (#199)
+
+## 2.2.0      2022-12-06
+* Handle changing tenant from `nil` to a value [#173](https://github.com/citusdata/activerecord-multi-tenant/pull/173)
+* Allow Partitioned tables to be created without a primary key [#172](https://github.com/citusdata/activerecord-multi-tenant/pull/172)
+* Only attempt to reload with MultiTenant when parition_key is present [#175](https://github.com/citusdata/activerecord-multi-tenant/pull/175)
+* Remove support for Ruby 2.5 & ActiveRecord 5.2
+
+## 2.1.6      2022-11-23
+* Fix undefined wrap_methods error & wrap_methods version check [#170](https://github.com/citusdata/activerecord-multi-tenant/pull/170)
+
+## 2.1.5      2022-11-20
+* Fix `MultiTenant.without` codegen bug in Rails 6.1+ [#168](https://github.com/citusdata/activerecord-multi-tenant/pull/168)
+
+## 2.1.4      2022-11-03
+* Fixes #166 where db:schema:dump is broken when using this gem with MySQL [#167](https://github.com/citusdata/activerecord-multi-tenant/pull/167)
+
+## 2.1.3      2022-10-27
+* Error when calling a method that takes keyword arguments with MultiTenant.wrap_methods [#164](https://github.com/citusdata/activerecord-multi-tenant/pull/164)
+
+## 2.1.2      2022-10-26
+* Fixes issue when wraping methods that require a block [#162](https://github.com/citusdata/activerecord-multi-tenant/pull/162)
+
+## 2.1.1      2022-10-20
+* Fix query building for models with mismatched partition_keys [#150](https://github.com/citusdata/activerecord-multi-tenant/pull/150)
+* Identify tenant even if class name is nonstandard [#152](https://github.com/citusdata/activerecord-multi-tenant/pull/152)
+* Add current_tenant_id to WHERE clauses when calling methods on activerecord instance or its associations [#154](https://github.com/citusdata/activerecord-multi-tenant/pull/154)
+* Make create_distributed_table, create_reference_table reversible & add ruby wrapper for rebalance_table_shards [#155](https://github.com/citusdata/activerecord-multi-tenant/pull/155)
+* Support create_distributed_table, create_reference_table in schema.rb [#156](https://github.com/citusdata/activerecord-multi-tenant/pull/156)
+* Add client and server sidekiq middleware to sidekiq middleware chain [#158](https://github.com/citusdata/activerecord-multi-tenant/pull/158)
+
+## 2.0.0      2022-05-19
+
+* Replace RequestStore with CurrentAttributes [#139](https://github.com/citusdata/activerecord-multi-tenant/pull/139)
+* Support changing table_name after calling multi_tenant [#128](https://github.com/citusdata/activerecord-multi-tenant/pull/128)
+* Allow to use uuid as primary key on partition table [#112](https://github.com/citusdata/activerecord-multi-tenant/pull/112)
+* Support latest Rails 5.2 [#145](https://github.com/citusdata/activerecord-multi-tenant/pull/145)
+* Support optional: true for belongs_to [#147](https://github.com/citusdata/activerecord-multi-tenant/pull/147)
+
+
+## 1.2.0      2022-03-29
+
+* Test Rails 7 & Ruby 3
+* Fix regression in 1.1.1 involving deleted tenants [#123](https://github.com/citusdata/activerecord-multi-tenant/pull/123)
+* Fix incorrect SQL generated when joining two models and one has a default scope [#132](https://github.com/citusdata/activerecord-multi-tenant/pull/132)
+* Update for Rails 5+ removal of type_cast_for_database [#135](https://github.com/citusdata/activerecord-multi-tenant/pull/135)
+
+
+## 1.1.1      2021-01-15
+
+* Add support for Rails 6.1 [#108](https://github.com/citusdata/activerecord-multi-tenant/pull/108)
+* Fix statement cache for has_many through relations [#103](https://github.com/citusdata/activerecord-multi-tenant/pull/103)
+
+
+## 1.1.0      2020-08-06
+
+* See commits for changes:
+  https://github.com/citusdata/activerecord-multi-tenant/commits/v1.1.0
+
+
+## 1.0.4      2019-10-30
+
+* Fix bug introduced in 1.0.3 for delete when table is reference or not distributed
+
+
+## 1.0.3      2019-10-28
+
+* Ensure that when using object.delete, we set the tenant
+
+
+## 1.0.2      2019-09-20
+
+* Compatibility  with rails 6
+* Remove support for rails 4.0  and  4.1
+* Fix bug when multiple databases are used
+
+
+## 1.0.1      2019-08-27
+
+* Ensure current tenant is present before adding tenant id filter in DatabaseStatements
+
+
+## 1.0.0      2019-07-05
+
+* Fix `RETURNING id` for distributed tables with no primary key
+* Include fix for partial select described in issue [#34](https://github.com/citusdata/activerecord-multi-tenant/issues/34).
+  - When doing a partial select without the tenant like `Project.select(:name).find(project.id)` it would raise `ActiveModel::MissingAttributeError (missing attribute: tenant_id)`
+
+
+## 0.11.0      2019-06-12
+
+* Fix queries with joins by including the tenant column when current tenant isn't set
+  - A common use case is having a filter on the tenant, but `MultiTenant.with` isn't used like `Project.where(account_id: 1).eager_load(:categories)`. This version fixes the ORM call to include in the `join`: `"project_categories"."account_id" = "projects"."account_id"`
+
+
+## 0.10.0      2019-05-31
+
+* Add `MultiTenant.without` to remove already set tenant context in a block [#45](https://github.com/citusdata/activerecord-multi-tenant/pull/45) [Jackson Miller](https://github.com/jaxn)
+* Fix uninitialized constant X::ActiveRecord::VERSION [#42](https://github.com/citusdata/activerecord-multi-tenant/pull/42) [vollnhals](https://github.com/vollnhals)
+* Fix find and find_by caching issues
+  - This builds on work by multiple contributors, and fixes issues where the
+    tenant_id would be cached across different tenant contexts for find
+    and find_by methods. This issue was only present with prepared
+    statements turned on
+  -  Note that the mechanism to solve this is slightly different for Rails 4
+    and 5:
+    - Rails 4: Disable any caching for find and find_by methods
+    - Rails 5: Explicitly add the current_tenant_id into the cache key
+  - This also ensures that we test both prepared statements on and off
+    on Travis
+* Added method to ensure that the current tenant is loaded [#49](https://github.com/citusdata/activerecord-multi-tenant/pull/49) [Stephen Bussey](https://github.com/sb8244)
+  - This is ideal for fully utilizing the ActiveRecord extensions, as they only take effect when the
+    current tenant is not an ID
+* Update loofah and rack to fix security warnings
+  - Note that loofah is not a direct dependency of the libary, so this only
+    applies when running the test suite
+* Remove monkey patch that previously disabled referential integrity (DISABLE/ENABLE TRIGGER ALL) [#53](https://github.com/citusdata/activerecord-multi-tenant/pull/53) [Rémi Piotaix](https://github.com/piotaixr)
+  - This was required for Citus compatibility, but the issue has been fixed in
+    Citus for over a year (https://github.com/citusdata/citus/issues/1080)
+
+
+## 0.9.0       2018-06-22
+
+* ActiveRecord 5.2 support [Nathan Stitt](https://github.com/nathanstitt) & [osyo-manga](https://github.com/osyo-manga)
+
+
+## 0.8.1       2017-10-06
+
+* Cast attribute name to a string to avoid double applying tenant clause [Ben Olive](https://github.com/sionide21)
+* Allow bulk delete/update with subqueries on joins [Kyle Bock](https://github.com/kwbock)
+
+
+## 0.8.0       2017-08-16
+
+* Significant improvements and simplifications of query rewriting
+  * Big thanks to [Kyle Bock](https://github.com/kwbock) and [Ben Olive](https://github.com/sionide21)
+    for (re-)writing this code and verifying it works well
+  * This fixes caching issues across multiple MultiTenant.with { } blocks when
+    interacting with the Rails statement cache
+* Drop support for Rails 3.2
+  * The arel version used in Rails 3.2 has caused more trouble than its worth -
+    it seems less troublesome to ask any users of this library to upgrade to at
+    least Rails 4.0
+
+
+## 0.7.0       2017-07-18
+
+* Switch back to Relation-based mechanism of hooking into ActiveRecord (this resolves issues for simple queries that didn't get rewritten)
+* Query rewriter improvements
+  * Handle OUTER JOIN correctly
+  * Correctly rewrite sub-selects
+* Model tenant method: Only return cached object if not loaded
+* Fix support for inherited model classes that only have `multi-tenant` on a higher level object [Aaron Severs](https://github.com/webandtech) [#13](https://github.com/citusdata/activerecord-multi-tenant/pull/13)
+* Sidekiq middleware: Don't automatically perform a find for the tenant object for every job [Scott Mitchell](https://github.com/smitchelus) [#14](https://github.com/citusdata/activerecord-multi-tenant/pull/14)
+* Fix automatic inverse of on singular associations [Kyle Bock](https://github.com/kwbock) [#15](https://github.com/citusdata/activerecord-multi-tenant/pull/15)
+* Fix bug that prevents fast truncate from running [Kyle Bock](https://github.com/kwbock) [#17](https://github.com/citusdata/activerecord-multi-tenant/pull/17)
+
+
+## 0.6.0       2017-06-09
+
+* Query rewriter
+  * Change hook from per-relation to be pre-SQL statement thats being output
+    - This should resolve issues where we added conditions in the wrong place
+  * Use table name to model klass registry
+* Improve tests for activerecord-multi-tenant
+  * Use lower shard count to speed up tests
+  * Drop database cleaner dependency
+
+
+## 0.5.0       2017-05-08
+
+* Write-only mode that enables step-by-step migrations
+* Add tenant_id to queries using rewrite instead of default_scope/unscoped [Ben Olive](https://github.com/sionide21)
+* Query monitor that warns you about missing tenant_id
+* Helper for fast truncation
+* Sidekiq middleware
+
+
+## 0.4.1       2017-03-23
+
+* Allow use outside of Rails, e.g. when using Sinatra with ActiveRecord [@nathanstitt](https://github.com/nathanstitt) [#5](https://github.com/citusdata/activerecord-multi-tenant/pull/5)
+
+
+## 0.4.0       2017-03-22
+
+* Infer multi_tenant setting from parent classes [@webandtech](https://github.com/webandtech) [#6](https://github.com/citusdata/activerecord-multi-tenant/pull/6)
+* Remove use of global tenant klass variable [@webandtech](https://github.com/webandtech) [#6](https://github.com/citusdata/activerecord-multi-tenant/pull/6)
+* Support passing ID values to MultiTenant.with directly [@webandtech](https://github.com/webandtech) [#6](https://github.com/citusdata/activerecord-multi-tenant/pull/6)
+  * This effectively deprecates with_id, but we'll keep it around for now
+* Remove unnecessary validation for invalid belongs_to association
+
+
+## 0.3.4       2017-02-22
+
+* Expand with_lock workaround to cover lock! as well
+* Enable trigger workaround on Rails 5 as well, otherwise it fails silently
+* Tests: Switch to database cleaner truncation strategy to avoid multi-shard transactions
+
+
+## 0.3.3       2017-02-21
+
+* Avoid warning about multi-column primary keys with Rails 5 [#2](https://github.com/citusdata/activerecord-multi-tenant/issues/2)
+* Fix odd bind errors for has_one/has_many through
+* Add MultiTenant.current_tenant_id= helper method
+
+
+## 0.3.2       2017-02-16
+
+* Support blocks passed into the unscoped method (this fixes reload, amongst other issues)
+* Make with_lock work by adding workaround for Citus #1236 (SELECT ... FOR UPDATE is not router-plannable)
+
+
+## 0.3.1       2017-02-13
+
+* Rails 5 API controller support [@mstahl](https://github.com/mstahl) [#4](https://github.com/citusdata/activerecord-multi-tenant/pull/4)
+* Citus 6.1 compatibility
+
+
+## 0.3.0       2016-12-30
+
+* Remove dependency on acts_as_tenant - instead copy the code thats necessary
+* Fix issue with callbacks having TenantIdWrapper instead of the actual object
+
+
+## 0.2.1       2016-12-29
+
+* Fix bug in CopyFromClient helper
+
+
+## 0.2.0       2016-12-27
+
+* Initial release

data/LICENSE

@@ -0,0 +1,18 @@
+Copyright (c) 2018, Citus Data Inc.
+
+Permission is hereby granted, free of charge, to any person obtaining a copy of
+this software and associated documentation files (the "Software"), to deal in
+the Software without restriction, including without limitation the rights to use,
+copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the
+Software, and to permit persons to whom the Software is furnished to do so,
+subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
+FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
+COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
+IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
+CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,121 @@
+# activerecord-multi-tenant-next
+[![Tests](https://github.com/tachyurgy/activerecord-multi-tenant/actions/workflows/active-record-multi-tenant-tests.yml/badge.svg)](https://github.com/tachyurgy/activerecord-multi-tenant/actions/workflows/active-record-multi-tenant-tests.yml) [![Gem Version](https://img.shields.io/gem/v/activerecord-multi-tenant-next.svg)](https://rubygems.org/gems/activerecord-multi-tenant-next)
+
+A maintained fork of [citusdata/activerecord-multi-tenant](https://github.com/citusdata/activerecord-multi-tenant).
+The upstream gem has had no release since 2.4.0 (September 2023) and no active maintainer
+([#268](https://github.com/citusdata/activerecord-multi-tenant/issues/268)), so no released
+version works on Rails 7.2, 8.0 or 8.1. This fork collects the fixes that were sitting in
+upstream pull requests (authors kept on their commits), adds the missing ones, and runs CI
+against every supported Rails/Ruby/Citus combination.
+
+It is a drop-in replacement: same `MultiTenant` API, same `require` path. Everything is
+also offered back upstream; if upstream starts releasing again, switching back is a one-line
+Gemfile change.
+
+Introduction Post: https://www.citusdata.com/blog/2017/01/05/easily-scale-out-multi-tenant-apps/
+
+ActiveRecord/Rails integration for multi-tenant databases, in particular the open-source [Citus](https://github.com/citusdata/citus) extension for PostgreSQL.
+
+Enables easy scale-out by adding the tenant context to your queries, enabling the database (e.g. Citus) to efficiently route queries to the right database node.
+
+## Installation
+
+Add the following to your Gemfile:
+
+```ruby
+gem 'activerecord-multi-tenant-next'
+```
+
+If you are coming from `activerecord-multi-tenant`, remove that line from your Gemfile; the two
+gems ship the same files and should not be installed together.
+
+## Supported Rails versions
+
+All Ruby on Rails versions starting with 6.0 or newer (up to 8.1) are supported, on
+Ruby 3.0 or newer. CI runs every combination that Rails itself supports
+(Rails 8.x needs Ruby 3.2+, Rails 7.2 needs Ruby 3.1+, Rails 6.x does not run on
+Ruby 3.4) against Citus 10, 11 and 12.
+
+This gem only supports ActiveRecord (the Rails default ORM), and not alternative ORMs like Sequel.
+
+## Usage
+
+It is required that you add `multi_tenant` definitions to your model in order to have full support for Citus, in particular when updating records.
+
+In the example of an analytics application, sharding on `customer_id`, annotate your models like this:
+
+```ruby
+class PageView < ActiveRecord::Base
+  multi_tenant :customer
+  belongs_to :site
+
+  # ...
+end
+
+class Site < ActiveRecord::Base
+  multi_tenant :customer
+  has_many :page_views
+
+  # ...
+end
+```
+
+and then wrap all code that runs queries/modifications in blocks like this:
+
+```ruby
+customer = Customer.find(session[:current_customer_id])
+# ...
+MultiTenant.with(customer) do
+  site = Site.find(params[:site_id])
+  site.update! last_accessed_at: Time.now
+  site.page_views.count
+end
+```
+
+Inside controllers you can use a before_action together with set_current_tenant, to set the tenant for the current request:
+
+```ruby
+class ApplicationController < ActionController::Base
+  set_current_tenant_through_filter # Required to opt into this behavior
+  before_action :set_customer_as_tenant
+
+  def set_customer_as_tenant
+    customer = Customer.find(session[:current_customer_id])
+    set_current_tenant(customer)
+  end
+end
+```
+
+## Rolling out activerecord-multi-tenant for your application (write-only mode)
+
+The library relies on tenant_id to be present and NOT NULL for all rows. However,
+its often useful to have the library set the tenant_id for new records, and then backfilling
+tenant_id for existing records as a background task.
+
+To support this, there is a write-only mode, in which tenant_id is not included in queries,
+but only set for new records. Include the following in an initializer to enable it:
+
+```ruby
+MultiTenant.enable_write_only_mode
+```
+
+Once you are ready to enforce tenancy, make your tenant_id column NOT NULL and simply remove that line.
+
+## Frequently Asked Questions
+
+* **What if I have a table that doesn't relate to my tenant?** (e.g. templates that are the same in every account)
+
+  We recommend not using activerecord-multi-tenant on these tables. In case only some records in a table are not associated to a tenant (i.e. your templates are in the same table as actual objects), we recommend setting the tenant_id to 0, and then using MultiTenant.with(0) to access these objects.
+
+* **What if my tenant model is not defined in my application?**
+
+  The tenant model does not have to be defined. Use the gem as if the model was present. `MultiTenant.with` accepts either a tenant id or model instance.
+
+## Credits
+
+This gem was initially based on [acts_as_tenant](https://github.com/ErwinM/acts_as_tenant), and still shares some code. We thank the authors for their efforts.
+
+## License
+
+Copyright (c) 2018, Citus Data Inc.<br>
+Licensed under the MIT license, see LICENSE file for details.

data/lib/activerecord-multi-tenant/arel_visitors_depth_first.rb

@@ -0,0 +1,209 @@
+# frozen_string_literal: true
+
+module MultiTenant
+  class ArelVisitorsDepthFirst < Arel::Visitors::Visitor
+    def initialize(block = nil)
+      @block = block || proc
+      super()
+    end
+
+    private
+
+    def visit(obj, _ = nil)
+      super
+      @block.call obj
+    end
+
+    def unary(obj)
+      visit obj.expr
+    end
+    alias visit_Arel_Nodes_Else              unary
+    alias visit_Arel_Nodes_Group             unary
+    alias visit_Arel_Nodes_Cube              unary
+    alias visit_Arel_Nodes_RollUp            unary
+    alias visit_Arel_Nodes_GroupingSet       unary
+    alias visit_Arel_Nodes_GroupingElement   unary
+    alias visit_Arel_Nodes_Grouping          unary
+    alias visit_Arel_Nodes_Having            unary
+    alias visit_Arel_Nodes_Lateral           unary
+    alias visit_Arel_Nodes_Limit             unary
+    alias visit_Arel_Nodes_Not               unary
+    alias visit_Arel_Nodes_Offset            unary
+    alias visit_Arel_Nodes_On                unary
+    alias visit_Arel_Nodes_Ordering          unary
+    alias visit_Arel_Nodes_Ascending         unary
+    alias visit_Arel_Nodes_Descending        unary
+    alias visit_Arel_Nodes_UnqualifiedColumn unary
+    alias visit_Arel_Nodes_OptimizerHints    unary
+    alias visit_Arel_Nodes_ValuesList        unary
+
+    def function(obj)
+      visit obj.expressions
+      visit obj.alias if obj.respond_to?(:alias)
+      visit obj.distinct
+    end
+    alias visit_Arel_Nodes_Avg    function
+    alias visit_Arel_Nodes_Exists function
+    alias visit_Arel_Nodes_Max    function
+    alias visit_Arel_Nodes_Min    function
+    alias visit_Arel_Nodes_Sum    function
+
+    # rubocop:disable Naming/MethodName
+
+    def visit_Arel_Nodes_NamedFunction(obj)
+      visit obj.name
+      visit obj.expressions
+      visit obj.distinct
+      visit obj.alias if obj.respond_to?(:alias)
+    end
+
+    def visit_Arel_Nodes_Count(obj)
+      visit obj.expressions
+      visit obj.alias if obj.respond_to?(:alias)
+      visit obj.distinct
+    end
+
+    def visit_Arel_Nodes_Case(obj)
+      visit obj.case
+      visit obj.conditions
+      visit obj.default
+    end
+
+    def nary(obj)
+      obj.children.each { |child| visit child }
+    end
+    alias visit_Arel_Nodes_And nary
+
+    def binary(obj)
+      visit obj.left
+      visit obj.right
+    end
+    alias visit_Arel_Nodes_As                 binary
+    alias visit_Arel_Nodes_Assignment         binary
+    alias visit_Arel_Nodes_Between            binary
+    alias visit_Arel_Nodes_Concat             binary
+    alias visit_Arel_Nodes_DeleteStatement    binary
+    alias visit_Arel_Nodes_DoesNotMatch       binary
+    alias visit_Arel_Nodes_Equality           binary
+    alias visit_Arel_Nodes_FullOuterJoin      binary
+    alias visit_Arel_Nodes_GreaterThan        binary
+    alias visit_Arel_Nodes_GreaterThanOrEqual binary
+    alias visit_Arel_Nodes_In                 binary
+    alias visit_Arel_Nodes_InfixOperation     binary
+    alias visit_Arel_Nodes_JoinSource         binary
+    alias visit_Arel_Nodes_InnerJoin          binary
+    alias visit_Arel_Nodes_LessThan           binary
+    alias visit_Arel_Nodes_LessThanOrEqual    binary
+    alias visit_Arel_Nodes_Matches            binary
+    alias visit_Arel_Nodes_NotEqual           binary
+    alias visit_Arel_Nodes_NotIn              binary
+    alias visit_Arel_Nodes_NotRegexp          binary
+    alias visit_Arel_Nodes_IsNotDistinctFrom  binary
+    alias visit_Arel_Nodes_IsDistinctFrom     binary
+    alias visit_Arel_Nodes_Or                 binary
+    alias visit_Arel_Nodes_OuterJoin          binary
+    alias visit_Arel_Nodes_Regexp             binary
+    alias visit_Arel_Nodes_RightOuterJoin     binary
+    alias visit_Arel_Nodes_TableAlias         binary
+    alias visit_Arel_Nodes_When               binary
+
+    def visit_Arel_Nodes_StringJoin(obj)
+      visit obj.left
+    end
+
+    def visit_Arel_Attribute(obj)
+      visit obj.relation
+      visit obj.name
+    end
+    alias visit_Arel_Attributes_Integer visit_Arel_Attribute
+    alias visit_Arel_Attributes_Float visit_Arel_Attribute
+    alias visit_Arel_Attributes_String visit_Arel_Attribute
+    alias visit_Arel_Attributes_Time visit_Arel_Attribute
+    alias visit_Arel_Attributes_Boolean visit_Arel_Attribute
+    alias visit_Arel_Attributes_Attribute visit_Arel_Attribute
+    alias visit_Arel_Attributes_Decimal visit_Arel_Attribute
+
+    def visit_Arel_Table(obj)
+      visit obj.name
+    end
+
+    def terminal(obj); end
+    alias visit_ActiveSupport_Multibyte_Chars terminal
+    alias visit_ActiveSupport_StringInquirer  terminal
+    alias visit_Arel_Nodes_Lock               terminal
+    alias visit_Arel_Nodes_Node               terminal
+    alias visit_Arel_Nodes_SqlLiteral         terminal
+    alias visit_Arel_Nodes_BindParam          terminal
+    alias visit_Arel_Nodes_Window             terminal
+    alias visit_Arel_Nodes_True               terminal
+    alias visit_Arel_Nodes_False              terminal
+    alias visit_BigDecimal                    terminal
+    alias visit_Class                         terminal
+    alias visit_Date                          terminal
+    alias visit_DateTime                      terminal
+    alias visit_FalseClass                    terminal
+    alias visit_Float                         terminal
+    alias visit_Integer                       terminal
+    alias visit_NilClass                      terminal
+    alias visit_String                        terminal
+    alias visit_Symbol                        terminal
+    alias visit_Time                          terminal
+    alias visit_TrueClass                     terminal
+
+    def visit_Arel_Nodes_InsertStatement(obj)
+      visit obj.relation
+      visit obj.columns
+      visit obj.values
+    end
+
+    def visit_Arel_Nodes_SelectCore(obj)
+      visit obj.projections
+      visit obj.source
+      visit obj.wheres
+      visit obj.groups
+      visit obj.windows
+      visit obj.havings
+    end
+
+    def visit_Arel_Nodes_SelectStatement(obj)
+      visit obj.cores
+      visit obj.orders
+      visit obj.limit
+      visit obj.lock
+      visit obj.offset
+    end
+
+    def visit_Arel_Nodes_UpdateStatement(obj)
+      visit obj.relation
+      visit obj.values
+      visit obj.wheres
+      visit obj.orders
+      visit obj.limit
+    end
+
+    def visit_Arel_Nodes_Comment(obj)
+      visit obj.values
+    end
+
+    def visit_Array(obj)
+      obj.each { |i| visit i }
+    end
+    alias visit_Set visit_Array
+
+    def visit_Hash(obj)
+      obj.each do |k, v|
+        visit(k)
+        visit(v)
+      end
+    end
+
+    DISPATCH = dispatch_cache
+
+    # rubocop:disable Naming/AccessorMethodName
+    def get_dispatch_cache
+      DISPATCH
+    end
+    # rubocop:enable Naming/AccessorMethodName
+    # rubocop:enable Naming/MethodName
+  end
+end

data/lib/activerecord-multi-tenant/controller_extensions.rb

@@ -0,0 +1,36 @@
+# frozen_string_literal: true
+
+# Extension to the controller to allow setting the current tenant
+# set_current_tenant and current_tenant methods are introduced
+# to set and get the current tenant in the controllers that uses
+# the MultiTenant module.
+module MultiTenant
+  module ControllerExtensions
+    def set_current_tenant_through_filter
+      class_eval do
+        helper_method :current_tenant if respond_to?(:helper_method)
+
+        private
+
+        # rubocop:disable Naming/AccessorMethodName
+        def set_current_tenant(current_tenant_object)
+          MultiTenant.current_tenant = current_tenant_object
+        end
+        # rubocop:enable Naming/AccessorMethodName
+
+        def current_tenant
+          MultiTenant.current_tenant
+        end
+      end
+    end
+  end
+end
+
+# This block is executed when the file is loaded and
+# makes the base class; ActionController::Base to
+# extend the ControllerExtensions module.
+# This will add the set_current_tenant and current_tenant
+# in all the controllers that inherit from ActionController::Base
+ActiveSupport.on_load(:action_controller) do |base|
+  base.extend MultiTenant::ControllerExtensions
+end