PL/pgSQL函数报错‘id列引用歧义’求排查及优化方案

嘿，这个歧义问题我之前踩过坑！咱们先搞清楚为什么会出错：你用PL/pgSQL写的函数定义了返回表table(id int)，这会在函数的局部作用域里自动生成一个名为id的输出变量。当你在WITH语句和最后的联合查询里写select id时，PL/pgSQL没法区分你指的是companies表的id列，还是这个返回用的id变量，所以就抛出了歧义错误。

下面给你几个可行的解决方案，按推荐程度排序：

方案1：改用SQL函数（最简洁） #

你的逻辑完全不需要PL/pgSQL的流程控制，纯SQL函数就能搞定，而且不会有变量作用域的歧义问题。直接把函数改成SQL语言：

create or replace function sp_get_insert_company(company_name varchar(100)) 
returns table(id int) as $$
with s as (select id from companies where name = company_name),
i as (insert into companies (name) select company_name where not exists (select 1 from s) returning id)
select id from i union all select id from s;
$$ language sql;

调用的时候用：

select * from sp_get_insert_company('TEST');

方案2：修改PL/pgSQL函数，明确限定列来源 #

如果你一定要用PL/pgSQL，那就在SQL语句里给所有id列加上表别名或表名限定，让PL/pgSQL能明确区分：

create or replace function sp_get_insert_company( in company_name varchar(100) ) 
returns table (id int) as $$ 
begin
    with s as (select c.id from companies c where c.name = company_name),
         i as (insert into companies (name) select company_name where not exists (select 1 from s) returning companies.id)
    select * from i union all select * from s;
end; 
$$ language plpgsql;

或者你也可以修改返回列的名称，避免和表列重名，比如把returns table(id int)改成returns table(company_id int)，然后查询时用select id as company_id，也能解决歧义。

方案3：优化并发安全（进阶） #

另外要提醒你：原逻辑在高并发场景下可能会出现重复插入的问题（比如两个会话同时检查同一个公司名不存在，然后都执行插入）。建议先给companies表的name字段加唯一约束：

alter table companies add constraint unique_company_name unique(name);

然后用INSERT ... ON CONFLICT语法实现更可靠的“存在则返回，不存在则插入”逻辑，比如：

create or replace function sp_get_insert_company(company_name varchar(100)) 
returns int as $$
with ins as (
    insert into companies (name) values (company_name)
    on conflict (name) do nothing
    returning id
)
select coalesce((select id from ins), (select id from companies where name = company_name));
$$ language sql;

这个版本能确保并发下不会重复插入，也能正确返回已存在的ID。

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