PostgreSQL 生成随机数字、字符串、日期、验证码以及 UUID

作者: 不剪发的Tony老师
毕业于北京航空航天大学,十多年数据库管理与开发经验,目前在一家全球性的金融公司从事数据库架构设计。CSDN学院签约讲师以及GitChat专栏作者。csdn上的博客收藏于以下地址:https://tonydong.blog.csdn.net

文章目录

        生成随机数字
            生成 0 到 1 之间的随机数
            生成指定范围内的随机数
            生成 6 位数字手机验证码
            生成遵循正态分布的随机数
        生成随机字符串
            生成固定长度的随机字符串
            生成可变长度的随机字符串
        生成随机日期和时间
        获取表中的随机记录
        生成 UUID
        总结

大家好,我是只谈技术不剪发的 Tony 老师。今天我们介绍一下如何在 PostgreSQL 中生成随机数据,包括随机数字、验证码、随机字符串以及随机日期和时间等。

生成随机数字
生成 0 到 1 之间的随机数

PostgreSQL 中的RANDOM()函数可以用于生成一个大于等于 0 小于 1 的随机数字。例如:

SELECT random();
random            |
------------------|
0.5573292311275964|

SELECT random();
random              |
--------------------|
0.017827744704202786|

该函数返回的数据类型为 double precision,每次调用都会返回不同的结果。

如果我们想要重现某个结果,需要生成相同的随机数;这种情况下可以使用SETSEED(d)函数设置一个随机数种子,d 的类型为 double precision,取值范围从 -1.0 到 1.0。例如:

SELECT setseed(0);

SELECT random();
random            |
------------------|
0.0000000000000391|

random            |
------------------|
0.0009853946746503|
...

SELECT setseed(0);

SELECT random();
random            |
------------------|
0.0000000000000391|

random            |
------------------|
0.0009853946746503|

从结果可以看出,设置相同的种子之后,随后的函数调用返回了一系列相同的随机数。
生成指定范围内的随机数

基于 RANDOM() 函数和一些运算,就可以返回任意两个数字之间的随机数:

low + RANDOM() * (high - low)

以上表达式将会返回一个大于等于 low,小于 high 的随机数。例如:

SELECT 10 + random() * 10 AS rd;
rd                |
------------------|
15.680596127871453|

以上示例返回了一个大于等于 10 且小于 20 的随机数字。

如果想要生成某个范围内的随机整数,可以加上 FLOOR 函数。例如:

SELECT floor(10 + random() * 10);
floor|
-----|
 12.0|

该语句返回了一个大于等于 10,小于等于 19(不是 20)的随机整数。
生成 6 位数字手机验证码

我们已经知道了如何获取指定范围内的随机数,再加上 TO_CHAR 格式化函数就可以生成由 6 位数字字符组成的手机验证码。例如:

SELECT to_char(random() * 1000000, '099999') AS captcha;
captcha|
-------|
 076774|

TO_CHAR 函数可以确保数据不够 6 位时在前面补足 0。
生成遵循正态分布的随机数

PostgreSQL 提供了一个扩展模块 tablefunc,可以用于生成遵循正态分布(normal distribution)的随机数。首先,输入以下命令启用该模块:

CREATE EXTENSION tablefunc;

然后使用该模块中的NORMAL_RAND(n,mean, stddev)函数返回 n 个均值为 mean,标准差为 stddev 的随机数。例如:

SELECT *
FROM normal_rand(10, 0, 1);
normal_rand         |
--------------------|
  0.0936639131151394|
   -1.26936035550923|
   2.006729235590952|
  0.7869592803653096|
 -1.5740650326039192|
-0.18656503408337746|
  1.0665080022417979|
 -1.1240167023021148|
  1.1073155396442795|
 0.09360901134478303|

以上查询返回了 10 个遵循标准正态分布的随机数。

我们也可以验证一下该函数是否遵循正态分布,例如:

SELECT count(*), avg(v), stddev(v)
FROM normal_rand(1000000, 0, 1) AS v;
count  |avg               |stddev            |
-------|------------------|------------------|
1000000|0.0001662571158423|0.9992607627843408|

另一种方法就创建一个存储函数来模拟正态分布的随机数:

CREATE OR REPLACE FUNCTION normal_distrib(mean DOUBLE PRECISION, stdev DOUBLE PRECISION)
RETURNS DOUBLE PRECISION
LANGUAGE plpgsql
AS $$
DECLARE
    x DOUBLE PRECISION;
    y DOUBLE PRECISION;
    rd DOUBLE PRECISION;
BEGIN
    x := random();
    y := random();
    rd = (sqrt(-2 * ln(x)) * cos(2 * pi() * y)) * stdev + mean;

    RETURN rd;

END $$;

该函数利用 Box-Muller 变换算法通过两个平均分布的随机数生成正态分布的随机数。以下语句通过 normal_distrib 函数生成了一个遵循正态分布的随机数:

SELECT normal_distrib(0,1);
normal_distrib   |
-----------------|
0.404847649020953|

以下语句可以用于验证 normal_distrib 函数是否遵循正态分布:

WITH RECURSIVE d(n, v) AS (
  SELECT 1 AS n, normal_distrib(0, 1) AS v
  UNION ALL
  SELECT n+1, normal_distrib(0, 1) FROM d WHERE n<1000000
)
count  |avg                  |stddev           |
-------|---------------------|-----------------|
1000000|-0.001243494839949032|0.999320444731066|

经过测试,自定义的 normal_distrib 函数执行时间为 12.5 s,normal_rand 函数只需要 1.5 s。
生成随机字符串
生成固定长度的随机字符串

