【postgresql】 简易订单编码生成函数

描述:根据前缀和当前日期生成订单编码到指定表的字段

生成结构:前缀+年+月+编号

逻辑:判断该字段是否有数据,如果没有数据或已有的数据里的月份小于当前日期所在的月份,则生成本月第一条订单,否则生成编号+1的订单

实现代码如下:

CREATE OR REPLACE FUNCTION "public"."create_order_number"("pn" text, "fn" text, "tn" text)
  RETURNS "pg_catalog"."varchar" AS $BODY$
	--pn:单号前缀prefix_name
	--pl:前缀长度prefix_length
	--od:订单号order_number
	--fn:单号的列名field_name
	--tn:单号的表名table_name
declare pl INTEGER:=length(pn);
declare str VARCHAR:='select max('||fn||') from '||tbname;
declare od varchar;
declare cod varchar;--生成的单号create_order_number
	BEGIN
	EXECUTE str into od;--执行str
	IF od is null or make_date(substring(od,pl+1,4)::integer, substr(od,pl+5,2)::integer, 1)	<cast(date_trunc('month',current_date) as date)
	THEN select pn||TO_CHAR(CURRENT_DATE,'yyyymm')||lpad('1', 3, '0') into cod;
ELSE 
	select (pn||TO_CHAR(CURRENT_DATE,'yyyymm')||lpad(((substr(dh,pl+7,3)::INTEGER)+1)::varchar,3,'0'))into cod;
END IF;
return cod;
END$BODY$
  LANGUAGE plpgsql VOLATILE
  COST 100
	

参数说明: pn 为要生成单号的前缀.

                  fn为要生成单号的表字段.

                  tn为要生成单号的表.

测试语句:

	select create_order_number('PO','po_num','purchase_order')

数据库小白编写,对数据库安全性问题未做了解,请大神指导.