在 SELECT 语句中使用 UDF

Posted 2023-04-13

【中文标题】在 SELECT 语句中使用 UDF

【英文标题】：Using UDF in SELECT statement

【发布时间】：2012-06-25 01:54:28

【问题描述】：

我为营业时间计算做了一个用户定义的函数。

这是我的 UDF。

CREATE FUNCTION  fn_GetBusinessHour (@date datetime, @addHours int)
RETURNS datetime
AS
BEGIN 
    DECLARE @CalcuatedDate datetime;
    DECLARE @addDayCount int, @addHourCount int, @addMinCount int;

    SET @addDayCount = @addHours / 8.5;
    SET @addHourCount = @addHours - (@addDayCount * 8.5);
    SET @addMinCount =  @addHours - (@addDayCount * 8.5) - @addHourCount;

    IF(@addDayCount != 0) 
        SET @CalcuatedDate = DATEADD(DD, @addDayCount, @date); 

    SET @CalcuatedDate = DATEADD(HH, @addHourCount, @CalcuatedDate);

    IF(@addMinCount != 0) 
        SET @CalcuatedDate = DATEADD(MM, @addMinCount, @CalcuatedDate); 

    RETURN @CalcuatedDate;
END

当我使用以下语句进行测试时，

SELECT dbo.fn_GetBusinessHour(GETDATE(), 40)

它显示了正确的结果。

但是，我是这样使用我的函数的，

SELECT TicketID
     , DateTimeLogged --Type: Datetime
     , Priority       --Type: int
     , [dbo].[fn_GetBusinessHour](DateTimeLogged, Priority)
  FROM TicketHeader

结果只显示 NULL 值。

TicketID    DateTimeLogged  Priority    (No column name)
1   2011-07-04 11:26:19.510     30  NULL
2   2011-07-04 13:58:45.683     30  NULL
3   2011-07-05 10:09:16.923     10  NULL
4   2011-07-05 13:13:30.237     30  NULL
5   2011-07-05 16:50:34.033     20  NULL

我尝试了 CONVERT，因为它在我给出值 40 时有效，但它也显示空值。

SELECT TicketID
         , DateTimeLogged --Type: Datetime
         , Priority       --Type: int
         , [dbo].[fn_GetBusinessHour](DateTimeLogged, CONVERT(int, Priority))
      FROM TicketHeader

如何解决此问题以使用我的 UDF？ 为什么会发生这种事情？ 我不明白 Priority 和 40 之间有什么不同。

提前谢谢你。

【参考方案1】：

对于优先级 > 8.5 的值，这对我来说似乎很好：

DECLARE @t TABLE(TicketID INT, DateTImeLogged DATETIME, Priority INT);

INSERT @t SELECT 1,'20110704 11:26:19.510',30
UNION ALL SELECT 2,'20110704 13:58:45.683',30
UNION ALL SELECT 3,'20110705 10:09:16.923',10
UNION ALL SELECT 4,'20110705 13:13:30.237',30
UNION ALL SELECT 5,'20110705 16:50:34.033',20;

SELECT TicketID
     , DateTimeLogged --Type: Datetime
     , Priority       --Type: int
     , [dbo].[fn_GetBusinessHour](DateTimeLogged, Priority)
  FROM @t;

产量：

TicketID  DateTimeLogged           Priority  (No column name)
--------  -----------------------  --------  -----------------------
1         2011-07-04 11:26:19.510  30        2011-07-07 15:26:19.510
2         2011-07-04 13:58:45.683  30        2011-07-07 17:58:45.683
3         2011-07-05 10:09:16.923  10        2011-07-06 11:09:16.923
4         2011-07-05 13:13:30.237  30        2011-07-08 17:13:30.237
5         2011-07-05 16:50:34.033  20        2011-07-07 19:50:34.033

如果我添加另一行优先级

INSERT @t SELECT 6,'20110705 13:13:30.237',5;

然后将这一行添加到结果中：

TicketID  DateTimeLogged           Priority  (No column name)
--------  -----------------------  --------  -----------------------
6         2011-07-05 13:13:30.237  5         NULL

换句话说，如果函数逻辑未分配@CalculatedDate，则函数将输出NULL，如果@addDayCount = 0，则会发生这种情况。在你说的函数中：

IF(@addDayCount != 0) 
    SET @CalcuatedDate = DATEADD(DD, @addDayCount, @date); 

由于 @addDayCount 是一个 INT，试试这个：

DECLARE @addDayCount INT;
SET @addDayCount = 5 / 8.5;
SELECT @addDayCount;

结果：

0

因此，由于最初没有为 @CalculatedDate 分配值，因此以下所有 DATEADD 操作都在执行 DATEADD(interval, number, NULL) 仍然产生 NULL。

所以也许你需要为函数中的变量使用不同的数据类型...

【讨论】：

非常感谢@aaronb。终于我知道了。我添加一行 DECLARE CalcuatedDate = date。它防止 DATEADD(interval, number, NULL)。再次感谢您。