使用Postgresql怎么自定義一個函數

本篇文章為大家展示了使用Postgresql怎么自定義一個函數，內容簡明扼要并且容易理解，絕對能使你眼前一亮，通過這篇文章的詳細介紹希望你能有所收獲。

PostgreSQL函數也稱為PostgreSQL存儲過程。 PostgreSQL函數或存儲過程是存儲在數據庫服務器上并可以使用SQL界面調用的一組SQL和過程語句(聲明，分配，循環，控制流程等)。

語法：

CREATE [OR REPLACE] FUNCTION function_name (arguments)  
RETURNS return_datatype AS $variable_name$ 
 DECLARE 
  declaration; 
  [...] 
 BEGIN 
  < function_body > 
  [...] 
  RETURN { variable_name | value } 
 END; LANGUAGE plpgsql;

參數說明

function_name:指定函數的名稱。

[OR REPLACE]:是可選的，它允許修改/替換現有函數。

DECLARE:定義參數(參數名寫在前面 類型寫在后面)。

BEGIN~END: 在中間寫方法主體。

RETURN:指定要從函數返回的數據類型(它可以是基礎，復合或域類型，或者也可以引用表列的類型)。

LANGUAGE:它指定實現該函數的語言的名稱。

下面我將創建一個簡單的函數：

CREATE OR REPLACE FUNCTION test(id uuid)
 RETURNS INTEGER
 LANGUAGE plpgsql
AS
$$
declare
 count integer;
begin
 count = (SELECT
       json_array_length(A.json::json -> 'features')
      FROM "Json" A
      WHERE A.uid = id);
 return count;
end;
 
$$

該函數的功能是用來返回json中數組的長度的

定義好該函數后，我們可以像調用其他的函數一樣調用它

SELECT test(id)

返回數組長度(integer)

補充：PostgreSQL中自定義函數（function）返回數據集

1.準備數據

建表

 create table city (
 cityId int, 
 cityName varchar(20)
 );

插入數據

insert into city values(1,'BeiJing'),(2,'NewYork'),(3,'Hong kong'),(4,'ShaingHai');

2. 函數實例

2.1 setof 表名

創建函數getCity()，它返回city表中所有的數據，返回值聲明為 setof 表名。

create or replace function getCity() returns setof city as 
$$
begin
return query select * from city;
end;
$$
language plpgsql;

執行函數。

mydb=# select getCity();
   getcity
-----------------
 (1,BeiJing)
 (2,NewYork)
 (3,"Hong kong")
 (4,ShaingHai)
(4 行記錄)

也可以用類似查詢表的方式。

mydb=# select * from getCity();
 cityid | cityname
--------+-----------
   1 | BeiJing
   2 | NewYork
   3 | Hong kong
   4 | ShaingHai
(4 行記錄)

PostgreSQL還支持對函數執行結果進行條件判斷并過濾。

mydb=# select * from getCity() where cityId > 3;
 cityid | cityname
--------+-----------
   4 | ShaingHai
(1 行記錄)

2.2 setof record

為了使函數更加通用，以解決動態返回數據集的問題。創建函數getRows(text)，將表名city作為函數參數，它返回表中所有的數據，返回值聲明為 setof record。

create or replace function getRows(text) returns setof record as
$$
declare
rec record;
begin
for rec in EXECUTE 'select * from ' || $1 loop
return next rec;
end loop;
return;
end
$$
language 'plpgsql';

執行函數。

mydb=# select * from getRows('city') as city(id int, name varchar(20));
 id |  name
----+-----------
 1 | BeiJing
 2 | NewYork
 3 | Hong kong
 4 | ShaingHai
(4 行記錄)

上述內容就是使用Postgresql怎么自定義一個函數，你們學到知識或技能了嗎？如果還想學到更多技能或者豐富自己的知識儲備，歡迎關注億速云行業資訊頻道。