除了随机数字之外,有时候我们也需要生成一些随机的字符串。PostgreSQL 没有提供专门生成随机字符串的函数,但是可以通过其他函数进行模拟。例如:

SELECT chr(floor(random() * 26)::integer + 65);
chr|
---|
V  |

以上查询返回了一个随机的大写字母,chr 函数用于将 ASCII 码转换为对应的字符。我们可以基于该查询进一步创建一个存储函数:

CREATE OR REPLACE FUNCTION random_string(
  num INTEGER,
  chars TEXT default '0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz'
) RETURNS TEXT
LANGUAGE plpgsql
AS $$
DECLARE
  res_str TEXT := '';
BEGIN
  IF num < 1 THEN
      RAISE EXCEPTION 'Invalid length';
  END IF;
  FOR __ IN 1..num LOOP
    res_str := res_str || substr(chars, floor(random() * length(chars))::int + 1, 1);
  END LOOP;
  RETURN res_str;
END $$;

random_string 函数可以返回由指定字符(默认为所有数字、大小写字母)组成的随机字符串。例如:

SELECT random_string(10);
random_string|
-------------|
etP3odkRgA   |

以上示例返回了一个由字母和数字组成、长度为 10 的随机字符串。以下语句也可以用于返回一个 6 位随机数字组成的手机验证码:

SELECT random_string(6, '0123456789');
random_string|
-------------|
082661       |

生成可变长度的随机字符串

那么,怎么返回一个长度可变的随机字符串呢?很简单,为 random_string 函数指定一个随机的长度参数即可。例如:

SELECT random_string(floor(10 + random() * 11)::int);
random_string   |
----------------|
8tz5zHcbKVKoVg4S|

以上示例返回了一个长度大于等于 10 且小于等于 20 的随机字符串。
生成随机日期和时间

将指定日期增加一个随机的数字,就可以得到随机的日期。例如:

SELECT current_date + floor((random() * 15))::int rand_date;
rand_date |
----------|
2020-11-04|

以上示例返回了当前日期 14 天之内的某个随机日期。以下语句则返回了一天中的某个随机时间:

SELECT make_time(floor((random() * 12))::int, floor((random() * 60))::int, floor((random() * 60))::int) AS rand_time;
rand_time|
---------|
 10:04:52|

其中,make_time 函数用于将代表时、分、秒的整数转换为时间。
获取表中的随机记录

对于返回多行数据的查询语句,RANDOM 函数每次都会返回不同的随机值。例如:

SELECT random() FROM employee;
random             |
-------------------|
0.10449782906204419|
 0.3345344734009643|
 0.7295074473683592|
...

利用这个特性,我们可以从表中返回随机的数据行。例如:

SELECT emp_id, emp_name
FROM employee
ORDER BY random()
LIMIT 5;
emp_id|emp_name |
------|---------|
     2|关羽      |
     9|赵云      |
    13|关兴      |
    25|孙乾      |
    17|马岱      |

以上示例从 employee 表中返回了 5 行随机记录。该方法需要为表中的每行数据都生成一个随机数,然后进行排序;所以会随着表中的数据量增加而逐渐变慢。

如果表中存在自增主键,也可以基于主键生成一个随机数据。例如:

SELECT round(rand() * (SELECT max(emp_id) FROM employee)) AS id;
id  |
----|
10.0|

然后基于这个随机数返回一条随机的记录:

SELECT e.emp_id, e.emp_name
FROM employee e
INNER JOIN (SELECT round(random() * (SELECT max(emp_id) FROM employee)) AS id
     ) AS t
ON e.emp_id >= t.id
LIMIT 1;
emp_id|emp_name|
------|--------|
    10|廖化     |

这种方法一次只能返回一条随机记录,而且只有当自增字段的值没有间隙时才会返回均匀分布的随机记录。

另外,PostgreSQL 中的查询语句支持 TABLESAMPLE 子句,可以实现数据的抽样。例如:

SELECT emp_id, emp_name
FROM employee
TABLESAMPLE BERNOULLI (10);
emp_id|emp_name|
------|--------|
     4|诸葛亮   |
    13|关兴     |

除了 BERNOULLI 之外,也可以指定 SYSTEM 抽样方法,参数代表了抽样近似百分比。
生成 UUID

UUID(Universal Unique Identifier)或者 GUID(Globally Unique Identifier)是一个 128 比特的数字,可以用于唯一标识每个网络对象或资源。由于它的生成机制,一个 UUID 可以保证几乎不会与其他 UUID 重复,因此常常用于生成数据库中的主键值。

PostgreSQL 提供了一个用于加/解密的扩展模块 pgcrypto,其中的 gen_random_uuid() 函数可以用于返回一个 version 4 的随机 UUID。首先,输入以下命令启用该模块(gen_random_uuid() 从 PostgreSQL 13 开始成为了一个内置函数):

CREATE EXTENSION pgcrypto;

然后,通过该函数返回一个 UUID:

SELECT gen_random_uuid();
gen_random_uuid                     |
------------------------------------|
2d757cf5-c18c-469c-8b5e-eed914eacc93|

该函数返回的数据类型为 uuid。如果想要生成没有中划线(-)的 UUID 字符串,可以使用 REPLACE 函数:

SELECT replace(gen_random_uuid()::text,'-','');
replace                         |
--------------------------------|
cabbfcdc62c54e2889bdd2b7095f1270|

总结

本文介绍了在 PostgreSQL 中生成随机数据的方法,包括随机数字、验证码、随机字符串以及随机日期和时间等,同时还介绍了如何从表中返回随机记录,以及如何生成 UUID。