Skip to content

Commit 32f474b

Browse files
committed
Address column sql fix
1 parent 018459e commit 32f474b

1 file changed

Lines changed: 13 additions & 8 deletions

File tree

lemonade-map/src/scripts/add_address_columns.sql

Lines changed: 13 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -3,19 +3,24 @@
33

44
-- Function to add a column if it doesn't exist
55
CREATE OR REPLACE FUNCTION add_column_if_not_exists(
6-
table_name text,
7-
column_name text,
8-
column_type text
6+
_table_name text,
7+
_column_name text,
8+
_column_type text
99
) RETURNS void AS $$
1010
BEGIN
1111
IF NOT EXISTS (
1212
SELECT 1
13-
FROM information_schema.columns
14-
WHERE table_name = add_column_if_not_exists.table_name
15-
AND column_name = add_column_if_not_exists.column_name
13+
FROM information_schema.columns AS c
14+
WHERE c.table_schema = 'public'
15+
AND c.table_name = _table_name
16+
AND c.column_name = _column_name
1617
) THEN
17-
EXECUTE format('ALTER TABLE %I ADD COLUMN %I %s',
18-
table_name, column_name, column_type);
18+
EXECUTE format(
19+
'ALTER TABLE public.%I ADD COLUMN IF NOT EXISTS %I %s',
20+
_table_name,
21+
_column_name,
22+
_column_type
23+
);
1924
END IF;
2025
END;
2126
$$ LANGUAGE plpgsql;

0 commit comments

Comments
 (0)