PostgreSQL触发器函数交替设置ws_source异常问题排查

咱们先拆解下你触发器函数里的几个核心问题，正是这些问题导致了交替分配的逻辑失效：

1. 多余的UPDATE操作+语法错误 #

你在触发器里直接执行update leads set ws_source = 'foo2', where ws_id = NEW.ws_id完全没必要——行级触发器中，直接修改NEW对象的字段就会生效，额外的UPDATE不仅冗余，还可能引发递归触发风险。另外这条语句还有语法错误：where前多了一个逗号，这会直接导致函数执行报错，只是你可能没留意到数据库日志。

2. 硬编码的查询条件限制 #

你的查询语句select ws_source into v_return from leads where ws_tsource = 'foobar' ...用了固定的'foobar'，但实际应该用NEW.ws_tsource，这样才能匹配当前插入记录对应的源，否则只有ws_tsource恰好是foobar的记录才会被处理，其他类似foobar2的场景根本不会触发逻辑。

3. 未处理初始无记录的场景 #

当某个ws_tsource是第一次插入记录时，v_return会是NULL，这时候你的两个if条件都不满足，导致ws_source无法被正确设置，还是保留原来的值。

4. 变量声明缺失 #

你用到了v_priority但没有在DECLARE块里声明，这会直接导致函数执行失败，必须补上变量声明语句。

修正后的触发器函数 #

下面是修复后的完整代码，我调整了逻辑、去掉冗余操作、修复语法问题，同时覆盖了初始无记录的场景：

CREATE OR REPLACE FUNCTION public.leads_ins() 
RETURNS trigger 
LANGUAGE plpgsql 
AS $function$ 
DECLARE 
    v_return text;
    v_priority integer; -- 补上缺失的变量声明
BEGIN 
    -- 先处理ws_source初始为foo的情况
    IF NEW.ws_source = 'foo' THEN 
        v_priority := 13821;
        
        -- 匹配当前插入记录的ws_tsource前缀（支持foobar、foobar2等）
        IF left(NEW.ws_tsource, 7) = 'foobar' THEN 
            -- 查询同ws_tsource的最后一条记录的ws_source
            SELECT ws_source 
            INTO v_return 
            FROM leads 
            WHERE ws_tsource = NEW.ws_tsource -- 用当前记录的ws_tsource，不要硬编码
            ORDER BY ws_id DESC 
            LIMIT 1;
            
            -- 根据最后一条的ws_source决定当前值，同时处理初始无记录的情况
            IF v_return = 'foo' THEN
                NEW.ws_source := 'foo2';
                v_priority := 16460;
            ELSE
                -- 包含v_return为foo2或NULL（第一条记录）的场景
                NEW.ws_source := 'foo';
                v_priority := 13821;
            END IF;
        END IF;
    END IF;
    
    -- 如果需要将v_priority赋值给表的priority字段，解开下面的注释
    -- NEW.priority := v_priority;
    
    RETURN NEW;
END; 
$function$;

额外注意事项 #

这种交替分配的需求完全可行，核心逻辑就是通过查询同组最后一条记录来决定当前值。如果你的业务是高并发插入场景，为了避免同时插入的两条记录查询到同一个最后一条导致分配重复，可以给查询语句加上行锁：

SELECT ws_source 
INTO v_return 
FROM leads 
WHERE ws_tsource = NEW.ws_tsource 
ORDER BY ws_id DESC 
LIMIT 1
FOR UPDATE; -- 加上行锁，防止并发冲突

内容的提问来源于stack exchange，提问作者João R.