SQL 触发函数在 INSERT 时更新每日移动平均线

Question

我正在尝试创建一个 SQL 触发器函数，该函数应该在将数据插入到表中时更新列。更新基于正在插入的值中存在的值。

我有下表来存储股票的每日 OHLC 数据。

CREATE TABLE daily_ohlc (
 cdate date,
 open numeric(8,2),
 high numeric(8,2),
 low numeric(8,2),
 close numeric(8,2),
 sma8 numeric(8,2)
);

插入命令：

INSERT INTO daily_ohlc (cdate, open, high, low, close) 
 values ('SYMBOL', 101, 110, 95, 108);

执行此命令时，我想根据正在插入的当前值和表中已有的值更新“sma8”列。

到目前为止，我正在使用以下 SQL 查询来计算每一行的值，然后使用结果来使用 python 更新“sma8”列。

SELECT sec.date, AVG(sec.close) 
 OVER(ORDER BY sec.date ROWS BETWEEN 7 PRECEDING AND CURRENT ROW) AS 
simple_mov_avg FROM daily_ohlc sec;

上述查询计算最近 8 条记录（包括当前行）的简单移动平均线。

使用此过程，我每次插入数据时都会更新“sma8”列中的每一行数据。我想通过使用触发器只更新最后一行（即插入的行）。如何做到这一点？

Answer 1

您可以在触发器中使用适当的联接来执行UPDATE FROM 选择查询。

create or replace function update_sma8() RETURNS TRIGGER AS
$$
 BEGIN

UPDATE daily_ohlc d SET sma8 = s.simple_mov_avg 
FROM
(
 SELECT  sec.cdate,AVG(sec.close)  
   OVER(ORDER BY sec.cdate ROWS BETWEEN 7 PRECEDING AND CURRENT ROW) AS 
    simple_mov_avg FROM daily_ohlc sec
)s where s.cdate = NEW.cdate  --The newly inserted cdate
     AND d.cdate = s.cdate;   
RETURN NULL;

END $$ language plpgsql;

Demo

使用此方法的唯一警告是，如果有人删除行或更新 close 列，则必须重新计算值，这不会发生在现有行上。只有插入的行会看到正确的重新计算值。

相反，您可以简单地创建 View 以在请求时从主表中为所有行计算 sma8 列。

Answer 2

你不能按照这些思路做点什么吗？

INSERT INTO daily_ohlc 
SELECT current_date, 101, 110, 95, 108, (COUNT(*)*AVG(close)+108)/(1+Count(*))
FROM daily_ohlc
WHERE cDate >= ANY (
    SELECT MIN(cdate)
    FROM (SELECT CDate, ROW_NUMBER() OVER (ORDER BY CDate DESC) as RowNum FROM daily_ohlc) a 
    WHERE RowNum <= 7
)

我很清楚，与触发器相比，它可能看起来很复杂。
但是，我试图避免您成功创建 ON INSERT 触发器并接下来要处理表中的更新的情况。在由同一表中的更新触发的过程中更新表并不是最好的主意。