使postgres进行阻塞等待

问题描述

我认为创建一个可以从plpgsql函数使用的阻塞wait（）可能很有趣（而且很有用）。我可以使它正常工作，但是我不确定它是否构想得很好。使用datagrip时会发生一个有趣的问题。如果我从函数调用pg_wait（）：

select util.wait_test_func() into var   --this calls my pg_wait()

该功能将正确阻止，直到我在其他查询控制台窗口中发出NOTIFY为止。实际上，如果函数wait_test_function（）调用了我的pg_wait（）3次，它将正确地阻塞3次，而我将不得不调用NOTIFY 3次以使函数完成。但是，一旦函数完成，如果我再次调用它，它将立即返回而不会阻塞。通知似乎仍在排队中，但我并不十分确定这是问题所在。如果我关闭了datagrip查询控制台，打开了一个新控制台，然后重新发出该函数调用，它将再次按预期工作，并正确地阻塞了。我可以一贯地重复这一点。新查询控制台窗口中的第一个调用始终会正确阻止，但随后的每个调用始终会立即返回。这是我在postgres中的第一个C函数，所以我想知道我是否在做根本上错误的事情。感谢您的任何帮助。我的代码如下。

C函数：

#ifdef WIN32
#include <windows.h>
#endif
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <errno.h>
#include <sys/time.h>
#include <sys/types.h>

#ifdef HAVE_SYS_SELECT_H
#include <sys/select.h>
#endif

#include "libpq-fe.h"
#include "postgres.h"
#include <limits.h>
#include <unistd.h>
#include <string.h>
#include "fmgr.h"
#include "utils/palloc.h"
#include "utils/elog.h"
#include "storage/bufpage.h"
#include "utils/builtins.h"

#ifdef PG_MODULE_MAGIC
PG_MODULE_MAGIC;
#endif

PG_FUNCTION_INFO_V1( pg_wait );

Datum pg_wait( PG_FUNCTION_ARGS );

Datum pg_wait( PG_FUNCTION_ARGS )
{
    PGconn     *conn;
    PGresult   *res;
    PGnotify   *notify;
    int         nnotifies;
    int         sock;
    fd_set      input_mask;
    
    char* conninfo = text_to_cstring(PG_GETARG_TEXT_PP(0));
    char* channel_name = text_to_cstring(PG_GETARG_TEXT_PP(1));
    
    char strlisten[50];
    strcpy(strlisten,"LISTEN ");
    strcat(strlisten,channel_name);

    conn = PQconnectdb(conninfo);
    res = PQexec(conn,strlisten);
    PQclear(res);

    sock = PQsocket(conn);
    FD_ZERO(&input_mask);
    FD_SET(sock,&input_mask);
    select(sock + 1,&input_mask,NULL,NULL);
    PQconsumeInput(conn);
    if ((notify = PQnotifies(conn)) != NULL)
    {
       PQfreemem(notify);
       PQconsumeInput(conn);
    }
    PQfinish(conn);

    PG_RETURN_TEXT_P( PG_GETARG_TEXT_PP(1) );
}

util.wait_test_func（）

create or replace function util.wait_test_func() returns integer
    parallel safe
    language plpgsql
as $$
DECLARE
    signal_name TEXT;
BEGIN
    RAISE NOTICE 'Before wait1';
    SELECT pg_wait('dbname=edw port=5432','CHANNEL1') INTO signal_name;
    RAISE NOTICE 'After wait1: %',signal_name;

    RAISE NOTICE 'Before wait2';
    SELECT pg_wait('dbname=edw port=5432','CHANNEL1') INTO signal_name;
    RAISE NOTICE 'After wait2: %',signal_name;

    RAISE NOTICE 'Before wait3';
    SELECT pg_wait('dbname=edw port=5432','CHANNEL1') INTO signal_name;
    RAISE NOTICE 'After wait3: %',signal_name;

    RETURN 0;
END
$$;

再次感谢您提供的任何帮助。

解决方法

您是否尝试过声明函数VOLATILE