如何在Postgresql中创建条件插入函数?

时间:2020-06-05 16:20:06

标签: sql postgresql sql-update case sql-function

我尝试根据该表中的值编写一个函数以在表中的column3中插入一些值,但是我不太熟悉在Postgresql 9.6中编写函数。

--Create some table

    CREATE TABLE test(column1 INT, column2 INT, column3 TEXT) 
    INSERT INTO test VALUES(-8020,15),(-200,1),(23,115)

--Build function

    CREATE OR REPLACE FUNCTION new_function()
    RETURNS TEXT AS 
    $$
    BEGIN

        IF test(column1) <= -7000 THEN INSERT INTO test(column3) VALUES('small');
        ELSIF test(column1) >= -7000 AND test(column2) <= 15 THEN INSERT INTO test(column3) VALUES('nohello');
        ELSIF test(column1) >= -7000 ANDtable(column2) >= 15 THEN INSERT INTO test(column3) VALUES('test');
        ELSE INSERT INTO test(column6) VALUES("nodata");
        END IF;

    END;
    $$
    LANGUAGE plpgsql;

结果应该是一个看起来像这样的表:

Column1 | Column2 | Column3
---------------------------
 -8020  |    15   |  small
  -200  |     1   |  nohello
    23  |   115   |  test

调用new_function时出现错误column1 doesn't exist.

2 个答案:

答案 0 :(得分:4)

您似乎实际上是在寻找update(用于更改现有行的值),而不是insert(用于创建新行)。

但是,最重要的是,我建议只使用一个计算列:

create table test(
    column1 int, 
    column2 int, 
    column3 text generated always as (
        case 
            when column1 <= -7000 then 'small'
            when column1 <= 15    then 'nohello'
            else 'nodata'
        end
    ) stored
);

在表中插入或更新行时,数据库会相应地自动调整计算列,因此它始终是最新的。

Demo on DB Fiddle

insert into test(column1, column2) values(-8020,15),(-200,1),(23,115);

select * from test;
column1 | column2 | column3
------: | ------: | :------
  -8020 |      15 | small  
   -200 |       1 | nohello
     23 |     115 | nodata 

请注意,生成的列仅从Postgres 12开始可用。在早期版本中,一种替代方法是在表中仅保留前两列,并创建一个视图以显示第三列:

create view myview as 
select
    column1,
    column2,
    case 
        when column1 <= -7000 then 'small'
        when column1 <= 15    then 'nohello'
        else 'nodata'
    end as column3
from mytable

然后您可以查询视图而不是查询表来显示数据。

答案 1 :(得分:1)

GMB的答案是一个完美的解决方案,尽管您可以通过使用CASE条件表达式来更新表,如下所示

update test
set column3 = case 
                when column1 <= - 7000 then 'small'
                when (column1 >= - 7000 and column2 <= 15) then 'nohello'
                when (column1 >= - 7000 and column2 >= 15) then 'test'
                else 'nodata'
              end;

Demo