PostgreSQL stable 函数调用次数

digoal 2018-06-29

742

作者

digoal

日期

2018-06-29

背景

immutable 函数，输入参数静态时，返回结果不变，所以它在plan前先计算，得到一个常量后，放到QUERY里面再生成执行计划。

stable 函数，输入参数静态时，如果在一个事务中多次调用它，返回结果不变。

显然一个immutable 函数在同一个SQL中，永远只调用一次，（在bind , execute的模式下则只有prepare的时候被调用一次）

但是一个stable函数，虽然输入参数静态时，如果在一个事务中多次调用它，返回结果不变。但是它可能被多次调用。

例如在同一个QUERY中，如果要扫描多条记录时，stable函数可能被多次调用。

1、全表扫描，扫描多少条记录，就需要调用多少次。（同时对于有静态参数输入或无参数的stable函数，需要再加一次执行计划时的评估）

2、索引扫描：

需要区分STABLE函数输入的是静态还是动态参数(或volatile函数的返回值作为参数)。

静态参数，索引扫描时只需要调用两次（不管扫描多少索引条目），一次是执行计划，一次是索引扫描前计算stable函数的返回值，然后就是索引扫了。

如果stable函数输入的是动态参数，不支持索引。所以只可能走全表。

例子

1、创建一个stable函数，里面通过raise来输出信息，判断这个函数被调用了多少次。

无输入参数

postgres=# create or replace function ff() returns int as $$ declare begin raise notice 'a'; return 1; end; $$ language plpgsql strict stable; CREATE FUNCTION

2、创建测试表

postgres=# create table test123(id int); CREATE TABLE postgres=# insert into test123 values (1),(2),(3); INSERT 0 3

3、生成执行计划时，stable函数被执行一次（指无动态参数的stable函数，有动态参数的，不执行）

```
postgres=# explain select * from test123 where id=ff();
NOTICE: a
QUERY PLAN

Seq Scan on test123 (cost=0.00..679.38 rows=13 width=4)
Filter: (id = ff())
(2 rows)
```

4、使用全表扫描，有多少记录就会被执行多少次，同时加上一次执行计划的评估，一共调用了4次

```
postgres=# select * from test123 where id=ff();
NOTICE: a
NOTICE: a
NOTICE: a
NOTICE: a
id

1
(1 row)
```

5、创建索引

postgres=# create index idx_test123 on test123(id); CREATE INDEX

6、强制使用索引扫描

postgres=# set enable_seqscan=off; SET

7、生成执行计划，调用一次

```
postgres=# explain select * from test123 where id=ff();
NOTICE: a
QUERY PLAN

Index Only Scan using idx_test123 on test123 (cost=0.38..2.60 rows=1 width=4)
Index Cond: (id = ff())
(2 rows)
```

8、执行+执行计划，分别调用一次

```
postgres=# select * from test123 where id=ff();
NOTICE: a 第一次生成执行计划
NOTICE: a 第二次是索引扫描前的函数值计算
id

1
(1 row)
```

9、新增一条数据

postgres=# insert into test123 values (1); INSERT 0 1

10、符合条件的数据有2条，不影响索引扫描时，无参数或无动态参数的stable函数的调用次数

```
postgres=# select * from test123 where id=ff();
NOTICE: a
NOTICE: a
id

1
1
(2 rows)
```

```
postgres=# select * from test123 where id>ff();
NOTICE: a
NOTICE: a
id

2
3
(2 rows)

postgres=# select * from test123 where id>=ff();
NOTICE: a
NOTICE: a
id

1
1
2
3
(4 rows)

postgres=# explain analyze select * from test123 where id>ff();
NOTICE: a
NOTICE: a
QUERY PLAN

Index Scan using idx_test123 on test123 (cost=0.38..2.60 rows=1 width=4) (actual time=0.067..0.070 rows=2 loops=1)
Index Cond: (id > ff())
Planning time: 0.185 ms
Execution time: 0.127 ms
(4 rows)
```

11、创建一个带参数的stable函数

create or replace function ff(int) returns int as $$ declare begin raise notice 'a'; return 1; end; $$ language plpgsql strict stable;

12、插入10条重复值

postgres=# insert into test123 select 1 from generate_series(1,10); INSERT 0 10

13、使用静态参数输入时，可以走索引，所以依旧只调用2次。

```
postgres=# select * from test123 where id>ff(1);
NOTICE: a 执行计划
NOTICE: a 索引扫描前固定stable函数值
id

2
3
(2 rows)
```

```
postgres=# explain select * from test123 where id>ff(1);
NOTICE: a
QUERY PLAN

Index Scan using idx_test123 on test123 (cost=0.38..2.60 rows=1 width=4)
Index Cond: (id > ff(1))
(2 rows)
```

14、改成动态参数，那么就不能走索引扫描，调用次数等于记录数，因为explain时也不会被调用

调用了14次。

```
postgres=# explain analyze select * from test123 where id>ff(id);
NOTICE: a
NOTICE: a
NOTICE: a
NOTICE: a
NOTICE: a
NOTICE: a
NOTICE: a
NOTICE: a
NOTICE: a
NOTICE: a
NOTICE: a
NOTICE: a
NOTICE: a
NOTICE: a
QUERY PLAN

Seq Scan on test123 (cost=10000000000.00..10000000001.79 rows=1 width=4) (actual time=0.085..0.159 rows=2 loops=1)
Filter: (id > ff(id))
Rows Removed by Filter: 12
Planning time: 0.085 ms
Execution time: 0.197 ms
(5 rows)

postgres=# select * from test123 where id>ff(id);
NOTICE: a
NOTICE: a
NOTICE: a
NOTICE: a
NOTICE: a
NOTICE: a
NOTICE: a
NOTICE: a
NOTICE: a
NOTICE: a
NOTICE: a
NOTICE: a
NOTICE: a
NOTICE: a
id

2
3
(2 rows)

postgres=# select count(*) from test123;
count

(1 row)
```