From e29fbb96b574bd15f0eba6075bba0041a8b26bac Mon Sep 17 00:00:00 2001
From: Nikita Kalyanov <nikitakalyanov@gmail.com>
Date: Fri, 17 Nov 2023 08:56:03 +0100
Subject: [PATCH] idempotency

---
 compute_tools/src/compute.rs | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/compute_tools/src/compute.rs b/compute_tools/src/compute.rs
index 3c3a935a68..3c5cd806b8 100644
--- a/compute_tools/src/compute.rs
+++ b/compute_tools/src/compute.rs
@@ -1056,9 +1056,12 @@ LIMIT 100",
             .batch_execute(&format!(
                 "BEGIN;
         ALTER TABLE {0} ENABLE ROW LEVEL SECURITY;
+        DROP POLICY IF EXISTS neon_row_level ON {0};
+        DROP ROLE IF EXISTS {1};
         CREATE USER {1} WITH PASSWORD '{2}' IN GROUP {3};
         CREATE POLICY neon_row_level ON {0} TO {3}
-            USING ({4} = current_user);
+            USING ({4} = current_user)
+            WITH CHECK ({4} = current_user);
         COMMIT;",
                 &params.table_name,
                 &params.user_name,