查找 SSIS 脚本组件中的最后一行

Question

我有一个 SSIS 2008 脚本组件，它使用以下方法

Input0_ProcessInputRow(Input0Buffer 行）

来处理行。有没有办法确定我正在读取的行是否是数据集中的最后一行。

Answer 1

if (Row.EndOfRowset()) { // this is the last row }

Answer 2

重写 InputRows_ProcessInput 的解决方案对我有用。但是，不要随意使用该代码。方法的名称取决于您的输入的名称。在我的例子中，方法签名是：

public override void Input0_ProcessInput(Input0Buffer Buffer)

找出正确方法的一个简单方法是进入脚本并输入“base.Input”intellisense 应该为您提供正确的重写方法。

public override void **Input0_ProcessInput**(**Input0Buffer** Buffer)
    {
        while (Buffer.NextRow())
        {
            Input0_ProcessInputRow(Buffer);
        }

        if (Buffer.EndOfRowset()) 
        {
            //do something 
        }
    }

仅仅检查“Buffer.EndOfRowset()”是不够的，并且它在“while (Buffer.NextRow())”循环中不起作用。所以我猜“EndOfRowset”发生在最后一行之后。

Answer 3

我以前没有尝试过，但在您的脚本组件之前，添加一个 RowCount 任务并在此任务中设置一个变量的值（称为 RowCnt，此处初始化为 0）。然后在您的脚本中，有类似这样的内容（确保变量声明的范围在类中）...

Public Class ScriptMain
Inherits UserComponent

Dim iRows As Integer

Public Overrides Sub Input0_ProcessInputRow(ByVal Row As Input0Buffer)
    iRows = iRows + 1
    If iRows = Variables.RowCnt Then
        'Do something here
    End If
End Sub

End Class

这应该允许您处理最后一行。

Answer 4

重写 InputRows_ProcessInput 方法，如下所示：

public override void InputRows_ProcessInput(InputRowsBuffer Buffer)
    {

        while (Buffer.NextRow())
        {
            InputRows_ProcessInputRow(Buffer);
        }

        if (Buffer.EndOfRowset()) 
        {
            //do something 
        }

    }

Answer 5

自定义脚本代码写在 ScriptMain 类中。该类继承自 UserComponent，其中包含一个虚拟方法 FinishOutputs()。您可以在 ScriptMain 中覆盖它以执行最后一行之后需要运行的代码。

UserComponent 中的相关代码。

public override void ProcessInput(int InputID, string InputName, PipelineBuffer Buffer, OutputNameMap OutputMap)
{

    if (InputName.Equals(@"Input 0", StringComparison.Ordinal))
    {
        Input0_ProcessInput(new Input0Buffer(Buffer, GetColumnIndexes(InputID), OutputMap));
    }

    if (Buffer.EndOfRowset)
    {
        InputsFinished = InputsFinished + 1;
        if (InputsFinished == 1)
        {
            FinishOutputs();
            MarkOutputsFinished();
        }
    }

}

public virtual void FinishOutputs()
{
}

请注意，此代码是自动生成的，无法更改。然而，它确实有效地为您提供了放置代码的位置，因为您可以在 ScriptMain 中的代码中重写 FinishOutputs()，如下所示：

public override void FinishOutputs()
{
   //do stuff
}