1. 程式人生 > 資料庫 >sql server使用公用表表達式CTE通過遞迴方式編寫通用函式自動生成連續數字和日期

sql server使用公用表表達式CTE通過遞迴方式編寫通用函式自動生成連續數字和日期

問題:

在資料庫指令碼開發中,有時需要生成一堆連續數字或者日期,例如yearly report就需要連續數字做年份,例如daily report就需要生成一定時間範圍內的每一天日期。

而自帶的系統表master..spt_values存在一定的侷限性,只是從0到2047(驗證指令碼:select * from master..spt_values b where b.type = 'P'),也不能直接生成連續日期。

可能大部分人會想到一個笨辦法,通過while迴圈去逐條插入資料到臨時表,每次數字加1或者日期加1天,但這樣和資料庫伺服器的互動就太頻繁了。如果生成1W個連續數字,那就要跟資料庫伺服器互動1W次,可怕!如果是有1000個客戶端都需要呼叫這個while迴圈,那就是1000W次!可怕!

解決方案:

可以使用公用表表達式CTE通過遞迴方式實現,並編寫為一個通用表值函式方便呼叫,封裝起來簡化使用,返回表格式資料。

CTE是在記憶體中準備好資料,而不是每次一條往返伺服器和客戶端一次。如果需要再插入到臨時表的話就是全部資料一次性插入。

如果傳入引數為數字,則生成連續數字;如果傳入引數為日期,則生成連續日期。
是不是覺得很方便呢?

函式指令碼:

if object_id('dbo.fun_ConcatStringsToTable') is not null drop function dbo.fun_ConcatStringsToTable
go
/*
  功能:連續字串(數字或日期)以table形式返回
  作者:zhang502219048 2018-12-10
  指令碼來源:https://www.cnblogs.com/zhang502219048/p/11108991.html 
-- 示例1(數字):
  select * from dbo.fun_ConcatStringsToTable(1,10000)
-- 示例2(數字文字):
  select * from dbo.fun_ConcatStringsToTable('1','10000')
-- 示例3(日期):
  declare @dateBegin datetime = '2009-1-1',@dateEnd datetime = '2018-12-31'
  select * from dbo.fun_ConcatStringsToTable(@dateBegin,@dateEnd)
-- 示例4(日期文字):
  select * from dbo.fun_ConcatStringsToTable('2009-1-1','2018-12-31')
**/
create function [dbo].[fun_ConcatStringsToTable]
(
  @strBegin as nvarchar(100),@strEnd as nvarchar(100)
)
returns @tempResult table (vid nvarchar(100))
as
begin
  --數字
  if isnumeric(@strBegin) = 1 and isnumeric(@strEnd) = 1
  begin
    --使用CTE遞迴批量插入數字資料
    ;with cte_table(id) as
    (
      select cast(@strBegin as int)
      union all
      select id + 1
      from cte_table
      where id < @strEnd
    )
    insert into @tempResult
    select cast(id as nvarchar(100))
    from cte_table 
    option (maxrecursion 0)
  end
  --日期
  else if isdate(@strBegin) = 1 and isdate(@strEnd) = 1
  begin
    --使用CTE遞迴批量插入日期資料
    ;with cte_table(CreatedDate) as
    (
      select cast(@strBegin as datetime)
      union all
      select dateadd(day,1,CreatedDate)
      from cte_table
      where CreatedDate < @strEnd
    )
    insert into @tempResult
    select convert(varchar(10),CreatedDate,120)
    from cte_table
    option (maxrecursion 0)
  end

  return;
end
go

呼叫函式示例:

-- 示例1(數字):
  select * from dbo.fun_ConcatStringsToTable(1,'2018-12-31')

指令碼執行結果:


結論:

從上面幾個圖可以看到,通過簡單呼叫fun_ConcatStringsToTable這個自定義表值函式,指定起止數字或日期,就達到了生成連續數字和日期的目的。

擴充套件:

如果想生成連續月份呢?博主在這裡也幫大家寫了一下指令碼,如果需要可以在此基礎上再自行做成表值函式:

with cte_table(CreatedDate) as
(
  select cast('2017-12-1' as datetime)
  union all
  select dateadd(month,CreatedDate)
  from cte_table
  where CreatedDate < '2018-04-01'
)
select convert(varchar(7),120) as YearMonth
from cte_table
option (maxrecursion 0)

總結

以上所述是小編給大家介紹的sql server使用公用表表達式CTE通過遞迴方式編寫通用函式自動生成連續數字和日期,希望對大家有所幫助,如果大家有任何疑問請給我留言,小編會及時回覆大家的。在此也非常感謝大家對我們網站的支援!如果你覺得本文對你有幫助,歡迎轉載,煩請註明出處,謝謝!