PostgreSQL的generate_series函数应用
一、简介
PostgreSQL 中有一个很有用处的内置函数generate_series，可以按不同的规则产生一系列的填充数据。
二、语法
| 函数 | 参数类型 | 返回类型 | 描述 |
|---|---|---|---|
| generate_series(start, stop) | int 或 bigint | setof int 或 setof bigint(与参数类型相同) | 生成一个数值序列，从start 到 stop，步进为一 |
| generate_series(start, stop, step) | int 或 bigint | setof int 或 setof bigint(与参数类型相同) | 生成一个数值序列，从start 到 stop，步进为step |
| generate_series(start, stop, step_interval) | timestamp or timestamp with time zone | timestamp 或 timestamp with time zone(same as argument type) | 生成一个数值序列，从start 到 stop，步进为step |
三、实例 型
a. 不写步进时默认为1
david=# select generate_series(1, 10);
generate_series
-----------------
1
2
3
4
5
6
7
8
9
10
(10 rows)
david=#
b. 设置步进
david=# select generate_series(1, 10, 3);
generate_series
-----------------
1
4
7
10
(4 rows)
david=#
c. 如果step 是正数，而start 大于stop，那么返回零行。相反，如果step 是负数，start 小于stop，则返回零行。如果是NULL 输入，也产生零行。step 为零则是一个错误。
david=# select generate_series(5,1); generate_series ----------------- (0 rows) david=#
NULL inputs
david=# select generate_series(5,null); generate_series ----------------- (0 rows) david=#
step 为零
david=# select generate_series(5,1,0); ERROR: step size cannot equal zero david=#
start 大于stop，step 是负数
david=# select generate_series(5,1,-1);
generate_series
-----------------
5
4
3
2
1
(5 rows)
david=#
3.2) 时间类型
david=# select generate_series(now(), now() + '7 days', '1 day');
generate_series
-------------------------------
2013-04-03 14:22:26.391852+08
2013-04-04 14:22:26.391852+08
2013-04-05 14:22:26.391852+08
2013-04-06 14:22:26.391852+08
2013-04-07 14:22:26.391852+08
2013-04-08 14:22:26.391852+08
2013-04-09 14:22:26.391852+08
2013-04-10 14:22:26.391852+08
(8 rows)
david=#
david=# select generate_series(to_date('20130403','yyyymmdd'), to_date('20130404','yyyymmdd'), '3 hours');
generate_series
------------------------
2013-04-03 00:00:00+08
2013-04-03 03:00:00+08
2013-04-03 06:00:00+08
2013-04-03 09:00:00+08
2013-04-03 12:00:00+08
2013-04-03 15:00:00+08
2013-04-03 18:00:00+08
2013-04-03 21:00:00+08
2013-04-04 00:00:00+08
(9 rows)
david=#
3.3) IP类型
a. 建表
david=# create table tbl_david(id int, ip_start inet, ip_stop inet); CREATE TABLE david=#
b. 插入数据
david=# insert into tbl_david values (1, '192.168.1.6', '192.168.1.10'); INSERT 0 1 david=# insert into tbl_david values (2, '192.168.2.16', '192.168.2.20'); INSERT 0 1 david=# insert into tbl_david values (3, '192.168.3.116', '192.168.3.120'); INSERT 0 1 david=#
c. 查看数据
david=# select * from tbl_david ; id | ip_start | ip_stop ----+---------------+--------------- 1 | 192.168.1.6 | 192.168.1.10 2 | 192.168.2.16 | 192.168.2.20 3 | 192.168.3.116 | 192.168.3.120 (3 rows) david=#
d. generate_series 生成序列
david=# select id, generate_series(0, ip_stop-ip_start)+ip_start as ip_new from tbl_david ; id | ip_new ----+--------------- 1 | 192.168.1.6 1 | 192.168.1.7 1 | 192.168.1.8 1 | 192.168.1.9 1 | 192.168.1.10 2 | 192.168.2.16 2 | 192.168.2.17 2 | 192.168.2.18 2 | 192.168.2.19 2 | 192.168.2.20 3 | 192.168.3.116 3 | 192.168.3.117 3 | 192.168.3.118 3 | 192.168.3.119 3 | 192.168.3.120 (15 rows) david=#
四、总结
PostgreSQL的generate_series函数对生成测试数据，批量更新一定规则的数据有比较多的应用场景，使用得当可提升开发效率，另外IP的序列生成也是PG的一个亮点。
五、参考
- PostgreSQL官方文档：http://www.postgresql.org/docs/9.2/static/functions-srf.html
- kenyon的个人页面：http://my.oschina.net/Kenyon/blog/75099