3

vip_until在一个文字游戏的 PostgreSQL 表中，我通过列或grand_until具有未来有效日期的列来跟踪付费玩家：

create table users (
        uid serial primary key,
        vip_until timestamp null,  -- date in future indicates paying customer
        grand_until timestamp null -- date in future indicates paying customer
);

我写了一个简短的存储过程来检查：

create or replace function is_vip(
    IN in_uid integer,
    OUT out_vip boolean
) as $BODY$
        BEGIN
                out_vip := exists(select 1 from users 
                           where uid = in_uid and 
                           greatest(vip_until, grand_until) > current_timestamp);
        END;
$BODY$ language plpgsql;

然后我尝试在另一个存储过程中使用上述函数：

create or replace function join_new_game(
    IN in_uid integer,
    IN in_letters varchar(130),
    IN in_style integer,
    OUT out_gid integer
) as $BODY$
        BEGIN
        /* maybe there is a new game already, just waiting for the player's 1st move*/
        select gid into out_gid from games 
        where (player1 = in_uid and stamp1 is null) 
        or (player2 = in_uid and stamp2 is null) limit 1;

        IF not found THEN
                /* try to find games having just 1 player (with different uid) */
                select gid into out_gid from games 
                where (player1 != in_uid and stamp1 is not null
                and player2 is null) limit 1;

                IF not found THEN
                        /* only allow board style 1 for non-paying customers */
                        IF not select is_vip(in_uid) THEN
                                in_style := 1;  -- the above line fails
                        END IF;

                        /* create new game with player1 = uid and stamp1 = null */
                        insert into games (
                                created, 
                                player1, 
                                stamp1, 
                                stamp2, 
                                letters1, 
                                letters2, 
                                letters, 
                                board, 
                                style 
                        ) values (
                                current_timestamp, 
                                in_uid, 
                                null, 
                                null, 
                                substring(in_letters, 1, 7), 
                                substring(in_letters, 8, 7), 
                                substring(in_letters, 15), 
                                rpad('', 225), -- fill 15x15 board
                                in_style
                        ) returning gid into out_gid;
                ELSE
                        update games set player2 = in_uid where gid = out_gid;
                END IF;
        END IF;
        END;
$BODY$ language plpgsql;

但我得到这个语法错误：

ERROR:  syntax error at or near "select"
LINE 21:                         IF not select is_vip(in_uid) TH...
                                        ^

如何is_vip()正确使用该功能？

4

3 回答 3

3

is_vip(in_uid)是一个返回布尔值的函数。你可以直接调用它：

IF not is_vip(in_uid) THEN

或者，如果您想使用选择：

IF not (select is_vip(in_uid)) THEN

如果要将标量查询用作值表达式，则必须将查询括起来。

于 2016-02-03T20:04:15.463 回答
2

如何正确编写函数is_vip()？

你的功能可以更有效率。使它成为一个STABLESQL 函数，所以它可以被内联。

CREATE OR REPLACE FUNCTION is_vip(in_uid integer)
  RETURNS boolean AS
$func$
SELECT EXISTS (
   SELECT 1 FROM users
   WHERE  uid = in_uid
   AND   (vip_until   > current_timestamp OR
          grand_until > current_timestamp)
   )
$func$ LANGUAGE sql STABLE;

理想情况下，您有一个多列索引以允许仅索引扫描：

(uid, vip_until, grand_until)

uid必须是第一列。

How to use the is_vip function properly?

@kin 已经为您的基本语法错误提供了修复：使用您自己的函数，就像任何其他 Postgres 函数一样。但这里还有很多：

我将最重要的部分格式化为粗体：

CREATE OR REPLACE FUNCTION join_new_game(
    IN in_uid integer,
    IN in_letters varchar(130),
    IN in_style integer,
    OUT out_gid integer) AS
$func$
BEGIN
   /* maybe there is a new game already, just waiting for the player's 1st move*/
   SELECT gid INTO out_gid
   FROM   games 
   WHERE (player1 = in_uid AND stamp1 IS NULL) 
   OR    (player2 = in_uid AND stamp2 IS NULL)
   LIMIT 1;

   IF NOT FOUND THEN
      /* try to find games having just 1 player (with different uid) */
      /* and UPDATE immediately using a smart locking strategy */
      UPDATE games g
      SET    player2 = in_uid
      FROM  (
         SELECT gid
         FROM   games 
         WHERE  player1 <> in_uid
         AND    stamp1 IS NOT NULL
         AND    player2 IS NULL
         LIMIT  1
         FOR    UPDATE SKIP LOCKED  -- see link below !!
         ) g1
      WHERE   g.gid = g1.gid
      RETURNING g.gid
      INTO    out_gid;

      IF NOT FOUND THEN
         /* create new game with player1 = uid and stamp1 = null */
         INSERT INTO games (created, player1, stamp1, stamp2, letters1, letters2, letters, board, style)
         VALUES (current_timestamp, in_uid, null, null
               , left(in_letters, 7)
               , substring(in_letters, 8, 7)
               , right(in_letters, -15)  -- guessing you want to start at pos 16!
               , rpad('', 225)  -- fill 15x15 board
               /* only allow board style 1 for non-paying customers */
               , CASE WHEN NOT is_vip(in_uid) THEN 1 END  -- defaults to NULL
               )
         RETURNING gid
         INTO  out_gid;
      END IF; 
   END IF;
END
$func$  LANGUAGE plpgsql;

幸运的是，您使用的是最新版本的 Postgres (9.5)，它引入了新的智能锁定策略，用于使用FOR UPDATE SKIP LOCKED. 详细解释：

is_vip()我还使用表达式将调用内联到INSERT查询中CASE。这样更有效率。

其他几个小的优化。

于 2016-02-04T04:45:49.977 回答
1

尝试使用 schema.function_name 进行更改，如下所示：

IF (<schema>.is_vip(in_uid) = 'false')
THEN
   ...
END IF;
于 2016-02-03T20:02:38.403 回答