rls_multi_tenant 0.2.7 → 0.2.8

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 50eff8f3ced091cac7b277aff79cfd435a0ce0701ee2ca3ca7163d57e6792406
-  data.tar.gz: 7739392cf8b866e5d6e86ccebeeccb3a3e913b7efbccf89a4441d79dfbc93f84
+  metadata.gz: 1ac751dd2c642fe57321ed7af2122805db1654353d933b12aba216cad22c0160
+  data.tar.gz: 947a7c120d823380bf29cb80881c64739bf662e284021805576fbf0baa733138
 SHA512:
-  metadata.gz: a8c6c0d9323251de3b6bc8321cb8bae781ae8248ea37dd9cf643aa7e4e0441234c42084d7391dcd73d4735e20080c27025969359e5dfe1c9f0bea0d30e5aa34d
-  data.tar.gz: 7788deda9e7deb6c49e7fd9359cfb06e648044c48d78ac9e1823b3cf91eb0deff90db6d95e94f2433adc726b0d6fd924b822958cab0c7882da0667d7af606def
+  metadata.gz: 27c78853768c5fd7e978184455ff9ba8b60401314d752e8b7140b775bf2b897820dcebca6eb7efc9467317b08de8adabbde8e49a36c0edcf3e09e775520871f9
+  data.tar.gz: ac3e1092d6a177323f08fcf7e2fb371bd90502af21afba051fb0c90917edc4497fabc950325035614acd8918a41f3076cd6d1e9e02132a599fb03d4dc8854806

data/lib/rls_multi_tenant/concerns/tenant_context.rb

@@ -8,19 +8,28 @@ module RlsMultiTenant
       SET_TENANT_ID_SQL = 'SET %s = %s'
       RESET_TENANT_ID_SQL = 'RESET %s'
 
+      # rubocop:disable Metrics/BlockLength
       class_methods do
         def tenant_session_var
           "rls.#{RlsMultiTenant.tenant_id_column}"
         end
 
+        def tenant_stack
+          Thread.current[:"#{name}_tenant_stack"] ||= []
+        end
+
         # Switch tenant context for a block
         def switch(tenant_or_id)
           tenant_id = extract_tenant_id(tenant_or_id)
           validate_tenant_exists!(tenant_id)
+
+          previous_tenant_id = current_tenant_id
+          tenant_stack.push(previous_tenant_id)
+
           connection.execute format(SET_TENANT_ID_SQL, tenant_session_var, connection.quote(tenant_id))
           yield
         ensure
-          reset!
+          restore_tenant_context!
         end
 
         # Switch tenant context permanently (until reset)
@@ -32,6 +41,7 @@ module RlsMultiTenant
 
         # Reset tenant context
         def reset!
+          tenant_stack.clear
           connection.execute format(RESET_TENANT_ID_SQL, tenant_session_var)
         end
 
@@ -51,6 +61,25 @@ module RlsMultiTenant
 
         private
 
+        def current_tenant_id
+          return nil unless connection.active?
+
+          result = connection.execute("SHOW #{tenant_session_var}")
+          result.first&.dig(tenant_session_var)
+        rescue ActiveRecord::StatementInvalid, PG::Error
+          nil
+        end
+
+        def restore_tenant_context!
+          previous_tenant_id = tenant_stack.pop
+
+          if previous_tenant_id.present?
+            connection.execute format(SET_TENANT_ID_SQL, tenant_session_var, connection.quote(previous_tenant_id))
+          else
+            connection.execute format(RESET_TENANT_ID_SQL, tenant_session_var)
+          end
+        end
+
         def extract_tenant_id(tenant_or_id)
           case tenant_or_id
           when ->(obj) { obj.is_a?(RlsMultiTenant.tenant_class) }
@@ -71,6 +100,7 @@ module RlsMultiTenant
           raise StandardError, "#{RlsMultiTenant.tenant_class_name} with id '#{tenant_id}' not found"
         end
       end
+      # rubocop:enable Metrics/BlockLength
 
       # Instance methods
       def switch(tenant_or_id, &block)

data/lib/rls_multi_tenant/middleware/subdomain_tenant_selector.rb

@@ -104,7 +104,7 @@ module RlsMultiTenant
 
       def excluded_subdomain?(subdomain)
         return false if subdomain.blank?
-        
+
         excluded = RlsMultiTenant.excluded_subdomains
         excluded.is_a?(Array) && excluded.include?(subdomain.to_s.downcase)
       end

data/lib/rls_multi_tenant/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module RlsMultiTenant
-  VERSION = '0.2.7'
+  VERSION = '0.2.8'
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: rls_multi_tenant
 version: !ruby/object:Gem::Version
-  version: 0.2.7
+  version: 0.2.8
 platform: ruby
 authors:
 - Coding Ways