而SQL Server 2008中的循环遍历日期范围然后INSERT

时间:2015-03-18 18:14:08

标签: sql sql-server-2008

我有一个包含几列的表,其中一列是Timestamp列。但目前在这张表中并没有每天的记录。意思是,有1月1日和1月2日的记录,但在时间戳字段中没有1月3日或1月4日的记录。但是,有记录在1月5日和1月6日继续,依此类推。基本上,周末和其他随机日子都不见了。

我正在尝试编写一个脚本,将这个表从StartDate扫描到EndDate(我选择的任何日期范围),并迭代这个日期范围,如果一条记录不存在,其中包含任何日期date-range,在Timestamp字段中插入具有该特定日期的新记录,但为其余字段插入空/ NULL数据。

这是我到目前为止所获得的伪代码,我认为这是正确的方法:

DECLARE @StartDate AS DATETIME
DECLARE @EndDate AS DATETIME
DECLARE @CurrentDate AS DATETIME

SET @StartDate = '2015-01-01'
SET @EndDate = GETDATE()
SET @CurrentDate = @StartDate

WHILE (@CurrentDate < @EndDate)
BEGIN
    SELECT * FROM myTable WHERE myTable.Timestamp = "@CurrentDate"
    IF @@ROWCOUNT < 1
        print @CurrentDate
        /*insert a new row query here*/

    SET @CurrentDate = convert(varchar(30), dateadd(day,1, @CurrentDate), 101); /*increment current date*/
END

这是SQLFiddle - http://sqlfiddle.com/#!6/06c73/1

我正在SQL Server Management Studio 2008中编写我的第一个脚本,我认为可能是针对中间用户的。我是一名PHP / MySQL开发人员,对这些技术非常熟悉,但我是SQL和VBScript的新手。我理解编程概念和逻辑,但这似乎与我以前的不同。

我非常感谢所有的帮助和见解!

2 个答案:

答案 0 :(得分:19)

SQL是一种基于集合的语言,循环应该是最后的手段。因此,基于集合的方法将首先生成所需的所有日期并一次性插入它们,而不是一次循环插入一个日期。 Aaron Bertrand撰写了一篇关于生成没有循环的集合或序列的精彩系列文章:

第3部分特别相关,因为它涉及日期。

假设您没有日历表,则可以使用堆叠CTE方法生成开始日期和结束日期之间的日期列表。

DECLARE @StartDate DATE = '2015-01-01',
        @EndDate DATE = GETDATE();

WITH N1 (N) AS (SELECT 1 FROM (VALUES (1), (1), (1), (1), (1), (1), (1), (1), (1), (1)) n (N)),
N2 (N) AS (SELECT 1 FROM N1 AS N1 CROSS JOIN N1 AS N2),
N3 (N) AS (SELECT 1 FROM N2 AS N1 CROSS JOIN N2 AS N2)
SELECT TOP (DATEDIFF(DAY, @StartDate, @EndDate) + 1)
        Date = DATEADD(DAY, ROW_NUMBER() OVER(ORDER BY N) - 1, @StartDate)
FROM N3;

我已经跳过了一些关于它是如何工作的细节,因为它在链接文章中有所涉及,实质上它以10行的硬编码表开始,然后将此表与自身连接以获得100行(10 x 10)然后将这100行的表连接到自身以获得10,000行(此时我停止了,但如果您需要更多行,则可以添加更多联接)。

在每一步中,输出都是一个名为N的列,值为1(为了简单起见)。在定义如何生成10,000行的同时,我实际上告诉SQL Server仅使用TOP生成所需的数字以及开始日期和结束日期之间的差异 - TOP(DATEDIFF(DAY, @StartDate, @EndDate) + 1)。这避免了不必要的工作。我必须在差异中加1,以确保包括两个日期。

使用排名函数ROW_NUMBER()我为每个生成的行添加一个增量编号,然后将此增量编号添加到开始日期以获取日期列表。由于ROW_NUMBER()从1开始,我需要从中扣除1以确保包含开始日期。

然后,仅使用NOT EXISTS排除已存在的日期。我已将上述查询的结果包含在他们自己的CTE中,名为dates

DECLARE @StartDate DATE = '2015-01-01',
        @EndDate DATE = GETDATE();

WITH N1 (N) AS (SELECT 1 FROM (VALUES (1), (1), (1), (1), (1), (1), (1), (1), (1), (1)) n (N)),
N2 (N) AS (SELECT 1 FROM N1 AS N1 CROSS JOIN N1 AS N2),
N3 (N) AS (SELECT 1 FROM N2 AS N1 CROSS JOIN N2 AS N2),
Dates AS
(   SELECT TOP (DATEDIFF(DAY, @StartDate, @EndDate) + 1)
            Date = DATEADD(DAY, ROW_NUMBER() OVER(ORDER BY N) - 1, @StartDate)
    FROM N3
)
INSERT INTO MyTable ([TimeStamp])
SELECT  Date
FROM    Dates AS d
WHERE NOT EXISTS (SELECT 1 FROM MyTable AS t WHERE d.Date = t.[TimeStamp])

