当我尝试运行以下查询时，我收到错误子查询必须只返回一列：

Question

CREATE or REPLACE PROCEDURE checkUpdateAdd(imei1 inout text, assetName1 inout text)
language 'plpgsql'
AS $BODY$
declare
begin
    PERFORM * from msdata;
if (select * from msdata where imei = imei1) then
    --UPDATE "public"."msdata"
    UPDATE public.msdata SET assetname1 = assetname where imei = imei1;
    -- return assetname;
-- SELECT * FROM msdata ORDER BY imei ASC;
elseif (select * from msdata where imei != imei1) then
    Insert into public.msdata(imei,assetname) values (imei1,assetname1);
    --return;
end if;
    -- return (null,null);
end;
$BODY$;

call checkUpdateAdd('123','abc1');
SELECT * FROM msdata; 

其中msdata 是包含imei 和assetname 列的现有表。

所以现在如果我必须创建一个在调用时创建或更新 msdata 的过程

Answer 1

我宁愿创建一个函数来更新或插入并在更新时返回 assetname 字段的旧值或在插入时返回 null（因为没有旧值）。请注意FOUND 预定义变量。

CREATE or REPLACE FUNCTION checkUpdateAdd(imei1 text, assetName1 text)
returns text language plpgsql AS
$BODY$
declare
  retval text;
begin
  select assetname into retval from msdata where imei = imei1;
  if FOUND then -- (imei = imei1) exists, return the old assetname value
    update msdata SET assetname = assetname1 where imei = imei1;
    return retval;
  else
    insert into msdata(imei, assetname) values (imei1, assetname1);
    return null;
  end if;
end;
$BODY$;

顺便说一句。如果imei 是主键或者如果它有unique 约束，那么任务就变得简单多了。您可以使用insert on conflict do update，不需要函数或过程。

更新作为一个程序：

CREATE or REPLACE PROCEDURE checkUpdateAdd(imei1 text, assetName1 text) 
language plpgsql AS
$BODY$
begin
  if exists (select from msdata where imei = imei1) then
    update msdata SET assetname = assetname1 where imei = imei1;
  else
    insert into msdata(imei, assetname) values (imei1, assetname1);
  end if;
end;
$BODY$;