如何在PostgreSQL中通过IF语句条件触发SQL脚本？

啊，这个坑我之前踩过！核心原因是你搞混了psql客户端元命令和服务器端PL/pgSQL代码的边界——\ir是psql专属的客户端命令，根本没法在服务器端执行的DO块里运行，这就是报错的根源。

为什么会报错？ #

你写的DO $$ ... $$块是运行在PostgreSQL服务器端的PL/pgSQL代码，而\ir是psql客户端的元命令（用来在客户端加载并执行外部SQL文件）。服务器端的PL/pgSQL解释器完全不认识\开头的这些客户端命令，自然会报语法错误。

直接执行\ir ./script.sql或者把脚本内容写进DO块能行，是因为前者是客户端直接处理文件加载，后者是把SQL代码直接交给服务器执行，都没有跨环境的问题。

正确的解决方案 #

根据你的需求，推荐两种最实用的方法：

方法1：用psql客户端的条件判断（PostgreSQL 9.6+ 支持）

这是最直接的方案，利用psql自带的\if元命令在客户端层面做条件判断：

-- 第一步：查询目标schema是否存在，把结果存入psql会话变量
SELECT EXISTS(SELECT 1 FROM information_schema.schemata WHERE schema_name = 'public') AS schema_exists;
\gset

-- 第二步：根据变量判断，满足条件就执行外部脚本
\if :schema_exists
  \ir ./script.sql
\endif

• \gset会把查询结果的列名作为变量名，将查询结果赋值给psql的会话变量；
• \if :schema_exists直接读取这个变量做条件判断，符合条件就触发\ir加载外部脚本。

方法2：用Shell脚本配合psql（兼容所有PostgreSQL版本）

如果你的PostgreSQL版本低于9.6（不支持\if元命令），可以用Shell脚本先做判断，再决定是否执行脚本：

#!/bin/bash
# 连接数据库查询schema是否存在，返回结果（注意结果前的空格）
SCHEMA_EXISTS=$(psql -d my_database -t -c "SELECT EXISTS(SELECT 1 FROM information_schema.schemata WHERE schema_name = 'public')")

# 判断结果，存在则执行脚本
if [ "$SCHEMA_EXISTS" = " t" ]; then
  psql -d my_database -f ./script.sql
fi

不推荐的方案：服务器端动态执行 #

如果一定要在服务器端处理（比如不能依赖客户端脚本），只能把script.sql里的SQL代码直接写到PL/pgSQL的EXECUTE语句里，但这种方式维护成本极高——每次修改外部脚本都要同步更新DO块，完全失去了外部脚本的灵活性，所以不推荐。

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