sql - 创建函数 : Syntax Error at or near $$

Question

我是 Postgres 的初学者，我想使用一个函数自动删除表中的列。但这给了我标题中提到的错误。

这是我的代码:

create function dropColumn(table_name text,col_name text) returns void as $$
ALTER TABLE $1 DROP COLUMN IF EXIST $2;
$$
language 'psql';

错误:

ERROR:  syntax error at or near "$$
language 'psql';
create function dropColumn(table_name text,col_name text) returns
void $$"
LINE 1: $$

有什么问题？我该如何解决这个问题？

Answer 1

您的函数几乎所有地方都是错误的。最重要的是，您不能在纯 SQL 中参数化标识符。您需要在 plpgsql 函数(或支持它的任何其他过程语言)中使用 EXECUTE 的动态 SQL。这将完成这项工作:

CREATE OR REPLACE FUNCTION drop_column(table_name text, col_name text)
  RETURNS void AS
$func$
BEGIN
   EXECUTE format('ALTER TABLE %I DROP COLUMN IF EXISTS %I'
                 , table_name, col_name);
END
$func$
LANGUAGE plpgsql;

调用:

SELECT drop_column('my_tbl', 'my_column');

开始 reading the manual here并研究一些 related questions and answers on SO.

特别注意正确防御SQL注入(inject):

• Table name as a PostgreSQL function parameter