pgbench 测试 postgresql 存储过程高并发锁表情况

CREATE TABLE test_table1 (    id SERIAL PRIMARY KEY,    bz INTEGER NOT NULL DEFAULT 0);
CREATE TABLE test_table2 (    id SERIAL PRIMARY KEY,    test_table1_id INTEGER REFERENCES test_table1(id),    other_data TEXT);
-- 插入示例数据INSERT INTO test_table1 (bz) VALUES (0), (0), (1);  -- id 为 1, 2, 3INSERT INTO test_table2 (test_table1_id, other_data) VALUES (1, 'data1'), (2, 'data2'), (3, 'data3');

CREATE OR REPLACE FUNCTION update_if_not_updated() RETURNS TEXT AS $$DECLARE    test_table1_record RECORD;    found_record BOOLEAN := FALSE;  -- 标志变量，记录是否找到匹配的记录BEGIN    -- 联表查询并只锁定 test_table1 表中的行    FOR test_table1_record IN        SELECT a.*        FROM test_table1 a, test_table2 b        WHERE a.id = b.test_table1_id AND a.bz = 0        FOR UPDATE OF a    LOOP        found_record := TRUE;  -- 找到匹配记录，设置标志为 TRUE                -- 检查是否需要更新        IF test_table1_record.bz = 0 THEN            -- 更新 test_table1 表的 bz 字段            UPDATE test_table1 SET bz = 1 WHERE id = test_table1_record.id;            RAISE NOTICE 'Thread %: 更新成功, id = %, bz = 1', pg_backend_pid(), test_table1_record.id;            RETURN '更新成功';        ELSE            RAISE NOTICE 'Thread %: 已被更新, id = %', pg_backend_pid(), test_table1_record.id;            RETURN '已被更新';        END IF;    END LOOP;
    -- 如果没有找到任何记录    IF NOT found_record THEN        RAISE NOTICE 'Thread %: 没有找到符合条件的记录', pg_backend_pid();        RETURN '没有找到符合条件的记录';    END IF;
    RETURN '没有可更新的记录';END;$$ LANGUAGE plpgsql;

SELECT update_if_not_updated();

pgbench -c 5 -j 2 -t 10 -f test_script.sql your_database_name

SELECT * FROM test_table1;