1. 程式人生 > 資料庫 >PostgreSQL 定義返回表函式的操作

PostgreSQL 定義返回表函式的操作

本文我們學習如何在PostgreSQL 開發返回表函式。

示例資料表

我們使用的示例資料庫表為film,如下圖所示:

PostgreSQL 定義返回表函式的操作

示例1

第一個函式發揮所有滿足條件film表記錄,這裡使用ilike操作,和like類似,但不區分大小寫:

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) 函式接收一個引數,為匹配title欄位的模式字串。

為了從函式中返回表,需要使用return table語法,以及表的欄位,每個欄位使用逗號分隔。

在函式中,我們返回一個查詢(select 語句)作為返回結果。注意select語句中的欄位必須和返回表的欄位型別一致。因為film表中release_year的資料型別不是integer,所以我們要使用cast函式轉換成integer。

下面進行測試該函式:

SELECT
 *
FROM
 get_film ('Al%');

我們呼叫該函式,獲取所有title以Al開頭的記錄:

PostgreSQL 定義返回表函式的操作

注意,我們也可以使用下面語句進行呼叫:

SELECT
 get_film ('Al%');

PostgreSQL 返回已一列陣列形式返回表。

PostgreSQL 定義返回表函式的操作

示例2

實際開發中,我們經常需要在返回函式結果集之前處理每一行記錄。下面通過示例說明:

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';

該函式與上一個名稱一樣get_film(varchar,int),但有兩個引數:

第一個引數匹配title欄位的模式字串。仍然使用ilike操作執行搜尋。

第二個引數是file的發行年度。

這兩個函式在PostgreSQL中稱為過載函式。我們想在返回結果之前處理每一行,使用 FOR LOOP語句進行處理。內部每個迭代中使用UPPER函式是film title 變為大寫,僅為了演示而已。

return next語句是增加一行至函式結果集中,不斷執行迴圈,在每次迭代中生成結果集。

下面進行測試:

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

PostgreSQL 定義返回表函式的操作

總結

現在你應該理解了如何開發返回表的函式,主要使用return query 和 return next 語句。

補充:Postgres自定義函式返回記錄集(虛擬表結構)

看例項吧~

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_name ;
end;$BODY$
 LANGUAGE plpgsql VOLATILE
 COST 100
 ROWS 10;
ALTER FUNCTION fun_get_real_inv_qty()
 OWNER TO postgres;

上面是例子,呼叫這個函式:

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定義 必須與函式輸出的列數量及每列資料型別一一對應.

以上為個人經驗,希望能給大家一個參考,也希望大家多多支援我們。如有錯誤或未考慮完全的地方,望不吝賜教。