Improve

hakuna · Mar 26, 2024 · 0893779 · 0893779
1 parent bb34254
commit 0893779
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/lib/tasks/rls.rake b/lib/tasks/rls.rake
@@ -45,10 +45,12 @@ namespace :rls do
       RLS.connection.execute <<~SQL
         DO $$
         BEGIN
-          CREATE ROLE "#{RLS.role}" WITH NOLOGIN;
-        EXCEPTION
-          WHEN DUPLICATE_OBJECT THEN
+          IF NOT EXISTS (SELECT 1 FROM pg_roles WHERE rolname = '#{RLS.role}') THEN
+            CREATE ROLE "#{RLS.role}" WITH NOLOGIN;
+            RAISE NOTICE 'Role "#{RLS.role}" created';
+          ELSE
             RAISE NOTICE 'Role "#{RLS.role}" already exists';
+          END IF;
         END
         $$;