PostgreSQL 定義返回表函式的操作
本文我們學習如何在PostgreSQL 開發返回表函式。
示例資料表
我們使用的示例資料庫表為film,如下圖所示:
示例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開頭的記錄:
注意,我們也可以使用下面語句進行呼叫:
SELECT get_film ('Al%');
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);
總結
現在你應該理解了如何開發返回表的函式,主要使用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定義 必須與函式輸出的列數量及每列資料型別一一對應.
以上為個人經驗,希望能給大家一個參考,也希望大家多多支援我們。如有錯誤或未考慮完全的地方,望不吝賜教。