仅在存在时重命名列

Question

PostgreSQL 不允许

ALTER TABLE t RENAME COLUMN IF EXISTS c1 TO c2

...或类似的东西。但是，能够编写修改数据库结构的脚本非常方便，可以再次运行而无需首先检查它是否已经运行.

我该如何编写一个 PostgreSQL 函数来做到这一点？

Answer 1

最好有两个函数，一个调用另一个：

CREATE OR REPLACE FUNCTION column_exists(ptable TEXT, pcolumn TEXT)
  RETURNS BOOLEAN AS $BODY$
DECLARE result bool;
BEGIN
    -- Does the requested column exist?
    SELECT COUNT(*) INTO result
    FROM information_schema.columns
    WHERE
      table_name = ptable and
      column_name = pcolumn;
    RETURN result;
END$BODY$
  LANGUAGE plpgsql VOLATILE;

CREATE OR REPLACE FUNCTION rename_column_if_exists(ptable TEXT, pcolumn TEXT, new_name TEXT)
  RETURNS VOID AS $BODY$
BEGIN
    -- Rename the column if it exists.
    IF column_exists(ptable, pcolumn) THEN
        EXECUTE FORMAT('ALTER TABLE %I RENAME COLUMN %I TO %I;',
            ptable, pcolumn, new_name);
    END IF;
END$BODY$
  LANGUAGE plpgsql VOLATILE;

Answer 2

@NessBird 2 函数方法很好，但是 Column_Exists 函数可以简化为 select 存在，避免计数，并且作为 SQL 函数而不是 plpgsql 函数。

create or replace function 
column_exists(ptable text, pcolumn text, pschema text default 'public')
  returns boolean 
  language sql stable strict  
as $body$
    -- does the requested table.column exist in schema?
    select exists 
         ( select null 
             from information_schema.columns 
             where table_name=ptable 
               and column_name=pcolumn 
               and table_schema=pschema
         ); 
$body$;

我添加了架构参数来处理具有相同 table 名称的多个架构。 rename_column_if_exists 除了可能添加架构外保持不变。

Answer 3

请阅读此article以获得详细说明。

DO $$
BEGIN
  IF EXISTS(SELECT *
    FROM information_schema.columns
    WHERE table_name='your_table' and column_name='your_column')
  THEN
      ALTER TABLE "public"."your_table" RENAME COLUMN "your_column" TO "your_new_column";
  END IF;
END $$;

Answer 4

您可以简单地处理 anonymous code block:

中可能引发的错误

DO
$$
    BEGIN
        ALTER TABLE t
            RENAME COLUMN c1 TO c2;
    EXCEPTION
        WHEN undefined_column THEN RAISE NOTICE 'column t.c1 does not exist';
    END;
$$;

您可以省略 THEN 之后的文字，什么也不做：

DO
$$
    BEGIN
        ALTER TABLE t
            RENAME COLUMN c1 TO c2;
    EXCEPTION
        WHEN undefined_column THEN
     END;
$$;

您可能只会在错误发生时得到一个数字。您可以从 here 中找到条件名称（WHEN 之后的错误名称）。确保您的数据库版本正确。

Answer 5

谢谢大家的宝贵建议！我已经为运行迁移编译了一个生产就绪的解决方案。我使用相同的函数来检查该列是否存在，但是我用过程替换了 rename_table_if_exists 函数：

CREATE OR REPLACE FUNCTION "product".column_exists (ptable text, pcolumn text, pschema text DEFAULT 'public')
    RETURNS boolean
    LANGUAGE sql
    STABLE STRICT
    AS $BODY$
    -- does the requested table.column exist in schema?
    SELECT
        EXISTS (
            SELECT
                NULL
            FROM
                information_schema.columns
            WHERE
                table_name = ptable
                AND column_name = pcolumn
                AND table_schema = pschema);
$BODY$;


CREATE OR REPLACE PROCEDURE "product".rename_column_if_exists (ptable TEXT, pcolumn TEXT, new_name TEXT, pschema text DEFAULT 'public')
LANGUAGE plpgsql
AS $$
BEGIN
    -- Rename the column if it exists.
    IF product.column_exists (ptable,
        pcolumn,
        pschema) THEN
    EXECUTE FORMAT('ALTER TABLE %I.%I RENAME COLUMN %I TO %I;', pschema, ptable, pcolumn, new_name);
END IF;
END
$$;

可以这样使用：

CALL <schema?>.rename_column_if_exists ('users', 'name', 'first_name', <schema name - optional>);

仅在存在时重命名列

Rename column only if exists

postgresql

ddl

alter-table