Enable RLS (Row Level Security)

nitzanpap · Aug 13, 2024 · d977fb2 · d977fb2
1 parent 2828075
commit d977fb2
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/server/internal/configs/dbConfigs/db.config.go b/server/internal/configs/dbConfigs/db.config.go
@@ -43,6 +43,7 @@ func ConnectToDBPool(dbURL string) (*pgxpool.Pool, error) {
 
 func InitDB(db *pgx.Conn) {
 	createDbTables(db)
+	enableRLS(db)
 	createPreparedStatements(db)
 }
 
@@ -81,6 +82,20 @@ func createDbTables(db *pgx.Conn) {
 	}
 }
 
+func enableRLS(db *pgx.Conn) {
+	// Enable RLS on the Users table
+	_, err := db.Exec(context.Background(), "ALTER TABLE users ENABLE ROW LEVEL SECURITY;")
+	if err != nil {
+		log.Fatalf(colors.Error("Unable to enable RLS on users table: %v\n"), err)
+	}
+
+	// Enable RLS on the URLs table
+	_, err = db.Exec(context.Background(), "ALTER TABLE urls ENABLE ROW LEVEL SECURITY;")
+	if err != nil {
+		log.Fatalf(colors.Error("Unable to enable RLS on urls table: %v\n"), err)
+	}
+}
+
 var PreparedStatements = preparedStatementsStruct{
 	CreateUserRow:   "createUserRow",
 	GetUserRow:      "getUserRow",