表达式的唯一索引不像我期望的那样工作

时间:2018-03-06 09:24:04

标签: postgresql unique-index

以下是我的问题示例:http://dbfiddle.uk/?rdbms=postgres_9.6&fiddle=ddb9cfd2da315ecf36cfffd66853f023

我使用这个功能:

CREATE OR REPLACE FUNCTION inListExistOrNull(list jsonb) RETURNS boolean AS
$BODY$
    DECLARE
        r TEXT;
        i boolean := false;
        vcount int;
    BEGIN
        FOR r IN SELECT * FROM jsonb_array_elements($1) LOOP
            vcount := (SELECT COUNT(*)
                       FROM table_example
                       WHERE data->>'test' LIKE '%' || r || '%');

            i := vcount > 0;

            IF i = true THEN
                EXIT;
            END IF;
        END LOOP;

        IF i = true THEN
            RETURN true;
        END IF;

        RETURN NULL;
    END;
$BODY$
LANGUAGE 'plpgsql'
IMMUTABLE
RETURNS NULL ON NULL INPUT;

我创建了一个表和这样的索引:

CREATE TABLE IF NOT EXISTS table_example(
   id bigserial primary key,
   data jsonb,
   txid bigint
);

CREATE INDEX IF NOT EXISTS table_example_txid_index
   ON table_example(txid);

CREATE UNIQUE INDEX IF NOT EXISTS unique_table_example
   ON table_example(inListExistOrNull(data->'test'));

我插入一些行:

INSERT INTO table_example (id, data, txid)
   VALUES (1, '{"test": ["https://example.com/test/123", "https://example.com/test/678"]}', 1);
INSERT INTO table_example (id, data, txid)
   VALUES (2, '{"test": ["https://example.com/test/b4b81fb221d4fa641", "https://example.com/test/624f3e10048245fb1"]}', 2);
INSERT INTO table_example (id, data, txid)
   VALUES (4, '{"test": ["https://example.com/test/ggg", "https://example.com/test/hhh"]}', 4);
INSERT INTO table_example (id, data, txid)
   VALUES (5, '{"test": ["https://example.com/test/ggg"]}', 5);

我不知道为什么我可以用id = 5创建一行。它应该被唯一索引捕获,但事实并非如此。

这就像我期望的那样:

INSERT INTO table_example (id, data, txid)
   VALUES (6, '{"test": ["https://example.com/test/b4b81fb221d4fa641", "https://example.com/test/624f3e10048245fb1"]}', 6);
ERROR:  duplicate key value violates unique constraint "unique_table_example"
DETAIL:  Key (inlistexistornull(data -> 'test'::text))=(t) already exists.

TRIGGER:

CREATE OR REPLACE FUNCTION inListExistOrNull() RETURNS TRIGGER AS
$BODY$
    DECLARE
        r TEXT;
        i boolean := false;
        vcount int;
        newData jsonb;
    BEGIN
        newData := NEW.data->'test';

        FOR r IN SELECT * FROM jsonb_array_elements(newData) LOOP
            vcount := (SELECT COUNT(*) FROM table_example WHERE data->>'test' LIKE '%' || r || '%');

            i := vcount > 0;

            IF i = true THEN
                RAISE 'Duplicate data: %', r USING ERRCODE = '23505';
            END IF;
        END LOOP;

        RETURN NEW;
    END;
$BODY$
LANGUAGE 'plpgsql'
STABLE;

CREATE TRIGGER inListExistOrNullTrigger
BEFORE INSERT OR UPDATE ON table_example
    FOR EACH ROW EXECUTE PROCEDURE inListExistOrNull();

UNIQUE INDEX:

CREATE OR REPLACE FUNCTION inListExistOrNull(list jsonb) RETURNS int AS
$BODY$
    DECLARE
        r TEXT;
        i boolean := false;
        vcount int;
    BEGIN
        FOR r IN SELECT * FROM jsonb_array_elements($1) LOOP
            vcount := (SELECT COUNT(*) FROM table_example WHERE data->>'test' LIKE '%' || r || '%');

            i := vcount > 0;

            IF i = true THEN
                RAISE 'Duplicate data: %', r USING ERRCODE = '23505';
            END IF;
        END LOOP;

        IF i = true
        THEN
            RETURN true;
        ELSE
            RETURN NULL;
        END IF;
    END;
$BODY$
LANGUAGE 'plpgsql'
IMMUTABLE
RETURNS NULL ON NULL INPUT;

CREATE UNIQUE INDEX IF NOT EXISTS unique_table_example ON table_example(inListExistOrNull(data->'test'));

2 个答案:

答案 0 :(得分:2)

一个问题是你的函数不是真正不可变的,当你将它标记为PostgreSQL时,你就是在撒谎。 IMMUTABLE意味着它始终必须为相同的参数返回相同的结果,无论数据库中有什么数据以及如何配置它。

另一个问题是索引不会做你想要的:它只会抱怨第二次你输入一个值,其中函数结果为TRUE 。这就是你在这里遇到的。

您可以使用具有类似功能的AFTER INSERT OR UPDATE触发器(如果遇到重复项会引发异常),而不是索引,这将执行您想要的操作。然后你可以按照你应该标记函数STABLE,一切都应该正常。

答案 1 :(得分:0)

因为你在函数的输出上创建索引所以......

  1. 你创建记录 - 函数返回null
  2. 你创建另一个(应该失败) - 函数返回true - 没有违反唯一约束
  3. 您创建第三条记录 - 函数返回true - 违反了唯一约束,并且未创建记录。
  4. 您应该添加raise expcetion或类似内容。