pg_rls 0.1.0 → 0.1.1
Sign up to get free protection for your applications and to get access to all the features.
- checksums.yaml +4 -4
- data/.rubocop.yml +3 -0
- data/Gemfile.lock +1 -1
- data/lib/pg_rls/middleware/sidekiq/client.rb +3 -3
- data/lib/pg_rls/tenant.rb +1 -5
- data/lib/pg_rls/version.rb +1 -1
- data/lib/pg_rls.rb +11 -0
- metadata +1 -1
checksums.yaml
CHANGED
@@ -1,7 +1,7 @@
|
|
1
1
|
---
|
2
2
|
SHA256:
|
3
|
-
metadata.gz:
|
4
|
-
data.tar.gz:
|
3
|
+
metadata.gz: 0b51a33bf2c5a0518a96142e8b07e303b32419719dfce1715ee237a0348d19c9
|
4
|
+
data.tar.gz: 3bba1aaa681ab9a6d0defff9110c5f2077b08ff6a07332ad429cd040d8c84a7a
|
5
5
|
SHA512:
|
6
|
-
metadata.gz:
|
7
|
-
data.tar.gz:
|
6
|
+
metadata.gz: 3ff073704aeca1d9e1b3e8673353deeca756d06b95a32f62fe85ac65cd3f145007c68d6b0b980b78221bdd0f4330421fc540d8efdc7e11600d8adaada1aa105c
|
7
|
+
data.tar.gz: 11711ea29d21eef5d0dccbdfc5916074fe1a98089752f7fc0095ca1d514b40f5c40cb4b0a7a56688160a8afeb67414c3902d88152732774ad62fcbd6c7294140
|
data/.rubocop.yml
CHANGED
data/Gemfile.lock
CHANGED
@@ -12,11 +12,11 @@ module PgRls
|
|
12
12
|
end
|
13
13
|
|
14
14
|
def load_tenant_attribute!(msg)
|
15
|
-
if PgRls.
|
15
|
+
if PgRls.admin_connection?
|
16
|
+
msg['admin'] = true
|
17
|
+
else
|
16
18
|
tenant = PgRls::Tenant.fetch!
|
17
19
|
msg['pg_rls'] = tenant.id
|
18
|
-
else
|
19
|
-
msg['admin'] = true
|
20
20
|
end
|
21
21
|
end
|
22
22
|
end
|
data/lib/pg_rls/tenant.rb
CHANGED
@@ -81,7 +81,7 @@ module PgRls
|
|
81
81
|
end
|
82
82
|
|
83
83
|
def find_tenant(resource)
|
84
|
-
raise PgRls::Errors::AdminUsername if
|
84
|
+
raise PgRls::Errors::AdminUsername if PgRls.admin_connection?
|
85
85
|
|
86
86
|
reset_rls!
|
87
87
|
|
@@ -95,10 +95,6 @@ module PgRls
|
|
95
95
|
raise PgRls::Errors::TenantNotFound if tenant.blank?
|
96
96
|
end
|
97
97
|
|
98
|
-
def admin_username?
|
99
|
-
PgRls.username != PgRls.current_db_username
|
100
|
-
end
|
101
|
-
|
102
98
|
def find_tenant_by_method(resource, method)
|
103
99
|
look_up_value = resource.is_a?(PgRls.main_model) ? resource.send(method) : resource
|
104
100
|
PgRls.main_model.send("find_by_#{method}!", look_up_value)
|
data/lib/pg_rls/version.rb
CHANGED
data/lib/pg_rls.rb
CHANGED
@@ -87,6 +87,8 @@ module PgRls
|
|
87
87
|
end
|
88
88
|
|
89
89
|
def on_each_tenant(&)
|
90
|
+
set_rls_connection! if admin_connection?
|
91
|
+
|
90
92
|
result = []
|
91
93
|
main_model.find_each do |tenant|
|
92
94
|
allowed_search_fields = search_methods.map(&:to_s).intersection(main_model.column_names)
|
@@ -100,6 +102,11 @@ module PgRls
|
|
100
102
|
result
|
101
103
|
end
|
102
104
|
|
105
|
+
def set_rls_connection!
|
106
|
+
self.as_db_admin = false
|
107
|
+
establish_new_connection!
|
108
|
+
end
|
109
|
+
|
103
110
|
def execute_rls_in_shards
|
104
111
|
connection_pool_list = PgRls.connection_class.connection_handler.connection_pool_list
|
105
112
|
result = []
|
@@ -123,6 +130,10 @@ module PgRls
|
|
123
130
|
@as_db_admin || false
|
124
131
|
end
|
125
132
|
|
133
|
+
def admin_connection?
|
134
|
+
current_db_username != username
|
135
|
+
end
|
136
|
+
|
126
137
|
private
|
127
138
|
|
128
139
|
attr_writer :as_db_admin
|