Rerturn從函數返回ID

Question

我基本上試圖從下面的函數返回更新的ID。以下是我的代碼的簡化版本：

CREATE FUNCTION confirmUser(_token varchar(255)) RETURNS integer AS $$ 
BEGIN 
    return (UPDATE users SET isConfirmed = true where _token = token RETURNING id)); 
END; 
$$ LANGUAGE plpgsql; 

這是什麼問題？

Answer 1

我不認爲你可以像這樣返回值。您需要將其存儲到一個變量中，然後返回該變量：

CREATE FUNCTION confirmUser(_token varchar(255)) RETURNS integer 
AS 
$$ 
declare 
    _id integer; 
BEGIN 
    UPDATE users SET isConfirmed = true where _token = token 
    RETURNING id 
    into _id; 

    return _id; 
END; 
$$ 
LANGUAGE plpgsql; 

但是，您不需要PL/pgSQL函數。一個普通的SQL函數將事情做的很好：

CREATE FUNCTION confirmUser(_token varchar(255)) RETURNS integer 
AS 
$$ 
    UPDATE users SET isConfirmed = true where _token = token 
    RETURNING id 
$$ LANGUAGE sql; 

但你必須確保token是唯一的，否則這將失敗。