PostgreSQL行级安全配置报错：current_user转integer语法无效

当切换至自行创建的rentaluser角色执行SELECT查询时，触发错误：

Invalid syntax for type integer: "rentaluser"

问题根源 #

错误出在行级安全策略的USING条件中：customer_id = current_user::integer。current_user返回的是字符串类型的用户名（比如'rentaluser'），而customer_id是整数类型，强行将字符串转成整数必然报错。更关键的是，用户名和customer_id的数值本身没有对应关系，这个逻辑从一开始就不成立。

解决方案 #

需要建立用户名与customer_id的关联映射，再修改行级安全策略，以下提供两种可行方法：

方法一：使用映射表存储关联关系

1. 创建用户名与客户ID的映射表：

CREATE TABLE user_customer_map (
    username text PRIMARY KEY,
    customer_id integer NOT NULL REFERENCES customer(customer_id)
);

2. 插入rentaluser对应的客户ID（示例中假设对应ID为1，根据实际情况修改）：

INSERT INTO user_customer_map VALUES ('rentaluser', 1);

3. 修改行级安全策略：

CREATE OR REPLACE POLICY rental_policy ON rental TO rentaluser
  USING (customer_id = (SELECT customer_id FROM user_customer_map WHERE username = current_user));

CREATE OR REPLACE POLICY payment_policy ON payment TO rentaluser
  USING (customer_id = (SELECT customer_id FROM user_customer_map WHERE username = current_user));

方法二：使用自定义配置参数绑定客户ID

1. 给rentaluser设置对应的客户ID参数：

ALTER ROLE rentaluser SET customer.id = '1'; -- 替换为实际对应的customer_id

2. 修改行级安全策略：

CREATE OR REPLACE POLICY rental_policy ON rental TO rentaluser
  USING (customer_id = current_setting('customer.id')::integer);

CREATE OR REPLACE POLICY payment_policy ON payment TO rentaluser
  USING (customer_id = current_setting('customer.id')::integer);

验证修改 #

切换角色后执行查询，确认错误消失：

SET ROLE rentaluser;
SELECT * FROM rental;
SELECT * FROM payment;

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