如何为PostgreSQL编写自己的全局锁定/解锁函数

Question

我有pergre的postgresql函数getTravelTime（integer，timestamp），它试图为指定的ID和时间戳选择数据。 如果没有数据或数据很旧，它将从外部服务器下载数据（下载时间约为300毫秒）。

多进程使用此数据库和此功能。 当两个进程找不到数据并下载并尝试在travel_time表中插入数据时，会出现错误（ID和时间戳对必须唯一）。 我想到了锁。 锁定整个表将阻止所有进程，并且仅允许一个进程继续进行。 我只需要锁定id和时间戳。 pg_advisory_lock似乎仅在“当前会话”中锁定。 但是我的进程使用自己的会话。

我试图编写自己的锁定/解锁功能。 我做对了吗？ 我使用主动等待，该如何忽略呢？ 也许有一种方法可以使用pg_advisory_lock（）作为全局锁？

我的代码：

CREATE TABLE travel_time_locks (
    id_key integer NOT NULL,
    time_key timestamp without time zone NOT NULL,
    UNIQUE (id_key, time_key) 
);

------------
-- Function: mylock(integer, timestamp)
DROP FUNCTION IF EXISTS mylock(integer, timestamp) CASCADE;
-- Usage: SELECT mylock(1, '2010-03-28T19:45');
-- function tries to do a global lock similar to pg_advisory_lock(key, key)
CREATE OR REPLACE FUNCTION mylock(id_input integer, time_input timestamp)
  RETURNS void AS
$BODY$
DECLARE
    rows int;
BEGIN
    LOOP

        BEGIN
            -- active waiting here !!!! :(
            INSERT INTO travel_time_locks (id_key, time_key) VALUES (id_input, time_input);
        EXCEPTION WHEN unique_violation THEN
            CONTINUE;
        END;
        EXIT;
    END LOOP;
END;
$BODY$ LANGUAGE 'plpgsql' VOLATILE
    COST 1;

------------
-- Function: myunlock(integer, timestamp)
DROP FUNCTION IF EXISTS myunlock(integer, timestamp) CASCADE;
-- Usage: SELECT myunlock(1, '2010-03-28T19:45');
-- function tries to do a global unlock similar to pg_advisory_unlock(key, key)
CREATE OR REPLACE FUNCTION myunlock(id_input integer, time_input timestamp)
  RETURNS integer AS
$BODY$
DECLARE
BEGIN
    DELETE FROM ONLY travel_time_locks WHERE id_key=id_input AND time_key=time_input;
    RETURN 1;
END;
$BODY$ LANGUAGE 'plpgsql' VOLATILE
    COST 1;

Answer 1

使用pg_advisory_lock() 。 正是您所需要的。

Answer 2

经过进一步测试，我发现“当前会话”是指PostgreSQL后端的整个实例，而不仅仅是单个连接。

我的解决方案是： PERFORM pg_advisory_lock(id, extract(epoch from my_time)::int ); （+解锁）在我的getTravelTime函数中。