<强> Example on SQL Fiddle


如果您要创建日历表(如链接文章中所述),则可能没有必要插入这些额外的行,您可以动态生成结果集,如:

SELECT  [Timestamp] = c.Date,
        t.[FruitType],
        t.[NumOffered],
        t.[NumTaken],
        t.[NumAbandoned],
        t.[NumSpoiled]
FROM    dbo.Calendar AS c
        LEFT JOIN dbo.MyTable AS t
            ON t.[Timestamp] = c.[Date]
WHERE   c.Date >= @StartDate
AND     c.Date < @EndDate;

<强>附录

要回答您的实际问题,您的循环将按如下方式编写:

DECLARE @StartDate AS DATETIME
DECLARE @EndDate AS DATETIME
DECLARE @CurrentDate AS DATETIME

SET @StartDate = '2015-01-01'
SET @EndDate = GETDATE()
SET @CurrentDate = @StartDate

WHILE (@CurrentDate < @EndDate)
BEGIN
    IF NOT EXISTS (SELECT 1 FROM myTable WHERE myTable.Timestamp = @CurrentDate)
    BEGIN
        INSERT INTO MyTable ([Timestamp])
        VALUES (@CurrentDate);
    END

    SET @CurrentDate = DATEADD(DAY, 1, @CurrentDate); /*increment current date*/
END

Example on SQL Fiddle

我不提倡这种做法,只是因为有些事情只是做了一次,并不意味着我不应该表现出正确的做法。


进一步解释

由于堆叠CTE方法可能使基于集合的方法过于复杂,因此我将使用未记录的系统表master..spt_values来简化它。如果您运行:

SELECT Number
FROM master..spt_values
WHERE Type = 'P';

您将看到从0到-2047获得所有数字。

现在如果你跑:

DECLARE @StartDate DATE = '2015-01-01',
        @EndDate DATE = GETDATE();


SELECT Date = DATEADD(DAY, number, @StartDate)
FROM master..spt_values
WHERE type = 'P';

您将获得从开始日期到将来的2047天的所有日期。如果您添加更多where子句,则可以将其限制为结束日期之前的日期:

DECLARE @StartDate DATE = '2015-01-01',
        @EndDate DATE = GETDATE();


SELECT Date = DATEADD(DAY, number, @StartDate)
FROM master..spt_values
WHERE type = 'P'
AND DATEADD(DAY, number, @StartDate) <= @EndDate;

现在,您可以在基于单个集的查询中获得所需的所有日期,您可以使用NOT EXISTS消除表中已存在的行

DECLARE @StartDate DATE = '2015-01-01',
        @EndDate DATE = GETDATE();


SELECT Date = DATEADD(DAY, number, @StartDate)
FROM master..spt_values
WHERE type = 'P'
AND DATEADD(DAY, number, @StartDate) <= @EndDate
AND NOT EXISTS (SELECT 1 FROM MyTable AS t WHERE t.[Timestamp] = DATEADD(DAY, number, @StartDate));

最后,您可以使用INSERT

将这些日期插入到表格中
DECLARE @StartDate DATE = '2015-01-01',
        @EndDate DATE = GETDATE();

INSERT YourTable ([Timestamp])
SELECT Date = DATEADD(DAY, number, @StartDate)
FROM master..spt_values
WHERE type = 'P'
AND DATEADD(DAY, number, @StartDate) <= @EndDate
AND NOT EXISTS (SELECT 1 FROM MyTable AS t WHERE t.[Timestamp] = DATEADD(DAY, number, @StartDate));

希望这可以说明基于集合的方法不仅效率更高,而且更简单。

答案 1 :(得分:0)

我使用的一个简单的解决方案,没有像 GarethD 这样的经验,

将假设您的表确实有以下数据

DECLARE @t TABLE(Dt Date,Name varchar(10))
INSERT INTO @t VALUES
('2021-05-18','hi'),('2021-05-20','heloo'),('2021-05-25','welocme'),('2021-05-27','goto')



DECLARE @startDate DATE, @endDate DATE
Set @startDate = '2021-05-18'
set @endDate = GETDATE()

;WITH Calender AS (
    SELECT @startDate AS YourDate
    UNION ALL
    SELECT DATEADD(day,1,YourDate) FROM Calender
    WHERE DATEADD(day,1,YourDate) <= @endDate
)
INSERT INTO @t SELECT
    Dt = YourDate,Name = DATENAME (WEEKDAY,YourDate)

FROM Calender c
LEFT JOIN @t t 
ON t.Dt = c.YourDate
WHERE t.dt IS NULL
option (maxrecursion 0)

SELECT * FROM @t ORDER BY dt