PostgreSQL 10 - 自定义聚合函数
比如,想解决一个很简单的问题:比如客户打车要付钱,假如3公里内13元,以后每公里需要付2.30元。那一共付多少钱呢?
当然,不用自定义聚合也能解决。不过,看看我怎么做。首先,增加测试数据:
postgres=# CREATE TABLE t_taxi (c_id int, km numeric);
CREATE TABLE
postgres=# INSERT INTO t_taxi
postgres-# VALUES (1, 4.0), (1, 3.2), (1, 4.5), (2, 1.9), (2, 4.5);
INSERT 0 5
可以使用CREATE AGGREGATE增加聚合。
第一件事是写聚合函数,它由每行调用。它的参数是两个中间值和所处理的行的数据:
CREATE OR REPLACE FUNCTION taxi_per_line (numeric, numeric)
RETURNS numeric AS
$$
BEGIN
RAISE NOTICE 'intermediate: %, per row: %', $1, $2;
IF $2 <= 3
THEN
RETURN $1 + 13;
ELSE
RETURN $1 + 13 + ($2 - 3) * 2.3;
END IF;
END;
$$
LANGUAGE 'plpgsql';
然后写一个聚合函数:
CREATE AGGREGATE taxi_pr