PostgreSQL客户端工具执行RAISE NOTICE及IF语句问题咨询

你遇到的问题不是工具不支持，而是PostgreSQL对过程化语句的执行规则和SSMS里的T-SQL不一样：

• 普通SQL语句（比如你执行的select * FROM pg_tables WHERE schemaname = 'schema_name' AND tablename = 'table_name'）可以直接在查询窗口运行，但RAISE NOTICE、IF这类控制流语句属于PL/pgSQL语法，不能单独执行，必须包裹在代码块或存储对象中。

解决方法： #

1. 用DO语句执行匿名代码块 #

这是临时测试PL/pgSQL逻辑最方便的方式，把代码放在DO $$ ... END $$结构里：

DO $$
BEGIN
  -- 单独执行RAISE NOTICE
  RAISE NOTICE 'Hi';
  
  -- 带IF语句的示例
  IF current_user = 'postgres' THEN
    RAISE NOTICE '当前用户是postgres';
  ELSE
    RAISE NOTICE '当前用户是%', current_user;
  END IF;
END $$;

这里的$$是自定义分隔符，用来避免和字符串中的单引号冲突，BEGIN...END包裹所有过程化逻辑。

2. 创建函数/存储过程复用逻辑 #

如果需要反复执行这段逻辑，可以把代码封装成函数：

CREATE OR REPLACE FUNCTION test_plpgsql_logic()
RETURNS void AS $$
BEGIN
  RAISE NOTICE 'Hi from function';
  
  IF extract(month from current_date) = 10 THEN
    RAISE NOTICE '现在是10月份';
  END IF;
END $$ LANGUAGE plpgsql;

-- 调用函数
SELECT test_plpgsql_logic();

和SSMS的差异说明 #

SSMS里的T-SQL允许直接执行PRINT或者单独的IF语句，但PostgreSQL的PL/pgSQL是独立的过程化扩展语言，必须在代码块（DO语句）、函数、存储过程这类容器中才能执行，普通查询窗口只识别标准SQL语句。

内容的提问来源于stack exchange，提问作者Eric