Spring Boot整合PostgreSQL：如何向触发器传递当前登录用户

要把Spring Boot中获取的当前登录用户传递给PostgreSQL触发器，最常用且优雅的方式是利用PostgreSQL的会话变量（Session Variable），以下是具体实现步骤：

方法一：使用PostgreSQL会话变量传递用户 #

1. 在Spring事务中设置会话变量 #

在执行数据更新操作前，先通过JPA或JDBC设置当前会话的自定义变量，把登录用户传递给PostgreSQL：

// 获取当前登录用户
Authentication auth = SecurityContextHolder.getContext().getAuthentication();
String currentUsername = auth.getName();

// 注入EntityManager，在事务中设置会话变量
@Autowired
private EntityManager entityManager;

// 执行更新逻辑前设置变量
entityManager.createNativeQuery("SET LOCAL app.current_user = :username")
             .setParameter("username", currentUsername)
             .executeUpdate();

// 执行你的数据更新操作
Data targetData = entityManager.find(Data.class, 1);
targetData.setValue(100);
entityManager.merge(targetData);

2. 修改触发器函数读取会话变量 #

修改你的触发器函数，不再使用current_user，而是读取我们自定义的会话变量：

CREATE OR REPLACE FUNCTION data_history() RETURNS TRIGGER AS
$BODY$
DECLARE
    changes text;
    login_user text;
BEGIN
    -- 读取自定义会话变量，第二个参数true表示变量不存在时返回null
    SELECT current_setting('app.current_user', true) INTO login_user;
    
    -- 处理变量为空的情况（比如未登录或外部工具直接操作数据库）
    IF login_user IS NULL THEN
        login_user = 'system';
    END IF;
    
    changes := '';
    IF old.value <> new.value THEN
        changes := CONCAT(old.value, ':', new.value);
        INSERT INTO changelog(changes, "user", changed_on)
        VALUES(changes, login_user, NOW());
    END IF;
    RETURN NEW;
END;
$BODY$
language plpgsql;

补充说明 #

• SET LOCAL 只会在当前事务生效，事务结束后变量自动失效，不会影响其他会话，避免用户信息串用。
• 如果需要全局生效（不推荐），可以用SET代替SET LOCAL，但要注意会话隔离问题。

备选方法：通过实体临时字段传递 #

如果不想用会话变量，可以给业务表添加一个临时字段（比如temp_modified_by），Spring更新时设置该字段的值，触发器读取后不保存这个字段：

1. 修改业务表：

ALTER TABLE data ADD COLUMN temp_modified_by text;

2. 修改实体类（添加字段但不持久化）：

@Entity
@Table(name = "data")
public class Data {
    // 其他字段...
    @Column(name = "temp_modified_by", updatable = false, insertable = false)
    private String tempModifiedBy;
    
    // getter/setter
}

3. Spring更新时设置字段：

targetData.setTempModifiedBy(currentUsername);
entityManager.merge(targetData);

4. 触发器中读取该字段：

INSERT INTO changelog(changes, "user", changed_on)
VALUES(changes, NEW.temp_modified_by, NOW());

这种方法需要修改实体和表结构，不如会话变量灵活，适合特殊场景。

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