PostgreSQL 定義返回表函數(shù)的操作

更新時(shí)間：2021年01月07日 11:35:25 作者：neweastsun

這篇文章主要介紹了PostgreSQL 定義返回表函數(shù)的操作，具有很好的參考價(jià)值，希望對大家有所幫助。一起跟隨小編過來看看吧

本文我們學(xué)習(xí)如何在PostgreSQL 開發(fā)返回表函數(shù)。

示例數(shù)據(jù)表

我們使用的示例數(shù)據(jù)庫表為film，如下圖所示：

示例1

第一個(gè)函數(shù)發(fā)揮所有滿足條件film表記錄，這里使用ilike操作，和like類似，但不區(qū)分大小寫：

CREATE OR REPLACE FUNCTION get_film (p_pattern VARCHAR) 
 RETURNS TABLE (
 film_title VARCHAR,
 film_release_year INT
) 
AS $$
BEGIN
 RETURN QUERY SELECT
 title,
 cast( release_year as integer)
 FROM
 film
 WHERE
 title ILIKE p_pattern ;
END; $$ 
 
LANGUAGE 'plpgsql';

get_film(varchar) 函數(shù)接收一個(gè)參數(shù)，為匹配title字段的模式字符串。

為了從函數(shù)中返回表，需要使用return table語法，以及表的字段，每個(gè)字段使用逗號分隔。

在函數(shù)中，我們返回一個(gè)查詢（select 語句）作為返回結(jié)果。注意select語句中的字段必須和返回表的字段類型一致。因?yàn)閒ilm表中release_year的數(shù)據(jù)類型不是integer，所以我們要使用cast函數(shù)轉(zhuǎn)換成integer。

下面進(jìn)行測試該函數(shù)：

SELECT
 *
FROM
 get_film ('Al%');

我們調(diào)用該函數(shù)，獲取所有title以Al開頭的記錄：

注意，我們也可以使用下面語句進(jìn)行調(diào)用：

SELECT
 get_film ('Al%');

PostgreSQL 返回已一列數(shù)組形式返回表。

示例2

實(shí)際開發(fā)中，我們經(jīng)常需要在返回函數(shù)結(jié)果集之前處理每一行記錄。下面通過示例說明：

CREATE OR REPLACE FUNCTION get_film (p_pattern VARCHAR,p_year INT) 
 RETURNS TABLE (
 film_title VARCHAR,
 film_release_year INT
) AS $$
DECLARE 
 var_r record;
BEGIN
 FOR var_r IN(SELECT title, release_year 
   FROM film 
   WHERE title ILIKE p_pattern 
   AND release_year = p_year) 
 LOOP
 film_title := upper(var_r.title) ; 
 film_release_year := var_r.release_year;
 
 RETURN NEXT;
 END LOOP;
END; $$ 
LANGUAGE 'plpgsql';

該函數(shù)與上一個(gè)名稱一樣get_film(varchar,int)，但有兩個(gè)參數(shù)：

第一個(gè)參數(shù)匹配title字段的模式字符串。仍然使用ilike操作執(zhí)行搜索。

第二個(gè)參數(shù)是file的發(fā)行年度。

這兩個(gè)函數(shù)在PostgreSQL中稱為重載函數(shù)。我們想在返回結(jié)果之前處理每一行，使用 FOR LOOP語句進(jìn)行處理。內(nèi)部每個(gè)迭代中使用UPPER函數(shù)是film title 變?yōu)榇髮?，僅為了演示而已。

return next語句是增加一行至函數(shù)結(jié)果集中，不斷執(zhí)行循環(huán)，在每次迭代中生成結(jié)果集。

下面進(jìn)行測試：

SELECT
 *
FROM
 get_film ('%er', 2006);

總結(jié)

現(xiàn)在你應(yīng)該理解了如何開發(fā)返回表的函數(shù)，主要使用return query 和 return next 語句。

補(bǔ)充：Postgres自定義函數(shù)返回記錄集(虛擬表結(jié)構(gòu))

看實(shí)例吧~

CREATE OR REPLACE FUNCTION fun_get_real_inv_qty(pvOrderId varchar)
 RETURNS SETOF record AS
$BODY$begin
--drop table if exists tmp_1 ;
--create temp table tmp_1 as 
return query 
	select fp_prod_id,fq_part_no,fq_name, 
	sum(case when fo_type='P' then -fp_qty 	 
		 else 0
	  end 
	) as purchase_qty,
	sum(case when fo_type='S' then -fp_qty 	 
		 else 0
	  end 
	) as saleqty,
	sum(case when fo_type='S' then -fp_qty 
		 when fo_type='P' then fp_qty 
		 else 0
	  end ) as surplus_qty from tp_send_det,to_send_note,tq_prod_mstr
	where fp_order_id=pvOrderId and fo_note_id=fp_note_id and fq_prod_id=fp_prod_id and fq_type='I'
	group by fp_prod_id,fq_part_no,fq_name ;
end;$BODY$
 LANGUAGE plpgsql VOLATILE
 COST 100
 ROWS 10;
ALTER FUNCTION fun_get_real_inv_qty()
 OWNER TO postgres;

上面是例子,調(diào)用這個(gè)函數(shù):

select * from fun_get_real_inv_qty('D302') f(fp_prod_id bigint,fq_part_no varchar ,fq_name varchar ,purchase_qty numeric ,saleqty numeric ,surplus_qty numeric );

f...后面帶的是記錄的column定義必須與函數(shù)輸出的列數(shù)量及每列數(shù)據(jù)類型一一對應(yīng).

以上為個(gè)人經(jīng)驗(yàn)，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教。

您可能感興趣的文章: