brandon/oam
1
0
Fork 0
mirror of https://gitea.com/mcereda/oam.git synced 2026-02-12 23:14:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore(postgresql): dump acquired knowledge

Browse Source
This commit is contained in:
Michele Cereda
2024-11-01 20:56:26 +01:00
parent 2e5e2920fc
commit df47fc3d7d
4 changed files with 102 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

47
snippets/postgres/functions.sql
View File

@@ -23,25 +23,34 @@ LANGUAGE plpgsql
AS $$
DECLARE affected_rows INTEGER;
BEGIN
EXECUTE format('
WITH
ctids AS (
SELECT
ROW_NUMBER() OVER (),
ctid
FROM %I
),
shuffled_values AS (
SELECT
ROW_NUMBER() OVER (ORDER BY RANDOM()),
%I AS new_value
FROM %I
)
UPDATE %I
SET %I = shuffled_values.new_value
FROM shuffled_values JOIN ctids ON shuffled_values.row_number = ctids.row_number
WHERE %I.ctid = ctids.ctid
', table_name, column_name, table_name, table_name, column_name, table_name);
SET CONSTRAINTS ALL DEFERRED;
EXECUTE format(
'
WITH
ctids AS (
SELECT
ROW_NUMBER() OVER (),
ctid
FROM %I
),
shuffled_values AS (
SELECT
ROW_NUMBER() OVER (ORDER BY RANDOM()),
%I AS new_value
FROM %I
)
UPDATE %I
SET %I = shuffled_values.new_value
FROM shuffled_values JOIN ctids ON shuffled_values.row_number = ctids.row_number
WHERE %I.ctid = ctids.ctid
',
table_name,
column_name,
table_name,
table_name,
column_name,
table_name
);
GET DIAGNOSTICS affected_rows = ROW_COUNT;
RETURN affected_rows;
END;