Тайм-аут на консультативные блокировки в postgresql

Я мигрирую из ORACLE. В настоящее время я пытаюсь перенести этот вызов:

lkstat := DBMS_LOCK.REQUEST(lkhndl, DBMS_LOCK.X_MODE, lktimeout, true);

Эта функция пытается получить блокировку lkhndl и возвращает 1, если не удается получить ее через timeout секунд.

В postgresql я использую

pg_advisory_xact_lock(lkhndl);

Однако кажется, что он ждет блокировки навсегда. pg_try_advisory_xact_lock немедленно возвращается в случае сбоя. Есть ли способ реализовать версию тайм-аута блокировки?

Существует настройка lock_timeout, но я не уверен, применима ли она к консультативным блокировкам и как pg_advisory_xact_lock будет вести себя после истечения времени ожидания.

Ответ 1

Это прототип оболочки, которая плохо имитирует DBMS_LOCK.REQUEST - ограничен только одним типом блокировки (консультативная блокировка области транзакции).

Чтобы сделать функцию полностью совместимой с Oracle, потребуется несколько сотен строк. Но это начало.

CREATE OR REPLACE FUNCTION
advisory_xact_lock_request(p_key bigint, p_timeout numeric)
RETURNS integer
LANGUAGE plpgsql AS $$
/*  Imitate DBMS_LOCK.REQUEST for PostgreSQL advisory lock. 
Return 0 on Success, 1 on Timeout, 3 on Parameter Error. */
DECLARE
    t0 timestamptz := clock_timestamp();
BEGIN
    IF p_timeout NOT BETWEEN 0 AND 86400 THEN
        RAISE WARNING 'Invalid timeout parameter';
        RETURN 3;
    END IF;
    LOOP
        IF pg_try_advisory_xact_lock(key) THEN
            RETURN 0;
        ELSIF clock_timestamp() > t0 + (p_timeout||' seconds')::interval THEN
            RAISE WARNING 'Could not acquire lock in % seconds', p_timeout;
            RETURN 1;
        ELSE
            PERFORM pg_sleep(0.01); /* 10 ms */
        END IF;
    END LOOP;
END;
$$;

Проверьте это с помощью этого кода:

SELECT CASE 
    WHEN advisory_xact_lock_request(1, 2.5) = 0
    THEN pg_sleep(120)
END; -- and repeat this in parallel session 

/* Usage in Pl/PgSQL */

lkstat := advisory_xact_lock_request(lkhndl, lktimeout);