产生PLPGSQL唯一的随机字符串(Generate unique random strings i

2019-09-29 18:44发布

我试图写一个函数来创建可变长度的唯一的随机令牌。 不过,我对PLPGSQL语法难住了。 我的目的是建立一个功能,

  • 以表和列作为输入
  • 产生给定长度的随机串,与给定的字符集
  • 检查字符串是否已经在科拉姆
  • 如果是的话(这预计将是罕见的),简单地生成一个新的随机字符串。
  • 否则,返回的随机字符串

我现在的尝试是这样的:

CREATE FUNCTION random_token(_table TEXT, _column TEXT, _length INTEGER) RETURNS text AS $$
DECLARE
  alphanum CONSTANT text := 'abcdefghijkmnopqrstuvwxyz23456789';
  range_head CONSTANT integer := 25;
  range_tail CONSTANT integer := 33;
  random_string text;
BEGIN
  REPEAT
    SELECT substring(alphanum from trunc(random() * range_head + 1)::integer for 1) ||
      array_to_string(array_agg(substring(alphanum from trunc(random() * range_tail + 1)::integer for 1)), '')
      INTO random_string FROM generate_series(1, _length - 1);
  UNTIL random_string NOT IN FORMAT('SELECT %I FROM %I WHERE %I = random_string;', _column, _table, _column)
  END REPEAT;
  RETURN random_string;
END
$$ LANGUAGE plpgsql;

然而,这并不工作,并给了我一个不是非常有帮助的错误:

DatabaseError:错误“ERROR:在语法错误或接近 “REPEAT”

我已经尝试了许多变化,但不知道什么语法错误是我难倒。 任何想法如何解决这个功能呢?

Answer 1:

有没有repeat的PLPGSQL声明。 使用简单loop

CREATE OR REPLACE FUNCTION random_token(_table TEXT, _column TEXT, _length INTEGER) RETURNS text AS $$
DECLARE
  alphanum CONSTANT text := 'abcdefghijkmnopqrstuvwxyz23456789';
  range_head CONSTANT integer := 25;
  range_tail CONSTANT integer := 33;
  random_string text;
  ct int;
BEGIN
  LOOP
    SELECT substring(alphanum from trunc(random() * range_head + 1)::integer for 1) ||
      array_to_string(array_agg(substring(alphanum from trunc(random() * range_tail + 1)::integer for 1)), '')
      INTO random_string FROM generate_series(1, _length - 1);
    EXECUTE FORMAT('SELECT count(*) FROM %I WHERE %I = %L', _table, _column, random_string) INTO ct;
    EXIT WHEN ct = 0;
  END LOOP;
  RETURN random_string;
END
$$ LANGUAGE plpgsql;

注意, random_string应该是一个参数format()

更新。 据Abelisto准确的提示,这应该是一个大的表快:

DECLARE
  dup boolean;
...
    EXECUTE FORMAT('SELECT EXISTS(SELECT 1 FROM %I WHERE %I = %L)', _table, _column, random_string) INTO dup;
    EXIT WHEN NOT dup;
...


Answer 2:

这是几乎可以肯定不是你想要的。 当你说,“检查字符串已经在列”你是不是指一些看起来独一无二的,你指的是什么,实际上是UNIQUE

相反,我会点你在这个答案我给了有关的UUID 。



文章来源: Generate unique random strings in plpgsql