拆分给定字符串并准备 case 语句

Question

表：表名

create table table_name
(
given_dates timestamp,
set_name varchar
);

插入记录:

insert into table_name values('2001-01-01'),('2001-01-05'),('2001-01-10'),
                 ('2001-01-15'),('2001-01-20'),('2001-01-25'),
                 ('2001-02-01'),('2001-02-05'),('2001-02-10'),
                 ('2001-02-15');

现在我想更新某些日期的 set_name 。

例如：

我想像这样更新表格：

given_dates    set_name 
----------------------
2001-01-01      s1
2001-01-05      s1
2001-01-10      s2
2001-01-15      s2
2001-01-20
2001-01-25
2001-02-01
2001-02-05
2001-02-10
2001-02-15

注意：

given_dates

和

set_name

是传递参数的，因为它们是动态的。我可能会通过2组 如上图

s1,s2

或者根据要求可以通过4组。

所以我需要动态 case 语句来更新

set_name

。

给定两个参数：

declare p_dates varchar := '2001-01-01to2001-01-05,2001-01-10to2001-01-15';

declare p_sets varchar := 's1,s2';

我可以使用以下静态脚本来做到这一点：

静态更新声明:

update table_name
SET set_name = 
CASE  
when given_dates between '2001-01-01' and '2001-01-05' then 's1'
when given_dates between '2001-01-10' and '2001-01-15' then 's2'
else '' 
end;

上面的更新语句完成了工作，但是是静态的。

就像更新表一样，我只想准备 case 语句，该语句应该是动态的，可以根据参数

(p_dates,p_sets)

更改而更改。

问题：

1. 如何分割给定的日期

   p_dates

   ？ （我在两个日期之间有

   to

   关键字。）
2. 如何分割给定的

   p_sets

   集合？ （我在两个 set_name 之间有“,”逗号。）
3. 拆分

   p_dates

   和

   p_sets

   后如何准备动态case语句？

---

这个问题与 使用 SQL Server 2008 R2 的动态 case 语句有关，这与 Microsoft SQL Server 是相同的。

Answer 1

清洁设置：

CREATE TABLE tbl (
  given_date date
, set_name varchar
);

使用单数术语作为单个值的列名称。
数据类型显然是

date

而不是

timestamp

。

将文本参数转换为有用的表格：

SELECT unnest(string_to_array('2001-01-01to2001-01-05,2001-01-10to2001-01-15', ',')) AS date_range
     , unnest(string_to_array('s1,s2', ',')) AS set_name;

“并行解除嵌套”很方便，但有其注意事项。 Postgres 9.4 添加了一个干净的解决方案，Postgres 10 最终净化了它的行为。见下文。

动态执行

准备好的声明

准备好的语句仅对创建会话可见并随之消失。 根据文档：

准备好的语句仅在当前数据库会话期间持续。

PREPARE

每节一次：

PREPARE upd_tbl AS
UPDATE tbl t
SET    set_name = s.set_name
FROM  (
   SELECT unnest(string_to_array($1, ',')) AS date_range
        , unnest(string_to_array($2, ',')) AS set_name
   ) s
WHERE t.given_date BETWEEN split_part(date_range, 'to', 1)::date
                       AND split_part(date_range, 'to', 2)::date;

或使用客户提供的工具来准备声明。
使用任意参数执行n次：

EXECUTE upd_tbl('2001-01-01to2001-01-05,2001-01-10to2001-01-15', 's1,s4');

服务器端功能

功能是持久的并且对所有会话可见。

CREATE FUNCTION

一次:

CREATE OR REPLACE FUNCTION f_upd_tbl(_date_ranges text, _names text)
  RETURNS void
  LANGUAGE sql AS
$func$
UPDATE tbl t
SET    set_name = s.set_name
FROM  (
   SELECT unnest(string_to_array($1, ',')) AS date_range
        , unnest(string_to_array($2, ',')) AS set_name
   ) s
WHERE  t.given_date BETWEEN split_part(date_range, 'to', 1)::date
                        AND split_part(date_range, 'to', 2)::date
$func$;

拨打n次：

SELECT f_upd_tbl('2001-01-01to2001-01-05,2001-01-20to2001-01-25', 's2,s5');

_旧sqlfiddle

卓越的设计

使用数组参数（仍然可以作为字符串文字提供）、

daterange

类型（均为第9.3页）和新并行

unnest()

（第9.4页）。

CREATE OR REPLACE FUNCTION f_upd_tbl(_dr daterange[], _n text[])
  RETURNS void
  LANGUAGE sql AS
$func$
UPDATE tbl t
SET    set_name = s.set_name
FROM   unnest($1, $2) s(date_range, set_name)
WHERE  t.given_date <@ s.date_range
$func$;

<@

是“元素包含于”运算符。

致电：

SELECT f_upd_tbl('{"[2001-01-01,2001-01-05]"
                  ,"[2001-01-20,2001-01-25]"}', '{s2,s5}');

详情：

• 并行解除多个数组的嵌套

Answer 2

字符串到数组

declare p_dates varchar[] := string_to_array('2001-01-01,2001-01-05,
2001-01-10,2001-01-15*2001-01-01,2001-01-05,2001-01-10,2001-01-15','*');
declare p_sets varchar[]  := string_to_array('s1,s2',',');
declare p_length integer=0;
declare p_str  varchar[];
declare i integer;
select array_length(p_dates ,1) into p_count;

for i in 1..p_count loop

  p_str  := string_to_array( p_dates[i],',')   

  execute 'update table_name
  SET set_name = 
  CASE  
  when given_dates between'''|| p_str  [1] ||''' and '''|| p_str  [2] 
  ||''' then ''' || p_sets[1] ||'''
  when given_dates between '''|| p_str  [3] ||''' and '''
  || p_str  [4] ||''' then ''' || p_sets[2] ||'''
  else '''' 
  end';
end loop;

Answer 3

现在我们可以使用

datemultirange

。

create or replace function f_upd_tbl_multirange(_dr datemultirange , _n text[])
    returns void as
$func$
UPDATE tbl t
SET    set_name = s.set_name
FROM   unnest($1,$2) s(date_range,set_name)
WHERE  t.given_date <@ s.date_range
$func$ language sql;

运行它。

SELECT f_upd_tbl_multirange(
    '{[''2022-01-01'',''2022-01-05''],[''2022-02-06'',''2022-02-25'']}', '{s2,s5}');