SQL對表值函數中的臨時表中的總行進行計數

Question

如果@total等於1，我需要在@t插入的@total 。

我怎樣才能做到這一點？

ALTER FUNCTION [dbo].[myfunction]
(   
    @ID int = NULL,
    @years int = NULL,
   ,@total BIT = 0 
)
RETURNS @t TABLE (
    RowNum int,
    ID int,
    years int,
) 
AS
BEGIN
    INSERT INTO @t
       SELECT  
          ROW_NUMBER() OVER(ORDER BY years) AS RowNum,
          ID,
          years,
       FROM dbo.mytable
       WHERE ..     

    RETURN 
END

結果的外觀應為：

Total   RowNum   ID   year
-------------------------------------
  3       1      101  2014
  3       2      102  2015
  3       3      103  2016 

謝謝！

Answer 1

RETURNS @t TABLE (
    Total int,
    RowNum int,
    ID int,
    years int,
) 

...

INSERT INTO @t
       SELECT  
          NULL,
          ROW_NUMBER() OVER(ORDER BY years) AS RowNum,
          ID,
          years,
       FROM dbo.mytable
       WHERE ..

...
IF(@total=1) BEGIN
    DECLARE @Count INT
    SELECT @Count=COUNT(*) FROM @t
    UPDATE @t SET Total=@Count
END
RETURN