如何在 PostgreSQL 表中使用自定义函数作为列默认值

Question

我创建了一个名为 Loan 的表，其中包含以下列：

loanID SERIAL, 
annualInterestRate INT, 
numberOfYears INT, 
loanAmount NUMERIC, 
monthlyPayment NUMERIC

monthlyPayment 的计算取决于numberOfYears、loanAmount 和annualInterestRate，按照以下公式：

monthlyPayment = (loanAmount * monthlyInterestRate) /
                (1 - (1/Math.pow(1 + monthlyInterestRate , numberOfYears * 12) ));

我创建了一个名为get_monthly_payment() 的函数，它可以毫无问题地返回monthlyPayment。对于每一行，我想将此函数的返回设为monthlyPayment 列的默认值。 我怎样才能做到这一点？

Answer 1

月利率可能不是您表格中的一列，但年利率是。但每年到每月是一个简单的计算。以下内容可用于 v12 生成的列和使用触发器的先前版本。此外，它还提供了一个通常有用的 Postgres 函数，用于计算每月付款。请参阅fiddle。

create or replace 
function monthly_loan_payment 
       ( amount  numeric
       , apr     numeric
       , term    numeric
       )
  returns numeric
  language sql 
  immutable strict  
/*  Given a loan amount, the Annual Percent (Interest) Rate and term (in years) 
 *  compute the monthly payment to service the loan. 
 *  Note. Monthly payment calculates correctly, but due to the exact terms of loan
 *        and date of payment receipt adjustments for end of loan payment may be 
 *        required.  
 */ 
as $$   
  with monthly (mrate) as (values ( (apr/100.00) / 12.00 ) )
  select round((amount * mrate)  /(1.0 - (1.0/ ((1.0 +mrate)^( term * 12.00)) ))::numeric,2) 
     from monthly;
$$; 

Answer 2

如果monthlyPayment 根据您的定义是固定的，因此完全依赖于功能，那么请考虑完全不保留该值。继续使用您的廉价（！）功能。整体上更清洁，更便宜。喜欢：

SELECT *, f_monthly_payment(l) AS monthly_payment
FROM loan l;

假设函数定义为f_monthly_payment(loan)（以表的行类型为参数）。否则相应地进行调整。

Postgres 12 或更高版本具有 STORED 生成的列，但这些只为昂贵的计算付费。持久化列占用空间并减慢表上的所有操作。

见：

• Computed / calculated / virtual / derived columns in PostgreSQL

如果您想允许手动更改，默认列是可行的方法（就像您实际问过的一样）。但是你不能使用你的函数，因为quoting the manual:

DEFAULT 子句为其指定的列分配默认数据值 它出现在列定义中。该值是任何无变量的 表达式（特别是对其他列的交叉引用） 不允许当前表）。

剩下的解决方案是在桌子上触发BEFORE INSERT，例如：

CREATE OR REPLACE FUNCTION trg_loan_default_rate()
  RETURNS trigger
  LANGUAGE plpgsql AS
$func$
BEGIN
   NEW.monthlyPayment := (NEW.loanAmount * monthlyInterestRate())
                    / (1 - (1/power(1 + monthlyInterestRate(), NEW.numberOfYears * 12)));

   RETURN NEW;
END
$func$;

CREATE TRIGGER loan_default_rate
BEFORE INSERT ON loan
FOR EACH ROW EXECUTE FUNCTION trg_loan_default_rate();

假设monthlyInterestRate() 是一个自定义函数。
我用内置的Postgres function power()替换了Math.pow。

NEW是触发函数中的特殊记录变量，指的是新插入的行。见：

• FOR EACH STATEMENT trigger example

EXECUTE FUNCTION 需要 Postgres 11。请参阅：

• Trigger function does not exist, but I am pretty sure it does

相关：

• PostgreSQL - set a default cell value according to another cell value

另外：考虑 Postgres 中合法的、小写的、不带引号的标识符。见：

• Are PostgreSQL column names case-sensitive?

Answer 3

在 Postgres 12+ 中，您已经生成了列：

生成的列是始终从其他列计算的特殊列。

“生成的列是一个特殊的列，它总是从其他列中计算出来的。...”

所以：

monthlyPayment numeric GENERATED ALWAYS AS(loanAmount * monthlyInterestRate) /
                (1 - (1/Math.pow(1 + monthlyInterestRate , numberOfYears * 12) ))  STORED

在版本 12 之前，您必须使用 CREATE TRIGGER 向表中添加一个触发器，该触发器调用包含上述公式的函数并设置列值。