我们如何使用约束来确保数组的值不为NULL?

时间:2018-04-19 05:35:34

标签: postgresql

假设我有下表:

CREATE TABLE test (
    arr_column VARCHAR[] NOT NULL
)

这不会阻止在插入行时将数组的值设置为NULL。所以,我想要一个约束来强制执行这条规则。我的尝试如下:

CREATE TABLE test (
    arr_column VARCHAR[] NOT NULL 
        CHECK (NOT (ARRAY[NULL]::VARCHAR[] <@ arr_column))
)

但不幸的是,如果我插入,这不会失败:

INSERT INTO test (ARRAY['some_string', NULL]::VARCHAR[])

2 个答案:

答案 0 :(得分:4)

进行此类检查的一种简单直接的方法是使用触发器,但您也可以简单地创建一个函数并在CHECK子句中使用它,就像您到目前为止所做的那样:

CREATE OR REPLACE FUNCTION check_null_element(arr TEXT[]) 
RETURNS BOOLEAN AS $BODY$
DECLARE j INT;
BEGIN
  FOR j IN 1 .. ARRAY_UPPER(arr, 1) LOOP
    IF arr[j] IS NULL THEN
       RETURN FALSE;
    END IF;
  END LOOP;
RETURN TRUE;
END;
$BODY$
LANGUAGE plpgsql;

因此,在创建表格时,您只需要:

CREATE temp TABLE test (
    arr_column VARCHAR[] NOT NULL 
        CHECK (check_null_element(arr_column))
);

尝试插入一个NULL值的数组:

db=# INSERT INTO test VALUES (ARRAY['some_string', NULL]::VARCHAR[]);
FEHLER:  neue Zeile für Relation »test« verletzt Check-Constraint »test_arr_column_check«
DETAIL:  Fehlgeschlagene Zeile enthält ({some_string,NULL}).

并且有效一个..

db=# INSERT INTO test VALUES (ARRAY['some_string', 'NOT NULL :-)']::VARCHAR[]);
INSERT 0 1

编辑:很高兴:

为了避免不必要的异常,您还可以检查参数本身是否为NULL - 这个问题是多余的,因为它已经在NOT NULL语句中使用CREATE TABLE约束进行了检查。这可以通过向函数添加以下条件来完成:IF arr IS NULL THEN RETURN FALSE; END IF;

CREATE OR REPLACE FUNCTION check_null_element(arr TEXT[]) 
RETURNS BOOLEAN AS $BODY$
DECLARE j INT;
BEGIN
  IF arr IS NULL THEN RETURN FALSE; END IF;
  FOR j IN 1 .. ARRAY_UPPER(arr, 1) LOOP
    IF arr[j] IS NULL THEN
       RETURN FALSE;
    END IF;
  END LOOP;
RETURN TRUE;
END;
$BODY$
LANGUAGE plpgsql;

答案 1 :(得分:1)

以下似乎可以解决问题,代码很少:

CHECK (array_position(arr_column, NULL) is NULL)